[Device] WarpSpeed enablement and single node CU and perf opt for MI350 (#2073)

[ROCm/rccl commit: d009ab144e]
Tento commit je obsažen v:
Mustafa Abduljabbar
2025-12-11 19:04:35 -05:00
odevzdal GitHub
rodič f17357d0d4
revize 2621e0254e
22 změnil soubory, kde provedl 424 přidání a 65 odebrání
+3
Zobrazit soubor
@@ -146,6 +146,9 @@ __device__ __forceinline__ void mscclRunInterpreter(
}
if (bytes) copyToShmem8(tid%WARP_SIZE, dst, src, bytes);
}
#ifdef ENABLE_WARP_SPEED
ncclShmem.warpComm = 0;
#endif
__syncthreads(); // publish shmem
#if defined(ENABLE_NPKIT)