tilelang.language.fill_op

Fill operations exposed on the TileLang language surface.

函式

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.

參數:

  • buffer (tilelang._typing.BufferLikeType) -- Either a TVM buffer or buffer region to be filled

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

回傳:

A TVM intrinsic call that performs the fill operation

回傳型別:

tvm.tir.PrimExpr

tilelang.language.fill_op.clear(buffer)

Clear a buffer by filling it with zeros.

參數:

buffer (tilelang._typing.BufferLikeType) -- Either a TVM buffer or a variable that contains a buffer region

回傳:

A fill operation that sets the buffer contents to zero

引發:

ValueError -- If the buffer variable contains an invalid buffer region

回傳型別:

tvm.tir.PrimExpr