From a52452e891d5dc07c83cf4edaea01ae4ab684b3a Mon Sep 17 00:00:00 2001
From: Rahul Vaidya <ravaidya@amd.com>
Date: Fri, 16 Jan 2026 11:28:40 -0800
Subject: [PATCH] Update AlltoAll and AlltoAllv API for 2.28.3 compatibility
 (#161)

Signed-off-by: ravaidya <ravaidya@amd.com>
---
 src/common.h | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/src/common.h b/src/common.h
index 4c6af89f1e..01d6251f41 100644
--- a/src/common.h
+++ b/src/common.h
@@ -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);