Skip to content

Commit aa5ecdc

Browse files
authored
Allow specifiying the use of persistent kernel
Differential Revision: D86579911 Pull Request resolved: #654
1 parent 9e859f6 commit aa5ecdc

File tree

1 file changed

+8
-0
lines changed

1 file changed

+8
-0
lines changed

tritonbench/operators/fp8_gemm_rowwise/operator.py

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -49,6 +49,14 @@ def parse_args(args: List[str]) -> argparse.Namespace:
4949
action="store_true",
5050
)
5151
parser.add_argument("--warp_specialization", action="store_true")
52+
53+
# Validate that both --use_persistent and --no_use_persistent are not specified together
54+
if "--use_persistent" in args and "--no_use_persistent" in args:
55+
parser.error(
56+
"Cannot specify both '--use_persistent' and '--no_use_persistent' at the same time. "
57+
"These options are mutually exclusive. Please use only one."
58+
)
59+
5260
parsed_args = parser.parse_args(args)
5361
if parsed_args.use_tma is None:
5462
# Default to True for CUDA, False for ROCm

0 commit comments

Comments
 (0)