custom all reduce support cuda graph (#2938)

* Support enabling cuda graph and custom all reduce at the same time, and fix the overwritten custom all reduce flag * rename communication_op to communication
2026-04-23 17:11:21 +08:00 · 2025-07-21 22:52:03 +08:00
parent ff4569f135
commit 0262ef7eb3
21 changed files with 88 additions and 51 deletions
@@ -498,7 +498,7 @@ def parse_args():
        help="enable prefix cache",
    )
    parser.add_argument(
-        "--enable-custom-all-reduce",
+        "--enable_custom_all_reduce",
        action="store_true",
        help="enable custom all-reduce",
    )