File tree Expand file tree Collapse file tree 2 files changed +13
-1
lines changed Expand file tree Collapse file tree 2 files changed +13
-1
lines changed Original file line number Diff line number Diff line change 20
20
""" This file declares the SyclDevice extension type.
21
21
"""
22
22
23
- from ._backend cimport *
23
+ from libcpp cimport bool
24
+ from libc.stdint cimport uint32_t
25
+ from ._backend cimport (
26
+ DPCTLSyclDeviceRef
27
+ )
24
28
25
29
26
30
cdef class SyclDevice:
@@ -52,3 +56,10 @@ cdef class SyclDevice:
52
56
cpdef get_max_num_sub_groups (self )
53
57
cpdef has_int64_base_atomics (self )
54
58
cpdef has_int64_extended_atomics (self )
59
+
60
+
61
+ cpdef select_accelerator_device()
62
+ cpdef select_cpu_device()
63
+ cpdef select_default_device()
64
+ cpdef select_gpu_device()
65
+ cpdef select_host_device()
Original file line number Diff line number Diff line change @@ -45,6 +45,7 @@ from ._backend cimport (
45
45
DPCTLDeviceSelector_Delete,
46
46
DPCTLSize_t_Array_Delete,
47
47
DPCTLSyclDeviceRef,
48
+ DPCTLSyclDeviceSelectorRef,
48
49
)
49
50
from . import device_type
50
51
You can’t perform that action at this time.
0 commit comments