Unit Test: support ncclAvg

2021-08-25 12:01:49 -07:00
parent 60ca7484c0
commit 1faff323b4
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,