From e331dee2b9b1bf2d4fe09487719f31cac17cd8fe Mon Sep 17 00:00:00 2001 From: unknown Date: Tue, 26 Jan 2021 10:59:53 +0800 Subject: [PATCH] Fix ut. --- tests/ut/ge/graph/load/kernel_ex_task_info_unittest.cc | 2 +- tests/ut/ge/graph/load/kernel_task_info_unittest.cc | 9 +++------ 2 files changed, 4 insertions(+), 7 deletions(-) diff --git a/tests/ut/ge/graph/load/kernel_ex_task_info_unittest.cc b/tests/ut/ge/graph/load/kernel_ex_task_info_unittest.cc index 7bc3faca..850a29ad 100644 --- a/tests/ut/ge/graph/load/kernel_ex_task_info_unittest.cc +++ b/tests/ut/ge/graph/load/kernel_ex_task_info_unittest.cc @@ -138,7 +138,7 @@ TEST_F(UtestKernelExTaskInfo, kernel_ex_task_info_calculate_args) { } TEST_F(UtestKernelExTaskInfo, kernel_ex_task_ext_info) { - const char ext_info[4] = {0, 0, 0, 4}; + const string ext_info = {1, 1, 1, 1, 0, 0, 0, 0}; const OpDescPtr op_desc = CreateOpDesc("FrameworkOp", "FrameworkOp"); KernelExTaskInfo kernel_ex_task_info; diff --git a/tests/ut/ge/graph/load/kernel_task_info_unittest.cc b/tests/ut/ge/graph/load/kernel_task_info_unittest.cc index c253666f..65783ac9 100644 --- a/tests/ut/ge/graph/load/kernel_task_info_unittest.cc +++ b/tests/ut/ge/graph/load/kernel_task_info_unittest.cc @@ -413,6 +413,9 @@ TEST_F(UtestKernelTaskInfo, init_kernel_taskInfo_with_aicpu_kernel_type_fail) { // rtMemcpy -> RT_ERROR_INVALID_VALUE EXPECT_EQ(kernel_task_info.Init(task_def, &model), SUCCESS); + const string ext_info = {1, 1, 1, 1, 0, 0, 0, 0}; + EXPECT_EQ(kernel_task_info.InitAicpuTaskExtInfo(ext_info), SUCCESS); + EXPECT_EQ(kernel_task_info.Distribute(), SUCCESS); EXPECT_EQ(kernel_task_info.Release(), SUCCESS); @@ -1195,10 +1198,4 @@ TEST_F(UtestKernelTaskInfo, kernel_task_info_super_kernel_info) { EXPECT_EQ(kernel_task_info.SKTFinalize(), SUCCESS); } -TEST_F(UtestKernelTaskInfo, kernel_ask_ext_info) { - const char ext_info[4] = {0, 0, 0, 4}; - KernelTaskInfo kernel_task_info; - EXPECT_EQ(kernel_task_info.InitAicpuTaskExtInfo(ext_info), SUCCESS); -} - } // namespace ge