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/AllToAllTests.cpp
+10 -10
View File
@@ -10,12 +10,12 @@
namespace RcclUnitTesting
{
TEST(AllToAll, OutOfPlace)
TEST(AlltoAll, OutOfPlace)
{
TestBed testBed;
// Configuration
std::vector<ncclFunc_t> const funcTypes = {ncclCollAllToAll};
std::vector<ncclFunc_t> const funcTypes = {ncclCollAlltoAll};
std::vector<ncclDataType_t> const dataTypes = {ncclFloat16, ncclFloat32};
std::vector<ncclRedOp_t> const redOps = {ncclSum};
std::vector<int> const roots = {0};
@@ -29,12 +29,12 @@ namespace RcclUnitTesting
testBed.Finalize();
}
TEST(AllToAll, OutOfPlaceGraph)
TEST(AlltoAll, OutOfPlaceGraph)
{
TestBed testBed;
// Configuration
std::vector<ncclFunc_t> const funcTypes = {ncclCollAllToAll};
std::vector<ncclFunc_t> const funcTypes = {ncclCollAlltoAll};
std::vector<ncclDataType_t> const dataTypes = {ncclFloat64, ncclBfloat16, ncclFloat8e4m3, ncclFloat8e5m2};
std::vector<ncclRedOp_t> const redOps = {ncclSum};
std::vector<int> const roots = {0};
@@ -48,12 +48,12 @@ namespace RcclUnitTesting
testBed.Finalize();
}
TEST(AllToAll, ManagedMem)
TEST(AlltoAll, ManagedMem)
{
TestBed testBed;
// Configuration
std::vector<ncclFunc_t> const funcTypes = {ncclCollAllToAll};
std::vector<ncclFunc_t> const funcTypes = {ncclCollAlltoAll};
std::vector<ncclDataType_t> const dataTypes = {ncclUint8};
std::vector<ncclRedOp_t> const redOps = {ncclSum};
std::vector<int> const roots = {0};
@@ -67,12 +67,12 @@ namespace RcclUnitTesting
testBed.Finalize();
}
TEST(AllToAll, ManagedMemGraph)
TEST(AlltoAll, ManagedMemGraph)
{
TestBed testBed;
// Configuration
std::vector<ncclFunc_t> const funcTypes = {ncclCollAllToAll};
std::vector<ncclFunc_t> const funcTypes = {ncclCollAlltoAll};
std::vector<ncclDataType_t> const dataTypes = {ncclUint32, ncclUint64};
std::vector<ncclRedOp_t> const redOps = {ncclSum};
std::vector<int> const roots = {0};
@@ -86,13 +86,13 @@ namespace RcclUnitTesting
testBed.Finalize();
}
TEST(AllToAll, Channels)
TEST(AlltoAll, Channels)
{
TestBed testBed;
if(testBed.ev.maxGpus >= 8) {
if(testBed.ev.isGfx94) {
// Configuration
std::vector<ncclFunc_t> const funcTypes = {ncclCollAllToAll};
std::vector<ncclFunc_t> const funcTypes = {ncclCollAlltoAll};
std::vector<ncclDataType_t> const dataTypes = {ncclBfloat16};
std::vector<ncclRedOp_t> const redOps = {ncclSum};
std::vector<int> const roots = {0};