Update AlltoAll and AlltoAllv API for 2.28.3 compatibility (#161)
Signed-off-by: ravaidya <ravaidya@amd.com>
This commit is contained in:
zatwierdzone przez
GitHub
rodzic
5272cd16ef
commit
a52452e891
@@ -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);
|
||||
|
||||
|
||||
Reference in New Issue
Block a user