Add another Rome model (#1354)

[ROCm/rccl commit: e453f1ced9]
This commit is contained in:
Wenkai Du
2024-10-01 15:41:27 -07:00
committed by GitHub
parent cf980e9b9c
commit 74aa13afbe
3 changed files with 19 additions and 1 deletions
+1 -1
View File
@@ -21,7 +21,7 @@
DIR="$(cd -P "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
for i in {0..87}
for i in {0..88}
do
if [[ $i -eq 50 ]] || [[ $i -eq 51 ]]
then
@@ -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);