From 14acee1cbc991439469a83a3c984db60238c2def Mon Sep 17 00:00:00 2001 From: zhujingjing Date: Wed, 21 Jul 2021 09:22:41 +0800 Subject: [PATCH] GE supports aicore DEPEND_SHAPE_RANGE op update outputs shape --- tests/ut/ge/CMakeLists.txt | 1 + tests/ut/ge/hybrid/ge_hybrid_unittest.cc | 2 -- 2 files changed, 1 insertion(+), 2 deletions(-) diff --git a/tests/ut/ge/CMakeLists.txt b/tests/ut/ge/CMakeLists.txt index 42fa6128..9d2ef050 100755 --- a/tests/ut/ge/CMakeLists.txt +++ b/tests/ut/ge/CMakeLists.txt @@ -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 diff --git a/tests/ut/ge/hybrid/ge_hybrid_unittest.cc b/tests/ut/ge/hybrid/ge_hybrid_unittest.cc index e8c13a82..42b1469d 100644 --- a/tests/ut/ge/hybrid/ge_hybrid_unittest.cc +++ b/tests/ut/ge/hybrid/ge_hybrid_unittest.cc @@ -807,7 +807,6 @@ TEST_F(UtestGeHybrid, TestParseDependencies) { } TEST_F(UtestGeHybrid, TestTaskExecuteAsync) { - dlog_setlevel(0, 0, 0); auto graph = make_shared("graph"); OpDescPtr op_desc = CreateOpDesc("Add", "Add"); GeShape shape({2, 16}); @@ -836,6 +835,5 @@ TEST_F(UtestGeHybrid, TestTaskExecuteAsync) { std::vector> tasks; AiCoreNodeTask node_task(std::move(tasks)); ASSERT_EQ(node_task.ExecuteAsync(task_context, nullptr), SUCCESS); - dlog_setlevel(0, 3, 0); } } // namespace ge \ No newline at end of file