Support bfloat16 on rest of the unit tests

2019-11-18 14:17:29 -08:00
parent bdac0256a5
commit 4ca05c1297
8 changed files with 18 additions and 6 deletions
@@ -101,7 +101,8 @@ namespace CorrectnessTests
                                                ncclUint64,
                                                //ncclFloat16,
                                                ncclFloat32,
-                                                ncclFloat64),
+                                                ncclFloat64,
+                                                ncclBfloat16),
                                // Number of elements
                                testing::Values(3072, 3145728),
                                // Number of devices