Browse Source

For aicpu all_shape compile.

pull/929/head
unknown 5 years ago
parent
commit
ff69d4fdab
3 changed files with 15 additions and 0 deletions
  1. +13
    -0
      ge/hybrid/node_executor/aicpu/aicpu_ext_info.cc
  2. +1
    -0
      ge/hybrid/node_executor/aicpu/aicpu_ext_info.h
  3. +1
    -0
      third_party/fwkacllib/inc/cce/fwk_adpt_struct.h

+ 13
- 0
ge/hybrid/node_executor/aicpu/aicpu_ext_info.cc View File

@@ -63,6 +63,9 @@ Status AicpuExtInfoHandler::Parse(const std::string &ext_info) {
case aicpu::FWKAdapter::FWK_ADPT_EXT_SESSION_INFO:
GE_CHK_STATUS_RET(ParseExtSessionInfo(aicpu_ext_info), "Parse ext session info failed.");
break;
case aicpu::FWKAdapter::FWK_ADPT_EXT_BITMAP:
GE_CHK_STATUS_RET(ParseExtBitMap(aicpu_ext_info), "Parse ext session info failed.");
break;
default:
GELOGD("Node[%s] ignore infoType=%d, infoLen=%u.",
node_name_.c_str(), aicpu_ext_info->infoType, aicpu_ext_info->infoLen);
@@ -139,6 +142,16 @@ Status AicpuExtInfoHandler::ParseExtSessionInfo(AicpuExtInfo *aicpu_ext_info) {
return SUCCESS;
}

Status AicpuExtInfoHandler::ParseExtBitMap(AicpuExtInfo *aicpu_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);

bit_map_ = reinterpret_cast<uint64_t *>(aicpu_ext_info->infoMsg);
GELOGI("Node[%s] bit_map info success infoLen=%u.", node_name_.c_str(), aicpu_ext_info->infoLen);
return SUCCESS;
}

Status AicpuExtInfoHandler::UpdateSessionInfo(uint64_t session_id, uint64_t kernel_id, bool sess_flag) {
if (session_info_ == nullptr) {
GELOGD("There is no session info in ext_info, no need update.");


+ 1
- 0
ge/hybrid/node_executor/aicpu/aicpu_ext_info.h View File

@@ -80,6 +80,7 @@ class AicpuExtInfoHandler {
const uint32_t output_num_;
UnknowShapeOpType unknown_type_;
AicpuSessionInfo *session_info_ = nullptr;
uint64_t *bit_map_ = nullptr;

std::unique_ptr<uint8_t[]> ext_info_;
size_t ext_info_len_ = 0;


+ 1
- 0
third_party/fwkacllib/inc/cce/fwk_adpt_struct.h View File

@@ -60,6 +60,7 @@ enum FWKTaskExtInfoType {
FWK_ADPT_EXT_UPDATE_ADDR,
FWK_ADPT_EXT_OP_NAME,
FWK_ADPT_EXT_SESSION_INFO,
FWK_ADPT_EXT_BITMAP,
FWK_ADPT_EXT_INVALID
};



Loading…
Cancel
Save