tilelang.language.copy module#

The language interface for tl programs.

tilelang.language.copy.buffer_load_to_tile_region(load: BufferLoad, access_type: str, extents: List[PrimExpr])#

Convert a buffer load operation to a tile region descriptor.

Parameters:

  • load (tir.BufferLoad) – The buffer load operation

  • access_type (str) – Type of access - ‘r’ for read, ‘w’ for write, ‘rw’ for read-write

  • extents (List[tir.PrimExpr]) – List of expressions defining the region size

Returns:

A region descriptor for the loaded area

Return type:

tir.Call

tilelang.language.copy.buffer_region_to_tile_region(buffer_region: BufferRegion, access_type: str)#

Convert a buffer region to a tile region descriptor.

Parameters:

  • buffer_region (tir.BufferRegion) – The buffer region to convert

  • access_type (str) – Type of access - ‘r’ for read, ‘w’ for write, ‘rw’ for read-write

Returns:

A region descriptor for the specified buffer region

Return type:

tir.Call

tilelang.language.copy.buffer_to_tile_region(buffer: Buffer, access_type: str)#

Convert a TVM buffer to a tile region descriptor.

Parameters:

  • buffer (tir.Buffer) – The buffer to convert

  • access_type (str) – Type of access - ‘r’ for read, ‘w’ for write, ‘rw’ for read-write

Returns:

A region descriptor covering the entire buffer

Return type:

tir.Call

tilelang.language.copy.c2d_im2col(img: Buffer, col: Buffer, nhw_step: PrimExpr, c_step: PrimExpr, kernel: int, stride: int, dilation: int, pad: int)#

Perform im2col transformation for 2D convolution.

Parameters:

  • img (tir.Buffer) – Input image buffer

  • col (tir.Buffer) – Output column buffer

  • nhw_step (tir.PrimExpr) – Step size for batch and spatial dimensions

  • c_step (tir.PrimExpr) – Step size for channel dimension

  • kernel (int) – Kernel size

  • stride (int) – Stride of the convolution

  • dilation (int) – Dilation rate

  • pad (int) – Padding size

Returns:

A handle to the im2col operation

Return type:

tir.Call

tilelang.language.copy.copy(src: Union[Buffer, BufferLoad, BufferRegion], dst: Union[Buffer, BufferLoad], coalesced_width: Optional[int] = None)#

Copy data between memory regions.

Parameters:

  • src (Union[tir.Buffer, tir.BufferLoad, tir.BufferRegion]) – Source memory region

  • dst (Union[tir.Buffer, tir.BufferLoad]) – Destination memory region

  • coalesced_width (Optional[int], optional) – Width for coalesced memory access. Defaults to None.

Raises:

TypeError – If copy extents cannot be deduced from arguments

Returns:

A handle to the copy operation

Return type:

tir.Call

tilelang.language.copy.region(buffer: BufferLoad, access_type: str, *args: PrimExpr)#

Create a memory region descriptor for tile operations.

Parameters:

  • buffer (tir.BufferLoad) – The buffer to create a region for

  • access_type (str) – Type of access - ‘r’ for read, ‘w’ for write, ‘rw’ for read-write

  • *args (tir.PrimExpr) – Extent expressions defining the region size

Returns:

A region descriptor for tile operations

Return type:

tir.Call