tilelang.language.proxy module#

The language interface for tl programs.

class tilelang.language.proxy.BaseTensorProxy#

Bases: object

Base proxy class for tensor types with configurable defaults.

This class serves as a foundation for different tensor proxy types, providing customizable default values for scope, alignment, and offset factors. It implements the core functionality for creating TIR buffers with specific memory configurations.

default_align = 0#
default_offset_factor = 0#
default_scope = 'global'#
from_ptr(pointer_var: Var, shape: tuple[tvm.ir.expr.PrimExpr, ...], dtype: str = 'float32') → Buffer#

Create a buffer from a pointer, shape, and data type.

Parameters:

  • pointer_var – The pointer variable

  • shape – The shape of the buffer

  • dtype – The data type of the buffer (default: float32)

Returns:

A buffer created from the given parameters

class tilelang.language.proxy.BufferProxy#

Bases: object

Buffer proxy class for constructing tir buffer.

from_ptr(pointer_var: Var, shape: tuple[tvm.ir.expr.PrimExpr, ...], dtype: str = 'float32') → <tilelang.language.proxy.BufferProxy object at 0x7fc573e792a0>#

Create a buffer from a pointer, shape, and data type.

Parameters:

  • pointer_var – The pointer variable

  • shape – The shape of the buffer

  • dtype – The data type of the buffer (default: float32)

Returns:

A buffer created from the given parameters

class tilelang.language.proxy.FragmentBufferProxy#

Bases: BaseTensorProxy

Proxy class for fragment memory buffers.

This class represents tensor proxies specifically for local fragment memory, typically used in GPU tensor core operations.

default_scope = 'local.fragment'#
class tilelang.language.proxy.LocalBufferProxy#

Bases: BaseTensorProxy

Proxy class for local memory buffers.

This class represents tensor proxies for local memory scope, typically used for temporary computations in GPU kernels.

default_scope = 'local'#
class tilelang.language.proxy.SharedBufferProxy#

Bases: BaseTensorProxy

Proxy class for shared memory buffers.

This class represents tensor proxies for dynamic shared memory, commonly used in GPU shared memory operations.

default_scope = 'shared.dyn'#
class tilelang.language.proxy.TensorProxy#

Bases: BaseTensorProxy

Main tensor proxy class for global scope buffers.

This class implements the default tensor proxy with global memory scope, inheriting all functionality from BaseTensorProxy without modifications.

tilelang.language.proxy.make_tensor(ptr: Var, shape: tuple[tvm.ir.expr.PrimExpr, ...], dtype: str = 'float32') → Buffer#
tilelang.language.proxy.ptr(dtype: Optional[str] = None, storage_scope: str = 'global', *, is_size_var: bool = False) → Var#

Create a TIR var that represents a pointer.

Parameters:

  • dtype (str) – The data type of the pointer.

  • storage_scope (str) – The storage scope of the pointer.

  • is_size_var (bool) – Whether or not to return a SizeVar instead of Var.

Returns:

res – The new tir.Var with type handle or casted expression with type handle.

Return type:

PrimExpr