Support bfloat16 on rest of the unit tests
[ROCm/rccl commit: 4ca05c1297]
This commit is contained in:
1 parent
7dc39b8928
commit
55ad3c801e
8 files changed
+18
-6
No files matched your search
@@ -110,7 +110,8 @@ namespace CorrectnessTests
|
||||
ncclUint64,
|
||||
//ncclFloat16,
|
||||
ncclFloat32,
|
||||
ncclFloat64),
|
||||
ncclFloat64,
|
||||
ncclBfloat16),
|
||||
// Number of elements
|
||||
testing::Values(3072, 3145728),
|
||||
// Number of devices
|
||||
|
||||
Reference in new issue
Block a user