diff --git a/tritonbench/utils/parser.py b/tritonbench/utils/parser.py index 29723e60..acf164a7 100644 --- 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.")