DeepSeek/DeepEP
1
0
Fork 0
mirror of https://github.com/deepseek-ai/DeepEP synced 2025-06-26 18:28:11 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Remove memory fence in NVLink barrier.

Browse Source
This commit is contained in:
Shangyan Zhou
2025-06-24 17:51:28 +08:00
parent a15faa9ff0
commit f1d7a7c89f
2 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
csrc/kernels/utils.cuh
View File

@@ -446,7 +446,6 @@ barrier_block(int** barrier_signal_ptrs, int rank) {
// Add self-ranks, sub other ranks
if (thread_id < kNumRanks) {
atomicAdd_system(barrier_signal_ptrs[rank] + thread_id, FINISHED_SUM_TAG);
memory_fence();
atomicSub_system(barrier_signal_ptrs[thread_id] + rank, FINISHED_SUM_TAG);
}
EP_DEVICE_ASSERT(kNumRanks <= blockDim.x);