microsoft/mscclpp
1
0
Fork 0
mirror of https://github.com/microsoft/mscclpp.git synced 2026-05-21 13:29:45 +00:00
Code Issues Packages Projects Releases Wiki Activity

Relaxed sync

Browse Source
This commit is contained in:
Changho Hwang
2023-03-09 07:24:09 +00:00
parent 38a2adfe97
commit 6ac3c4c90f
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
tests/p2p_test.cu
View File

@@ -67,13 +67,13 @@ __global__ void kernel(int rank, int world_size)
// Each warp receives data from different ranks
#if (USE_DMA_FOR_P2P == 1)
// Wait until the proxy have sent my data and flag
while (*trig != 0) {}
// Trigger sending data and flag
uint64_t dataOffset = rank * sizeof(int);
uint64_t dataSize = sizeof(int);
*trig = TRIGGER_VALUE(mscclppSync | mscclppFlag | mscclppData, dataOffset, dataSize);
// Wait until the proxy have sent my data and flag
while (*trig != 0) {}
*trig = TRIGGER_VALUE(mscclppFlag | mscclppData, dataOffset, dataSize);
// Wait for receiving data from remote rank
while (*proxyFlag == baseFlag) {}