Skip to content

Commit 1c3598a

Browse files
Support DLPack on Windows
Since DPC++ 2024.2.0 implements support for default platform context on Windows, this enables proper support for DLPack exchange on Windows. Remove special case logic avoiding use of platform.default_context on Windows.
1 parent 0306b10 commit 1c3598a

File tree

1 file changed

+2
-11
lines changed

1 file changed

+2
-11
lines changed

dpctl/tensor/_dlpack.pyx

Lines changed: 2 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -34,8 +34,6 @@ from .._backend cimport (
3434
)
3535
from ._usmarray cimport USM_ARRAY_C_CONTIGUOUS, USM_ARRAY_WRITABLE, usm_ndarray
3636

37-
from platform import system as sys_platform
38-
3937
import numpy as np
4038

4139
import dpctl
@@ -44,10 +42,6 @@ import dpctl.memory as dpmem
4442
from ._device import Device
4543

4644

47-
cdef bint _IS_LINUX = sys_platform() == "Linux"
48-
49-
del sys_platform
50-
5145
cdef extern from 'dlpack/dlpack.h' nogil:
5246
cdef int DLPACK_MAJOR_VERSION
5347

@@ -170,12 +164,9 @@ cdef void _managed_tensor_versioned_deleter(DLManagedTensorVersioned *dlmv_tenso
170164

171165
cdef object _get_default_context(c_dpctl.SyclDevice dev) except *:
172166
try:
173-
if _IS_LINUX:
174-
default_context = dev.sycl_platform.default_context
175-
else:
176-
default_context = None
167+
default_context = dev.sycl_platform.default_context
177168
except RuntimeError:
178-
# RT does not support default_context, e.g. Windows
169+
# RT does not support default_context
179170
default_context = None
180171

181172
return default_context

0 commit comments

Comments
 (0)