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
cdecf866db
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, node, davinci_model);
known_node_executor.SettingDaviciModel(hybrid_model, node, davinci_model);
EXPECT_EQ(davinci_model->global_step_addr_, 520);
}

Loading…
Cancel
Save