|
|
|
@@ -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<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) { |
|
|
|
HybridModelExecutor::ExecuteArgs args; |
|
|
|
GeTensorDescPtr ge_tensor = make_shared<GeTensorDesc>(GeTensorDesc()); |
|
|
|
|