AI/rocm-systems
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Merge commit '3d4813d99196bb349eccd50a925e2addc8f1622c' into develop

Browse Source
This commit is contained in:
Ameya Keshava Mallya
2026-01-21 20:28:14 +00:00
parent 12ab8df3bc 3d4813d991
commit 8d996cc05f
295 changed files with 27704 additions and 3800 deletions
Download Patch File Download Diff File Expand all files Collapse all files
projects/rccl/test/AllToAllVTests.cpp
+8 -8
View File
@@ -74,7 +74,7 @@ namespace RcclUnitTesting
}
}
TEST(AllToAllv, OutOfPlace)
TEST(AlltoAllv, OutOfPlace)
{
TestBed testBed;
@@ -100,7 +100,7 @@ namespace RcclUnitTesting
const std::vector<int>& gpuPriorityOrder = testBed.ev.GetGpuPriorityOrder();
testBed.InitComms(TestBed::GetDeviceIdsList(numProcesses, totalRanks, gpuPriorityOrder));
// Prepare AllToAllV options
// Prepare AlltoAllV options
std::vector<size_t> numInputElements;
std::vector<size_t> numOutputElements;
PrepareCounts(totalRanks, 256, options, numInputElements, numOutputElements, 40);
@@ -110,14 +110,14 @@ namespace RcclUnitTesting
if (testBed.ev.showNames)
{
std::string name = testBed.GetTestCaseName(totalRanks, isMultiProcess,
ncclCollAllToAllv, dataTypes[dataIdx],
ncclCollAlltoAllv, dataTypes[dataIdx],
ncclSum, -1, inPlace, useManagedMem, useHipGraph);
INFO("%s\n", name.c_str());
}
for (int rank = 0; rank < totalRanks; ++rank)
{
testBed.SetCollectiveArgs(ncclCollAllToAllv,
testBed.SetCollectiveArgs(ncclCollAlltoAllv,
dataTypes[dataIdx],
numInputElements[rank],
numOutputElements[rank],
@@ -138,7 +138,7 @@ namespace RcclUnitTesting
}
TEST(AllToAllv, OutOfPlaceGraph)
TEST(AlltoAllv, OutOfPlaceGraph)
{
TestBed testBed;
@@ -164,7 +164,7 @@ namespace RcclUnitTesting
const std::vector<int>& gpuPriorityOrder = testBed.ev.GetGpuPriorityOrder();
testBed.InitComms(TestBed::GetDeviceIdsList(numProcesses, totalRanks, gpuPriorityOrder));
// Prepare AllToAllV options
// Prepare AlltoAllV options
std::vector<size_t> numInputElements;
std::vector<size_t> numOutputElements;
PrepareCounts(totalRanks, 256, options, numInputElements, numOutputElements, 60);
@@ -174,14 +174,14 @@ namespace RcclUnitTesting
if (testBed.ev.showNames)
{
std::string name = testBed.GetTestCaseName(totalRanks, isMultiProcess,
ncclCollAllToAllv, dataTypes[dataIdx],
ncclCollAlltoAllv, dataTypes[dataIdx],
ncclSum, -1, inPlace, useManagedMem, useHipGraph);
INFO("%s\n", name.c_str());
}
for (int rank = 0; rank < totalRanks; ++rank)
{
testBed.SetCollectiveArgs(ncclCollAllToAllv,
testBed.SetCollectiveArgs(ncclCollAlltoAllv,
dataTypes[dataIdx],
numInputElements[rank],
numOutputElements[rank],