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
@@ -59,7 +59,8 @@ namespace CorrectnessTests
                                                ncclUint64,
                                                //ncclFloat16,
                                                ncclFloat32,
-                                                ncclFloat64),
+                                                ncclFloat64,
+                                                ncclBfloat16),
                                // Number of elements
                                testing::Values(1024, 1048576),
                                // Number of devices