tilelang.contrib.ptodsl.mixed_kernel

PTODSL compatibility helpers for mixed Cube/Vector kernels.

Attributes

MixedKernelRole

Functions

mixed_kernel_section(role)	Emit one PTO physical section in the active kernel entry.
finalize_mixed_kernel(kernel)	Select PTODSL's flat module layout before the first specialization.

Module Contents

tilelang.contrib.ptodsl.mixed_kernel.MixedKernelRole

tilelang.contrib.ptodsl.mixed_kernel.mixed_kernel_section(role)

Emit one PTO physical section in the active kernel entry.

The public context records the physical core while tracing. Helpers such as pto.init_core() use that context to specialize their Cube and Vector implementations, while the emitted section remains a direct child of the entry function for vpto-split-cv-module.

Parameters:

role (MixedKernelRole)

Return type:

collections.abc.Iterator[None]

tilelang.contrib.ptodsl.mixed_kernel.finalize_mixed_kernel(kernel)

Select PTODSL’s flat module layout before the first specialization.