diff --git a/tests/ut/ge/hybrid/node_executor/aicpu/aicpu_node_executor_unittest.cc b/tests/ut/ge/hybrid/node_executor/aicpu/aicpu_node_executor_unittest.cc index d8d39e11..153dda85 100644 --- a/tests/ut/ge/hybrid/node_executor/aicpu/aicpu_node_executor_unittest.cc +++ b/tests/ut/ge/hybrid/node_executor/aicpu/aicpu_node_executor_unittest.cc @@ -151,16 +151,15 @@ TEST_F(UtestAicpuNodeExecutor, aicpu_tf_node_task) { domi::KernelExDef *kernel_def = task_def2.mutable_kernel(); kernel_def->set_args(reinterpret_cast(&args), args.head.length); kernel_def->set_args_size(args.head.length); - AicpuExtInfo aicpu_ext_info; - aicpu_ext_info.infoType = aicpu::FWKAdapter::FWK_ADPT_EXT_SHAPE_TYPE; - aicpu_ext_info.infoLen = sizeof(int32_t); + AicpuExtInfo aicpu_ext_info2; + aicpu_ext_info2.infoType = aicpu::FWKAdapter::FWK_ADPT_EXT_SHAPE_TYPE; + aicpu_ext_info2.infoLen = sizeof(int32_t); int32_t type = node_item->shape_inference_type; - 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)); - std::string ext_info(ext_mem, sizeof(AicpuExtInfo) + sizeof(int32_t)); - kernel_def->set_kernel_ext_info(ext_info.c_str(), ext_info.length()); - kernel_def->set_kernel_ext_info_size(ext_info.length()); + memcpy_s(aicpu_ext_info2.infoMsg, sizeof(int32_t), &type, sizeof(int32_t)); + char *ext_mem2 = (char*)malloc(sizeof(AicpuExtInfo) + sizeof(int32_t)); + memcpy_s(ext_mem2, sizeof(AicpuExtInfo) + sizeof(int32_t), &aicpu_ext_info2, sizeof(AicpuExtInfo) + sizeof(int32_t)); + kernel_def->set_kernel_ext_info(ext_mem2, sizeof(AicpuExtInfo) + sizeof(int32_t)); + kernel_def->set_kernel_ext_info_size(sizeof(AicpuExtInfo) + sizeof(int32_t)); hybrid_model.task_defs_[node] = std::vector({task_def2, task_def2}); AicpuNodeTask aicpu_node_task(node_item, task_def2);