Skip to content

Commit 8f2b44c

Browse files
committed
Enable tests with a cross device
1 parent 6d8d9b4 commit 8f2b44c

File tree

1 file changed

+4
-7
lines changed

1 file changed

+4
-7
lines changed

dpnp/tests/test_sycl_queue.py

Lines changed: 4 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -133,10 +133,8 @@ def test_array_creation_from_array(func, args, device_x, device_y):
133133
assert_sycl_queue_equal(y.sycl_queue, x.sycl_queue)
134134

135135
# cross device
136-
# TODO: include geomspace when issue dpnp#2352 is resolved
137-
if func != "geomspace":
138-
y = getattr(dpnp, func)(*args, device=device_y)
139-
assert_sycl_queue_equal(y.sycl_queue, x.to_device(device_y).sycl_queue)
136+
y = getattr(dpnp, func)(*args, device=device_y)
137+
assert_sycl_queue_equal(y.sycl_queue, x.to_device(device_y).sycl_queue)
140138

141139

142140
@pytest.mark.parametrize("device_x", valid_dev, ids=dev_ids)
@@ -148,10 +146,9 @@ def test_array_creation_logspace_base(device_x, device_y):
148146
y = dpnp.logspace(0, 8, 4, base=x[1:3])
149147
assert_sycl_queue_equal(y.sycl_queue, x.sycl_queue)
150148

151-
# TODO: include geomspace when issue dpnp#2353 is resolved
152149
# cross device
153-
# y = dpnp.logspace(0, 8, 4, base=x[1:3], device=device_y)
154-
# assert_sycl_queue_equal(y.sycl_queue, x.to_device(device_y).sycl_queue)
150+
y = dpnp.logspace(0, 8, 4, base=x[1:3], device=device_y)
151+
assert_sycl_queue_equal(y.sycl_queue, x.to_device(device_y).sycl_queue)
155152

156153

157154
@pytest.mark.parametrize("device", valid_dev + [None], ids=dev_ids + [None])

0 commit comments

Comments
 (0)