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, *args[, 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¶

返回类型:: int

property bytes: int¶

返回类型:: int

as_torch()¶

返回类型:: torch.dtype

tilelang.language.dtypes.AnyDType¶

tilelang.language.dtypes.dtype_name = 'uint16'¶

tilelang.language.dtypes.int_¶

tilelang.language.dtypes.__dtype_call__(self, *args, is_size_var=False)¶

参数:

self (dtype)
is_size_var (bool)

返回类型:

tvm.tir.Var

tilelang.language.dtypes.__dtype_as_torch__(self)¶

Convert TileLang dtype to PyTorch dtype.

参数:: self (dtype)
返回类型:: torch.dtype

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

参数:: value (AnyDType)
返回类型:: dtype

tilelang.language.dtypes.__dtype_bytes__(self)¶

Return the number of bytes for this dtype.

参数:: self (dtype)
返回类型:: int

tilelang.language.dtypes.get_tvm_dtype(value)¶

参数:: value (AnyDType)
返回类型:: 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__¶