Skip to content

Commit 7a3dbb4

Browse files
committed
Purge node cache on destroy to prevent stale identity lookups
Made-with: Cursor
1 parent 6411881 commit 7a3dbb4

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

cuda_core/cuda/core/_graph/_graph_def/_graph_node.pyx

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -153,6 +153,7 @@ cdef class GraphNode:
153153
return
154154
with nogil:
155155
HANDLE_RETURN(cydriver.cuGraphDestroyNode(node))
156+
_node_cache.pop(<uintptr_t>self._h_node.get(), None)
156157
invalidate_graph_node_handle(self._h_node)
157158

158159
@property

0 commit comments

Comments
 (0)