ops-tune-workbench-plugin 0.0.0-stage → 0.0.1-beta.2

package/skills/tool-ops-tune/ops-family/references/conversion/templates/dav3510/transpose_n_last.template

@@ -0,0 +1,340 @@
+/**
+ * Copyright (c) 2025 Huawei Technologies Co., Ltd.
+ * This program is free software, you can redistribute it and/or modify it under the terms and conditions of
+ * CANN Open Software License Agreement Version 2.0 (the "License").
+ * Please refer to the License for details. You may not use this file except in compliance with the License.
+ * THIS SOFTWARE IS PROVIDED ON AN "AS IS" BASIS, WITHOUT WARRANTIES OF ANY KIND, EITHER EXPRESS OR IMPLIED,
+ * INCLUDING BUT NOT LIMITED TO NON-INFRINGEMENT, MERCHANTABILITY, OR FITNESS FOR A PARTICULAR PURPOSE.
+ * See LICENSE in the root of the software repository for the full text of the License.
+ */
+
+/*!
+ * \file transpose_n_last.h
+ * \brief transpose_n_last
+ */
+#ifndef TRANSPOSE_N_LAST_H
+#define TRANSPOSE_N_LAST_H
+
+#include "transpose_base.h"
+
+namespace Transpose {
+using namespace AscendC;
+
+template <typename T>
+class TransposeNLast : public TransposeBase<T> {
+public:
+    __aicore__ inline TransposeNLast(){};
+    __aicore__ inline void Init(GM_ADDR x, GM_ADDR y, const TransposeOpTilingData* tilingData, TPipe* pipe);
+    __aicore__ inline void Process();
+
+private:
+    __aicore__ inline void ParseTilingData();
+    __aicore__ inline void GetBlockLoopNum();
+    __aicore__ inline void GetLoopParams(int64_t n);
+    __aicore__ inline int64_t GetDstAddressOffset(int64_t loopIdx, int64_t permSize, const int64_t mixedBase[],
+                                                  const int64_t loopNum[]);
+    __aicore__ inline void ProcessPerCore();
+    __aicore__ inline void CopyIn(int64_t loopIdx, int64_t inputBlockLen, int64_t inCutLoopSize, int64_t inUbFactor,
+                                  int64_t inTailFactor, int64_t permSize);
+    __aicore__ inline void CopyOut(int64_t loopIdx, int64_t loopSize[], int64_t loopSrcStride[],
+                                   int64_t loopDstStride[], int64_t permSize, const int64_t mixedBase[],
+                                   const int64_t loopNum[]);
+
+private:
+    int64_t blockIdx_;
+
+    // buffer
+    TQueBind<QuePosition::VECIN, QuePosition::VECOUT, 1> vecQue_;
+    GlobalTensor<T> inputGM_;
+    GlobalTensor<T> outputGM_;
+
+    // tiling params
+    const TransposeOpTilingData* tiling_;
+    int64_t reducedInputShape_[TRANSPOSE_MAX_AXIS_NUM] = {0, 0, 0, 0, 0, 0, 0, 0};
+    int64_t reducedOutputShape_[TRANSPOSE_MAX_AXIS_NUM] = {0, 0, 0, 0, 0, 0, 0, 0};
+    int64_t inUbInputShapeMain_[TRANSPOSE_MAX_AXIS_NUM] = {0, 0, 0, 0, 0, 0, 0, 0};
+    int64_t inUbOutputShapeMain_[TRANSPOSE_MAX_AXIS_NUM] = {0, 0, 0, 0, 0, 0, 0, 0};
+    int64_t inUbInputShapeTail_[TRANSPOSE_MAX_AXIS_NUM] = {0, 0, 0, 0, 0, 0, 0, 0};
+    int64_t inUbOutputShapeTail_[TRANSPOSE_MAX_AXIS_NUM] = {0, 0, 0, 0, 0, 0, 0, 0};
+    int64_t outputBlockLenMain_ = 0;
+    int64_t outputBlockLenTail_ = 0;
+
+    // core params
+    int64_t blkProcessNum_ = 0;
+    int64_t blkProcessIdxStart_ = 0;
+    int64_t blkProcessIdxEnd_ = 0;
+
+    // addressOffset params
+    int64_t dstAddressOffsetMixedBase_[TRANSPOSE_MAX_AXIS_NUM] = {0, 0, 0, 0, 0, 0, 0, 0};
+    int64_t dstLoopNumArray_[TRANSPOSE_MAX_AXIS_NUM] = {0, 0, 0, 0, 0, 0, 0, 0};
+
+    // dataCopy params
+    int64_t loopSizeMain_[TRANSPOSE_MAX_AXIS_NUM] = {1, 1, 1, 1, 1, 1, 1, 1};
+    int64_t loopSrcStrideMain_[TRANSPOSE_MAX_AXIS_NUM] = {0, 0, 0, 0, 0, 0, 0, 0};
+    int64_t loopDstStrideMain_[TRANSPOSE_MAX_AXIS_NUM] = {0, 0, 0, 0, 0, 0, 0, 0};
+    int64_t loopSizeTail_[TRANSPOSE_MAX_AXIS_NUM] = {1, 1, 1, 1, 1, 1, 1, 1};
+    int64_t loopSrcStrideTail_[TRANSPOSE_MAX_AXIS_NUM] = {0, 0, 0, 0, 0, 0, 0, 0};
+    int64_t loopDstStrideTail_[TRANSPOSE_MAX_AXIS_NUM] = {0, 0, 0, 0, 0, 0, 0, 0};
+};
+
+template <typename T>
+__aicore__ inline void TransposeNLast<T>::Init(GM_ADDR x, GM_ADDR y, const TransposeOpTilingData* tilingData,
+                                               TPipe* pipe)
+{
+    blockIdx_ = GetBlockIdx();
+    tiling_ = tilingData;
+    ParseTilingData();
+    inputGM_.SetGlobalBuffer((__gm__ T*)x);
+    outputGM_.SetGlobalBuffer((__gm__ T*)y);
+    pipe->InitBuffer(vecQue_, BUFFER_NUM, tiling_->ubSize / BUFFER_NUM);
+}
+
+template <typename T>
+__aicore__ inline void TransposeNLast<T>::Process()
+{
+    if (!ParseMultiCoreRange(blockIdx_, tiling_->realCoreNum, tiling_->blkFactor, tiling_->blkTailFactor,
+                             blkProcessNum_, blkProcessIdxStart_, blkProcessIdxEnd_)) {
+        return;
+    }
+    ProcessPerCore();
+}
+
+template <typename T>
+__aicore__ inline void TransposeNLast<T>::ParseTilingData()
+{
+    // input in ub shape main and tail
+    for (int64_t i = 0; i < tiling_->permSize; i++) {
+        reducedInputShape_[i] = tiling_->inputShape[i];
+        if (i > tiling_->inCutIndex) {
+            inUbInputShapeMain_[i] = reducedInputShape_[i];
+            inUbInputShapeTail_[i] = reducedInputShape_[i];
+        } else {
+            inUbInputShapeMain_[i] = 1;
+            inUbInputShapeTail_[i] = 1;
+        }
+    }
+    inUbInputShapeMain_[tiling_->inCutIndex] = tiling_->inUbFactor;
+    inUbInputShapeTail_[tiling_->inCutIndex] = tiling_->inTailFactor;
+
+    // output in ub shape main and tail
+    for (int64_t i = 0; i < tiling_->permSize; i++) {
+        reducedOutputShape_[i] = reducedInputShape_[tiling_->perm[i]];
+        inUbOutputShapeMain_[i] = inUbInputShapeMain_[tiling_->perm[i]];
+        inUbOutputShapeTail_[i] = inUbInputShapeTail_[tiling_->perm[i]];
+    }
+
+    outputBlockLenMain_ = inUbOutputShapeMain_[tiling_->permSize - 1];
+    outputBlockLenTail_ = inUbOutputShapeTail_[tiling_->permSize - 1];
+}
+
+template <typename T>
+__aicore__ inline void TransposeNLast<T>::GetBlockLoopNum()
+{
+    const int64_t permSize = tiling_->permSize;
+    int64_t dstLoopNumArrayTmp[TRANSPOSE_MAX_AXIS_NUM];
+    int64_t rightProducts[TRANSPOSE_MAX_AXIS_NUM + 1];
+    rightProducts[permSize] = 1;
+
+    for (int64_t i = permSize - 1; i >= 0; i--) {
+        rightProducts[i] = rightProducts[i + 1] * reducedOutputShape_[i];
+    }
+    for (int64_t i = 0; i < permSize; i++) {
+        dstAddressOffsetMixedBase_[i] = Ops::Base::CeilDiv(reducedInputShape_[i], inUbInputShapeMain_[i]);
+        dstLoopNumArrayTmp[i] = inUbOutputShapeMain_[i] * rightProducts[i + 1];
+    }
+    for (int64_t i = 0; i < permSize; i++) {
+        dstLoopNumArray_[tiling_->perm[i]] = dstLoopNumArrayTmp[i];
+    }
+}
+
+template <typename T>
+__aicore__ inline void TransposeNLast<T>::GetLoopParams(int64_t n)
+{
+    const int64_t permSize = tiling_->permSize;
+    const int64_t reverseIdx = permSize - 1 - n;
+    // main and tail loopSize
+    loopSizeMain_[n] = inUbInputShapeMain_[reverseIdx];
+    loopSizeTail_[n] = inUbInputShapeTail_[reverseIdx];
+    // main and tail last axis aligned to ub block size
+    int64_t alignedStrideMain = Ops::Base::CeilAlign(static_cast<int64_t>(loopSizeMain_[0] * sizeof(T)),
+                                                     BLOCK_SIZE_BYTE) /
+                                sizeof(T);
+    int64_t alignedStrideTail = Ops::Base::CeilAlign(static_cast<int64_t>(loopSizeTail_[0] * sizeof(T)),
+                                                     BLOCK_SIZE_BYTE) /
+                                sizeof(T);
+    // main loopSrcStride and loopDstStride
+    if (loopSizeMain_[n] != 1) {
+        loopSrcStrideMain_[n] = alignedStrideMain;
+        if (n == 0) {
+            loopSrcStrideMain_[n] = 1;
+        }
+        for (int64_t i = 1; i < n; i++) {
+            loopSrcStrideMain_[n] *= inUbInputShapeMain_[permSize - 1 - i];
+        }
+        loopDstStrideMain_[n] = 1;
+        for (int64_t i = permSize - 1; i >= 0; i--) {
+            if (tiling_->perm[i] != reverseIdx) {
+                loopDstStrideMain_[n] *= reducedOutputShape_[i];
+            } else {
+                break;
+            }
+        }
+    }
+    // tail loopSrcStride and loopDstStride
+    if (tiling_->inTailFactor != 0 && loopSizeTail_[n] != 1) {
+        loopSrcStrideTail_[n] = alignedStrideTail;
+        for (int64_t i = 1; i < n; i++) {
+            loopSrcStrideTail_[n] *= inUbInputShapeTail_[permSize - 1 - i];
+        }
+        loopDstStrideTail_[n] = 1;
+        for (int64_t i = permSize - 1; i >= 0; i--) {
+            if (tiling_->perm[i] != reverseIdx) {
+                loopDstStrideTail_[n] *= reducedOutputShape_[i];
+            } else {
+                break;
+            }
+        }
+    }
+}
+
+template <typename T>
+__aicore__ inline int64_t TransposeNLast<T>::GetDstAddressOffset(int64_t loopIdx, int64_t permSize,
+                                                                 const int64_t mixedBase[], const int64_t loopNum[])
+{
+    // Fuse DecimalToMixed + accumulation: no intermediate array, no zero-init overhead (P9)
+    int64_t dstAddressOffset = 0;
+    if (loopIdx != 0) {
+        int64_t num = loopIdx;
+        for (int64_t i = permSize - 1; i >= 0 && num != 0; i--) {
+            dstAddressOffset += (num % mixedBase[i]) * loopNum[i];
+            num /= mixedBase[i];
+        }
+    }
+    return dstAddressOffset;
+}
+
+template <typename T>
+__aicore__ inline void TransposeNLast<T>::CopyIn(int64_t loopIdx, int64_t inputBlockLen, int64_t inCutLoopSize,
+                                                 int64_t inUbFactor, int64_t inTailFactor, int64_t permSize)
+{
+    int64_t inputBlockLenMain = inputBlockLen * inUbFactor;
+    int64_t inputBlockLenTail = inputBlockLen * inTailFactor;
+    int64_t srcAddressOffset = loopIdx * inputBlockLenMain;
+    if (inTailFactor != 0) {
+        srcAddressOffset = (loopIdx - loopIdx / inCutLoopSize) * inputBlockLenMain +
+                           loopIdx / inCutLoopSize * inputBlockLenTail;
+    }
+    int64_t blockLen = inputBlockLenMain;
+    int64_t lastAxisLen = inUbInputShapeMain_[permSize - 1];
+    if (inTailFactor != 0 && (loopIdx + 1) % inCutLoopSize == 0) {
+        blockLen = inputBlockLenTail;
+        lastAxisLen = inUbInputShapeTail_[permSize - 1];
+    }
+    LocalTensor<T> bindLocalIn = vecQue_.AllocTensor<T>();
+    DataCopyExtParams copyInParams{1, 0, 0, 0, 0};
+    DataCopyPadExtParams<T> padParams{false, 0, 0, 0};
+
+    copyInParams.blockLen = blockLen * sizeof(T);
+    copyInParams.blockCount = 1;
+    if ((lastAxisLen * sizeof(T)) % BLOCK_SIZE_BYTE != 0) {
+        copyInParams.blockLen = lastAxisLen * sizeof(T);
+        copyInParams.blockCount = blockLen / lastAxisLen;
+    }
+    DataCopyPad(bindLocalIn, inputGM_[srcAddressOffset], copyInParams, padParams);
+    vecQue_.EnQue(bindLocalIn);
+}
+
+template <typename T>
+__aicore__ inline void TransposeNLast<T>::CopyOut(int64_t loopIdx, int64_t loopSize[], int64_t loopSrcStride[],
+                                                  int64_t loopDstStride[], int64_t permSize, const int64_t mixedBase[],
+                                                  const int64_t loopNum[])
+{
+    int64_t dstAddressOffset = GetDstAddressOffset(loopIdx, permSize, mixedBase, loopNum);
+
+    // P4: cache array elements into locals to avoid alias-induced reload after DataCopyPad stores
+    const int64_t ls4 = loopSize[4], ls5 = loopSize[5], ls6 = loopSize[6], ls7 = loopSize[7];
+    const int64_t lss4 = loopSrcStride[4], lss5 = loopSrcStride[5];
+    const int64_t lss6 = loopSrcStride[6], lss7 = loopSrcStride[7];
+    const int64_t lds4 = loopDstStride[4], lds5 = loopDstStride[5];
+    const int64_t lds6 = loopDstStride[6], lds7 = loopDstStride[7];
+
+    DataCopyExtParams copyOutParams{1, 0, 0, 0, 0};
+    copyOutParams.blockLen = loopSize[0] * sizeof(T);
+    copyOutParams.blockCount = loopSize[1];
+    copyOutParams.srcStride = 0;
+    copyOutParams.dstStride = copyOutParams.blockCount == 1 ? 0 : (loopDstStride[1] - loopSize[0]) * sizeof(T);
+
+    LoopModeParams loopParams;
+    loopParams.loop1Size = loopSize[2];
+    loopParams.loop1SrcStride = loopSrcStride[2] * sizeof(T);
+    loopParams.loop1DstStride = loopDstStride[2] * sizeof(T);
+    loopParams.loop2Size = loopSize[3];
+    loopParams.loop2SrcStride = loopSrcStride[3] * sizeof(T);
+    loopParams.loop2DstStride = loopDstStride[3] * sizeof(T);
+
+    LocalTensor<T> bindLocalOut = vecQue_.DeQue<T>();
+    // Move SetLoopModePara outside the loop: loopParams is constant across all iterations
+    SetLoopModePara(loopParams, DataCopyMVType::UB_TO_OUT);
+    for (int64_t loop7Idx = 0; loop7Idx < ls7; loop7Idx++) {
+        for (int64_t loop6Idx = 0; loop6Idx < ls6; loop6Idx++) {
+            for (int64_t loop5Idx = 0; loop5Idx < ls5; loop5Idx++) {
+                for (int64_t loop4Idx = 0; loop4Idx < ls4; loop4Idx++) {
+                    DataCopyPad(outputGM_[dstAddressOffset + loop7Idx * lds7 + loop6Idx * lds6 + loop5Idx * lds5 +
+                                          loop4Idx * lds4],
+                                bindLocalOut[loop7Idx * lss7 + loop6Idx * lss6 + loop5Idx * lss5 + loop4Idx * lss4],
+                                copyOutParams);
+                }
+            }
+        }
+    }
+    ResetLoopModePara(DataCopyMVType::UB_TO_OUT);
+    vecQue_.FreeTensor(bindLocalOut);
+}
+
+template <typename T>
+__aicore__ inline void TransposeNLast<T>::ProcessPerCore()
+{
+    // P6: cache tiling pointer fields into local variables to avoid repeated double-Load
+    const int64_t permSize = tiling_->permSize;
+    const int64_t inUbFactor = tiling_->inUbFactor;
+    const int64_t inTailFactor = tiling_->inTailFactor;
+    const int64_t inCutIndex = tiling_->inCutIndex;
+
+    for (int64_t i = 0; i < permSize; i++) {
+        GetLoopParams(i);
+    }
+    GetBlockLoopNum();
+    int64_t inputBlockLen = 1;
+    for (int64_t i = permSize - 1; i > inCutIndex; i--) {
+        inputBlockLen *= reducedInputShape_[i];
+    }
+    int64_t inCutLoopSize = Ops::Base::CeilDiv(reducedInputShape_[inCutIndex], inUbFactor);
+
+    // P4: cache member variables used only in this function
+    const int64_t loopStart = blkProcessIdxStart_;
+    const int64_t loopEnd = blkProcessIdxEnd_;
+
+    // P1: cache member arrays to stack to break alias pollution from struct dynamic indexing
+    int64_t localMixedBase[TRANSPOSE_MAX_AXIS_NUM];
+    int64_t localLoopNum[TRANSPOSE_MAX_AXIS_NUM];
+    for (int64_t i = 0; i < permSize; i++) {
+        localMixedBase[i] = dstAddressOffsetMixedBase_[i];
+        localLoopNum[i] = dstLoopNumArray_[i];
+    }
+
+    // P2: main/tail loop separation to eliminate per-iteration branch in hot path
+    for (int64_t loopIdx = loopStart; loopIdx < loopEnd; loopIdx++) {
+        if (inTailFactor != 0 && (loopIdx + 1) % inCutLoopSize == 0) {
+            CopyIn(loopIdx, inputBlockLen, inCutLoopSize, inUbFactor, inTailFactor, permSize);
+            CopyOut(loopIdx, loopSizeTail_, loopSrcStrideTail_, loopDstStrideTail_, permSize, localMixedBase,
+                    localLoopNum);
+        } else {
+            CopyIn(loopIdx, inputBlockLen, inCutLoopSize, inUbFactor, inTailFactor, permSize);
+            CopyOut(loopIdx, loopSizeMain_, loopSrcStrideMain_, loopDstStrideMain_, permSize, localMixedBase,
+                    localLoopNum);
+        }
+    }
+}
+} // namespace Transpose
+
+#endif // TRANSPOSE_N_LAST_H

