tilelang.language.fill module

The language interface for tl programs.

tilelang.language.fill.clear(buffer: Union[Buffer, Var])

Clear a buffer by filling it with zeros.

Parameters:

buffer – Either a TVM buffer or a variable that contains a buffer region

Returns:

A fill operation that sets the buffer contents to zero

Raises:

ValueError – If the buffer variable contains an invalid buffer region

tilelang.language.fill.fill(buffer: Union[Buffer, BufferRegion], value: PrimExpr)

Fill a buffer or buffer region with a specified value.

Parameters:

• buffer – Either a TVM buffer or buffer region to be filled

• value – The value to fill the buffer with

Returns:

A TVM intrinsic call that performs the fill operation