Print repro code on autotune success (#1203)

yf225 · web-flow · commit 1f2593c719d8 · 2025-12-03T18:48:10.000-08:00
diff --git a/helion/autotuner/base_search.py b/helion/autotuner/base_search.py
@@ -669,6 +669,7 @@ def autotune(self, *, skip_cache: bool = False) -> Config:
             f"    {kernel_decorator}\n",
             level=logging.INFO + 5,
         )
+        self.kernel.maybe_log_repro(self.log.warning, self.args, best)
         if self.settings.print_output_code:
             triton_code = self.kernel.to_triton_code(best)
             print(triton_code, file=sys.stderr)

Original file line number	Diff line number	Diff line change
`@@ -669,6 +669,7 @@ def autotune(self, *, skip_cache: bool = False) -> Config:`
`669`	`669`	`f" {kernel_decorator}\n",`
`670`	`670`	`level=logging.INFO + 5,`
`671`	`671`	`)`
	`672`	`+ self.kernel.maybe_log_repro(self.log.warning, self.args, best)`
`672`	`673`	`if self.settings.print_output_code:`
`673`	`674`	`triton_code = self.kernel.to_triton_code(best)`
`674`	`675`	`print(triton_code, file=sys.stderr)`