Skip to content

Commit 85d417f

Browse files
committed
Expose DLDeviceType in dpctl.tensor
1 parent fe71e99 commit 85d417f

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

dpctl/tensor/__init__.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -93,7 +93,7 @@
9393
from dpctl.tensor._reshape import reshape
9494
from dpctl.tensor._search_functions import where
9595
from dpctl.tensor._statistical_functions import mean, std, var
96-
from dpctl.tensor._usmarray import usm_ndarray
96+
from dpctl.tensor._usmarray import DLDeviceType, usm_ndarray
9797
from dpctl.tensor._utility_functions import all, any, diff
9898

9999
from ._accumulation import cumulative_logsumexp, cumulative_prod, cumulative_sum
@@ -376,4 +376,5 @@
376376
"nextafter",
377377
"diff",
378378
"count_nonzero",
379+
"DLDeviceType",
379380
]

0 commit comments

Comments
 (0)