We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 35d1722 commit 2ddbe6bCopy full SHA for 2ddbe6b
1 file changed
cuda_core/cuda/core/system/_event.pxi
@@ -95,6 +95,8 @@ cdef class DeviceEvents:
95
cdef intptr_t _device_handle
96
97
def __init__(self, device_handle: intptr_t, events: EventType | str | list[EventType | str]):
98
+ self._event_set = 0
99
+
100
cdef unsigned long long event_bitmask
101
if isinstance(events, (str, EventType)):
102
events = [events]
@@ -120,7 +122,8 @@ cdef class DeviceEvents:
120
122
nvml.device_register_events(self._device_handle, event_bitmask, self._event_set)
121
123
124
def __dealloc__(self):
- nvml.event_set_free(self._event_set)
125
+ if self._event_set != 0:
126
+ nvml.event_set_free(self._event_set)
127
128
def wait(self, timeout_ms: int = 0) -> EventData:
129
"""
0 commit comments