Browse Source

For aicpu all_shape compile.

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

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

@@ -48,8 +48,8 @@ Status KernelExTaskInfo::InitTaskExtInfo(const std::string &ext_info) {
GELOGD("Ext infoType=%d, infoLen=%u.", aicpu_ext_info->infoType, aicpu_ext_info->infoLen);
if (aicpu_ext_info->infoType == aicpu::FWKAdapter::FWK_ADPT_EXT_BITMAP) {
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);
"Parse bit_map info failed as infoLen must be %zu but %u.",
sizeof(uint64_t), aicpu_ext_info->infoLen);
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.");


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

@@ -988,8 +988,8 @@ Status KernelTaskInfo::InitAicpuTaskExtInfo(const std::string &ext_info) {
GELOGD("Update aicpu_task ext_info session_info session_id is %lu", session_info->sessionId);
} else if (aicpu_ext_info->infoType == aicpu::FWKAdapter::FWK_ADPT_EXT_BITMAP) {
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);
"Parse bit_map info failed as infoLen must be %zu but %u.",
sizeof(uint64_t), aicpu_ext_info->infoLen);
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