package/skills/tool-ops-tune/ops-family/references/conversion/templates/dav3510/transpose_small_shape.template

@@ -0,0 +1,321 @@
+/**
+ * Copyright (c) 2025 Huawei Technologies Co., Ltd.
+ * This program is free software, you can redistribute it and/or modify it under the terms and conditions of
+ * CANN Open Software License Agreement Version 2.0 (the "License").
+ * Please refer to the License for details. You may not use this file except in compliance with the License.
+ * THIS SOFTWARE IS PROVIDED ON AN "AS IS" BASIS, WITHOUT WARRANTIES OF ANY KIND, EITHER EXPRESS OR IMPLIED,
+ * INCLUDING BUT NOT LIMITED TO NON-INFRINGEMENT, MERCHANTABILITY, OR FITNESS FOR A PARTICULAR PURPOSE.
+ * See LICENSE in the root of the software repository for the full text of the License.
+ */
+
+/*!
+ * \file transpose_small_shape.h
+ * \brief transpose_small_shape
+ */
+#ifndef TRANSPOSE_SMALL_SHAPE_H
+#define TRANSPOSE_SMALL_SHAPE_H
+
+#include "transpose_base.h"
+#include "simt_api/asc_simt.h"
+
+#ifdef __DAV_FPGA__
+constexpr int32_t THREAD_DIM = 512;
+constexpr int32_t HALF_THREAD_DIM = 512;
+constexpr int32_t QUARTER_THREAD_DIM = 512;
+constexpr int32_t AN_EIGHTH_THREAD_DIM = 256;
+#else
+constexpr int32_t THREAD_DIM = 2048;
+constexpr int32_t HALF_THREAD_DIM = 1024;
+constexpr int32_t QUARTER_THREAD_DIM = 512;
+constexpr int32_t AN_EIGHTH_THREAD_DIM = 256;
+#endif
+
+namespace Transpose {
+using namespace AscendC;
+
+template <typename T>
+class TransposeSmallShape : public TransposeBase<T> {
+public:
+    __aicore__ inline TransposeSmallShape(){};
+    __aicore__ inline void Init(GM_ADDR x, GM_ADDR y, const TransposeOpTilingData* tilingData);
+    __aicore__ inline void Process();
+
+private:
+    GlobalTensor<T> inputGM_;
+    GlobalTensor<T> outputGM_;
+    int32_t blockIdx_;
+    const TransposeOpTilingData* tilingData_;
+};
+
+template <typename T>
+__aicore__ inline void TransposeSmallShape<T>::Init(GM_ADDR x, GM_ADDR y, const TransposeOpTilingData* tilingData)
+{
+    blockIdx_ = GetBlockIdx();
+    tilingData_ = tilingData;
+    inputGM_.SetGlobalBuffer((__gm__ T*)x);
+    outputGM_.SetGlobalBuffer((__gm__ T*)y);
+}
+
+template <typename T>
+__simt_vf__ LAUNCH_BOUND(THREAD_DIM) __aicore__
+    void SimtComputeDimTwo(__gm__ T* inputGM, __gm__ volatile T* outputGM, uint32_t coreFactor, uint32_t coreOffset,
+                           uint32_t outputShape0, uint32_t outputShape1, uint32_t m0, uint32_t m1, uint32_t s0,
+                           uint32_t s1)
+{
+    for (uint32_t idx = static_cast<uint32_t>(threadIdx.x); idx < coreFactor;
+         idx += static_cast<uint32_t>(blockDim.x)) {
+        uint32_t yIdx = coreOffset + idx;
+        uint32_t inputIndex0 = yIdx - Simt::UintDiv(yIdx, m0, s0) * outputShape0;
+        yIdx = Simt::UintDiv(yIdx, m0, s0);
+        uint32_t inputIndex1 = yIdx - Simt::UintDiv(yIdx, m1, s1) * outputShape1;
+        uint32_t xIdx = inputIndex0 * outputShape1 + inputIndex1;
+        outputGM[coreOffset + idx] = inputGM[xIdx];
+    }
+}
+
+template <typename T>
+__simt_vf__ LAUNCH_BOUND(THREAD_DIM) __aicore__
+    void SimtComputeDimThree(__gm__ T* inputGM, __gm__ volatile T* outputGM, uint32_t coreFactor, uint32_t coreOffset,
+                             uint32_t outputShape0, uint32_t outputShape1, uint32_t outputShape2, uint32_t dstStride0,
+                             uint32_t dstStride1, uint32_t dstStride2, uint32_t m0, uint32_t m1, uint32_t m2,
+                             uint32_t s0, uint32_t s1, uint32_t s2)
+{
+    for (uint32_t idx = static_cast<uint32_t>(threadIdx.x); idx < coreFactor;
+         idx += static_cast<uint32_t>(blockDim.x)) {
+        uint32_t yIdx = coreOffset + idx;
+        uint32_t inputIndex0 = yIdx - Simt::UintDiv(yIdx, m0, s0) * outputShape0;
+        yIdx = Simt::UintDiv(yIdx, m0, s0);
+        uint32_t inputIndex1 = yIdx - Simt::UintDiv(yIdx, m1, s1) * outputShape1;
+        yIdx = Simt::UintDiv(yIdx, m1, s1);
+        uint32_t inputIndex2 = yIdx - Simt::UintDiv(yIdx, m2, s2) * outputShape2;
+        uint32_t xIdx = inputIndex0 * dstStride0 + inputIndex1 * dstStride1 + inputIndex2 * dstStride2;
+        outputGM[coreOffset + idx] = inputGM[xIdx];
+    }
+}
+
+template <typename T>
+__simt_vf__ LAUNCH_BOUND(THREAD_DIM) __aicore__
+    void SimtComputeDimFour(__gm__ T* inputGM, __gm__ volatile T* outputGM, uint32_t coreFactor, uint32_t coreOffset,
+                            uint32_t outputShape0, uint32_t outputShape1, uint32_t outputShape2, uint32_t outputShape3,
+                            uint32_t dstStride0, uint32_t dstStride1, uint32_t dstStride2, uint32_t dstStride3,
+                            uint32_t m0, uint32_t m1, uint32_t m2, uint32_t m3, uint32_t s0, uint32_t s1, uint32_t s2,
+                            uint32_t s3)
+{
+    for (uint32_t idx = static_cast<uint32_t>(threadIdx.x); idx < coreFactor;
+         idx += static_cast<uint32_t>(blockDim.x)) {
+        uint32_t yIdx = coreOffset + idx;
+        uint32_t inputIndex0 = yIdx - Simt::UintDiv(yIdx, m0, s0) * outputShape0;
+        yIdx = Simt::UintDiv(yIdx, m0, s0);
+        uint32_t inputIndex1 = yIdx - Simt::UintDiv(yIdx, m1, s1) * outputShape1;
+        yIdx = Simt::UintDiv(yIdx, m1, s1);
+        uint32_t inputIndex2 = yIdx - Simt::UintDiv(yIdx, m2, s2) * outputShape2;
+        yIdx = Simt::UintDiv(yIdx, m2, s2);
+        uint32_t inputIndex3 = yIdx - Simt::UintDiv(yIdx, m3, s3) * outputShape3;
+        uint32_t xIdx = inputIndex0 * dstStride0 + inputIndex1 * dstStride1 + inputIndex2 * dstStride2 +
+                        inputIndex3 * dstStride3;
+        outputGM[coreOffset + idx] = inputGM[xIdx];
+    }
+}
+
+template <typename T>
+__simt_vf__ LAUNCH_BOUND(THREAD_DIM) __aicore__
+    void SimtComputeDimFive(__gm__ T* inputGM, __gm__ volatile T* outputGM, uint32_t coreFactor, uint32_t coreOffset,
+                            uint32_t outputShape0, uint32_t outputShape1, uint32_t outputShape2, uint32_t outputShape3,
+                            uint32_t outputShape4, uint32_t dstStride0, uint32_t dstStride1, uint32_t dstStride2,
+                            uint32_t dstStride3, uint32_t dstStride4, uint32_t m0, uint32_t m1, uint32_t m2,
+                            uint32_t m3, uint32_t m4, uint32_t s0, uint32_t s1, uint32_t s2, uint32_t s3, uint32_t s4)
+{
+    for (uint32_t idx = static_cast<uint32_t>(threadIdx.x); idx < coreFactor;
+         idx += static_cast<uint32_t>(blockDim.x)) {
+        uint32_t yIdx = coreOffset + idx;
+        uint32_t inputIndex0 = yIdx - Simt::UintDiv(yIdx, m0, s0) * outputShape0;
+        yIdx = Simt::UintDiv(yIdx, m0, s0);
+        uint32_t inputIndex1 = yIdx - Simt::UintDiv(yIdx, m1, s1) * outputShape1;
+        yIdx = Simt::UintDiv(yIdx, m1, s1);
+        uint32_t inputIndex2 = yIdx - Simt::UintDiv(yIdx, m2, s2) * outputShape2;
+        yIdx = Simt::UintDiv(yIdx, m2, s2);
+        uint32_t inputIndex3 = yIdx - Simt::UintDiv(yIdx, m3, s3) * outputShape3;
+        yIdx = Simt::UintDiv(yIdx, m3, s3);
+        uint32_t inputIndex4 = yIdx - Simt::UintDiv(yIdx, m4, s4) * outputShape4;
+        uint32_t xIdx = inputIndex0 * dstStride0 + inputIndex1 * dstStride1 + inputIndex2 * dstStride2 +
+                        inputIndex3 * dstStride3 + inputIndex4 * dstStride4;
+        outputGM[coreOffset + idx] = inputGM[xIdx];
+    }
+}
+
+template <typename T>
+__simt_vf__ LAUNCH_BOUND(THREAD_DIM) __aicore__
+    void SimtComputeDimSix(__gm__ T* inputGM, __gm__ volatile T* outputGM, uint32_t coreFactor, uint32_t coreOffset,
+                           uint32_t outputShape0, uint32_t outputShape1, uint32_t outputShape2, uint32_t outputShape3,
+                           uint32_t outputShape4, uint32_t outputShape5, uint32_t dstStride0, uint32_t dstStride1,
+                           uint32_t dstStride2, uint32_t dstStride3, uint32_t dstStride4, uint32_t dstStride5,
+                           uint32_t m0, uint32_t m1, uint32_t m2, uint32_t m3, uint32_t m4, uint32_t m5, uint32_t s0,
+                           uint32_t s1, uint32_t s2, uint32_t s3, uint32_t s4, uint32_t s5)
+{
+    for (uint32_t idx = static_cast<uint32_t>(threadIdx.x); idx < coreFactor;
+         idx += static_cast<uint32_t>(blockDim.x)) {
+        uint32_t yIdx = coreOffset + idx;
+        uint32_t inputIndex0 = yIdx - Simt::UintDiv(yIdx, m0, s0) * outputShape0;
+        yIdx = Simt::UintDiv(yIdx, m0, s0);
+        uint32_t inputIndex1 = yIdx - Simt::UintDiv(yIdx, m1, s1) * outputShape1;
+        yIdx = Simt::UintDiv(yIdx, m1, s1);
+        uint32_t inputIndex2 = yIdx - Simt::UintDiv(yIdx, m2, s2) * outputShape2;
+        yIdx = Simt::UintDiv(yIdx, m2, s2);
+        uint32_t inputIndex3 = yIdx - Simt::UintDiv(yIdx, m3, s3) * outputShape3;
+        yIdx = Simt::UintDiv(yIdx, m3, s3);
+        uint32_t inputIndex4 = yIdx - Simt::UintDiv(yIdx, m4, s4) * outputShape4;
+        yIdx = Simt::UintDiv(yIdx, m4, s4);
+        uint32_t inputIndex5 = yIdx - Simt::UintDiv(yIdx, m5, s5) * outputShape5;
+        uint32_t xIdx = inputIndex0 * dstStride0 + inputIndex1 * dstStride1 + inputIndex2 * dstStride2 +
+                        inputIndex3 * dstStride3 + inputIndex4 * dstStride4 + inputIndex5 * dstStride5;
+        outputGM[coreOffset + idx] = inputGM[xIdx];
+    }
+}
+
+template <typename T>
+__simt_vf__ LAUNCH_BOUND(THREAD_DIM) __aicore__
+    void SimtComputeDimSeven(__gm__ T* inputGM, __gm__ volatile T* outputGM, uint32_t coreFactor, uint32_t coreOffset,
+                             uint32_t outputShape0, uint32_t outputShape1, uint32_t outputShape2, uint32_t outputShape3,
+                             uint32_t outputShape4, uint32_t outputShape5, uint32_t outputShape6, uint32_t dstStride0,
+                             uint32_t dstStride1, uint32_t dstStride2, uint32_t dstStride3, uint32_t dstStride4,
+                             uint32_t dstStride5, uint32_t dstStride6, uint32_t m0, uint32_t m1, uint32_t m2,
+                             uint32_t m3, uint32_t m4, uint32_t m5, uint32_t m6, uint32_t s0, uint32_t s1, uint32_t s2,
+                             uint32_t s3, uint32_t s4, uint32_t s5, uint32_t s6)
+{
+    for (uint32_t idx = static_cast<uint32_t>(threadIdx.x); idx < coreFactor;
+         idx += static_cast<uint32_t>(blockDim.x)) {
+        uint32_t yIdx = coreOffset + idx;
+        uint32_t inputIndex0 = yIdx - Simt::UintDiv(yIdx, m0, s0) * outputShape0;
+        yIdx = Simt::UintDiv(yIdx, m0, s0);
+        uint32_t inputIndex1 = yIdx - Simt::UintDiv(yIdx, m1, s1) * outputShape1;
+        yIdx = Simt::UintDiv(yIdx, m1, s1);
+        uint32_t inputIndex2 = yIdx - Simt::UintDiv(yIdx, m2, s2) * outputShape2;
+        yIdx = Simt::UintDiv(yIdx, m2, s2);
+        uint32_t inputIndex3 = yIdx - Simt::UintDiv(yIdx, m3, s3) * outputShape3;
+        yIdx = Simt::UintDiv(yIdx, m3, s3);
+        uint32_t inputIndex4 = yIdx - Simt::UintDiv(yIdx, m4, s4) * outputShape4;
+        yIdx = Simt::UintDiv(yIdx, m4, s4);
+        uint32_t inputIndex5 = yIdx - Simt::UintDiv(yIdx, m5, s5) * outputShape5;
+        yIdx = Simt::UintDiv(yIdx, m5, s5);
+        uint32_t inputIndex6 = yIdx - Simt::UintDiv(yIdx, m6, s6) * outputShape6;
+        uint32_t xIdx = inputIndex0 * dstStride0 + inputIndex1 * dstStride1 + inputIndex2 * dstStride2 +
+                        inputIndex3 * dstStride3 + inputIndex4 * dstStride4 + inputIndex5 * dstStride5 +
+                        inputIndex6 * dstStride6;
+        outputGM[coreOffset + idx] = inputGM[xIdx];
+    }
+}
+
+template <typename T>
+__simt_vf__ LAUNCH_BOUND(THREAD_DIM) __aicore__
+    void SimtComputeDimEight(__gm__ T* inputGM, __gm__ volatile T* outputGM, uint32_t coreFactor, uint32_t coreOffset,
+                             uint32_t outputShape0, uint32_t outputShape1, uint32_t outputShape2, uint32_t outputShape3,
+                             uint32_t outputShape4, uint32_t outputShape5, uint32_t outputShape6, uint32_t outputShape7,
+                             uint32_t dstStride0, uint32_t dstStride1, uint32_t dstStride2, uint32_t dstStride3,
+                             uint32_t dstStride4, uint32_t dstStride5, uint32_t dstStride6, uint32_t dstStride7,
+                             uint32_t m0, uint32_t m1, uint32_t m2, uint32_t m3, uint32_t m4, uint32_t m5, uint32_t m6,
+                             uint32_t m7, uint32_t s0, uint32_t s1, uint32_t s2, uint32_t s3, uint32_t s4, uint32_t s5,
+                             uint32_t s6, uint32_t s7)
+{
+    for (uint32_t idx = static_cast<uint32_t>(threadIdx.x); idx < coreFactor;
+         idx += static_cast<uint32_t>(blockDim.x)) {
+        uint32_t yIdx = coreOffset + idx;
+        uint32_t inputIndex0 = yIdx - Simt::UintDiv(yIdx, m0, s0) * outputShape0;
+        yIdx = Simt::UintDiv(yIdx, m0, s0);
+        uint32_t inputIndex1 = yIdx - Simt::UintDiv(yIdx, m1, s1) * outputShape1;
+        yIdx = Simt::UintDiv(yIdx, m1, s1);
+        uint32_t inputIndex2 = yIdx - Simt::UintDiv(yIdx, m2, s2) * outputShape2;
+        yIdx = Simt::UintDiv(yIdx, m2, s2);
+        uint32_t inputIndex3 = yIdx - Simt::UintDiv(yIdx, m3, s3) * outputShape3;
+        yIdx = Simt::UintDiv(yIdx, m3, s3);
+        uint32_t inputIndex4 = yIdx - Simt::UintDiv(yIdx, m4, s4) * outputShape4;
+        yIdx = Simt::UintDiv(yIdx, m4, s4);
+        uint32_t inputIndex5 = yIdx - Simt::UintDiv(yIdx, m5, s5) * outputShape5;
+        yIdx = Simt::UintDiv(yIdx, m5, s5);
+        uint32_t inputIndex6 = yIdx - Simt::UintDiv(yIdx, m6, s6) * outputShape6;
+        yIdx = Simt::UintDiv(yIdx, m6, s6);
+        uint32_t inputIndex7 = yIdx - Simt::UintDiv(yIdx, m7, s7) * outputShape7;
+        uint32_t xIdx = inputIndex0 * dstStride0 + inputIndex1 * dstStride1 + inputIndex2 * dstStride2 +
+                        inputIndex3 * dstStride3 + inputIndex4 * dstStride4 + inputIndex5 * dstStride5 +
+                        inputIndex6 * dstStride6 + inputIndex7 * dstStride7;
+        outputGM[coreOffset + idx] = inputGM[xIdx];
+    }
+}
+
+template <typename T>
+__aicore__ inline void TransposeSmallShape<T>::Process()
+{
+    if (blockIdx_ >= tilingData_->realCoreNum) {
+        return;
+    }
+    uint32_t blkProcessNum = tilingData_->blkFactor;
+    uint32_t blkStartOffset = blockIdx_ * tilingData_->blkFactor;
+    if (blockIdx_ == tilingData_->realCoreNum - 1 && tilingData_->blkTailFactor != 0) {
+        blkProcessNum = tilingData_->blkTailFactor;
+    }
+
+    uint32_t permSize = tilingData_->permSize;
+    uint32_t outputShape[8] = {0, 0, 0, 0, 0, 0, 0, 0};
+    uint32_t dstStride[8] = {1, 1, 1, 1, 1, 1, 1, 1};
+    uint32_t dstStrideTmp[8] = {1, 1, 1, 1, 1, 1, 1, 1};
+
+    for (uint32_t i = 1; i < permSize; i++) {
+        dstStrideTmp[i] = tilingData_->inputShape[permSize - i] * dstStrideTmp[i - 1];
+    }
+    for (uint32_t i = 0; i < permSize; i++) {
+        outputShape[i] = tilingData_->outputShape[permSize - 1 - i];
+        dstStride[i] = dstStrideTmp[permSize - 1 - tilingData_->perm[permSize - 1 - i]];
+    }
+
+    uint32_t shift[8];
+    uint32_t m[8];
+    for (uint32_t i = 0; i < permSize; i++) {
+        GetUintDivMagicAndShift(m[i], shift[i], outputShape[i]);
+    }
+
+    if (permSize == DIM_TWO) {
+        asc_vf_call<SimtComputeDimTwo<T>>(dim3(THREAD_DIM), (__gm__ T*)(inputGM_.GetPhyAddr()),
+                                          (__gm__ volatile T*)(outputGM_.GetPhyAddr()), blkProcessNum, blkStartOffset,
+                                          outputShape[0], outputShape[1], m[0], m[1], shift[0], shift[1]);
+    } else if (permSize == DIM_THREE) {
+        asc_vf_call<SimtComputeDimThree<T>>(dim3(THREAD_DIM), (__gm__ T*)(inputGM_.GetPhyAddr()),
+                                            (__gm__ volatile T*)(outputGM_.GetPhyAddr()), blkProcessNum, blkStartOffset,
+                                            outputShape[0], outputShape[1], outputShape[2], dstStride[0], dstStride[1],
+                                            dstStride[2], m[0], m[1], m[2], shift[0], shift[1], shift[2]);
+    } else if (permSize == DIM_FOUR) {
+        asc_vf_call<SimtComputeDimFour<T>>(
+            dim3(THREAD_DIM), (__gm__ T*)(inputGM_.GetPhyAddr()), (__gm__ volatile T*)(outputGM_.GetPhyAddr()),
+            blkProcessNum, blkStartOffset, outputShape[0], outputShape[1], outputShape[2], outputShape[3], dstStride[0],
+            dstStride[1], dstStride[2], dstStride[3], m[0], m[1], m[2], m[3], shift[0], shift[1], shift[2], shift[3]);
+    } else if (permSize == DIM_FIVE) {
+        asc_vf_call<SimtComputeDimFive<T>>(
+            dim3(THREAD_DIM), (__gm__ T*)(inputGM_.GetPhyAddr()), (__gm__ volatile T*)(outputGM_.GetPhyAddr()),
+            blkProcessNum, blkStartOffset, outputShape[0], outputShape[1], outputShape[2], outputShape[3],
+            outputShape[4], dstStride[0], dstStride[1], dstStride[2], dstStride[3], dstStride[4], m[0], m[1], m[2],
+            m[3], m[4], shift[0], shift[1], shift[2], shift[3], shift[4]);
+    } else if (permSize == DIM_SIX) {
+        asc_vf_call<SimtComputeDimSix<T>>(dim3(THREAD_DIM), (__gm__ T*)(inputGM_.GetPhyAddr()),
+                                          (__gm__ volatile T*)(outputGM_.GetPhyAddr()), blkProcessNum, blkStartOffset,
+                                          outputShape[0], outputShape[1], outputShape[2], outputShape[3],
+                                          outputShape[4], outputShape[5], dstStride[0], dstStride[1], dstStride[2],
+                                          dstStride[3], dstStride[4], dstStride[5], m[0], m[1], m[2], m[3], m[4], m[5],
+                                          shift[0], shift[1], shift[2], shift[3], shift[4], shift[5]);
+    } else if (permSize == DIM_SEVEN) {
+        asc_vf_call<SimtComputeDimSeven<T>>(
+            dim3(THREAD_DIM), (__gm__ T*)(inputGM_.GetPhyAddr()), (__gm__ volatile T*)(outputGM_.GetPhyAddr()),
+            blkProcessNum, blkStartOffset, outputShape[0], outputShape[1], outputShape[2], outputShape[3],
+            outputShape[4], outputShape[5], outputShape[6], dstStride[0], dstStride[1], dstStride[2], dstStride[3],
+            dstStride[4], dstStride[5], dstStride[6], m[0], m[1], m[2], m[3], m[4], m[5], m[6], shift[0], shift[1],
+            shift[2], shift[3], shift[4], shift[5], shift[6]);
+    } else if (permSize == DIM_EIGHT) {
+        asc_vf_call<SimtComputeDimEight<T>>(
+            dim3(THREAD_DIM), (__gm__ T*)(inputGM_.GetPhyAddr()), (__gm__ volatile T*)(outputGM_.GetPhyAddr()),
+            blkProcessNum, blkStartOffset, outputShape[0], outputShape[1], outputShape[2], outputShape[3],
+            outputShape[4], outputShape[5], outputShape[6], outputShape[7], dstStride[0], dstStride[1], dstStride[2],
+            dstStride[3], dstStride[4], dstStride[5], dstStride[6], dstStride[7], m[0], m[1], m[2], m[3], m[4], m[5],
+            m[6], m[7], shift[0], shift[1], shift[2], shift[3], shift[4], shift[5], shift[6], shift[7]);
+    }
+}
+
+} // namespace Transpose
+
+#endif // TRANSPOSE_SMALL_SHAPE_H