Skip to content

Commit 4e53753

Browse files
Handle case when default_context is not available (WIN)
1 parent 5e1ff70 commit 4e53753

File tree

1 file changed

+6
-2
lines changed

1 file changed

+6
-2
lines changed

dpctl/tests/test_usm_ndarray_dlpack.py

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -222,9 +222,13 @@ def test_dlpack_from_subdevice():
222222
pytest.skip("Default device can not be partitioned")
223223
assert isinstance(sdevs, list) and len(sdevs) > 0
224224
try:
225-
q = dpctl.SyclQueue(sdevs[0].sycl_platform.default_context, sdevs[0])
225+
ctx = sdevs[0].sycl_platform.default_context
226+
except dpctl.SyclContextCreationError:
227+
pytest.skip("Platform's default_context is not available")
228+
try:
229+
q = dpctl.SyclQueue(ctx, sdevs[0])
226230
except dpctl.SyclQueueCreationError:
227-
pytest.skip("Default device can not be partitioned")
231+
pytest.skip("Queue could not be created")
228232

229233
ar = dpt.arange(n, dtype=dpt.int32, sycl_queue=q)
230234
ar2 = dpt.from_dlpack(ar)

0 commit comments

Comments
 (0)