tilelang.language.logical

Logical operations exposed on the TileLang language surface.

Functions

any_of(buffer)	Check if any element in the buffer is true.
all_of(buffer)	Check if all elements in the buffer are true.

Module Contents

tilelang.language.logical.any_of(buffer)

Check if any element in the buffer is true.

参数:

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

返回:

A TVM intrinsic call that performs the any operation

返回类型:

tvm.tir.PrimExpr

tilelang.language.logical.all_of(buffer)

Check if all elements in the buffer are true.

参数:

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

返回:

A TVM intrinsic call that performs the any operation

返回类型:

tvm.tir.PrimExpr