Merged
reducemin bugfix #3889
houyuyangAdd on Jul 2
reducemin bugfix #3889
Merged
Merged fromhouyuyang/asc-devkitminbugfixintocann/asc-devkit9.1.0
Total: 1 files changed+3-0
| @@ -454,6 +454,9 @@ __aicore__ inline void WholeReduceMinImpl(__ubuf__ T *dstLocal, __ubuf__ T *srcL | |||
| 454 | if (sizeof(T) == 2 && order == ReduceOrder::ORDER_ONLY_INDEX) { | 454 | if (sizeof(T) == 2 && order == ReduceOrder::ORDER_ONLY_INDEX) { |
| 455 | oneRepOffset = 2; | 455 | oneRepOffset = 2; |
| 456 | } | 456 | } |
| 457 | + | ||
| 458 | + ReduceTemplate<isSetMask, false, Reg::ReduceMin<T, Reg::MaskMergeMode::ZEROING, Reg::RegTensor<T>>>( | ||
| 459 | + dstLocal, srcLocal, repeat, dstRepStride, oneRepOffset, srcBlkStride, srcRepStride, maskReg, order); | ||
| 457 | } | 460 | } |
| 458 | 461 | ||
| 459 | template <typename T, bool isSetMask = true, typename U = T> | 462 | template <typename T, bool isSetMask = true, typename U = T> |