Browse Source

support fwk offline inference when ge_lib is not initialized

pull/1636/head
lichun 4 years ago
parent
commit
fd2c95a493
1 changed files with 4 additions and 4 deletions
  1. +4
    -4
      tests/ut/ge/hybrid/node_executor/aicore/aicore_task_compiler_unittest.cc

+ 4
- 4
tests/ut/ge/hybrid/node_executor/aicore/aicore_task_compiler_unittest.cc View File

@@ -39,16 +39,16 @@ TEST_F(UtestAiCoreTaskCompiler, test_aicore_task_compiler_init) {
AicoreTaskCompiler aicore_task_compiler; AicoreTaskCompiler aicore_task_compiler;
NodePtr node = nullptr; NodePtr node = nullptr;
std::vector<domi::TaskDef> tasks{}; std::vector<domi::TaskDef> tasks{};
ASSERT_EQ(aicore_task_compiler.Initialize(), FAILED); // cause: ge lib is nullptr
ASSERT_EQ(aicore_task_compiler.CompileOp(node, tasks), FAILED); // cause: aicore task compiler init failed.
EXPECT_EQ(aicore_task_compiler.Initialize(), FAILED); // cause: ge lib is nullptr
EXPECT_EQ(aicore_task_compiler.CompileOp(node, tasks), FAILED); // cause: aicore task compiler init failed.


std::shared_ptr<GELib> ge_lib_ptr = MakeShared<GELib>(); std::shared_ptr<GELib> ge_lib_ptr = MakeShared<GELib>();
ge_lib_ptr->init_flag_ = true; ge_lib_ptr->init_flag_ = true;
OpsKernelManager OpsKernelManagerObj; OpsKernelManager OpsKernelManagerObj;
OpsKernelInfoStorePtr ops_kernel_info_store = MakeShared<OpsKernelInfoStore>(new OpsKernelInfoStore()); OpsKernelInfoStorePtr ops_kernel_info_store = MakeShared<OpsKernelInfoStore>(new OpsKernelInfoStore());
OpsKernelManagerObj.ops_kernel_store_.insert("AIcoreEngine", ops_kernel_info_store); OpsKernelManagerObj.ops_kernel_store_.insert("AIcoreEngine", ops_kernel_info_store);
ASSERT_EQ(aicore_task_compiler.CompileOp(node, tasks), FAILED); // cause: node is nullptr
ASSERT_EQ(aicore_task_compiler.is_initialized_, true); // though CompileOp failed since node is nullptr,
EXPECT_EQ(aicore_task_compiler.CompileOp(node, tasks), FAILED); // cause: node is nullptr
EXPECT_EQ(aicore_task_compiler.is_initialized_, true); // though CompileOp failed since node is nullptr,
// but aicore_task_compiler init success. // but aicore_task_compiler init success.
} }
} // namespace ge } // namespace ge


Loading…
Cancel
Save