Skip to content

Commit 0a4a7fa

Browse files
authored
Avoid setting default --input-sample-mode to equally-spaced-k (#922)
1 parent 6ad53ad commit 0a4a7fa

File tree

2 files changed

+2
-8
lines changed

2 files changed

+2
-8
lines changed

.github/workflows/benchmark.yml

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -150,6 +150,7 @@ jobs:
150150
--only $IMPLS \
151151
--only-match-mode prefix-with-baseline \
152152
--baseline $BASELINE \
153+
--input-sample-mode equally-spaced-k \
153154
--keep-going
154155
155156
# Relax the GPU
@@ -164,6 +165,7 @@ jobs:
164165
--only $IMPLS \
165166
--only-match-mode prefix-with-baseline \
166167
--baseline $BASELINE \
168+
--input-sample-mode equally-spaced-k \
167169
--output "$TEST_REPORTS_DIR/helionbench.json" \
168170
--append-to-output \
169171
--keep-going

benchmarks/run.py

Lines changed: 0 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -814,14 +814,6 @@ def run_kernel_variants(
814814
file=sys.stderr,
815815
)
816816

817-
# Use equally-spaced-k input sample mode if not otherwise specified
818-
if "--input-sample-mode" not in tritonbench_args:
819-
tritonbench_args.extend(["--input-sample-mode", "equally-spaced-k"])
820-
print(
821-
f"Using input-sample-mode=equally-spaced-k for {operator_name}",
822-
file=sys.stderr,
823-
)
824-
825817
# Parse known args and collect unknown ones for operator
826818
tb_args, unknown_args = tb_parser.parse_known_args(tritonbench_args)
827819

0 commit comments

Comments
 (0)