tilelang.tools.cuda.iket.metadata

Self-contained IKET event metadata carried through TIR.

Attributes

TOKEN_PREFIX
MAX_EVENT_NAME_BYTES

Classes

Event

Metadata for one IKET event emitted by a TileLang kernel.

Functions

encode_event(event)	Encode an event as a source-safe token stored in a TIR StringImm.
decode_event(token)	Decode and validate an event token recovered from generated CUDA.

Module Contents

tilelang.tools.cuda.iket.metadata.TOKEN_PREFIX = '__tl_iket_v1_'

tilelang.tools.cuda.iket.metadata.MAX_EVENT_NAME_BYTES = 32

class tilelang.tools.cuda.iket.metadata.Event

Metadata for one IKET event emitted by a TileLang kernel.

name: str

event_id: int

kind: str

range_id: int = 0

payload_type: str = 'NoPayload'

payload_iket_id: int = 0

tilelang.tools.cuda.iket.metadata.encode_event(event)

Encode an event as a source-safe token stored in a TIR StringImm.

Parameters:

event (Event)

Return type:

str

tilelang.tools.cuda.iket.metadata.decode_event(token)

Decode and validate an event token recovered from generated CUDA.

Parameters:

token (str)

Return type:

Event