@@ -20,12 +20,12 @@ def __init__(self):
2020 self ._lock = threading .RLock ()
2121
2222 def acquire (self ):
23- ev .start_event ("numba. cuda:compiler_lock" )
23+ ev .start_event ("numba- cuda:compiler_lock" )
2424 self ._lock .acquire ()
2525
2626 def release (self ):
2727 self ._lock .release ()
28- ev .end_event ("numba. cuda:compiler_lock" )
28+ ev .end_event ("numba- cuda:compiler_lock" )
2929
3030 def __enter__ (self ):
3131 self .acquire ()
@@ -70,14 +70,12 @@ def __init__(self, cuda_lock, numba_lock):
7070 self ._numba_lock = numba_lock
7171
7272 def acquire (self ):
73- if self ._numba_lock :
74- self ._numba_lock .acquire ()
73+ self ._numba_lock .acquire ()
7574 self ._cuda_lock .acquire ()
7675
7776 def release (self ):
7877 self ._cuda_lock .release ()
79- if self ._numba_lock :
80- self ._numba_lock .release ()
78+ self ._numba_lock .release ()
8179
8280 def __enter__ (self ):
8381 self .acquire ()
@@ -86,10 +84,7 @@ def __exit__(self, exc_val, exc_type, traceback):
8684 self .release ()
8785
8886 def is_locked (self ):
89- cuda_locked = self ._cuda_lock .is_locked ()
90- if self ._numba_lock :
91- return cuda_locked and self ._numba_lock .is_locked ()
92- return cuda_locked
87+ return self ._cuda_lock .is_locked () and self ._numba_lock .is_locked ()
9388
9489 def __call__ (self , func ):
9590 @functools .wraps (func )
@@ -107,9 +102,3 @@ def _acquire_compile_lock(*args, **kwargs):
107102 )
108103else :
109104 global_compiler_lock = _numba_cuda_compiler_lock
110-
111-
112- def require_global_compiler_lock ():
113- """Sentry that checks the global_compiler_lock is acquired."""
114- # Use assert to allow turning off this check
115- assert global_compiler_lock .is_locked ()
0 commit comments