Rate this Page

Runtime Module#

The helion.runtime module handles kernel execution and configuration management.

Key Classes#

The runtime module provides the core execution infrastructure. For detailed documentation of individual classes, see:

Utility Functions#

helion.runtime.get_num_sm(device, *, reserved_sms=0)[source]#

Number of SMs (persistent-kernel grid size) for any Helion device.

Adds the CPU (Pallas-interpret) and TPU cases on top of the dependency-free GPU helper helion.runtime.triton.launcher.get_num_sm(). See that function for argument/return semantics.

Parameters:
Return type:

int

helion.runtime.set_triton_allocator()[source]#
Return type:

None