Support bfloat16 on rest of the unit tests

[ROCm/rccl commit: 4ca05c1297]
This commit is contained in:
Wenkai Du committed 2019-11-18 14:18:34 -08:00
1 parent 7dc39b8928
commit 55ad3c801e
8 files changed
+18 -6

No files matched your search

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