reverted the syncLDS back to syncthreads (#1554)

Этот коммит содержится в:
akolliasAMD
2025-02-19 10:44:32 -07:00
коммит произвёл GitHub
родитель baaa2ac64d
Коммит aedbc95735
2 изменённых файлов: 11 добавлений и 19 удалений
+2 -2
Просмотреть файл
@@ -108,7 +108,7 @@ __device__ __forceinline__ void mscclRunInterpreter(
threadBlockCopy(
(uint32_t *)&mscclShmem.mscclTB, (uint32_t *)(algo->mscclTBs + bid),
sizeof(struct mscclThreadBlock) / sizeof(uint32_t), tid, nthreads);
__synclds(); // publish mscclShmem.mscclTB.channelId
__syncthreads(); // publish mscclShmem.mscclTB.channelId
// initialize ncclShmem and mscclShmem.work
int channelId = mscclShmem.mscclTB.channelId;
@@ -146,7 +146,7 @@ __device__ __forceinline__ void mscclRunInterpreter(
}
if (bytes) copyToShmem8(tid%WARP_SIZE, dst, src, bytes);
}
__synclds(); // publish shmem
__syncthreads(); // publish shmem
#if defined(ENABLE_NPKIT)
int npKitCtxIdx = bid;