Preprocessor symbol RCCL_BFLOAT16 is used as feature indicator
Этот коммит содержится в:
Wenkai Du
2019-11-15 10:39:48 -08:00
родитель 58a6e535f6
Коммит 5e109ed400
10 изменённых файлов: 370 добавлений и 9 удалений
+1
Просмотреть файл
@@ -48,6 +48,7 @@ static __inline__ int ncclTypeSize(ncclDataType_t type) {
case ncclUint8:
return 1;
case ncclFloat16:
case ncclBfloat16:
return 2;
case ncclInt32:
case ncclUint32: