Skip to content

Commit 7dc7734

Browse files
Handle SyclDeviceException
1 parent 47f3b8c commit 7dc7734

File tree

1 file changed

+13
-3
lines changed

1 file changed

+13
-3
lines changed

dpctl/tests/test_usm_ndarray_operators.py

Lines changed: 13 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -16,6 +16,7 @@
1616

1717
import pytest
1818

19+
import dpctl
1920
import dpctl.tensor as dpt
2021

2122

@@ -48,7 +49,10 @@ def multiply(a, b):
4849

4950
@pytest.mark.parametrize("namespace", [None, Dummy()])
5051
def test_fp_ops(namespace):
51-
X = dpt.ones(1)
52+
try:
53+
X = dpt.ones(1)
54+
except dpctl.SyclDeviceCreationError:
55+
pytest.skip("No SYCL devices available")
5256
X._set_namespace(namespace)
5357
assert X.__array_namespace__() is namespace
5458
X[0] = -2.5
@@ -79,7 +83,10 @@ def test_fp_ops(namespace):
7983

8084
@pytest.mark.parametrize("namespace", [None, Dummy()])
8185
def test_int_ops(namespace):
82-
X = dpt.usm_ndarray(1, "i4")
86+
try:
87+
X = dpt.usm_ndarray(1, "i4")
88+
except dpctl.SyclDeviceCreationError:
89+
pytest.skip("No SYCL devices available")
8390
X._set_namespace(namespace)
8491
assert X.__array_namespace__() is namespace
8592
X.__lshift__(2)
@@ -108,7 +115,10 @@ def test_int_ops(namespace):
108115

109116
@pytest.mark.parametrize("namespace", [None, Dummy()])
110117
def test_mat_ops(namespace):
111-
M = dpt.eye(3, 3)
118+
try:
119+
M = dpt.eye(3, 3)
120+
except dpctl.SyclDeviceCreationError:
121+
pytest.skip("No SYCL devices available")
112122
M._set_namespace(namespace)
113123
assert M.__array_namespace__() is namespace
114124
M.__matmul__(M)

0 commit comments

Comments
 (0)