From d119c0886efcd174491ca85e795c0696ff5ef759 Mon Sep 17 00:00:00 2001
From: akolliasAMD <99202231+akolliasAMD@users.noreply.github.com>
Date: Tue, 21 Feb 2023 16:21:24 -0700
Subject: [PATCH] UnitTests: made reduceScatter run a smaller amount of tests
 (#691)

---
 test/ReduceScatter_InPlace.cpp    | 6 +++---
 test/ReduceScatter_ManagedMem.cpp | 2 +-
 test/ReduceScatter_OutOfPlace.cpp | 6 +++---
 3 files changed, 7 insertions(+), 7 deletions(-)

diff --git a/test/ReduceScatter_InPlace.cpp b/test/ReduceScatter_InPlace.cpp
index 2db1d46662..649b2bf524 100644
--- a/test/ReduceScatter_InPlace.cpp
+++ b/test/ReduceScatter_InPlace.cpp
@@ -13,10 +13,10 @@ namespace RcclUnitTesting
 
     // Configuration
     std::vector<ncclFunc_t>     const funcTypes       = {ncclCollReduceScatter};
-    std::vector<ncclDataType_t> const dataTypes       = {ncclInt8, ncclInt32, ncclInt64, ncclFloat16};
-    std::vector<ncclRedOp_t>    const redOps          = {ncclSum, ncclProd};
+    std::vector<ncclDataType_t> const dataTypes       = {ncclInt8, ncclInt32, ncclInt64};
+    std::vector<ncclRedOp_t>    const redOps          = {ncclProd};
     std::vector<int>            const roots           = {0};
-    std::vector<int>            const numElements     = {1048576, 53327, 5461, 1024};
+    std::vector<int>            const numElements     = {1048576, 53327, 1024};;
     std::vector<bool>           const inPlaceList     = {true};
     std::vector<bool>           const managedMemList  = {false};
     std::vector<bool>           const useHipGraphList = {false, true};
diff --git a/test/ReduceScatter_ManagedMem.cpp b/test/ReduceScatter_ManagedMem.cpp
index 0b9ce80c8b..87659a98ee 100644
--- a/test/ReduceScatter_ManagedMem.cpp
+++ b/test/ReduceScatter_ManagedMem.cpp
@@ -14,7 +14,7 @@ namespace RcclUnitTesting
     // Configuration
     std::vector<ncclFunc_t>     const funcTypes       = {ncclCollReduceScatter};
     std::vector<ncclDataType_t> const dataTypes       = {ncclUint8, ncclUint32, ncclUint64};
-    std::vector<ncclRedOp_t>    const redOps          = {ncclSum};
+    std::vector<ncclRedOp_t>    const redOps          = {ncclAvg};
     std::vector<int>            const roots           = {0};
     std::vector<int>            const numElements     = {1048576, 53327, 1024};
     std::vector<bool>           const inPlaceList     = {false};
diff --git a/test/ReduceScatter_OutOfPlace.cpp b/test/ReduceScatter_OutOfPlace.cpp
index b674dc68d8..714f7c3db9 100644
--- a/test/ReduceScatter_OutOfPlace.cpp
+++ b/test/ReduceScatter_OutOfPlace.cpp
@@ -13,10 +13,10 @@ namespace RcclUnitTesting
 
     // Configuration
     std::vector<ncclFunc_t>     const funcTypes       = {ncclCollReduceScatter};
-    std::vector<ncclDataType_t> const dataTypes       = {ncclFloat16, ncclFloat32, ncclFloat64, ncclBfloat16};
-    std::vector<ncclRedOp_t>    const redOps          = {ncclMin, ncclMax, ncclAvg};
+    std::vector<ncclDataType_t> const dataTypes       = {ncclFloat32, ncclFloat64, ncclBfloat16};
+    std::vector<ncclRedOp_t>    const redOps          = {ncclMax};
     std::vector<int>            const roots           = {0};
-    std::vector<int>            const numElements     = {1048576, 53327, 5461, 1024};
+    std::vector<int>            const numElements     = {1048576, 53327, 1024};
     std::vector<bool>           const inPlaceList     = {false};
     std::vector<bool>           const managedMemList  = {false};
     std::vector<bool>           const useHipGraphList = {false, true};