(cherry picked from commit 94da229a7788d74685d1591a4e75a8341de64f41)
This commit is contained in:
Wenkai Du
2023-07-21 07:31:27 -07:00
committato da GitHub
parent 59db7b8722
commit a7fcd58a97
15 ha cambiato i file con 159 aggiunte e 56 eliminazioni
+2
Vedi File
@@ -151,6 +151,8 @@ NodeModelDesc model_descs[] = {
{2, "topo_8p1h_4.xml", "2 nodes 8P1H Alt."},
{1, "topo_8p1h_5.xml", "Single node 8P1H Alt."},
{2, "topo_8p1h_5.xml", "2 nodes 8P1H Alt."},
{1, "topo_8p_940.xml", "Single node gfx940 8P"},
{2, "topo_4p_940.xml", "2 nodes gfx940 4P"},
};
NCCL_PARAM(MaxCTAs, "MAX_CTAS", MAXCHANNELS);