Skip to content

Commit 5df55b6

Browse files
committed
update Cython definitions of DLPack data type codes
1 parent 9ca567b commit 5df55b6

File tree

1 file changed

+11
-0
lines changed

1 file changed

+11
-0
lines changed

dpctl/tensor/_dlpack.pyx

Lines changed: 11 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -88,6 +88,17 @@ cdef extern from "dlpack/dlpack.h" nogil:
8888
kDLBfloat
8989
kDLComplex
9090
kDLBool
91+
kDLFloat8_e3m4
92+
kDLFloat8_e4m3
93+
kDLFloat8_e4m3b11fnuz
94+
kDLFloat8_e4m3fn
95+
kDLFloat8_e4m3fnuz
96+
kDLFloat8_e5m2
97+
kDLFloat8_e5m2fnuz
98+
kDLFloat8_e8m0fnu
99+
kDLFloat6_e2m3fn
100+
kDLFloat6_e3m2fn
101+
kDLFloat4_e2m1fn
91102

92103
ctypedef struct DLDataType:
93104
uint8_t code

0 commit comments

Comments
 (0)