Revert "Remove unused code path (#1527)" (#1530)

This reverts commit 091bf899a1.
此提交包含在:
Wenkai Du
2025-02-04 13:14:43 -08:00
提交者 GitHub
父節點 3646b1de43
當前提交 d00e903d72
+2 -2
查看文件
@@ -369,7 +369,7 @@ private:
// worker perf is the limiter, perf-wise this loop is effectively unentered,
// hence just a single branch insn.
#pragma unroll 1
while (false) {
while (slice < SlicePerChunk) {
sliceSize = sliceSize < nelem-offset ? sliceSize : nelem-offset;
{ // Only workers could have Wait roles so we know the slice must be empty
// since we've exited the loop above.
@@ -942,4 +942,4 @@ private:
__device__ __forceinline__ void localCopy(T* srcs, T* dsts, int eltN) {
return mscclGenericOp<0,1,0,0>(&srcs, 1, &dsts, 1, eltN);
}
};
};