Skip to content

Commit 6684c7b

Browse files
Merge pull request #125 from IntelPython/expose-get-current-queue-on-C-side
dpctl.get_current_queue is now cimportable
2 parents 519312f + 55a1c82 commit 6684c7b

File tree

2 files changed

+8
-2
lines changed

2 files changed

+8
-2
lines changed

dpctl/_sycl_core.pxd

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -118,3 +118,6 @@ cdef class SyclQueue:
118118
cpdef void wait (self)
119119
cdef DPPLSyclQueueRef get_queue_ref (self)
120120
cpdef memcpy (self, dest, src, int count)
121+
122+
123+
cpdef SyclQueue get_current_queue()

dpctl/sycl_core.pyx

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -626,7 +626,7 @@ cdef class _SyclRTManager:
626626
'''
627627
return self.get_current_queue().get_sycl_device().get_device_type()
628628

629-
def get_current_queue (self):
629+
cpdef SyclQueue get_current_queue (self):
630630
''' Returns the activated SYCL queue as a PyCapsule.
631631
'''
632632
return SyclQueue._create(DPPLQueueMgr_GetCurrentQueue())
@@ -736,7 +736,6 @@ _mgr = _SyclRTManager()
736736

737737
# Global bound functions
738738
dump = _mgr.dump
739-
get_current_queue = _mgr.get_current_queue
740739
get_current_device_type = _mgr.get_current_device_type
741740
get_num_platforms = _mgr.get_num_platforms
742741
get_num_activated_queues = _mgr.get_num_activated_queues
@@ -747,6 +746,10 @@ has_sycl_platforms = _mgr.has_sycl_platforms
747746
set_default_queue = _mgr.set_default_queue
748747
is_in_device_context = _mgr.is_in_device_context
749748

749+
cpdef SyclQueue get_current_queue():
750+
''' Obtain current Sycl Queue from Data Parallel Control package '''
751+
return _mgr.get_current_queue()
752+
750753

751754
def create_program_from_source (SyclQueue q, unicode source, unicode copts=""):
752755
''' Creates a Sycl interoperability program from an OpenCL source string.

0 commit comments

Comments
 (0)