AI/rocm-systems
2
0
Garpu 0
Kode Masalah Tarik Permintaan Actions Packages Projects Rilis Wiki Kegiatan

Unit Test: support ncclAvg

Telusuri Sumber 
[ROCm/rccl commit: 1faff323b4]
This commit is contained in:
Wenkai Du
2021-08-25 12:01:49 -07:00
orang tua da7ac55e6f
melakukan bd03ec2a45
8 mengubah file dengan 80 tambahan dan 36 penghapusan
Download Patch File Download Diff File Expand all files Collapse all files
projects/rccl/test/test_AllReduce.cpp
+1 -1
Melihat File
@@ -40,7 +40,7 @@ namespace CorrectnessTests
AllReduceCorrectnessTest,
testing::Combine(
// Reduction operator
testing::Values(ncclSum, ncclProd, ncclMax, ncclMin),
testing::Values(ncclSum, ncclProd, ncclMax, ncclMin, ncclAvg),
// Data types
testing::Values(ncclInt8,
ncclUint8,