Merge remote-tracking branch 'nccl/v2.19' into develop

Этот коммит содержится в:
BertanDogancay
2024-01-24 15:25:33 -08:00
родитель 18a57bac10 8c6c595185
Коммит 81ddf9de89
111 изменённых файлов: 4608 добавлений и 2075 удалений
-23
Просмотреть файл
@@ -31,29 +31,6 @@
ret func(args)
#endif // end PROFAPI
static __inline__ int ncclTypeSize(ncclDataType_t type) {
switch (type) {
case ncclInt8:
case ncclUint8:
return 1;
case ncclFloat16:
#if defined(RCCL_BFLOAT16)
case ncclBfloat16:
#endif
return 2;
case ncclInt32:
case ncclUint32:
case ncclFloat32:
return 4;
case ncclInt64:
case ncclUint64:
case ncclFloat64:
return 8;
default:
return -1;
}
}
#include "debug.h"
#include "checks.h"
#include "rocmwrap.h"