tilelang.language.fill

The language interface for tl programs.

Functions

fill(buffer, value)	Fill a buffer or buffer region with a specified value.
clear(buffer)	Clear a buffer by filling it with zeros.

Module Contents

tilelang.language.fill.fill(buffer, value)

Fill a buffer or buffer region with a specified value.

Parameters:

• buffer (Union[tvm.tir.Buffer, tvm.tir.BufferRegion]) – Either a TVM buffer or buffer region to be filled

• value (tvm.tir.PrimExpr) – The value to fill the buffer with

Returns:

A TVM intrinsic call that performs the fill operation

tilelang.language.fill.clear(buffer)

Clear a buffer by filling it with zeros.

Parameters:

buffer (Union[tvm.tir.Buffer, tvm.tir.Var]) – 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