Support bfloat16 on rest of the unit tests

[ROCm/rccl commit: 4ca05c1297]

projects/rccl/test/test_AllGather.cpp

@@ -101,7 +101,8 @@ namespace CorrectnessTests
                                                 ncclUint64,
                                                 //ncclFloat16,
                                                 ncclFloat32,
-                                                ncclFloat64),
+                                                ncclFloat64,
+                                                ncclBfloat16),
                                 // Number of elements
                                 testing::Values(3072, 3145728),
                                 // Number of devices