Skip to content

Commit 8b3be06

Browse files
oleksandr-pavlykDiptorup Deb
authored andcommitted
fix Cython error due to mismatch in declaration of DPCTLDevice_CreateSubDevicesEqually and others
1 parent a3feba5 commit 8b3be06

File tree

1 file changed

+15
-0
lines changed

1 file changed

+15
-0
lines changed

dpctl/_backend.pxd

Lines changed: 15 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -126,6 +126,14 @@ cdef extern from "dpctl_sycl_types.h":
126126
ctypedef DPCTLOpaqueSyclUSM *DPCTLSyclUSMRef
127127

128128

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+
129137
cdef extern from "dpctl_sycl_device_interface.h":
130138
cdef bool DPCTLDevice_AreEq(const DPCTLSyclDeviceRef DRef1,
131139
const DPCTLSyclDeviceRef DRef2)
@@ -163,6 +171,13 @@ cdef extern from "dpctl_sycl_device_interface.h":
163171
cdef uint32_t DPCTLDevice_GetPreferredVectorWidthHalf(const DPCTLSyclDeviceRef DRef)
164172
cpdef bool DPCTLDevice_HasAspect(
165173
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)
166181

167182

168183
cdef extern from "dpctl_sycl_device_manager.h":

0 commit comments

Comments
 (0)