We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 674e050 commit 9298b54Copy full SHA for 9298b54
1 file changed
cuda_core/cuda/core/system/_system_events.pyx
@@ -81,13 +81,15 @@ cdef class RegisteredSystemEvents:
81
82
initialize()
83
84
+ self._event_set = 0
85
self._event_set = nvml.system_event_set_create()
86
# If this raises, the event needs to be freed and this is handled by
87
# this class's __dealloc__ method.
88
nvml.system_register_events(event_bitmask, self._event_set)
89
90
def __dealloc__(self):
- nvml.system_event_set_free(self._event_set)
91
+ if self._event_set != 0:
92
+ nvml.system_event_set_free(self._event_set)
93
94
def wait(self, timeout_ms: int = 0, buffer_size: int = 1) -> SystemEvents:
95
"""
0 commit comments