Update AlltoAll and AlltoAllv API for 2.28.3 compatibility (#161)

Signed-off-by: ravaidya <ravaidya@amd.com>
This commit is contained in:
Rahul Vaidya
2026-01-16 11:28:40 -08:00
zatwierdzone przez GitHub
rodzic 5272cd16ef
commit a52452e891
+6
Wyświetl plik
@@ -30,6 +30,12 @@
#define ncclFloat8e5m2 ncclFp8E5M2
#endif
// Forward compatibility for AlltoAll API changes in 2.28.3
#if NCCL_VERSION_CODE >= NCCL_VERSION(2,28,3)
#define ncclAllToAll ncclAlltoAll
#define ncclAllToAllv ncclAlltoAllv
#endif
// For nccl.h < 2.13 since we define a weak fallback
extern "C" char const* ncclGetLastError(ncclComm_t comm);