diff --git a/impl/activation/softmax/v220/softmax_impl.h b/impl/activation/softmax/v220/softmax_impl.h index ceff15413924a033d22946321c2b3eb343805be2..10d33f8ea27e6a33d986decd0b4ce0ac8f50e57d 100644 --- a/impl/activation/softmax/v220/softmax_impl.h +++ b/impl/activation/softmax/v220/softmax_impl.h @@ -784,7 +784,7 @@ __aicore__ inline void SoftMaxNDImpl(const LocalTensor& dst, const LocalTe splitK = AlignUp(config.oriSrcK, FLOAT_NUM_PER_BLK); } ReduceLastND reduceParam = { tiling.splitM, config.oriSrcK, tiling.splitM, splitK, tiling.reduceM, - DEFAULT_REPEAT_STRIDE * HALF_FACTOR }; + DEFAULT_REPEAT_STRIDE }; SoftMaxNDExtImpl(dst, sumTensor, maxTensor, src, workLocal, originalSrcShape, tiling, reduceParam); } }