Skip to content

Commit c749e4c

Browse files
committed
use sycl_device instead of sycldevice in dldevice conversion functions
1 parent c5fb62a commit c749e4c

File tree

3 files changed

+14
-14
lines changed

3 files changed

+14
-14
lines changed

dpctl/tensor/__init__.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -60,8 +60,8 @@
6060
)
6161
from dpctl.tensor._device import Device
6262
from dpctl.tensor._dldevice_conversions import (
63-
dldevice_to_sycldevice,
64-
sycldevice_to_dldevice,
63+
dldevice_to_sycl_device,
64+
sycl_device_to_dldevice,
6565
)
6666
from dpctl.tensor._dlpack import from_dlpack
6767
from dpctl.tensor._indexing_functions import (
@@ -392,6 +392,6 @@
392392
"take_along_axis",
393393
"put_along_axis",
394394
"top_k",
395-
"dldevice_to_sycldevice",
396-
"sycldevice_to_dldevice",
395+
"dldevice_to_sycl_device",
396+
"sycl_device_to_dldevice",
397397
]

dpctl/tensor/_dldevice_conversions.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -18,7 +18,7 @@
1818
from ._usmarray import DLDeviceType
1919

2020

21-
def dldevice_to_sycldevice(dl_dev: tuple):
21+
def dldevice_to_sycl_device(dl_dev: tuple):
2222
if isinstance(dl_dev, tuple):
2323
if len(dl_dev) != 2:
2424
raise ValueError("dldevice tuple must have length 2")
@@ -31,7 +31,7 @@ def dldevice_to_sycldevice(dl_dev: tuple):
3131
return SyclDevice(str(dl_dev[1]))
3232

3333

34-
def sycldevice_to_dldevice(dev: SyclDevice):
34+
def sycl_device_to_dldevice(dev: SyclDevice):
3535
if not isinstance(dev, SyclDevice):
3636
raise TypeError(
3737
"dev is expected to be a SyclDevice, got " f"{type(dev)}"

dpctl/tests/test_usm_ndarray_dlpack.py

Lines changed: 8 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -828,36 +828,36 @@ def test_generic_container():
828828
assert Z.device == X.device
829829

830830

831-
def test_sycldevice_to_dldevice(all_root_devices):
831+
def test_sycl_device_to_dldevice(all_root_devices):
832832
for sycl_dev in all_root_devices:
833-
dev = dpt.sycldevice_to_dldevice(sycl_dev)
833+
dev = dpt.sycl_device_to_dldevice(sycl_dev)
834834
assert type(dev) is tuple
835835
assert len(dev) == 2
836836
assert dev[0] == device_oneAPI
837837
assert dev[1] == all_root_devices.index(sycl_dev)
838838

839839

840-
def test_dldevice_to_sycldevice(all_root_devices):
840+
def test_dldevice_to_sycl_device(all_root_devices):
841841
for sycl_dev in all_root_devices:
842842
dldev = dpt.empty(0, device=sycl_dev).__dlpack_device__()
843-
dev = dpt.dldevice_to_sycldevice(dldev)
843+
dev = dpt.dldevice_to_sycl_device(dldev)
844844
assert type(dev) is dpctl.SyclDevice
845845
assert dev == all_root_devices[dldev[1]]
846846

847847

848848
def test_dldevice_conversion_arg_validation():
849849
bad_dldevice_type = (dpt.DLDeviceType.kDLCPU, 0)
850850
with pytest.raises(ValueError):
851-
dpt.dldevice_to_sycldevice(bad_dldevice_type)
851+
dpt.dldevice_to_sycl_device(bad_dldevice_type)
852852

853853
bad_dldevice_len = bad_dldevice_type + (0,)
854854
with pytest.raises(ValueError):
855-
dpt.dldevice_to_sycldevice(bad_dldevice_len)
855+
dpt.dldevice_to_sycl_device(bad_dldevice_len)
856856

857857
bad_dldevice = dict()
858858
with pytest.raises(TypeError):
859-
dpt.dldevice_to_sycldevice(bad_dldevice)
859+
dpt.dldevice_to_sycl_device(bad_dldevice)
860860

861861
bad_sycldevice = dict()
862862
with pytest.raises(TypeError):
863-
dpt.sycldevice_to_dldevice(bad_sycldevice)
863+
dpt.sycl_device_to_dldevice(bad_sycldevice)

0 commit comments

Comments
 (0)