tilelang.language.builtin module#
The language interface for tl programs.
- tilelang.language.builtin.CreateListofMBarrierOp(*args)#
Create a list of memory barrier operations.
- Parameters:
*args – Variable arguments passed to the memory barrier creation operation
- Returns:
A handle to the created list of memory barriers
- Return type:
tir.Call
- tilelang.language.builtin.CreateTMADescriptorOp(*args)#
Create a Tensor Memory Access (TMA) descriptor.
- Parameters:
*args – Variable arguments defining the TMA descriptor configuration
- Returns:
A handle to the created TMA descriptor
- Return type:
tir.Call
- tilelang.language.builtin.FenceProxyAsyncOp(*args)#
Create a fence for asynchronous proxy operations.
- Parameters:
*args – Variable arguments for fence configuration
- Returns:
A handle to the fence operation
- Return type:
tir.Call
- tilelang.language.builtin.GetMBarrierOp(*args)#
Retrieve a memory barrier operation.
- Parameters:
*args – Variable arguments to specify which memory barrier to retrieve
- Returns:
A handle to the requested memory barrier
- Return type:
tir.Call
- tilelang.language.builtin.MBarrierExpectTX(*args)#
Set expected transaction count for memory barrier.
- Parameters:
*args – Variable arguments specifying the expected transaction count
- Returns:
A handle to the barrier expectation operation
- Return type:
tir.Call
- tilelang.language.builtin.MBarrierWaitParity(*args)#
Wait for memory barrier parity condition.
- Parameters:
*args – Variable arguments specifying the parity wait condition
- Returns:
A handle to the barrier wait operation
- Return type:
tir.Call
- tilelang.language.builtin.NoSetMaxNReg(*args)#
Disable the maximum register limit setting.
- Parameters:
*args – Variable arguments for the operation
- Returns:
A handle to the register limit disable operation
- Return type:
tir.Call
- tilelang.language.builtin.SetMaxNReg(*args)#
Set the maximum number of registers to use.
- Parameters:
*args – Variable arguments specifying register allocation limits
- Returns:
A handle to the register setting operation
- Return type:
tir.Call
- tilelang.language.builtin.TMALoadOp(*args)#
Perform a Tensor Memory Access (TMA) load operation.
- Parameters:
*args – Variable arguments specifying the TMA load parameters
- Returns:
A handle to the TMA load operation
- Return type:
tir.Call
- tilelang.language.builtin.TMAStoreArrive(*args)#
Signal the arrival of a TMA store operation.
- Parameters:
*args – Variable arguments for the store arrival operation
- Returns:
A handle to the store arrive operation
- Return type:
tir.Call
- tilelang.language.builtin.TMAStoreWait(*args)#
Wait for completion of TMA store operations.
- Parameters:
*args – Variable arguments specifying which store operations to wait for
- Returns:
A handle to the store wait operation
- Return type:
tir.Call
- tilelang.language.builtin.WaitWgmma(*args)#
Wait for WGMMA (Warp Group Matrix Multiply-Accumulate) operations to complete.
- Parameters:
*args – Variable arguments specifying which operations to wait for
- Returns:
A handle to the WGMMA wait operation
- Return type:
tir.Call