Profiling AR version 10

pull/576/head
taoxiangdong 4 years ago
parent 9567678168
commit 216a6f3b6a

@ -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 = ge::OpsKernelBuilderManager::Instance().RegProfReporterCallBack(func); rt_ret = ge::OpsKernelBuilderManager::Instance().RegProfReporterCallBack(reinterpret_cast<void *>(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;

Loading…
Cancel
Save