@@ -126,6 +126,14 @@ cdef extern from "dpctl_sycl_types.h":
126
126
ctypedef DPCTLOpaqueSyclUSM * DPCTLSyclUSMRef
127
127
128
128
129
+ cdef extern from " dpctl_sycl_device_manager.h" :
130
+ cdef struct DPCTLDeviceVector
131
+ ctypedef DPCTLDeviceVector * DPCTLDeviceVectorRef
132
+ ctypedef struct DPCTL_DeviceAndContextPair:
133
+ DPCTLSyclDeviceRef DRef
134
+ DPCTLSyclContextRef CRef
135
+
136
+
129
137
cdef extern from " dpctl_sycl_device_interface.h" :
130
138
cdef bool DPCTLDevice_AreEq(const DPCTLSyclDeviceRef DRef1,
131
139
const DPCTLSyclDeviceRef DRef2)
@@ -163,6 +171,13 @@ cdef extern from "dpctl_sycl_device_interface.h":
163
171
cdef uint32_t DPCTLDevice_GetPreferredVectorWidthHalf(const DPCTLSyclDeviceRef DRef)
164
172
cpdef bool DPCTLDevice_HasAspect(
165
173
const DPCTLSyclDeviceRef DRef, DPCTLSyclAspectType AT)
174
+ cdef DPCTLDeviceVectorRef DPCTLDevice_CreateSubDevicesEqually(
175
+ const DPCTLSyclDeviceRef DRef, size_t count)
176
+ cdef DPCTLDeviceVectorRef DPCTLDevice_CreateSubDevicesByCounts(
177
+ const DPCTLSyclDeviceRef DRef, size_t * counts, size_t ncounts)
178
+ cdef DPCTLDeviceVectorRef DPCTLDevice_CreateSubDevicesByAffinity(
179
+ const DPCTLSyclDeviceRef DRef,
180
+ DPCTLPartitionAffinityDomainType PartitionAffinityDomainTy)
166
181
167
182
168
183
cdef extern from " dpctl_sycl_device_manager.h" :
0 commit comments