tilelang.language.logical module

The language interface for tl programs.

tilelang.language.logical.all_of(buffer: ~typing.Union[<tilelang.language.proxy.TensorProxy object at 0x7ff64756fdc0>, ~tvm.tir.stmt.BufferRegion])

Check if all elements in the buffer are true.

Parameters:

buffer – Either a TVM buffer or buffer region to be checked

Returns:

A TVM intrinsic call that performs the any operation

tilelang.language.logical.any_of(buffer: ~typing.Union[<tilelang.language.proxy.TensorProxy object at 0x7ff64756fdc0>, ~tvm.tir.stmt.BufferRegion])

Check if any element in the buffer is true.

Parameters:

buffer – Either a TVM buffer or buffer region to be checked

Returns:

A TVM intrinsic call that performs the any operation