[UT] Start supporting multiple group calls and graphs (#1151)

* Start supporting multiple group calls UT
2024-04-25 11:11:16 -06:00
parent efe99057b0
commit 0ec41f1386
8 changed files with 566 additions and 240 deletions
@@ -34,7 +34,7 @@ namespace RcclUnitTesting
    {
      int const numProcesses = isMultiProcess ? totalRanks : 1;
      // Initialize communicators in non-blocking mode
-      testBed.InitComms(TestBed::GetDeviceIdsList(numProcesses, totalRanks), 1, useBlocking);
+      testBed.InitComms(TestBed::GetDeviceIdsList(numProcesses, totalRanks), 1, 1, 1, useBlocking);

      // Loop over various collective functions
      for (auto funcType : funcTypes)