diff --git a/runtime/acl_rt.cpp b/runtime/acl_rt.cpp index 804e4d76127b57faa926540f4ea245fbda05ea4a..c6b59049e6b07e7e3b381f98f601f9e454d9bd02 100755 --- a/runtime/acl_rt.cpp +++ b/runtime/acl_rt.cpp @@ -1325,46 +1325,55 @@ aclError aclmdlRIDestroy(aclmdlRI modelRI) aclError aclmdlRICaptureBegin(aclrtStream stream, aclmdlRICaptureMode mode) { + ACL_PROFILING_REG(acl::AclProfType::AclmdlRICaptureBegin); return aclmdlRICaptureBeginImpl(stream, mode); } aclError aclmdlRICaptureGetInfo(aclrtStream stream, aclmdlRICaptureStatus *status, aclmdlRI *modelRI) { + ACL_PROFILING_REG(acl::AclProfType::AclmdlRICaptureGetInfo); return aclmdlRICaptureGetInfoImpl(stream, status, modelRI); } aclError aclmdlRICaptureEnd(aclrtStream stream, aclmdlRI *modelRI) { + ACL_PROFILING_REG(acl::AclProfType::AclmdlRICaptureEnd); return aclmdlRICaptureEndImpl(stream, modelRI); } aclError aclmdlRIDebugPrint(aclmdlRI modelRI) { + ACL_PROFILING_REG(acl::AclProfType::AclmdlRIDebugPrint); return aclmdlRIDebugPrintImpl(modelRI); } aclError aclmdlRICaptureThreadExchangeMode(aclmdlRICaptureMode *mode) { + ACL_PROFILING_REG(acl::AclProfType::AclmdlRICaptureThreadExchangeMode); return aclmdlRICaptureThreadExchangeModeImpl(mode); } aclError aclmdlRICaptureTaskGrpBegin(aclrtStream stream) { + ACL_PROFILING_REG(acl::AclProfType::AclmdlRICaptureTaskGrpBegin); return aclmdlRICaptureTaskGrpBeginImpl(stream); } aclError aclmdlRICaptureTaskGrpEnd(aclrtStream stream, aclrtTaskGrp *handle) { + ACL_PROFILING_REG(acl::AclProfType::AclmdlRICaptureTaskGrpEnd); return aclmdlRICaptureTaskGrpEndImpl(stream, handle); } aclError aclmdlRICaptureTaskUpdateBegin(aclrtStream stream, aclrtTaskGrp handle) { + ACL_PROFILING_REG(acl::AclProfType::AclmdlRICaptureTaskUpdateBegin); return aclmdlRICaptureTaskUpdateBeginImpl(stream, handle); } aclError aclmdlRICaptureTaskUpdateEnd(aclrtStream stream) { + ACL_PROFILING_REG(acl::AclProfType::AclmdlRICaptureTaskUpdateEnd); return aclmdlRICaptureTaskUpdateEndImpl(stream); } diff --git a/toolchain/profiling_manager.cpp b/toolchain/profiling_manager.cpp index 4487b0510a8b318943c63efd155a5910454669be..bb0df7fdd0955196b3e80c805352e362b5a68f30 100755 --- a/toolchain/profiling_manager.cpp +++ b/toolchain/profiling_manager.cpp @@ -45,6 +45,15 @@ namespace acl { {AclProfType::AclmdlUnload, "aclmdlUnload"}, {AclProfType::AclmdlRIExecuteAsync, "aclmdlRIExecuteAsync"}, {AclProfType::AclmdlRIDestroy, "aclmdlRIDestroy"}, + {AclProfType::AclmdlRICaptureBegin, "aclmdlRICaptureBegin"}, + {AclProfType::AclmdlRICaptureGetInfo, "aclmdlRICaptureGetInfo"}, + {AclProfType::AclmdlRICaptureEnd, "aclmdlRICaptureEnd"}, + {AclProfType::AclmdlRIDebugPrint, "aclmdlRIDebugPrint"}, + {AclProfType::AclmdlRICaptureThreadExchangeMode, "aclmdlRICaptureThreadExchangeMode"}, + {AclProfType::AclmdlRICaptureTaskGrpBegin, "aclmdlRICaptureTaskGrpBegin"}, + {AclProfType::AclmdlRICaptureTaskGrpEnd, "aclmdlRICaptureTaskGrpEnd"}, + {AclProfType::AclmdlRICaptureTaskUpdateBegin, "aclmdlRICaptureTaskUpdateBegin"}, + {AclProfType::AclmdlRICaptureTaskUpdateEnd, "aclmdlRICaptureTaskUpdateEnd"}, {AclProfType::AclopLoad, "aclopLoad"}, {AclProfType::AclopExecute, "aclopExecute"}, {AclProfType::AclopCreateHandle, "aclopCreateHandle"}, diff --git a/toolchain/profiling_manager.h b/toolchain/profiling_manager.h index 783fd64947e95e2cb22b53f978925d941053ae77..00e9615675bb778737290f0fdbd5ac5b0139d74d 100755 --- a/toolchain/profiling_manager.h +++ b/toolchain/profiling_manager.h @@ -43,6 +43,15 @@ enum AclProfType { AclmdlUnload, AclmdlRIExecuteAsync, AclmdlRIDestroy, + AclmdlRICaptureBegin, + AclmdlRICaptureGetInfo, + AclmdlRICaptureEnd, + AclmdlRIDebugPrint, + AclmdlRICaptureThreadExchangeMode, + AclmdlRICaptureTaskGrpBegin, + AclmdlRICaptureTaskGrpEnd, + AclmdlRICaptureTaskUpdateBegin, + AclmdlRICaptureTaskUpdateEnd, ModelTypeEnd, OpTypeStart = MSPROF_REPORT_ACL_OP_BASE_TYPE, AclopLoad,