add reduce/broadcast algo/proto selection table for multi-node gfx940 (#1889)

2025-09-10 14:25:23 -05:00
parent c2bccf9156
commit 9c36439354
4 changed files with 9 additions and 3 deletions
@@ -291,7 +291,9 @@ int main(int argc,char* argv[])
  std::vector<ncclFunc_t> ncclFuncTypes = {
    ncclFuncAllReduce,
    ncclFuncReduceScatter,
-    ncclFuncAllGather
+    ncclFuncAllGather,
+    ncclFuncReduce,
+    ncclFuncBroadcast
  };

  std::cout << "Running fp32 production choices for algorithm/protocol/maxChannels" << std::endl;