Merge pull request #1781 from IntelPython/feature/dlpack-kdlcpu-support

Support `kDLCPU` DLDeviceType for `from_dlpack` and `__dlpack__`

Author: ndgrigorian

docs/doc_sources/api_reference/dpctl/tensor.constants.rst

@@ -0,0 +1,35 @@
+.. _dpctl_tensor_constants:
+
+Constants
+========================
+
+The following constants are defined in :py:mod:`dpctl.tensor`:
+
+.. currentmodule:: dpctl.tensor
+
+.. autodata:: DLDeviceType
+
+.. data:: e
+
+   ``float``:
+        IEEE 754 floating-point representation of Euler's constant.
+
+.. data:: inf
+
+   ``float``:
+        IEEE 754 floating-point representation of (positive) infinity.
+
+.. data:: nan
+
+   ``float``:
+        IEEE 754 floating-point representation of Not a Number (NaN).
+
+.. data:: newaxis
+
+   ``NoneType``:
+        Alias for ``None`` which is useful for indexing.
+
+.. data:: pi
+
+   ``float``:
+        IEEE 754 floating-point representation of the mathematical constant π.

docs/doc_sources/api_reference/dpctl/tensor.rst

@@ -29,6 +29,7 @@ This module contains:
 * :ref:`sorting functions <dpctl_tensor_sorting_functions>`
 * :ref:`statistical functions <dpctl_tensor_statistical_functions>`
 * :ref:`utility functions <dpctl_tensor_utility_functions>`
+* :ref:`constants <dpctl_tensor_constants>`
 
 
 .. toctree::
@@ -48,3 +49,4 @@ This module contains:
     tensor.sorting_functions
     tensor.statistical_functions
     tensor.utility_functions
+    tensor.constants

dpctl/tensor/__init__.py

@@ -100,7 +100,7 @@
 from dpctl.tensor._reshape import reshape
 from dpctl.tensor._search_functions import where
 from dpctl.tensor._statistical_functions import mean, std, var
-from dpctl.tensor._usmarray import usm_ndarray
+from dpctl.tensor._usmarray import DLDeviceType, usm_ndarray
 from dpctl.tensor._utility_functions import all, any, diff
 
 from ._accumulation import cumulative_logsumexp, cumulative_prod, cumulative_sum
@@ -383,5 +383,6 @@
     "nextafter",
     "diff",
     "count_nonzero",
+    "DLDeviceType",
     "take_along_axis",
 ]

dpctl/tensor/_dlpack.pxd

@@ -18,6 +18,10 @@
 # cython: language_level=3
 # cython: linetrace=True
 
+cdef extern from "numpy/npy_no_deprecated_api.h":
+    pass
+from numpy cimport ndarray
+
 from .._sycl_device cimport SyclDevice
 from ._usmarray cimport usm_ndarray
 
@@ -40,7 +44,8 @@ cdef extern from 'dlpack/dlpack.h' nogil:
 
 cpdef object to_dlpack_capsule(usm_ndarray array) except +
 cpdef object to_dlpack_versioned_capsule(usm_ndarray array, bint copied) except +
-cpdef usm_ndarray from_dlpack_capsule(object dltensor) except +
+cpdef object numpy_to_dlpack_versioned_capsule(ndarray array, bint copied) except +
+cpdef object from_dlpack_capsule(object dltensor) except +
 
 cdef int get_parent_device_ordinal_id(SyclDevice dev) except *