tilelang.metal.utils

Functions

is_metal_cooperative_tensor(buffer)	Check if the buffer is in the Metal cooperative tensor scope.
is_metal_simdgroup(buffer)	Check if the buffer is in the Metal simdgroup scope.

Module Contents

tilelang.metal.utils.is_metal_cooperative_tensor(buffer)

Check if the buffer is in the Metal cooperative tensor scope.

Parameters:

buffer (tilelang._typing.BufferLikeType)

Return type:

bool

tilelang.metal.utils.is_metal_simdgroup(buffer)

Check if the buffer is in the Metal simdgroup scope.

Parameters:

buffer (tilelang._typing.BufferLikeType)

Return type:

bool