Skip to content

Commit a189a04

Browse files
etotmenioleksandr-pavlyk
etotmeni
authored andcommitted
Change funcs to cpdef
1 parent 6684c7b commit a189a04

File tree

2 files changed

+11
-3
lines changed

2 files changed

+11
-3
lines changed

dpctl/_sycl_core.pxd

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -121,3 +121,5 @@ cdef class SyclQueue:
121121

122122

123123
cpdef SyclQueue get_current_queue()
124+
cpdef get_current_device_type ()
125+
cpdef get_current_backend()

dpctl/sycl_core.pyx

Lines changed: 9 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -616,12 +616,12 @@ cdef class _SyclRTManager:
616616
"""
617617
print(self._backend_str_ty_dict.keys())
618618

619-
def get_current_backend (self):
619+
cpdef get_current_backend (self):
620620
""" Returns the backend for the current queue as `backend_type` enum
621621
"""
622622
return self.get_current_queue().get_sycl_backend()
623623

624-
def get_current_device_type (self):
624+
cpdef get_current_device_type (self):
625625
''' Returns current device type as `device_type` enum
626626
'''
627627
return self.get_current_queue().get_sycl_device().get_device_type()
@@ -736,7 +736,6 @@ _mgr = _SyclRTManager()
736736

737737
# Global bound functions
738738
dump = _mgr.dump
739-
get_current_device_type = _mgr.get_current_device_type
740739
get_num_platforms = _mgr.get_num_platforms
741740
get_num_activated_queues = _mgr.get_num_activated_queues
742741
get_num_queues = _mgr.get_num_queues
@@ -750,6 +749,13 @@ cpdef SyclQueue get_current_queue():
750749
''' Obtain current Sycl Queue from Data Parallel Control package '''
751750
return _mgr.get_current_queue()
752751

752+
cpdef get_current_device_type():
753+
''' Obtain current device type from Data Parallel Control package '''
754+
return _mgr.get_current_device_type()
755+
756+
cpdef get_current_backend():
757+
''' Obtain current backend type from Data Parallel Control package '''
758+
return _mgr.get_current_backend()
753759

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

0 commit comments

Comments
 (0)