Этот коммит содержится в:
akolliasAMD
2024-07-10 13:32:09 -06:00
коммит произвёл GitHub
родитель 7e78641dc1
Коммит 63e4d76e23
3 изменённых файлов: 14 добавлений и 3 удалений
+5 -1
Просмотреть файл
@@ -50,7 +50,11 @@ extern __shared__ struct mscclShmemData mscclShmem;
inline __device__ static void barrier(int nthreads) {
#if defined(__HIP_PLATFORM_AMD__) || defined(__HCC__) || defined(__HIPCC__)
assert(nthreads == NCCL_MAX_NTHREADS);
__asm__ __volatile__("s_waitcnt vmcnt(0) lgkmcnt(0)\ns_barrier");
#ifdef __GFX12__
__asm__ __volatile__("s_waitcnt vmcnt(0) lgkmcnt(0)\ns_barrier_signal -1\ns_barrier_wait -1");
#else
__asm__ __volatile__("s_waitcnt vmcnt(0) lgkmcnt(0)\ns_barrier");
#endif
#else
asm volatile ("bar.sync %1, %0;" :: "r"(nthreads), "r"(15));
#endif