Insert barrier after loading work items to LDS (#1551)

[ROCm/rccl commit: baaa2ac64d]
这个提交包含在:
Wenkai Du 编写, GitHub 于 2025-02-18 10:17:27 -08:00 提交
1 父节点 35987b9170
当前提交 7eff149ceb
1 个文件已更改
+1 -1
+1 -1
查看文件
@@ -560,8 +560,8 @@ __device__ __forceinline__ void ncclKernelMain(struct ncclDevKernelArgs const* a
default:
break;
}
__synclds();
loadWorkBatchToShmem(tid%WARP_SIZE, tn, args, batchIx);
__synclds();
// Check whether the last operation was aborted and make sure all threads exit
bool aborted = false;