Add new GPU model (#1080)

This commit is contained in:
Wenkai Du
2024-02-23 12:19:42 -08:00
committed by GitHub
parent c5ab37211b
commit 74f9e5db64
2 changed files with 55 additions and 5 deletions
+1
View File
@@ -156,6 +156,7 @@ NodeModelDesc model_descs[] = {
{4, "topo_8p_940.xml", "4 nodes gfx940 8P"},
{1, "topo_8p_940vm.xml", "single node gfx940 VM"},
{2, "topo_8p_940vm.xml", "2 nodes gfx940 VM"},
{2, "topo_8p_940_16n.xml", "2 nodes gfx940 16 NICs"},
};
NCCL_PARAM(MaxCTAs, "MAX_CTAS", MAXCHANNELS);