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

Add another Rome model (#1354)

Browse Source
This commit is contained in:
Wenkai Du authored and GitHub committed 2024-10-01 17:41:27 -05:00
1 parent 7830af5844
commit e453f1ced9
3 files changed
+19 -1
Download Patch File Download Diff File Expand all files Collapse all files

No files matched your search

tools/topo_expl/topo_expl.cpp
+1
View File
@@ -160,6 +160,7 @@ NodeModelDesc model_descs[] = {
{2, "topo_8p1h_6.xml", "2 nodes 8P1H Alt."},
{5, "topo_8p_940.xml", "5 nodes gfx940 8P"},
{2, "topo_8p_942.xml", "2 nodes gfx942 8P"},
{2, "topo_8p_942_1.xml", "2 nodes gfx942 8P Alt."},
};
NCCL_PARAM(MaxCTAs, "MAX_CTAS", MAXCHANNELS);