[UnitTests] Check process mask for custom tests (#507)

[ROCm/rccl commit: 699dc30f05]
2022-03-02 17:24:14 -07:00
parent 2419a950fe
commit c802f53282
3 changed files with 9 additions and 4 deletions
@@ -26,6 +26,8 @@ namespace RcclUnitTesting
    for (int totalRanks = testBed.ev.minGpus; totalRanks <= testBed.ev.maxGpus && isCorrect; ++totalRanks)
    for (int isMultiProcess = 0; isMultiProcess <= 1; ++isMultiProcess)
    {
+      if (!(testBed.ev.processMask & (1 << isMultiProcess))) continue;
+
      int const numProcesses = isMultiProcess ? totalRanks : 1;
      testBed.InitComms(TestBed::GetDeviceIdsList(numProcesses, totalRanks));