Unit Test: support ncclAvg

[ROCm/rccl commit: 1faff323b4]
2021-08-25 12:01:49 -07:00
@@ -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,