diff --git a/3rdparty/dlpack b/3rdparty/dlpack index e2bdd3bee8cb..3ea601bb4130 160000 --- a/3rdparty/dlpack +++ b/3rdparty/dlpack @@ -1 +1 @@ -Subproject commit e2bdd3bee8cb6501558042633fa59144cc8b7f5f +Subproject commit 3ea601bb413074c49a77c4ce3218bc08f8c4703c diff --git a/include/tvm/runtime/data_type.h b/include/tvm/runtime/data_type.h index 40664f0c40a1..06de90eb7cb2 100644 --- a/include/tvm/runtime/data_type.h +++ b/include/tvm/runtime/data_type.h @@ -56,9 +56,17 @@ class DataType { kFloat = kDLFloat, kHandle = TVMArgTypeCode::kTVMOpaqueHandle, kBFloat = kDLBfloat, - kFloat8_e4m3fn = 6U, - kFloat8_e5m2 = 7U, - kFloat4_e2m1fn = 8U, + kFloat8_e3m4 = kDLFloat8_e3m4, + kFloat8_e4m3 = kDLFloat8_e4m3, + kFloat8_e4m3b11fnuz = kDLFloat8_e4m3b11fnuz, + kFloat8_e4m3fn = kDLFloat8_e4m3fn, + kFloat8_e4m3fnuz = kDLFloat8_e4m3fnuz, + kFloat8_e5m2 = kDLFloat8_e5m2, + kFloat8_e5m2fnuz = kDLFloat8_e5m2fnuz, + kFloat8_e8m0fnu = kDLFloat8_e8m0fnu, + kFloat6_e2m3fn = kDLFloat6_e2m3fn, + kFloat6_e3m2fn = kDLFloat6_e3m2fn, + kFloat4_e2m1fn = kDLFloat4_e2m1fn, kCustomBegin = 129 }; /*! \brief default constructor */ diff --git a/python/tvm/_ffi/runtime_ctypes.py b/python/tvm/_ffi/runtime_ctypes.py index 8a9c231617a2..d49d3bf9ab03 100644 --- a/python/tvm/_ffi/runtime_ctypes.py +++ b/python/tvm/_ffi/runtime_ctypes.py @@ -66,9 +66,17 @@ class DataTypeCode(object): FLOAT = 2 HANDLE = 3 BFLOAT = 4 - Float8E4M3FN = 6 - Float8E5M2 = 7 - Float4E2M1FN = 8 + Float8E3M4 = 7 + Float8E4M3 = 8 + Float8E4M3B11FNUZ = 9 + Float8E4M3FN = 10 + Float8E4M3FNUZ = 11 + Float8E5M2 = 12 + Float8E5M2FNUZ = 13 + Float8E8M0FNU = 14 + Float6E2M3FN = 15 + Float6E3M2FN = 16 + Float4E2M1FN = 17 class DataType(ctypes.Structure):