Unit Test: support ncclAvg

[ROCm/rccl commit: 1faff323b4]
2021-08-25 12:01:49 -07:00
parent da7ac55e6f
commit bd03ec2a45
8 changed files with 80 additions and 36 deletions
@@ -48,7 +48,7 @@ namespace CorrectnessTests
                             ReduceCorrectnessTest,
                             testing::Combine(
                                // Reduction operator
-                                testing::Values(ncclSum, ncclProd, ncclMax, ncclMin),
+                                testing::Values(ncclSum, ncclProd, ncclMax, ncclMin, ncclAvg),
                                // Data types
                                testing::Values(ncclInt8,
                                                ncclUint8,