tilelang.language.dtypes

Attributes

AnyDType
dtype_name
int_
__all__

Classes

dtype
bool

Functions

__dtype_call__(self[, expr, is_size_var])
__dtype_as_torch__(self)	Convert TileLang dtype to PyTorch dtype.
__dtype_new__(cls, value)
__dtype_bytes__(self)	Return the number of bytes for this dtype.
get_tvm_dtype(value)

Module Contents

class tilelang.language.dtypes.dtype

Bases: Generic[_T]

property bits: int

Return type:

int

property bytes: int

Return type:

int

as_torch()

Return type:

torch.dtype

tilelang.language.dtypes.AnyDType

tilelang.language.dtypes.dtype_name = 'float8_e4m3fn'

tilelang.language.dtypes.int_

tilelang.language.dtypes.__dtype_call__(self, expr=None, is_size_var=False)

Parameters:

• self (dtype)

• is_size_var (bool)

Return type:

tvm.tir.Var

tilelang.language.dtypes.__dtype_as_torch__(self)

Convert TileLang dtype to PyTorch dtype.

Parameters:

self (dtype)

Return type:

torch.dtype

tilelang.language.dtypes.__dtype_new__(cls, value)

Parameters:

value (AnyDType)

Return type:

dtype

tilelang.language.dtypes.__dtype_bytes__(self)

Return the number of bytes for this dtype.

Parameters:

self (dtype)

Return type:

int

tilelang.language.dtypes.get_tvm_dtype(value)

Parameters:

value (AnyDType)

Return type:

dtype

class tilelang.language.dtypes.bool

Bases: dtype

tilelang.language.dtypes.__all__