| @@ -393,10 +393,11 @@ TEST_F(UtestSingleOpModel, build_dynamic_task02) { | |||||
| aicpu_ext_info.infoLen = sizeof(int32_t); | aicpu_ext_info.infoLen = sizeof(int32_t); | ||||
| int32_t type = ge::DEPEND_COMPUTE; | int32_t type = ge::DEPEND_COMPUTE; | ||||
| memcpy_s(aicpu_ext_info.infoMsg, sizeof(int32_t), &type, sizeof(int32_t)); | 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(); | domi::KernelContext *context = kernel_def->mutable_context(); | ||||
| context->set_kernel_type(6); // ccKernelType::AI_CPU | context->set_kernel_type(6); // ccKernelType::AI_CPU | ||||