From ce31407363d2723231de8a1ab4f5692596b7a4fe Mon Sep 17 00:00:00 2001
From: jiaxingli <43110891+li126com@users.noreply.github.com>
Date: Wed, 28 Feb 2024 11:27:32 +0800
Subject: [PATCH] feat(switch topology): add control switch (#55)

---
 internlm/initialize/launch.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/internlm/initialize/launch.py b/internlm/initialize/launch.py
index 443a53d6..946316c2 100644
--- a/internlm/initialize/launch.py
+++ b/internlm/initialize/launch.py
@@ -55,6 +55,10 @@ def get_default_parser():
     parser.add_argument("--backend", type=str, default="nccl", help="backend for distributed communication")
     parser.add_argument("--seed", type=int, default=1024)
     parser.add_argument("--profiling", default=False, action="store_true", help="enable/disable profiling.")
+    parser.add_argument("--enable_ali_topology", default=False, action="store_true", help="enable ali switch topology.")
+    parser.add_argument(
+        "--disable_volc_topology", default=False, action="store_true", help="disable volc switch topology."
+    )
     return parser