Skip to content

Commit ec12380

Browse files
authored
[Autotuner] Better error message for default config error (#1092)
1 parent a3d8f44 commit ec12380

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

helion/autotuner/base_search.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -193,6 +193,8 @@ def _compute_baseline(self) -> tuple[object, bool, Sequence[object] | None]:
193193
"Default config failed while computing baseline.\n"
194194
f"Default config: {decorator}\n"
195195
f"{SUPPRESSED_TRITON_CODE_MSG}\n"
196+
"To work around this error, you could set `@helion.kernel(autotune_baseline_fn=...)` "
197+
"to provide a custom baseline function (e.g. PyTorch eager implementation of your kernel)."
196198
) from e
197199

198200
original_args_flat, _ = tree_flatten(self._original_args)

0 commit comments

Comments
 (0)