|
|
@@ -807,6 +807,7 @@ TEST_F(UtestGeHybrid, TestParseDependencies) { |
|
|
|
} |
|
|
|
|
|
|
|
TEST_F(UtestGeHybrid, TestTaskExecuteAsync) { |
|
|
|
dlog_setlevel(0, 0, 0); |
|
|
|
auto graph = make_shared<ComputeGraph>("graph"); |
|
|
|
OpDescPtr op_desc = CreateOpDesc("Add", "Add"); |
|
|
|
GeShape shape({2, 16}); |
|
|
@@ -821,6 +822,8 @@ TEST_F(UtestGeHybrid, TestTaskExecuteAsync) { |
|
|
|
node_item->output_start = 0; |
|
|
|
|
|
|
|
GraphExecutionContext execution_context; |
|
|
|
execution_context.callback_manager = |
|
|
|
std::unique_ptr<CallbackManager>(new (std::nothrow) CallbackManager()); |
|
|
|
GraphItem graph_item; |
|
|
|
SubgraphContext subgraph_context(&graph_item, &execution_context); |
|
|
|
ASSERT_EQ(subgraph_context.Init(), SUCCESS); |
|
|
@@ -833,5 +836,6 @@ TEST_F(UtestGeHybrid, TestTaskExecuteAsync) { |
|
|
|
std::vector<std::unique_ptr<AiCoreOpTask>> tasks; |
|
|
|
AiCoreNodeTask node_task(std::move(tasks)); |
|
|
|
ASSERT_EQ(node_task.ExecuteAsync(task_context, nullptr), SUCCESS); |
|
|
|
dlog_setlevel(0, 3, 0); |
|
|
|
} |
|
|
|
} // namespace ge |