diff --git a/impl/kfc/kfc_register_obj.h b/impl/kfc/kfc_register_obj.h index 13f058eb06d50dcdae00f3731dcd80927adc63a1..a5054dc18b81dc70ff2293b4dc155615b791c951 100644 --- a/impl/kfc/kfc_register_obj.h +++ b/impl/kfc/kfc_register_obj.h @@ -132,7 +132,7 @@ __aicore__ inline void InitCurObj(AscendC::TPipe* tpipe, T& a, Args&&... b) #ifdef ASCENDC_CUBE_ONLY #define REGIST_CUBE_OBJ(tpipe, workspace, ...) \ AscendC::InitCurObj(tpipe, __VA_ARGS__); \ - AscendC::AscendCTimeStamp(static_cast(AscendC::TimeStampId::TIME_STAMP_MATMUL_SERVER_OBJ)) + AscendC::PrintTimeStamp(static_cast(AscendC::TimeStampId::TIME_STAMP_MATMUL_SERVER_OBJ)) #define REGIST_CUBE_OBJ_REMOTE(tpipe, workspace, ...) #else @@ -145,11 +145,11 @@ __aicore__ inline void InitCurObj(AscendC::TPipe* tpipe, T& a, Args&&... b) static_assert(asEnableMixDualMaster != AscendC::CUBEOBJ_MIX_MODE, \ "enableMixDualMaster must be consistent for all cube objects."); \ AscendC::KfcServer server; \ - AscendC::AscendCTimeStamp(static_cast(AscendC::TimeStampId::TIME_STAMP_MATMUL_SERVER)); \ + AscendC::PrintTimeStamp(static_cast(AscendC::TimeStampId::TIME_STAMP_MATMUL_SERVER)); \ server.Init(workspace); \ - AscendC::AscendCTimeStamp(static_cast(AscendC::TimeStampId::TIME_STAMP_MATMUL_SERVER_INIT)); \ + AscendC::PrintTimeStamp(static_cast(AscendC::TimeStampId::TIME_STAMP_MATMUL_SERVER_INIT)); \ server.InitObj(tpipe, __VA_ARGS__); \ - AscendC::AscendCTimeStamp(static_cast(AscendC::TimeStampId::TIME_STAMP_MATMUL_SERVER_OBJ)); \ + AscendC::PrintTimeStamp(static_cast(AscendC::TimeStampId::TIME_STAMP_MATMUL_SERVER_OBJ)); \ if constexpr (!asEnableMixDualMaster) { \ while (server.isRun()) { \ server.Run(__VA_ARGS__); \ @@ -173,22 +173,22 @@ __aicore__ inline void InitCurObj(AscendC::TPipe* tpipe, T& a, Args&&... b) static_assert(asEnableMixDualMaster != AscendC::CUBEOBJ_MIX_MODE, \ "enableMixDualMaster must be consistent for all cube objects."); \ AscendC::KfcCommClient __kfcClient__(workspace, AscendC::GetSubBlockIdx(), asEnableMixDualMaster); \ - AscendC::AscendCTimeStamp(static_cast(AscendC::TimeStampId::TIME_STAMP_MATMUL_CLIENT_KFC)); \ + AscendC::PrintTimeStamp(static_cast(AscendC::TimeStampId::TIME_STAMP_MATMUL_CLIENT_KFC)); \ if constexpr (!asEnableMixDualMaster) { \ AscendC::g_kfcClient = &__kfcClient__; \ } \ AscendC::SetMatrixKfc(tpipe, &__kfcClient__, 0, workspace, __VA_ARGS__); \ - AscendC::AscendCTimeStamp(static_cast(AscendC::TimeStampId::TIME_STAMP_MATMUL_MATRIX_KFC)); \ + AscendC::PrintTimeStamp(static_cast(AscendC::TimeStampId::TIME_STAMP_MATMUL_MATRIX_KFC)); \ if constexpr (!asEnableMixDualMaster) { \ AscendC::WaitEvent(AscendC::WORKSPACE_SYNC_ID); \ } \ - AscendC::AscendCTimeStamp(static_cast(AscendC::TimeStampId::TIME_STAMP_MATMUL_WAIT_EVE)) + AscendC::PrintTimeStamp(static_cast(AscendC::TimeStampId::TIME_STAMP_MATMUL_WAIT_EVE)) #endif #else #define REGIST_CUBE_OBJ(tpipe, workspace, ...) \ AscendC::InitCurObj(tpipe, __VA_ARGS__); \ - AscendC::AscendCTimeStamp(static_cast(AscendC::TimeStampId::TIME_STAMP_MATMUL_OBJ)) + AscendC::PrintTimeStamp(static_cast(AscendC::TimeStampId::TIME_STAMP_MATMUL_OBJ)) #endif #endif