[DEV] Configure functions in RCCL (#986)

* configure functions in rccl

test/CMakeLists.txt

@@ -37,10 +37,18 @@ if(BUILD_TESTS)
     )
 
   # Collect source files for tests
-  if(BUILD_ALLREDUCE_ONLY)
-    set(TEST_SOURCE_FILES
-      AllReduce_Tests.cpp
-    )
+  if(ONLY_FUNCS)
+    # Convert input string to a list
+    string(REPLACE "|" ";" CONFIG_LIST ${ONLY_FUNCS})
+
+    # For each config in config list
+    foreach(item ${CONFIG_LIST})
+      string(REPLACE " " ";" CONFIG_PARAMS ${item})
+      list(GET CONFIG_PARAMS 0 COLL)
+
+      set(TEST_FILE "${COLL}Tests.cpp")
+      list(APPEND TEST_SOURCE_FILES ${TEST_FILE})
+    endforeach()
   else()
     set(TEST_SOURCE_FILES
       AllGatherTests.cpp

test/common/EnvVars.cpp

@@ -65,12 +65,9 @@ namespace RcclUnitTesting
     useInteractive = GetEnvVar("UT_INTERACTIVE",  0);
     timeoutUs      = GetEnvVar("UT_TIMEOUT_US" ,  5000000);
 
-    // Limit number of supported reduction operators to just ncclSum if only allReduce is built
-#ifdef BUILD_ALLREDUCE_ONLY
-    int numOps = 1;
-#else
+    // Total number of reduction ops
     int numOps = ncclNumOps;
-#endif
+
     std::vector<std::string> redOpStrings = GetEnvVarsList("UT_REDOPS");
     for (auto s : redOpStrings)
     {
@@ -98,12 +95,7 @@ namespace RcclUnitTesting
       {
         if (!strcmp(s.c_str(), ncclDataTypeNames[i]))
         {
-#ifdef BUILD_ALLREDUCE_ONLY
-          if (i == ncclFloat32)
-#endif
-          {
-            dataTypes.push_back((ncclDataType_t)i);
-          }
+          dataTypes.push_back((ncclDataType_t)i);
         }
       }
     }
@@ -112,8 +104,6 @@ namespace RcclUnitTesting
     if (dataTypes.empty())
     {
       dataTypes.push_back(ncclFloat32);
-      // Skip all but 32-bit floats if only AllReduce is being built
-#ifndef BUILD_ALLREDUCE_ONLY
       dataTypes.push_back(ncclInt8);
       dataTypes.push_back(ncclUint8);
       dataTypes.push_back(ncclInt32);
@@ -124,7 +114,6 @@ namespace RcclUnitTesting
       dataTypes.push_back(ncclFloat32);
       dataTypes.push_back(ncclFloat64);
       dataTypes.push_back(ncclBfloat16);
-#endif
     }
 
     // Build list of possible # GPU ranks based on env vars