diff --git a/ge/graph/load/new_model_manager/task_info/kernel_ex_task_info.cc b/ge/graph/load/new_model_manager/task_info/kernel_ex_task_info.cc index 35eade12..ad5c5b30 100644 --- a/ge/graph/load/new_model_manager/task_info/kernel_ex_task_info.cc +++ b/ge/graph/load/new_model_manager/task_info/kernel_ex_task_info.cc @@ -50,7 +50,7 @@ Status KernelExTaskInfo::InitTaskExtInfo(const std::string &ext_info) { GE_CHK_BOOL_RET_STATUS(aicpu_ext_info->infoLen == sizeof(uint64_t), PARAM_INVALID, "Node[%s] parse bit_map info failed as infoLen must be %zu but %u.", node_name_.c_str(), sizeof(uint64_t), aicpu_ext_info->infoLen); - uint64_t bit_map = reinterpret_cast(aicpu_ext_info->infoMsg); + uint64_t *bit_map = reinterpret_cast(aicpu_ext_info->infoMsg); *(bit_map) |= 1; GELOGD("Update aicpu_task ext_info bit_map to 1."); } diff --git a/ge/graph/load/new_model_manager/task_info/kernel_task_info.cc b/ge/graph/load/new_model_manager/task_info/kernel_task_info.cc index a526d716..db2e3f7f 100755 --- a/ge/graph/load/new_model_manager/task_info/kernel_task_info.cc +++ b/ge/graph/load/new_model_manager/task_info/kernel_task_info.cc @@ -990,7 +990,7 @@ Status KernelTaskInfo::InitAicpuTaskExtInfo(const std::string &ext_info) { GE_CHK_BOOL_RET_STATUS(aicpu_ext_info->infoLen == sizeof(uint64_t), PARAM_INVALID, "Node[%s] parse bit_map info failed as infoLen must be %zu but %u.", node_name_.c_str(), sizeof(uint64_t), aicpu_ext_info->infoLen); - uint64_t bit_map = reinterpret_cast(aicpu_ext_info->infoMsg); + uint64_t *bit_map = reinterpret_cast(aicpu_ext_info->infoMsg); *(bit_map) |= 1; GELOGD("Update aicpu_task ext_info bit_map to 1."); }