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

* Start supporting multiple group calls UT

test/SendRecvTests.cpp

@@ -16,6 +16,7 @@ namespace RcclUnitTesting
     std::vector<int>            const  numElements     = {1048576, 53327, 1024, 0};
     bool                        const  inPlace         = false;
     bool                        const  useManagedMem   = false;
+    int                         const  groupCallId     = 0;
 
     OptionalColArgs options;
     bool isCorrect = true;
@@ -42,11 +43,12 @@ namespace RcclUnitTesting
                                     numElements[numIdx],
                                     options,
                                     0,
+                                    groupCallId,
                                     sendRank);
           if (recvRank == 0)
           {
-            testBed.AllocateMem(inPlace, useManagedMem, 0, sendRank);
-            testBed.PrepareData(0, sendRank);
+            testBed.AllocateMem(inPlace, useManagedMem, groupCallId, 0, sendRank);
+            testBed.PrepareData(groupCallId, 0, sendRank);
           }
           if (recvRank  != sendRank)
           {
@@ -65,15 +67,16 @@ namespace RcclUnitTesting
                                       numElements[numIdx],
                                       options,
                                       0,
+                                      groupCallId,
                                       recvRank);
-            testBed.AllocateMem(inPlace, useManagedMem, 0, recvRank);
-            testBed.PrepareData(0, recvRank);
+            testBed.AllocateMem(inPlace, useManagedMem, groupCallId, 0, recvRank);
+            testBed.PrepareData(groupCallId, 0, recvRank);
             testBed.ExecuteCollectives({sendRank, recvRank});
-            testBed.ValidateResults(isCorrect, 0, recvRank);
-            testBed.DeallocateMem(0, recvRank);
+            testBed.ValidateResults(isCorrect, groupCallId, 0, recvRank);
+            testBed.DeallocateMem(groupCallId, 0, recvRank);
           }
         }
-        testBed.DeallocateMem(0, sendRank);
+        testBed.DeallocateMem(groupCallId, 0, sendRank);
       }
       testBed.DestroyComms();
     }