已合并
split_v偏移计算修正 #2865
split_v偏移计算修正 #2865
已合并
pengyiming7创建于 5月21日
1 个文件变更+1-1
@@ -478,8 +478,8 @@ __aicore__ inline void SplitVUbSplit<T, U, Y>::AllCopyOut(int64_t mOffset, int64
478 }478 }
479 if (ubOffset + mnAlignSize > ubSizeNum_) {479 if (ubOffset + mnAlignSize > ubSizeNum_) {
480 // 兜底分支,防止ub写越界480 // 兜底分支,防止ub写越界
481- curNFactor = curNFactor - (splitEndLine_ - curNOffset_);
482 splitEndLine_ = curNOffset_;481 splitEndLine_ = curNOffset_;
482+ curNFactor = splitEndLine_ - nOffset - nBlockOffset_;
483 break;483 break;
484 }484 }
485 if (curSplitNSizeAlign > maxGatherNum_) {485 if (curSplitNSizeAlign > maxGatherNum_) {