From 031ca47316ec01eb62b6162f48f34d81ac23f75d 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 14aa5f36..47926ecd 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.")