gfx12 initial enablement (#1219)

[ROCm/rccl commit: 63e4d76e23]
Bu işleme şunda yer alıyor:
akolliasAMD
2024-07-10 13:32:09 -06:00
işlemeyi yapan: GitHub
ebeveyn 9644767ead
işleme 942555fd21
3 değiştirilmiş dosya ile 14 ekleme ve 3 silme
+5 -1
Dosyayı Görüntüle
@@ -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