Exported dpctl.tensor.from_dlpack

Modified signature of dpctl.tensor.from_numpy renamed keyword argument
queue to sycl_queue

Author: oleksandr-pavlyk

dpctl/tensor/__init__.py

@@ -34,6 +34,7 @@
 from dpctl.tensor._copy_utils import copy_to_numpy as asnumpy
 from dpctl.tensor._copy_utils import copy_to_numpy as to_numpy
 from dpctl.tensor._ctors import asarray, empty
+from dpctl.tensor._dlpack import from_dlpack
 from dpctl.tensor._reshape import reshape
 from dpctl.tensor._usmarray import usm_ndarray
 
@@ -47,4 +48,5 @@
     "from_numpy",
     "to_numpy",
     "asnumpy",
+    "from_dlpack",
 ]

dpctl/tensor/_copy_utils.py

@@ -93,12 +93,12 @@ def copy_to_numpy(ary):
     )
 
 
-def copy_from_numpy(np_ary, usm_type="device", queue=None):
+def copy_from_numpy(np_ary, usm_type="device", sycl_queue=None):
     "Copies numpy array `np_ary` into a new usm_ndarray"
     # This may peform a copy to meet stated requirements
     Xnp = np.require(np_ary, requirements=["A", "O", "C", "E"])
-    if queue:
-        ctor_kwargs = {"queue": queue}
+    if sycl_queue:
+        ctor_kwargs = {"queue": sycl_queue}
     else:
         ctor_kwargs = dict()
     Xusm = dpt.usm_ndarray(

dpctl/tests/test_tensor_asarray.py

@@ -163,7 +163,7 @@ def test_asarray_copy_false():
         q = dpctl.SyclQueue()
     except dpctl.SyclQueueCreationError:
         pytest.skip("Could not create a queue")
-    X = dpt.from_numpy(np.random.randn(10, 4), usm_type="device", queue=q)
+    X = dpt.from_numpy(np.random.randn(10, 4), usm_type="device", sycl_queue=q)
     Y1 = dpt.asarray(X, copy=False, order="K")
     assert Y1 is X
     Y1c = dpt.asarray(X, copy=True, order="K")