From 4c58f7b30922600d511f54d711657a8ec35bc13d Mon Sep 17 00:00:00 2001 From: unknown Date: Sun, 10 Jan 2021 20:54:23 +0800 Subject: [PATCH] For aicpu all_shape compile. --- .../load/new_model_manager/task_info/kernel_ex_task_info.cc | 2 +- ge/graph/load/new_model_manager/task_info/kernel_task_info.cc | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) 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."); }