Skip to content

Commit f774228

Browse files
committed
update the tests
1 parent aef6de3 commit f774228

File tree

2 files changed

+24
-5
lines changed

2 files changed

+24
-5
lines changed

dpctl/tests/elementwise/test_add.py

Lines changed: 12 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -188,9 +188,18 @@ def test_add_dtype_out_keyword(op1_dtype, op2_dtype):
188188

189189

190190
def test_add_errors():
191-
ar1 = dpt.ones(2, dtype="float32", device="gpu")
192-
ar2 = dpt.ones_like(ar1, dtype="int32", device="gpu")
193-
y = dpt.empty_like(ar1, device="cpu")
191+
try:
192+
gpu_queue = dpctl.SyclQueue("gpu")
193+
except dpctl.SyclQueueCreationError:
194+
pytest.skip("SyclQueue('gpu') failed, skipping")
195+
try:
196+
cpu_queue = dpctl.SyclQueue("cpu")
197+
except dpctl.SyclQueueCreationError:
198+
pytest.skip("SyclQueue('cpu') failed, skipping")
199+
200+
ar1 = dpt.ones(2, dtype="float32", sycl_queue=gpu_queue)
201+
ar2 = dpt.ones_like(ar1, sycl_queue=gpu_queue)
202+
y = dpt.empty_like(ar1, sycl_queue=cpu_queue)
194203
assert_raises_regex(
195204
TypeError,
196205
"Input and output allocation queues are not compatible",

dpctl/tests/elementwise/test_cos.py

Lines changed: 12 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -4,6 +4,7 @@
44
import pytest
55
from numpy.testing import assert_raises_regex
66

7+
import dpctl
78
import dpctl.tensor as dpt
89
from dpctl.tests.helper import get_queue_or_skip, skip_if_dtype_not_supported
910

@@ -109,8 +110,17 @@ def test_cos_out_keyword(dtype):
109110

110111

111112
def test_cos_errors():
112-
x = dpt.zeros(2, device="gpu")
113-
y = dpt.empty_like(x, device="cpu")
113+
try:
114+
gpu_queue = dpctl.SyclQueue("gpu")
115+
except dpctl.SyclQueueCreationError:
116+
pytest.skip("SyclQueue('gpu') failed, skipping")
117+
try:
118+
cpu_queue = dpctl.SyclQueue("cpu")
119+
except dpctl.SyclQueueCreationError:
120+
pytest.skip("SyclQueue('cpu') failed, skipping")
121+
122+
x = dpt.zeros(2, sycl_queue=gpu_queue)
123+
y = dpt.empty_like(x, sycl_queue=cpu_queue)
114124
assert_raises_regex(
115125
TypeError,
116126
"Input and output allocation queues are not compatible",

0 commit comments

Comments
 (0)