From cdecf866db4f328e73e016aec3dd58b685be71b8 Mon Sep 17 00:00:00 2001 From: lichun Date: Thu, 1 Jul 2021 17:54:49 +0800 Subject: [PATCH] add global step info for known subgraph in unknown model and generate om for remained cases when some single op cases run atc failed --- tests/ut/ge/hybrid/known_node_executor_unittest.cc | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/ut/ge/hybrid/known_node_executor_unittest.cc b/tests/ut/ge/hybrid/known_node_executor_unittest.cc index a8367130..435928ee 100644 --- a/tests/ut/ge/hybrid/known_node_executor_unittest.cc +++ b/tests/ut/ge/hybrid/known_node_executor_unittest.cc @@ -141,6 +141,6 @@ TEST_F(UnknownNodeExecutorTest, TestSetGlobalStep) { KnownNodeExecutor known_node_executor; std::shared_ptr davinci_model = MakeShared(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); }