This is an automated email from the ASF dual-hosted git repository.
syfeng pushed a commit to branch main
in repository https://gitbox.apache.org/repos/asf/tvm.git
The following commit(s) were added to refs/heads/main by this push:
new ba9f174363 [3rdparty] Bump DLPack to v1.1 for float8/6/4 dtype
supports (#17831)
ba9f174363 is described below
commit ba9f17436315d36e2d1c122c954bd6f2820971b8
Author: Ruihang Lai <[email protected]>
AuthorDate: Thu Apr 17 22:00:58 2025 -0400
[3rdparty] Bump DLPack to v1.1 for float8/6/4 dtype supports (#17831)
This PR bumps the 3rdparty dlpack version to v1.1, which brings
the support of float8, float6 and float4.
---
3rdparty/dlpack | 2 +-
include/tvm/runtime/data_type.h | 14 +++++++++++---
python/tvm/_ffi/runtime_ctypes.py | 14 +++++++++++---
3 files changed, 23 insertions(+), 7 deletions(-)
diff --git a/3rdparty/dlpack b/3rdparty/dlpack
index e2bdd3bee8..3ea601bb41 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 40664f0c40..06de90eb7c 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 8a9c231617..d49d3bf9ab 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):
