Skip to content

Commit 5027b91

Browse files
author
Diptorup Deb
committed
Add declarations to pxd file.
1 parent 7a2070f commit 5027b91

File tree

2 files changed

+13
-1
lines changed

2 files changed

+13
-1
lines changed

dpctl/_sycl_device.pxd

Lines changed: 12 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -20,7 +20,11 @@
2020
""" This file declares the SyclDevice extension type.
2121
"""
2222

23-
from ._backend cimport *
23+
from libcpp cimport bool
24+
from libc.stdint cimport uint32_t
25+
from ._backend cimport (
26+
DPCTLSyclDeviceRef
27+
)
2428

2529

2630
cdef class SyclDevice:
@@ -52,3 +56,10 @@ cdef class SyclDevice:
5256
cpdef get_max_num_sub_groups (self)
5357
cpdef has_int64_base_atomics (self)
5458
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()

dpctl/_sycl_device.pyx

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -45,6 +45,7 @@ from ._backend cimport (
4545
DPCTLDeviceSelector_Delete,
4646
DPCTLSize_t_Array_Delete,
4747
DPCTLSyclDeviceRef,
48+
DPCTLSyclDeviceSelectorRef,
4849
)
4950
from . import device_type
5051

0 commit comments

Comments
 (0)