Browse Source

add global step info for known subgraph in unknown model and generate om for remained cases when some single op cases run atc failed

tags/v1.5.1
lichun 3 years ago
parent
commit
27dae9195c
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      tests/ut/ge/hybrid/known_node_executor_unittest.cc

+ 1
- 1
tests/ut/ge/hybrid/known_node_executor_unittest.cc View File

@@ -135,7 +135,7 @@ TEST_F(UnknownNodeExecutorTest, TestSetGlobalStep) {
HybridModel hybrid_model(ge_root_model);
auto *step_id = new int64_t[1];
step_id[0] = 520;
std::unique_ptr<TensorValue> tensor_buf;
std::unique_ptr<TensorBuffer> tensor_buf;
tensor_buf = tensor_buf->Create((void *)step_id, sizeof(int64_t));
hybrid_model.global_step_ = std::move(tensor_buf);
KnownNodeExecutor known_node_executor;


Loading…
Cancel
Save