Skip to content

Fix cuda.bindings: lock lazy init of _cyb_func_ptrs in driver _inspect_function_pointer #3406

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
lijinf2 wants to merge 4 commits into
base: main
Choose a base branch
from
Open

Fix cuda.bindings: lock lazy init of _cyb_func_ptrs in driver _inspect_function_pointer #3406

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
4 commits
Select commit Hold shift + click to select a range
8672ee4
add locking for _cyb_func_ptrs
lijinf2 Nov 6, 2026
3f2a5c8
use cybind gen
lijinf2 Nov 6, 2026
ab8ac53
let read do not use lock
lijinf2 Nov 7, 2026
45c27f2
reroute both entry point _inspect_function_pointer(name) and _inspect…
lijinf2 Nov 7, 2026
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
23 changes: 13 additions & 10 deletions cuda_bindings_13/cuda/bindings/_internal/driver_linux.pyx
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -3,7 +3,8 @@
# SPDX-License-Identifier: Apache-2.0
#
# This code was automatically generated across versions from 12.9.0 to 13.5.0. Do not modify it directly.
# CYTHON-BINDINGS-GENERATED-DO-NOT-MODIFY-THIS-FILE: format=1; content-sha256=855d1d0530ab58f150ca5fb8b9ec43262cf0df681f507e2b53feeb6cf77c7c17
# !!! WARNING: THIS FILE CONTAINS PRERELEASE APIs !!!
# CYTHON-BINDINGS-GENERATED-DO-NOT-MODIFY-THIS-FILE: format=1; content-sha256=bcb0306e1a5e714a98ec08dcce8b16f01459e21d6d2df027c1e7b5df0e392198


# <<<< PREAMBLE CONTENT >>>>
Expand Down Expand Up @@ -52,6 +53,7 @@ ctypedef int (*_cyb_cuGetProcAddress_v2_T)(const char *, void **, int, cuuint64_

cdef int _cyb___py_driver_init = 0
cdef dict _cyb_func_ptrs = None
cdef object _cyb_func_ptrs_lock = _cyb_threading.Lock()
cdef object _cyb_symbol_lock = _cyb_threading.Lock()

# <<<< END OF PREAMBLE CONTENT >>>>
Expand Down Expand Up @@ -2201,11 +2203,7 @@ cdef inline int _check_or_init_driver() except -1 nogil:
return _init_driver()


cpdef dict _inspect_function_pointers():
global _cyb_func_ptrs
if _cyb_func_ptrs is not None:
return _cyb_func_ptrs

cdef dict _build_function_pointers():
_check_or_init_driver()
cdef dict data = {}
global __cuGetErrorString
Expand Down Expand Up @@ -3785,15 +3783,20 @@ cpdef dict _inspect_function_pointers():

global __cuTensorMapSetAllocationClass
data["__cuTensorMapSetAllocationClass"] = <intptr_t>__cuTensorMapSetAllocationClass
_cyb_func_ptrs = data
return data


cpdef _inspect_function_pointer(str name):
cpdef dict _inspect_function_pointers():
global _cyb_func_ptrs
if _cyb_func_ptrs is None:
_cyb_func_ptrs = _inspect_function_pointers()
return _cyb_func_ptrs[name]
with _cyb_func_ptrs_lock:
if _cyb_func_ptrs is None:
_cyb_func_ptrs = _build_function_pointers()
return _cyb_func_ptrs


cpdef _inspect_function_pointer(str name):
return _inspect_function_pointers()[name]



Expand Down
23 changes: 13 additions & 10 deletions cuda_bindings_13/cuda/bindings/_internal/driver_windows.pyx
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -3,7 +3,8 @@
# SPDX-License-Identifier: Apache-2.0
#
# This code was automatically generated across versions from 12.9.0 to 13.5.0. Do not modify it directly.
# CYTHON-BINDINGS-GENERATED-DO-NOT-MODIFY-THIS-FILE: format=1; content-sha256=265db3de415ec26f7b9cfd9961c846938d258ff2c840d500316a3fd24266d6ac
# !!! WARNING: THIS FILE CONTAINS PRERELEASE APIs !!!
# CYTHON-BINDINGS-GENERATED-DO-NOT-MODIFY-THIS-FILE: format=1; content-sha256=8b56814f7e4876ba0386c167c8bd1f0f024f3c5c5350557b4dfc72446fe38979


# <<<< PREAMBLE CONTENT >>>>
Expand Down Expand Up @@ -53,6 +54,7 @@ ctypedef int (*_cyb_cuGetProcAddress_v2_T)(const char *, void **, int, cuuint64_

cdef int _cyb___py_driver_init = 0
cdef dict _cyb_func_ptrs = None
cdef object _cyb_func_ptrs_lock = _cyb_threading.Lock()
cdef object _cyb_symbol_lock = _cyb_threading.Lock()

# <<<< END OF PREAMBLE CONTENT >>>>
Expand Down Expand Up @@ -2204,11 +2206,7 @@ cdef inline int _check_or_init_driver() except -1 nogil:
return _init_driver()


cpdef dict _inspect_function_pointers():
global _cyb_func_ptrs
if _cyb_func_ptrs is not None:
return _cyb_func_ptrs

cdef dict _build_function_pointers():
_check_or_init_driver()
cdef dict data = {}
global __cuGetErrorString
Expand Down Expand Up @@ -3788,15 +3786,20 @@ cpdef dict _inspect_function_pointers():

global __cuTensorMapSetAllocationClass
data["__cuTensorMapSetAllocationClass"] = <intptr_t>__cuTensorMapSetAllocationClass
_cyb_func_ptrs = data
return data


cpdef _inspect_function_pointer(str name):
cpdef dict _inspect_function_pointers():
global _cyb_func_ptrs
if _cyb_func_ptrs is None:
_cyb_func_ptrs = _inspect_function_pointers()
return _cyb_func_ptrs[name]
with _cyb_func_ptrs_lock:
if _cyb_func_ptrs is None:
_cyb_func_ptrs = _build_function_pointers()
return _cyb_func_ptrs


cpdef _inspect_function_pointer(str name):
return _inspect_function_pointers()[name]



Expand Down
Loading