Skip to content

Commit 6120377

Browse files
Exported data types, finfo and iinfo symbols
1 parent 7b368a1 commit 6120377

File tree

1 file changed

+34
-0
lines changed

1 file changed

+34
-0
lines changed

dpctl/tensor/__init__.py

Lines changed: 34 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -21,6 +21,8 @@
2121
2222
"""
2323

24+
from numpy import dtype, finfo, iinfo
25+
2426
from dpctl.tensor._copy_utils import asnumpy, astype, copy, from_numpy, to_numpy
2527
from dpctl.tensor._ctors import (
2628
arange,
@@ -54,6 +56,21 @@
5456
from dpctl.tensor._reshape import reshape
5557
from dpctl.tensor._usmarray import usm_ndarray
5658

59+
bool = dtype("bool")
60+
int8 = dtype("int8")
61+
int16 = dtype("int16")
62+
int32 = dtype("int32")
63+
int64 = dtype("int64")
64+
uint8 = dtype("uint8")
65+
uint16 = dtype("uint16")
66+
uint32 = dtype("uint32")
67+
uint64 = dtype("uint64")
68+
float16 = dtype("float16")
69+
float32 = dtype("float32")
70+
float64 = dtype("float64")
71+
complex64 = dtype("complex64")
72+
complex128 = dtype("complex128")
73+
5774
__all__ = [
5875
"Device",
5976
"usm_ndarray",
@@ -87,4 +104,21 @@
87104
"from_dlpack",
88105
"tril",
89106
"triu",
107+
"dtype",
108+
"bool",
109+
"int8",
110+
"uint8",
111+
"int16",
112+
"uint16",
113+
"int32",
114+
"uint32",
115+
"int64",
116+
"uint64",
117+
"float16",
118+
"float32",
119+
"float64",
120+
"complex64",
121+
"complex128",
122+
"iinfo",
123+
"finfo",
90124
]

0 commit comments

Comments
 (0)