tilelang.language.fill_op

Fill operations exposed on the TileLang language surface.

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_op.fill(buffer, value)

Fill a buffer or buffer region with a specified value.

Parameters:

• buffer (tvm.tir.Buffer | tvm.tir.BufferRegion | tvm.tir.BufferLoad) – 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_op.clear(buffer)

Clear a buffer by filling it with zeros.

Parameters:

buffer (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