diff --git a/operator/ascendc/2_features/13_matmul_api_ibshare/MatmulABshareInvocation/matmul_ABshare_custom.cpp b/operator/ascendc/2_features/13_matmul_api_ibshare/MatmulABshareInvocation/matmul_ABshare_custom.cpp index 1c5d97c10dfb58e694ddd5eda25a25e123c90589..f0c3e610e7ea87129fc7dcdd1c6dec91a785362a 100644 --- a/operator/ascendc/2_features/13_matmul_api_ibshare/MatmulABshareInvocation/matmul_ABshare_custom.cpp +++ b/operator/ascendc/2_features/13_matmul_api_ibshare/MatmulABshareInvocation/matmul_ABshare_custom.cpp @@ -24,9 +24,9 @@ __aicore__ inline void CopyTiling(TCubeTiling *tiling, GM_ADDR tilingGM) // copy return; } -template class MatmutABshareKernel { +template class MatmulABshareKernel { public: - __aicore__ inline MatmutABshareKernel(){}; + __aicore__ inline MatmulABshareKernel(){}; __aicore__ inline void Init(GM_ADDR a, GM_ADDR b, GM_ADDR c, GM_ADDR workspace, const TCubeTiling &tiling, AscendC::TPipe *pipe); __aicore__ inline void Process(AscendC::TPipe *pipe); @@ -45,7 +45,7 @@ public: }; template -__aicore__ inline void MatmutABshareKernel::Init(GM_ADDR a, GM_ADDR b, GM_ADDR c, +__aicore__ inline void MatmulABshareKernel::Init(GM_ADDR a, GM_ADDR b, GM_ADDR c, GM_ADDR workspace,const TCubeTiling &tiling, AscendC::TPipe *pipe) { this->tiling = tiling; @@ -61,7 +61,7 @@ __aicore__ inline void MatmutABshareKernel::Init(GM_ADDR a, } template -__aicore__ inline void MatmutABshareKernel::Process(AscendC::TPipe *pipe) +__aicore__ inline void MatmulABshareKernel::Process(AscendC::TPipe *pipe) { matmulObj.Init(&tiling); matmulObj.SetTensorA(aGlobal); @@ -73,7 +73,7 @@ __aicore__ inline void MatmutABshareKernel::Process(AscendC template __aicore__ inline void -MatmutABshareKernel::CalcOffset(int32_t blockIdx, const TCubeTiling &tiling, +MatmulABshareKernel::CalcOffset(int32_t blockIdx, const TCubeTiling &tiling, int32_t &offsetA, int32_t &offsetB, int32_t &offsetC) { offsetA = 0; @@ -88,8 +88,8 @@ extern "C" __global__ __aicore__ void matmul_ABshare_custom(GM_ADDR a, GM_ADDR b TCubeTiling tiling; CopyTiling(&tiling, tilingGm); - MatmutABshareKernel MatmutABshareKernel; - MatmutABshareKernel.Init(a, b, c, workspace, tiling, &pipe); - REGIST_MATMUL_OBJ(&pipe, GetSysWorkSpacePtr(), MatmutABshareKernel.matmulObj, &MatmutABshareKernel.tiling); - MatmutABshareKernel.Process(&pipe); + MatmulABshareKernel MatmulABshareKernel; + MatmulABshareKernel.Init(a, b, c, workspace, tiling, &pipe); + REGIST_MATMUL_OBJ(&pipe, GetSysWorkSpacePtr(), MatmulABshareKernel.matmulObj, &MatmulABshareKernel.tiling); + MatmulABshareKernel.Process(&pipe); } \ No newline at end of file