Browse Source

GE supports aicore DEPEND_SHAPE_RANGE op update outputs shape

pull/2008/head
zhujingjing 3 years ago
parent
commit
14acee1cbc
2 changed files with 1 additions and 2 deletions
  1. +1
    -0
      tests/ut/ge/CMakeLists.txt
  2. +0
    -2
      tests/ut/ge/hybrid/ge_hybrid_unittest.cc

+ 1
- 0
tests/ut/ge/CMakeLists.txt View File

@@ -738,6 +738,7 @@ set(HYBRID_TEST_FILES
"hybrid/executor/hybrid_model_async_executor_unittest.cc"
"hybrid/executor/hybrid_model_pipeline_executor_unittest.cc"
"hybrid/node_executor/aicore/aicore_task_compiler_unittest.cc"
"hybrid/node_executor/aicore/aicore_op_task_unittest.cc"
)

set(OTHERS_TEST_FILES


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

@@ -807,7 +807,6 @@ 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});
@@ -836,6 +835,5 @@ 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