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); } } diff --git a/tests/activation/softmax/test_operator_softmax_v220.cpp b/tests/activation/softmax/test_operator_softmax_v220.cpp index 371524f14ad81a3a7ba96fcc0bd9805f298b953e..609062321002c6e8164dfa200af05dc31abeb525 100644 --- a/tests/activation/softmax/test_operator_softmax_v220.cpp +++ b/tests/activation/softmax/test_operator_softmax_v220.cpp @@ -1,4 +1,4 @@ -/** +** * Copyright (c) 2024 Huawei Technologies Co., Ltd. * This file is a part of the CANN Open Software. * Licensed under CANN Open Software License Agreement Version 1.0 (the "License").