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

此提交包含在:
gilbertlee-amd authored and GitHub committed 2022-03-02 17:24:14 -07:00
1 父節點 ff54e79799
當前提交 699dc30f05
3 file changed
+9 -4

No files matched your search

+2
查看文件
@@ -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));