Browse Source

single_op doesn't execute atomic.

tags/v1.2.0
unknown 3 years ago
parent
commit
70beda36e4
1 changed files with 5 additions and 2 deletions
  1. +5
    -2
      ge/hybrid/node_executor/aicore/aicore_op_task.cc

+ 5
- 2
ge/hybrid/node_executor/aicore/aicore_op_task.cc View File

@@ -131,7 +131,7 @@ Status AiCoreOpTask::RegisterKernelHandle(const OpDesc &op_desc) {
}

void *bin_handle = nullptr;
GELOGD("Start to register kernel for node: [%s].", op_desc.GetName().c_str());
GELOGD("Start to register kernel for node[%s].", op_desc.GetName().c_str());
rtDevBinary_t binary;
std::string json_string;
GE_IF_BOOL_EXEC(AttrUtils::GetStr(&op_desc, TVM_ATTR_NAME_MAGIC, json_string),
@@ -149,7 +149,7 @@ Status AiCoreOpTask::RegisterKernelHandle(const OpDesc &op_desc) {
binary.version = 0;
binary.data = tbe_kernel->GetBinData();
binary.length = tbe_kernel->GetBinDataSize();
GELOGI("TBE: binary.length: %lu", binary.length);
GELOGI("TBE: binary.length: %lu.", binary.length);
GE_CHK_RT_RET(rtRegisterAllKernel(&binary, &bin_handle));
handle_ = bin_handle;
auto holder = std::unique_ptr<TbeHandleHolder>(new (std::nothrow) TbeHandleHolder(handle_));
@@ -351,6 +351,9 @@ Status AiCoreOpTask::CalcTilingInfo(const NodePtr &node, OpRunInfo &tiling_info)
GE_CHK_STATUS_RET(OpParaCalculate(*node, tiling_info),
"Failed calc tiling data of node %s.",
node->GetName().c_str());
if (is_single_op_) {
tiling_info.clear_atomic = false;
}
GELOGD("[%s] Done invoking OpParaCalculate successfully.", node->GetName().c_str());
return SUCCESS;
}


Loading…
Cancel
Save