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

Unit Test: support ncclAvg

Telusuri Sumber
This commit is contained in:
Wenkai Du
2021-08-25 12:01:49 -07:00
orang tua 60ca7484c0
melakukan 1faff323b4
8 mengubah file dengan 80 tambahan dan 36 penghapusan
Download Patch File Download Diff File Expand all files Collapse all files
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,