Browse Source

For aicpu all_shape compile.

pull/929/head
unknown 5 years ago
parent
commit
4c58f7b309
2 changed files with 2 additions and 2 deletions
  1. +1
    -1
      ge/graph/load/new_model_manager/task_info/kernel_ex_task_info.cc
  2. +1
    -1
      ge/graph/load/new_model_manager/task_info/kernel_task_info.cc

+ 1
- 1
ge/graph/load/new_model_manager/task_info/kernel_ex_task_info.cc View File

@@ -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<uint64_t *>(aicpu_ext_info->infoMsg);
uint64_t *bit_map = reinterpret_cast<uint64_t *>(aicpu_ext_info->infoMsg);
*(bit_map) |= 1;
GELOGD("Update aicpu_task ext_info bit_map to 1.");
}


+ 1
- 1
ge/graph/load/new_model_manager/task_info/kernel_task_info.cc View File

@@ -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<uint64_t *>(aicpu_ext_info->infoMsg);
uint64_t *bit_map = reinterpret_cast<uint64_t *>(aicpu_ext_info->infoMsg);
*(bit_map) |= 1;
GELOGD("Update aicpu_task ext_info bit_map to 1.");
}


Loading…
Cancel
Save