Support bfloat16 on rest of the unit tests
Этот коммит содержится в:
1 родитель
bdac0256a5
Коммит
4ca05c1297
8 files changed
+18
-6
No files matched your search
@@ -57,7 +57,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