Unit Tests for testing channels (#1222)

[ROCm/rccl commit: e170f41ddd]
This commit is contained in:
saurabhAMD authored and GitHub committed 2024-06-25 10:10:10 -05:00
1 parent ebcac26530
commit de7ea612d7
5 files changed
+53 -38

No files matched your search

+5 -1
View File
@@ -616,7 +616,8 @@ namespace RcclUnitTesting
std::vector<int> const& numElements,
std::vector<bool> const& inPlaceList,
std::vector<bool> const& managedMemList,
std::vector<bool> const& useHipGraphList)
std::vector<bool> const& useHipGraphList,
bool const& enableSweep)
{
// Sort numElements in descending order to cut down on # of allocations
std::vector<int> sortedN = numElements;
@@ -662,6 +663,9 @@ namespace RcclUnitTesting
// Test either single process all GPUs, or 1 process per GPU
int const numChildren = isMultiProcess ? numGpus : 1;
int const numRanks = numGpus*ranksPerGpu;
if(enableSweep == false && (numGpus < 8 || numRanks < 8)) {
continue;
}
this->InitComms(TestBed::GetDeviceIdsList(numChildren, numGpus, ranksPerGpu));
if (testing::Test::HasFailure())
{