From 6c0c6867b0d87109dd87c1eb174622315dfa1500 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 277e9dbb..7a8b5a7c 100644 --- 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.")