Browse Source

GE supports aicore DEPEND_SHAPE_RANGE op update outputs shape

pull/2008/head
zhujingjing 3 years ago
parent
commit
1b6349833f
1 changed files with 4 additions and 0 deletions
  1. +4
    -0
      tests/ut/ge/hybrid/ge_hybrid_unittest.cc

+ 4
- 0
tests/ut/ge/hybrid/ge_hybrid_unittest.cc View File

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

Loading…
Cancel
Save