Support bfloat16 on rest of the unit tests

Этот коммит содержится в:
Wenkai Du
2019-11-18 14:17:29 -08:00
родитель bdac0256a5
Коммит 4ca05c1297
8 изменённых файлов: 18 добавлений и 6 удалений
+2 -1
Просмотреть файл
@@ -57,7 +57,8 @@ namespace CorrectnessTests
ncclUint64,
//ncclFloat16,
ncclFloat32,
ncclFloat64),
ncclFloat64,
ncclBfloat16),
// Number of elements
testing::Values(3072, 3145728),
// Number of devices