Support bfloat16 on rest of the unit tests

[ROCm/rccl commit: 4ca05c1297]
2019-11-18 14:17:29 -08:00
parent 7dc39b8928
commit 55ad3c801e
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