Fix parser

pytorch-labs · Nov 6, 2024 · 0e39cd2 · 0e39cd2
1 parent e5c2e2c
commit 0e39cd2
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/tritonbench/utils/parser.py b/tritonbench/utils/parser.py
@@ -154,6 +154,11 @@ def get_parser(args=None):
         action="store_true",
         help="Run each operator in a separate child process. By default, it will always continue on failure.",
     )
+    parser.add_argument(
+        "--child",
+        action="store_true",
+        help="Flag option that it is running in the child process.",
+    )
 
     if IS_FBCODE:
         parser.add_argument("--log-scuba", action="store_true", help="Log to scuba.")