Enable gfx94x (#808) (#816)

(cherry picked from commit 94da229a7788d74685d1591a4e75a8341de64f41)
2023-07-21 07:31:27 -07:00
parent 59db7b8722
commit a7fcd58a97
15 changed files with 159 additions and 56 deletions
@@ -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);