Skip to content

Commit 6874366

Browse files
to_device array method impl (#1143)
* to_device array method impl
1 parent 67ac3c6 commit 6874366

File tree

2 files changed

+22
-0
lines changed

2 files changed

+22
-0
lines changed

dpnp/dpnp_array.py

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -97,6 +97,13 @@ def T(self):
9797
else:
9898
return dpnp.transpose(self)
9999

100+
def to_device(self, target_device):
101+
"""
102+
Transfer array to target device
103+
"""
104+
105+
return dpnp_array(shape=self.shape, buffer=self.get_array().to_device(target_device))
106+
100107
def __abs__(self):
101108
return dpnp.abs(self)
102109

tests/test_sycl_queue.py

Lines changed: 15 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -83,3 +83,18 @@ def test_2in_1out(func, device):
8383

8484
assert_sycl_queue_equal(result_queue, expected_queue)
8585
assert result_queue.sycl_device == expected_queue.sycl_device
86+
87+
88+
@pytest.mark.parametrize("device_from",
89+
valid_devices,
90+
ids=[device.filter_string for device in valid_devices])
91+
@pytest.mark.parametrize("device_to",
92+
valid_devices,
93+
ids=[device.filter_string for device in valid_devices])
94+
def test_to_device(device_from, device_to):
95+
data = [1., 1., 1., 1., 1.]
96+
97+
x = dpnp.array(data, device=device_from)
98+
y = x.to_device(device_to)
99+
100+
assert y.get_array().sycl_device == device_to

0 commit comments

Comments
 (0)