tilelang.language.dtypes¶

Attributes¶

`AnyDType`
`dtype_name`
`int_`
`__all__`

Classes¶

`dtype`	Abstract base class for generic types.
`bool`	Abstract base class for generic types.

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]

Abstract base class for generic types.

A generic type is typically declared by inheriting from this class parameterized with one or more type variables. For example, a generic mapping type might be defined as:

class Mapping(Generic[KT, VT]):
    def __getitem__(self, key: KT) -> VT:
        ...
    # Etc.

This class can then be used as follows:

def lookup_name(mapping: Mapping[KT, VT], key: KT, default: VT) -> VT:
    try:
        return mapping[key]
    except KeyError:
        return default

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

Abstract base class for generic types.

A generic type is typically declared by inheriting from this class parameterized with one or more type variables. For example, a generic mapping type might be defined as:

class Mapping(Generic[KT, VT]):
    def __getitem__(self, key: KT) -> VT:
        ...
    # Etc.

This class can then be used as follows:

def lookup_name(mapping: Mapping[KT, VT], key: KT, default: VT) -> VT:
    try:
        return mapping[key]
    except KeyError:
        return default

tilelang.language.dtypes.__all__¶