Add bfloat16 all reduce unit test

2019-11-16 21:37:20 -08:00
parent 5e109ed400
commit bdac0256a5
3 changed files with 14 additions and 1 deletions
@@ -50,7 +50,8 @@ namespace CorrectnessTests
                                                ncclUint64,
                                                //ncclFloat16,
                                                ncclFloat32,
-                                                ncclFloat64),
+                                                ncclFloat64,
+                                                ncclBfloat16),
                                // Number of elements
                                testing::Values(1024, 1048576),
                                // Number of devices