Unit Tests for testing channels (#1222)

test/AllReduceTests.cpp

@@ -105,23 +105,26 @@ namespace RcclUnitTesting
   TEST(AllReduce, Channels)
   {
     TestBed testBed;
-    if(testBed.ev.isGfx94) {
-      // Configuration
-      std::vector<ncclFunc_t>     const funcTypes       = {ncclCollAllReduce};
-      std::vector<ncclDataType_t> const dataTypes       = {ncclBfloat16, ncclHalf};
-      std::vector<ncclRedOp_t>    const redOps          = {ncclSum};
-      std::vector<int>            const roots           = {0};
-      std::vector<int>            const numElements     = {64 * 1024 * 1024, 1024};
-      std::vector<bool>           const inPlaceList     = {false};
-      std::vector<bool>           const managedMemList  = {false};
-      std::vector<bool>           const useHipGraphList = {false, true};
-      std::vector<char *>         const channelList     = {"56", "84", "112"}; 
-      for (auto channel : channelList) {
-        setenv("NCCL_MIN_NCHANNELS", channel, 1);
-        testBed.RunSimpleSweep(funcTypes, dataTypes, redOps, roots, numElements,
-                              inPlaceList, managedMemList, useHipGraphList);
-        testBed.Finalize();
-        unsetenv("NCCL_MIN_NCHANNELS");
+    if(testBed.ev.maxGpus >= 8) {
+      if(testBed.ev.isGfx94) {
+        // Configuration
+        std::vector<ncclFunc_t>     const funcTypes       = {ncclCollAllReduce};
+        std::vector<ncclDataType_t> const dataTypes       = {ncclBfloat16};
+        std::vector<ncclRedOp_t>    const redOps          = {ncclSum};
+        std::vector<int>            const roots           = {0};
+        std::vector<int>            const numElements     = {64 * 1024 * 1024, 1024};
+        std::vector<bool>           const inPlaceList     = {false};
+        std::vector<bool>           const managedMemList  = {false};
+        std::vector<bool>           const useHipGraphList = {false, true};
+        std::vector<char *>         const channelList     = {"84", "112"};
+        bool                        const enableSweep     = false; 
+        for (auto channel : channelList) {
+          setenv("NCCL_MIN_NCHANNELS", channel, 1);
+          testBed.RunSimpleSweep(funcTypes, dataTypes, redOps, roots, numElements,
+                                inPlaceList, managedMemList, useHipGraphList, enableSweep);
+          testBed.Finalize();
+          unsetenv("NCCL_MIN_NCHANNELS");
+        }
       }
     }
   }