Add support for new datatype: bfloat16

Этот коммит содержится в:
David Addison
2021-03-15 14:44:06 -07:00
родитель 0b30de583f
Коммит e37545e491
2 изменённых файлов: 40 добавлений и 4 удалений
+3
Просмотреть файл
@@ -213,6 +213,9 @@ static size_t wordSize(ncclDataType_t type) {
#endif
return 1;
case ncclHalf:
#if defined(__CUDA_BF16_TYPES_EXIST__)
case ncclBfloat16:
#endif
//case ncclFloat16:
return 2;
case ncclInt: