Browse Source

Fix ut.

pull/1442/head
unknown 4 years ago
parent
commit
769cdbd2e8
1 changed files with 0 additions and 15 deletions
  1. +0
    -15
      tests/ut/ge/hybrid/ge_hybrid_unittest.cc

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

@@ -426,21 +426,6 @@ TEST_F(UtestGeHybrid, TestTaskContext) {
ASSERT_EQ(new_desc.GetShape().GetDims(), new_shape.GetDims()); ASSERT_EQ(new_desc.GetShape().GetDims(), new_shape.GetDims());
} }


TEST_F(UtestGeHybrid, test_key_for_kernel_bin) {
auto aicore_task = std::unique_ptr<hybrid::AiCoreOpTask>(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<hybrid::AtomicAddrCleanOpTask>(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) { TEST_F(UtestGeHybrid, hybrid_model_executor_check_shape) {
HybridModelExecutor::ExecuteArgs args; HybridModelExecutor::ExecuteArgs args;
GeTensorDescPtr ge_tensor = make_shared<GeTensorDesc>(GeTensorDesc()); GeTensorDescPtr ge_tensor = make_shared<GeTensorDesc>(GeTensorDesc());


Loading…
Cancel
Save