We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 39c085c commit 2102c1eCopy full SHA for 2102c1e
1 file changed
cuda_core/cuda/core/system/_event.pxi
@@ -144,13 +144,15 @@ cdef class DeviceEvents:
144
raise TypeError("events must be an EventType, str, or list of EventType or str")
145
146
self._device_handle = device_handle
147
+ self._event_set = 0
148
self._event_set = nvml.event_set_create()
149
# If this raises, the event needs to be freed and this is handled by
150
# this class's __dealloc__ method.
151
nvml.device_register_events(self._device_handle, event_bitmask, self._event_set)
152
153
def __dealloc__(self):
- nvml.event_set_free(self._event_set)
154
+ if self._event_set != 0:
155
+ nvml.event_set_free(self._event_set)
156
157
def wait(self, timeout_ms: int = 0) -> EventData:
158
"""
0 commit comments