Support bfloat16 on rest of the unit tests

Этот коммит содержится в:
Wenkai Du committed 2019-11-18 14:18:34 -08:00
1 родитель bdac0256a5
Коммит 4ca05c1297
8 files changed
+18 -6

No files matched your search

+2 -1
Просмотреть файл
@@ -57,7 +57,8 @@ namespace CorrectnessTests
ncclUint64,
//ncclFloat16,
ncclFloat32,
ncclFloat64),
ncclFloat64,
ncclBfloat16),
// Number of elements
testing::Values(3072, 3145728),
// Number of devices