tilelang.utils.language module

tilelang.utils.language.array_reduce(array: List[int]) → int

Reduce an array of integers to a single integer.

Parameters:

array (List[int]) – The array of integers to reduce.

Returns:

The reduced integer.

Return type:

int

tilelang.utils.language.is_fragment(buffer: Buffer) → bool

Check if the buffer is a fragment (e.g., for matrix multiplication operations).

Parameters:

buffer (Buffer) – The TVM buffer to check.

Returns:

True if the buffer is a fragment, False otherwise.

Return type:

bool

tilelang.utils.language.is_global(buffer: Buffer) → bool

Check if the buffer is in the global memory scope.

Parameters:

buffer (Buffer) – The TVM buffer to check.

Returns:

True if the buffer is in global memory, False otherwise.

Return type:

bool

tilelang.utils.language.is_local(buffer: Buffer) → bool

Check if the buffer is in the local memory scope.

Parameters:

buffer (Buffer) – The TVM buffer to check.

Returns:

True if the buffer is in local memory, False otherwise.

Return type:

bool

tilelang.utils.language.is_shared(buffer: Buffer, allow_dynamic: bool = True) → bool

Check if the buffer is in the shared memory scope.

Parameters:

buffer (Buffer) – The TVM buffer to check.

Returns:

True if the buffer is in shared memory, False otherwise.

Return type:

bool

tilelang.utils.language.is_shared_dynamic(buffer: Buffer) → bool

Check if the buffer is in the dynamic shared memory scope.

Parameters:

buffer (Buffer) – The TVM buffer to check.

Returns:

True if the buffer is in dynamic shared memory, False otherwise.

Return type:

bool

tilelang.utils.language.retrieve_func_from_module(ir_module: IRModule) → PrimFunc

Retrieve the single PrimFunc from an IRModule.

Parameters:

ir_module (IRModule) – The TVM IRModule to extract the function from. The module should contain exactly one global function.

Returns:

The single function contained in the module.

Return type:

PrimFunc

Raises:

• ValueError – If ir_module is not an IRModule.

• AssertionError – If the module contains more than one global function.