From 0e39cd208d11d2ee4a68e71127608f1488153104 Mon Sep 17 00:00:00 2001 From: Xu Zhao Date: Wed, 6 Nov 2024 11:12:37 -0500 Subject: [PATCH] Fix parser --- tritonbench/utils/parser.py | 5 +++++ 1 file changed, 5 insertions(+) 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.")