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.
162
+
Provide JSON via ``HELION_AUTOTUNE_CONFIG_OVERRIDES='{"num_warps": 4}'`` for global overrides.
153
163
154
164
.. autoattribute:: Settings.autotune_effort
155
165
@@ -183,6 +193,7 @@ See :class:`helion.autotuner.LocalAutotuneCache` for details on cache keys and b
183
193
.. autoattribute:: Settings.ignore_warnings
184
194
185
195
List of warning types to suppress during compilation. Default is an empty list.
196
+
Accepts comma-separated warning class names from ``helion.exc`` via ``HELION_IGNORE_WARNINGS`` (for example, ``HELION_IGNORE_WARNINGS=TensorOperationInWrapper``).
186
197
187
198
.. autoattribute:: Settings.debug_dtype_asserts
188
199
@@ -207,6 +218,7 @@ See :class:`helion.autotuner.LocalAutotuneCache` for details on cache keys and b
207
218
.. autoattribute:: Settings.autotuner_fn
208
219
209
220
Override the callable that constructs autotuner instances. Accepts the same signature as :func:`helion.runtime.settings.default_autotuner_fn`.
221
+
Pass a replacement callable via ``@helion.kernel(..., autotuner_fn=...)`` or ``helion.kernel(autotuner_fn=...)`` at definition time.
210
222
```
211
223
212
224
Built-in values for ``HELION_AUTOTUNER`` include ``"PatternSearch"``, ``"DifferentialEvolutionSearch"``, ``"FiniteSearch"``, and ``"RandomSearch"``.
@@ -222,9 +234,12 @@ Built-in values for ``HELION_AUTOTUNER`` include ``"PatternSearch"``, ``"Differe
0 commit comments