Skip to content

Commit 6e08c9d

Browse files
authored
Avoid bumping certain driver API to avoid future breakage (#185)
Co-authored-by: isVoid <[email protected]>
1 parent 1185852 commit 6e08c9d

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

numba_cuda/numba/cuda/cudadrv/driver.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -365,6 +365,9 @@ def _find_api(self, fname):
365365
else:
366366
variants = ('_v2', '')
367367

368+
if fname in ("cuCtxGetDevice", "cuCtxSynchronize"):
369+
return getattr(self.lib, fname)
370+
368371
for variant in variants:
369372
try:
370373
return getattr(self.lib, f'{fname}{variant}')

0 commit comments

Comments
 (0)