Unit Test: support ncclAvg

[ROCm/rccl commit: 1faff323b4]
Tento commit je obsažen v:
Wenkai Du
2021-08-25 12:01:49 -07:00
rodič da7ac55e6f
revize bd03ec2a45
8 změnil soubory, kde provedl 80 přidání a 36 odebrání
+1 -1
Zobrazit soubor
@@ -46,7 +46,7 @@ namespace CorrectnessTests
ReduceScatterCorrectnessTest,
testing::Combine(
// Reduction operator
testing::Values(ncclSum, ncclProd, ncclMax, ncclMin),
testing::Values(ncclSum, ncclProd, ncclMax, ncclMin, ncclAvg),
// Data types
testing::Values(ncclInt8,
ncclUint8,