tilelang.language.dtypes

Attributes

AnyDType
dtype_name
int_
__all__

類別

dtype
bool

函式

__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

回傳型別:

int

property bytes: int

回傳型別:

int

as_torch()

回傳型別:

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)

參數:

• 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

tilelang.language.dtypes.__all__