This document is relevant for: Trn2, Trn3

nki.compiler.SpecializationConfig#

class nki.compiler.SpecializationConfig[source]#

Per-kernel specialization-time config passed to @nki.jit(specialization_config=...).

Options that tune the specialization phase (turning the kernel into IR).

Methods

Attributes

allow_nested_config_override

When this kernel is the top-level compiled kernel, warn (instead of raising) if a nested @nki.jit sub-kernel's config disagrees, and let it inline under this kernel's config.
allow_nested_config_override: bool = Ellipsis#

When this kernel is the top-level compiled kernel, warn (instead of raising) if a nested @nki.jit sub-kernel’s config disagrees, and let it inline under this kernel’s config.

This document is relevant for: Trn2, Trn3