AI/rocm-systems
2
0
Tükrözés 0
Kód Hibajegyek Egyesítési kérések Actions Packages Projektek Kiadások Wiki Tevékenység

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

Forráskód böngészése 
* Start supporting multiple group calls UT
This commit is contained in:
Bertan Dogancay
2024-04-25 11:11:16 -06:00
committed by GitHub
szülő efe99057b0
commit 0ec41f1386
8 fájl változott, egészen pontosan 566 új sor hozzáadva és 240 régi sor törölve
Download Patch File Download Diff File Expand all files Collapse all files
test/SendRecvTests.cpp
+10 -7
Fájl megtekintése
@@ -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();
}