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