Support bfloat16 on rest of the unit tests

Esse commit está contido em:
Wenkai Du
2019-11-18 14:17:29 -08:00
commit 4ca05c1297
8 arquivos alterados com 18 adições e 6 exclusões
+2 -1
Ver Arquivo
@@ -59,7 +59,8 @@ namespace CorrectnessTests
ncclUint64,
//ncclFloat16,
ncclFloat32,
ncclFloat64),
ncclFloat64,
ncclBfloat16),
// Number of elements
testing::Values(1024, 1048576),
// Number of devices