tilelang.language.logical

Logical operations exposed on the TileLang language surface.

函式

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