File tree Expand file tree Collapse file tree 2 files changed +2
-8
lines changed Expand file tree Collapse file tree 2 files changed +2
-8
lines changed Original file line number Diff line number Diff 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
Original file line number Diff line number Diff 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
You can’t perform that action at this time.
0 commit comments