tvm_tirx.backend.cuda.codegen.PTXDataType#
- class tvm_tirx.backend.cuda.codegen.PTXDataType(*values)#
Bases:
EnumA Python equivalent of the provided C++ DataType enum class.
Inherits from IntEnum so that members behave both as enum members and as integers, mirroring the C++ behavior.
see also src/backend/cuda/codegen/ptx.cc
- INT4 = 0#
- UINT4 = 1#
- INT8 = 2#
- UINT8 = 3#
- INT16 = 4#
- UINT16 = 5#
- INT32 = 6#
- UINT32 = 7#
- INT64 = 8#
- UINT64 = 9#
- FLOAT4_E2M1FN = 10#
- FLOAT6_E2M3FN = 11#
- FLOAT6_E3M2FN = 12#
- FLOAT8_E4M3FN = 13#
- FLOAT8_E4M3FNUZ = 14#
- FLOAT8_E5M2 = 15#
- FLOAT8_E8M0FNU = 16#
- FLOAT16 = 17#
- BFLOAT16 = 18#
- FLOAT16X2 = 19#
- FLOAT32 = 20#
- TENSOR_FLOAT32 = 21#
- FLOAT64 = 22#
- BIT1 = 23#
- BIT8 = 24#
- BIT16 = 25#
- BIT32 = 26#
- BIT64 = 27#
- classmethod from_string(s_type)#
- Parameters:
s_type (str)
- Return type:
- to_string()#
- Return type:
str