From e74b4e0f4f23ab25fd2795e3e96f05eee49ac683 Mon Sep 17 00:00:00 2001 From: guopeian Date: Mon, 19 Jul 2021 16:45:29 +0800 Subject: [PATCH] fix ut --- tests/ut/ge/single_op/single_op_model_unittest.cc | 9 +++++---- 1 file changed, 5 insertions(+), 4 deletions(-) diff --git a/tests/ut/ge/single_op/single_op_model_unittest.cc b/tests/ut/ge/single_op/single_op_model_unittest.cc index 8e876456..2a3a3f93 100644 --- a/tests/ut/ge/single_op/single_op_model_unittest.cc +++ b/tests/ut/ge/single_op/single_op_model_unittest.cc @@ -393,10 +393,11 @@ TEST_F(UtestSingleOpModel, build_dynamic_task02) { aicpu_ext_info.infoLen = sizeof(int32_t); int32_t type = ge::DEPEND_COMPUTE; memcpy_s(aicpu_ext_info.infoMsg, sizeof(int32_t), &type, sizeof(int32_t)); - char *ext_mem = (char*)malloc(sizeof(AicpuExtInfo) + sizeof(int32_t)); - memcpy_s(ext_mem, sizeof(AicpuExtInfo) + sizeof(int32_t), &aicpu_ext_info, sizeof(AicpuExtInfo) + sizeof(int32_t)); - kernel_def->set_kernel_ext_info(ext_mem, sizeof(AicpuExtInfo) + sizeof(int32_t)); - kernel_def->set_kernel_ext_info_size(sizeof(AicpuExtInfo) + sizeof(int32_t)); + char *ext_mem = (char*)malloc(sizeof(ge::hybrid::AicpuExtInfo) + sizeof(int32_t)); + memcpy_s(ext_mem, sizeof(ge::hybrid::AicpuExtInfo) + sizeof(int32_t), &aicpu_ext_info, + sizeof(ge::hybrid::AicpuExtInfo) + sizeof(int32_t)); + kernel_def->set_kernel_ext_info(ext_mem, sizeof(ge::hybrid::AicpuExtInfo) + sizeof(int32_t)); + kernel_def->set_kernel_ext_info_size(sizeof(ge::hybrid::AicpuExtInfo) + sizeof(int32_t)); domi::KernelContext *context = kernel_def->mutable_context(); context->set_kernel_type(6); // ccKernelType::AI_CPU