From 0b2e1a62d9e657e5e5ca36d37b0907f9f4856188 Mon Sep 17 00:00:00 2001 From: Will Feng Date: Thu, 6 Nov 2025 13:36:36 -0800 Subject: [PATCH] [Autotuner] Better error message for default config error --- helion/autotuner/base_search.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/helion/autotuner/base_search.py b/helion/autotuner/base_search.py index a9e841f68..b6d3ab4b0 100644 --- a/helion/autotuner/base_search.py +++ b/helion/autotuner/base_search.py @@ -193,6 +193,8 @@ def _compute_baseline(self) -> tuple[object, bool, Sequence[object] | None]: "Default config failed while computing baseline.\n" f"Default config: {decorator}\n" f"{SUPPRESSED_TRITON_CODE_MSG}\n" + "To work around this error, you could set `@helion.kernel(autotune_baseline_fn=...)` " + "to provide a custom baseline function (e.g. PyTorch eager implementation of your kernel)." ) from e original_args_flat, _ = tree_flatten(self._original_args)