Skip to content

Commit 9298b54

Browse files
committed
Avoid possible free of NULL pointer on Windows
1 parent 674e050 commit 9298b54

1 file changed

Lines changed: 3 additions & 1 deletion

File tree

cuda_core/cuda/core/system/_system_events.pyx

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -81,13 +81,15 @@ cdef class RegisteredSystemEvents:
8181

8282
initialize()
8383

84+
self._event_set = 0
8485
self._event_set = nvml.system_event_set_create()
8586
# If this raises, the event needs to be freed and this is handled by
8687
# this class's __dealloc__ method.
8788
nvml.system_register_events(event_bitmask, self._event_set)
8889

8990
def __dealloc__(self):
90-
nvml.system_event_set_free(self._event_set)
91+
if self._event_set != 0:
92+
nvml.system_event_set_free(self._event_set)
9193

9294
def wait(self, timeout_ms: int = 0, buffer_size: int = 1) -> SystemEvents:
9395
"""

0 commit comments

Comments
 (0)