|
@@ -151,7 +151,7 @@ ge::Status RegProfReporterCallback(MsprofReporterCallback func) { |
|
|
return rt_ret; |
|
|
return rt_ret; |
|
|
} |
|
|
} |
|
|
// Pass MsprofReporterCallback to hccl in opskernel so initialize |
|
|
// Pass MsprofReporterCallback to hccl in opskernel so initialize |
|
|
rt_ret = OpsKernelBuilderManager::Instance().RegProfReporterCallBack(func); |
|
|
|
|
|
|
|
|
rt_ret = ge::OpsKernelBuilderManager::Instance().RegProfReporterCallBack(func); |
|
|
if (rt_ret != ge::SUCCESS) { |
|
|
if (rt_ret != ge::SUCCESS) { |
|
|
GELOGE(rt_ret, "Pass MsprofReporterCallback to hccl failed."); |
|
|
GELOGE(rt_ret, "Pass MsprofReporterCallback to hccl failed."); |
|
|
return rt_ret; |
|
|
return rt_ret; |
|
|