Improve failing input error message (#543)

yf225 · web-flow · commit 3cce79171413 · 2025-10-10T18:43:18.000-04:00
diff --git a/tritonbench/utils/triton_op.py b/tritonbench/utils/triton_op.py
@@ -1142,7 +1142,7 @@ def _reduce_benchmarks(acc, bm_name: str):
             )
             if getattr(self, "_cur_input_id", None) is not None:
                 logger.warning(
-                    "Failing input: --input-id %s --num-inputs 1",
+                    "Failing input: --input-id %s --num-inputs 1 --input-sample-mode first-k",
                     self._cur_input_id,
                 )
             if self.tb_args.exit_on_exception:

Original file line number	Diff line number	Diff line change
`@@ -1142,7 +1142,7 @@ def _reduce_benchmarks(acc, bm_name: str):`
`1142`	`1142`	`)`
`1143`	`1143`	`if getattr(self, "_cur_input_id", None) is not None:`
`1144`	`1144`	`logger.warning(`
`1145`		`- "Failing input: --input-id %s --num-inputs 1",`
	`1145`	`+ "Failing input: --input-id %s --num-inputs 1 --input-sample-mode first-k",`
`1146`	`1146`	`self._cur_input_id,`
`1147`	`1147`	`)`
`1148`	`1148`	`if self.tb_args.exit_on_exception:`