You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
The data type used for index variables in generated code. Default is ``torch.int32``.
84
+
Override via ``HELION_INDEX_DTYPE=int64`` (or any ``torch.<dtype>`` name).
77
85
78
86
.. autoattribute:: Settings.dot_precision
79
87
80
88
Precision mode for dot product operations. Default is ``"tf32"``. Controlled by ``TRITON_F32_DEFAULT`` environment variable.
81
89
82
90
.. autoattribute:: Settings.static_shapes
83
91
84
-
When enabled, tensor shapes are treated as compile-time constants for optimization. Default is ``True``. Set this to ``False`` if you need a single compiled kernel instance to serve many shape variants.
92
+
When enabled, tensor shapes are treated as compile-time constants for optimization. Default is ``True``.
93
+
Set ``HELION_STATIC_SHAPES=0`` the default if you need a compiled kernel instance to serve many shape variants.
Dict of config key/value pairs to force during autotuning. Useful for disabling problematic candidates or pinning experimental options.
158
+
Provide JSON via ``HELION_AUTOTUNE_CONFIG_OVERRIDES='{"num_warps": 4}'`` for global overrides.
149
159
150
160
.. autoattribute:: Settings.autotune_effort
151
161
@@ -179,6 +189,7 @@ See :class:`helion.autotuner.LocalAutotuneCache` for details on cache keys and b
179
189
.. autoattribute:: Settings.ignore_warnings
180
190
181
191
List of warning types to suppress during compilation. Default is an empty list.
192
+
Accepts comma-separated warning class names from ``helion.exc`` via ``HELION_IGNORE_WARNINGS`` (for example, ``HELION_IGNORE_WARNINGS=TensorOperationInWrapper``).
182
193
183
194
.. autoattribute:: Settings.debug_dtype_asserts
184
195
@@ -203,6 +214,7 @@ See :class:`helion.autotuner.LocalAutotuneCache` for details on cache keys and b
203
214
.. autoattribute:: Settings.autotuner_fn
204
215
205
216
Override the callable that constructs autotuner instances. Accepts the same signature as :func:`helion.runtime.settings.default_autotuner_fn`.
217
+
Pass a replacement callable via ``@helion.kernel(..., autotuner_fn=...)`` or ``helion.kernel(autotuner_fn=...)`` at definition time.
206
218
```
207
219
208
220
Built-in values for ``HELION_AUTOTUNER`` include ``"PatternSearch"``, ``"DifferentialEvolutionSearch"``, ``"FiniteSearch"``, and ``"RandomSearch"``.
@@ -218,9 +230,12 @@ Built-in values for ``HELION_AUTOTUNER`` include ``"PatternSearch"``, ``"Differe
0 commit comments