|
|
|
@ -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;
|
|
|
|
|
}
|
|
|
|
|