Fix parser

pytorch-labs · Nov 23, 2024 · 031ca47 · 031ca47
1 parent ed4b558
commit 031ca47
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.")