Print static_shapes settings value along with config for accurate repro (#649)

Author: yf225

helion/autotuner/base_search.py

@@ -135,7 +135,10 @@ def benchmark_function(self, config: Config, fn: CompiledConfig) -> float:
         except Exception as e:
             action = classify_triton_exception(e)
             if action == "raise":
-                raise exc.TritonError(f"{type(e).__qualname__}: {e}", config) from e
+                raise exc.TritonError(
+                    f"{type(e).__qualname__}: {e}",
+                    self.kernel.format_kernel_decorator(config, self.settings),
+                ) from e
             if action == "warn":
                 self.log.warning(format_triton_compile_failure(config, e))
             else:
@@ -182,9 +185,8 @@ def extract_launcher(
                 return PrecompileFuture.skip(self, config, True)
         except Exception:
             log.warning(
-                "Helion autotuner precompile error for config %r, settings %r",
-                config,
-                self.settings,
+                "Helion autotuner precompile error for %s",
+                self.kernel.format_kernel_decorator(config, self.settings),
                 exc_info=True,
             )
             raise
@@ -240,10 +242,11 @@ def autotune(self) -> Config:
         self.log.reset()
         best = self._autotune()
         end = time.perf_counter()
+        kernel_decorator = self.kernel.format_kernel_decorator(best, self.settings)
         self.log(
             f"Autotuning complete in {end - start:.1f}s after searching {self.counters['benchmark']} configs.\n"
             "One can hardcode the best config and skip autotuning with:\n"
-            f"    @helion.kernel(config={best!r})\n",
+            f"    {kernel_decorator}\n",
             level=logging.INFO + 5,
         )
         if self.settings.print_output_code:

helion/runtime/kernel.py

@@ -382,6 +382,12 @@ def configs(self) -> list[Config]:
         """
         return self.kernel.configs
 
+    def format_kernel_decorator(self, config: Config, settings: Settings) -> str:
+        """Return the @helion.kernel decorator snippet capturing configs and settings that influence Triton code generation."""
+        return (
+            f"@helion.kernel(config={config!r}, static_shapes={settings.static_shapes})"
+        )
+
     def to_triton_code(
         self, config: ConfigLike | None = None, emit_repro_caller: bool = False
     ) -> str:
@@ -432,9 +438,8 @@ def compile_config(
             module = PyCodeCache.load(triton_code)
         except Exception:
             log.warning(
-                "Helion compiler triton codegen error for config %r, settings %r",
-                config,
-                self.settings,
+                "Helion compiler triton codegen error for %s",
+                self.format_kernel_decorator(config, self.settings),
                 exc_info=True,
             )
             raise
@@ -563,7 +568,8 @@ def _implicit_config(self) -> Config | None:
             return configs[0]
         if len(configs) == 0 and self.kernel.settings.use_default_config:
             config = self.config_spec.default_config()
-            print(f"Using default config: {config}", file=sys.stderr)
+            kernel_decorator = self.format_kernel_decorator(config, self.settings)
+            print(f"Using default config: {kernel_decorator}", file=sys.stderr)
             return config
         return None