Fix parser

pytorch-labs · Nov 21, 2024 · 75e29fc · 75e29fc
1 parent 2b6065d
commit 75e29fc
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/tritonbench/utils/parser.py b/tritonbench/utils/parser.py
@@ -159,6 +159,11 @@ def get_parser(args=None):
         action="store_true",
         help="bypass and continue on operator 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.")