Update AlltoAll and AlltoAllv API for 2.28.3 compatibility (#161)
Signed-off-by: ravaidya <ravaidya@amd.com>
[ROCm/rccl-tests commit: a52452e891]
This commit is contained in:
@@ -30,6 +30,12 @@
|
|||||||
#define ncclFloat8e5m2 ncclFp8E5M2
|
#define ncclFloat8e5m2 ncclFp8E5M2
|
||||||
#endif
|
#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
|
// For nccl.h < 2.13 since we define a weak fallback
|
||||||
extern "C" char const* ncclGetLastError(ncclComm_t comm);
|
extern "C" char const* ncclGetLastError(ncclComm_t comm);
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user