Support bfloat16 on rest of the unit tests

Αυτή η υποβολή περιλαμβάνεται σε:
Wenkai Du
2019-11-18 14:17:29 -08:00
γονέας bdac0256a5
υποβολή 4ca05c1297
8 αρχεία άλλαξαν με 18 προσθήκες και 6 διαγραφές
@@ -89,7 +89,8 @@ namespace CorrectnessTests
ncclUint64,
//ncclFloat16,
ncclFloat32,
ncclFloat64),
ncclFloat64,
ncclBfloat16),
// Number of elements
testing::Values(3072, 3145728),
// Number of devices