release gil for record

Author: leofang

cuda_core/cuda/core/experimental/_event.pxd

@@ -0,0 +1,14 @@
+from cuda.bindings cimport cydriver
+
+
+cdef class Event:
+
+    cdef:
+        cydriver.CUevent _handle
+        bint _timing_disabled
+        bint _busy_waited
+        int _device_id
+        object _ctx_handle
+
+    cdef _shutdown_safe_close(self, is_shutting_down=*)
+    cpdef close(self)

cuda_core/cuda/core/experimental/_event.pyx

@@ -81,13 +81,6 @@ cdef class Event:
     and they should instead be created through a :obj:`~_stream.Stream` object.
 
     """
-    cdef:
-        cydriver.CUevent _handle
-        bint _timing_disabled
-        bint _busy_waited
-        int _device_id
-        object _ctx_handle
-
     def __cinit__(self):
         self._handle = <cydriver.CUevent>(NULL)
 

cuda_core/cuda/core/experimental/_stream.pyx

@@ -8,6 +8,7 @@ from libc.stdint cimport uintptr_t
 
 from cuda.bindings cimport cydriver
 
+from cuda.core.experimental._event cimport Event as cyEvent
 from cuda.core.experimental._utils.cuda_utils cimport (
     check_or_create_options,
     HANDLE_RETURN,
@@ -292,8 +293,9 @@ cdef class Stream:
         if event is None:
             self._get_device_and_context()
             event = Event._init(self._device_id, self._ctx_handle, options)
-        # TODO: revisit after Event is cythonized
-        HANDLE_RETURN(cydriver.cuEventRecord(<cydriver.CUevent><uintptr_t>(event.handle), self._handle))
+        cdef cydriver.CUevent e = (<cyEvent?>(event))._handle
+        with nogil:
+            HANDLE_RETURN(cydriver.cuEventRecord(e, self._handle))
         return event
 
     def wait(self, event_or_stream: Union[Event, Stream]):

cuda_core/tests/test_stream.py

@@ -51,7 +51,7 @@ def test_stream_record(init_cuda):
 
 def test_stream_record_invalid_event(init_cuda):
     stream = Device().create_stream(options=StreamOptions())
-    with pytest.raises(AttributeError):
+    with pytest.raises(TypeError):
         stream.record(event="invalid_event")