From b01248d4d69a67b1afc2dd2139e82230413b8f2c Mon Sep 17 00:00:00 2001 From: isVoid Date: Tue, 1 Apr 2025 15:17:29 -0700 Subject: [PATCH] Avoid bumping certain driver API to avoid future breakage --- numba_cuda/numba/cuda/cudadrv/driver.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/numba_cuda/numba/cuda/cudadrv/driver.py b/numba_cuda/numba/cuda/cudadrv/driver.py index 1641bf779..233e9ed7f 100644 --- a/numba_cuda/numba/cuda/cudadrv/driver.py +++ b/numba_cuda/numba/cuda/cudadrv/driver.py @@ -365,6 +365,9 @@ def _find_api(self, fname): else: variants = ('_v2', '') + if fname in ("cuCtxGetDevice", "cuCtxSynchronize"): + return getattr(self.lib, fname) + for variant in variants: try: return getattr(self.lib, f'{fname}{variant}')