tilelang.tools.cuda.iket.frontend

Kernel-side IKET event helpers.

Classes

PayloadSpec

IKET payload value descriptor for a TileLang CUDA event.

Functions

reset()	Reset frontend event-id allocation for kernels built afterward.
event_table()	Return events registered while constructing recent kernels.
payload(expr[, dtype])	Attach a supported scalar payload value to an IKET event.
mark(name[, payload])	Emit an IKET instant marker at the current program point.
range_push(name[, payload])	Emit an IKET range-start event at the current program point.
range_pop([name])	Emit an IKET range-end event for the current warp-local range.
range_start(name[, payload])	Alias for range_push().
range_end([name])	Alias for range_pop().
range(name[, payload])	Return a scope that emits IKET range-start and range-end events.

Module Contents

class tilelang.tools.cuda.iket.frontend.PayloadSpec

IKET payload value descriptor for a TileLang CUDA event.

expr: Any

dtype: str

iket_id: int

tilelang.tools.cuda.iket.frontend.reset()

Reset frontend event-id allocation for kernels built afterward.

Return type:

None

tilelang.tools.cuda.iket.frontend.event_table()

Return events registered while constructing recent kernels.

Return type:

list[dict[str, int | str]]

tilelang.tools.cuda.iket.frontend.payload(expr, dtype=None)

Attach a supported scalar payload value to an IKET event.

Parameters:

• expr (Any)

• dtype (str | None)

Return type:

PayloadSpec

tilelang.tools.cuda.iket.frontend.mark(name, payload=None)

Emit an IKET instant marker at the current program point.

Parameters:

• name (str)

• payload (Any)

tilelang.tools.cuda.iket.frontend.range_push(name, payload=None)

Emit an IKET range-start event at the current program point.

Parameters:

• name (str)

• payload (Any)

tilelang.tools.cuda.iket.frontend.range_pop(name=None)

Emit an IKET range-end event for the current warp-local range.

Parameters:

name (str | None)

tilelang.tools.cuda.iket.frontend.range_start(name, payload=None)

Alias for range_push().

Parameters:

• name (str)

• payload (Any)

tilelang.tools.cuda.iket.frontend.range_end(name=None)

Alias for range_pop().

Parameters:

name (str | None)

tilelang.tools.cuda.iket.frontend.range(name, payload=None)

Return a scope that emits IKET range-start and range-end events.

Parameters:

• name (str)

• payload (Any)

Return type:

_RangeScope