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
d74118080c
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

@@ -141,6 +141,6 @@ TEST_F(UnknownNodeExecutorTest, TestSetGlobalStep) {

KnownNodeExecutor known_node_executor;
std::shared_ptr<DavinciModel> davinci_model = MakeShared<DavinciModel>(0, nullptr);
known_node_executor.SettingDaviciModel(hybrid_model, node, davinci_model);
known_node_executor.SetDaviciModel(hybrid_model, node, davinci_model);
EXPECT_EQ(*(static_cast<int64_t*>(davinci_model->global_step_addr_)), 520);
}

Loading…
Cancel
Save