From 769cdbd2e82cf579bdbb6b9672a03ff9e098b8d3 Mon Sep 17 00:00:00 2001 From: unknown Date: Fri, 2 Apr 2021 19:27:58 +0800 Subject: [PATCH] Fix ut. --- tests/ut/ge/hybrid/ge_hybrid_unittest.cc | 15 --------------- 1 file changed, 15 deletions(-) diff --git a/tests/ut/ge/hybrid/ge_hybrid_unittest.cc b/tests/ut/ge/hybrid/ge_hybrid_unittest.cc index fba42b5c..8d1c844a 100644 --- a/tests/ut/ge/hybrid/ge_hybrid_unittest.cc +++ b/tests/ut/ge/hybrid/ge_hybrid_unittest.cc @@ -426,21 +426,6 @@ TEST_F(UtestGeHybrid, TestTaskContext) { ASSERT_EQ(new_desc.GetShape().GetDims(), new_shape.GetDims()); } -TEST_F(UtestGeHybrid, test_key_for_kernel_bin) { - auto aicore_task = std::unique_ptr(new(std::nothrow)hybrid::AiCoreOpTask()); - OpDesc op_desc("Sum", "Sum"); - EXPECT_EQ(aicore_task->GetKeyForTbeKernel(), OP_EXTATTR_NAME_TBE_KERNEL); - EXPECT_EQ(aicore_task->GetKeyForTvmMagic(), TVM_ATTR_NAME_MAGIC); - EXPECT_EQ(aicore_task->GetKeyForTvmMetaData(), TVM_ATTR_NAME_METADATA); - EXPECT_EQ(aicore_task->GetKeyForKernelName(op_desc), "Sum_kernelname"); - - auto atomic_task = std::unique_ptr(new(std::nothrow)hybrid::AtomicAddrCleanOpTask()); - EXPECT_EQ(atomic_task->GetKeyForTbeKernel(), EXT_ATTR_ATOMIC_TBE_KERNEL); - EXPECT_EQ(atomic_task->GetKeyForTvmMagic(), ATOMIC_ATTR_TVM_MAGIC); - EXPECT_EQ(atomic_task->GetKeyForTvmMetaData(), ATOMIC_ATTR_TVM_METADATA); - EXPECT_EQ(atomic_task->GetKeyForKernelName(op_desc), "Sum_atomic_kernelname"); -} - TEST_F(UtestGeHybrid, hybrid_model_executor_check_shape) { HybridModelExecutor::ExecuteArgs args; GeTensorDescPtr ge_tensor = make_shared(GeTensorDesc());