| @@ -131,7 +131,7 @@ Status AiCoreOpTask::RegisterKernelHandle(const OpDesc &op_desc) { | |||||
| } | } | ||||
| void *bin_handle = nullptr; | 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; | rtDevBinary_t binary; | ||||
| std::string json_string; | std::string json_string; | ||||
| GE_IF_BOOL_EXEC(AttrUtils::GetStr(&op_desc, TVM_ATTR_NAME_MAGIC, 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.version = 0; | ||||
| binary.data = tbe_kernel->GetBinData(); | binary.data = tbe_kernel->GetBinData(); | ||||
| binary.length = tbe_kernel->GetBinDataSize(); | 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)); | GE_CHK_RT_RET(rtRegisterAllKernel(&binary, &bin_handle)); | ||||
| handle_ = bin_handle; | handle_ = bin_handle; | ||||
| auto holder = std::unique_ptr<TbeHandleHolder>(new (std::nothrow) TbeHandleHolder(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), | GE_CHK_STATUS_RET(OpParaCalculate(*node, tiling_info), | ||||
| "Failed calc tiling data of node %s.", | "Failed calc tiling data of node %s.", | ||||
| node->GetName().c_str()); | node->GetName().c_str()); | ||||
| if (is_single_op_) { | |||||
| tiling_info.clear_atomic = false; | |||||
| } | |||||
| GELOGD("[%s] Done invoking OpParaCalculate successfully.", node->GetName().c_str()); | GELOGD("[%s] Done invoking OpParaCalculate successfully.", node->GetName().c_str()); | ||||
| return SUCCESS; | return SUCCESS; | ||||
| } | } | ||||