Skip to content

Commit b5403a4

Browse files
authored
Verify compiled kernels in subprocess (#914)
This is to handle configs that hang when we run them. It should also fix the IMA issues we have been seeing.
1 parent b77301f commit b5403a4

File tree

10 files changed

+551
-143
lines changed

10 files changed

+551
-143
lines changed

docs/api/settings.md

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -125,7 +125,12 @@ with helion.set_default_settings(
125125
126126
.. autoattribute:: Settings.autotune_precompile
127127
128-
Whether to precompile kernels before autotuning. Default is ``True`` on non-Windows systems, ``False`` on Windows.
128+
Select the autotuner precompile mode, which adds parallelism and
129+
checks for errors/timeouts. ``"spawn"`` (default) runs kernel
130+
warm-up in a fresh process including running to check for errors,
131+
``"fork"`` is faster but does not include the error check run,
132+
or None to disables precompile checks altogether. Controlled by
133+
``HELION_AUTOTUNE_PRECOMPILE``.
129134
130135
.. autoattribute:: Settings.autotune_random_seed
131136

0 commit comments

Comments
 (0)