Skip to content

Commit ce31407

Browse files
authored
feat(switch topology): add control switch (#55)
1 parent bd40db1 commit ce31407

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

internlm/initialize/launch.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -55,6 +55,10 @@ def get_default_parser():
5555
parser.add_argument("--backend", type=str, default="nccl", help="backend for distributed communication")
5656
parser.add_argument("--seed", type=int, default=1024)
5757
parser.add_argument("--profiling", default=False, action="store_true", help="enable/disable profiling.")
58+
parser.add_argument("--enable_ali_topology", default=False, action="store_true", help="enable ali switch topology.")
59+
parser.add_argument(
60+
"--disable_volc_topology", default=False, action="store_true", help="disable volc switch topology."
61+
)
5862
return parser
5963

6064

0 commit comments

Comments
 (0)