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

* Start supporting multiple group calls UT [ROCm/rccl commit: 0ec41f1386]
2024-04-25 11:11:16 -06:00
parent 9a203f439c
commit dea5e83940
8 changed files with 566 additions and 240 deletions
@@ -98,6 +98,7 @@ namespace RcclUnitTesting
                                    numOutputElements[rank],
                                    options,
                                    -1,
+                                    0,
                                    rank);
        }
        testBed.AllocateMem(inPlace, useManagedMem);
@@ -154,6 +155,7 @@ namespace RcclUnitTesting
                                    numOutputElements[rank],
                                    options,
                                    -1,
+                                    0,
                                    rank);
        }
        testBed.AllocateMem(inPlace, useManagedMem);