From b185131db9707840e4ca1246bea7a06db8614dfd Mon Sep 17 00:00:00 2001 From: Graham Markall Date: Fri, 18 Apr 2025 06:46:50 +0100 Subject: [PATCH] Restore PR #185 (Stop Certain Driver API Discovery for "v2") This was accidentally deleted in #145. --- 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 9c5237b30..2741d6ae7 100644 --- a/numba_cuda/numba/cuda/cudadrv/driver.py +++ b/numba_cuda/numba/cuda/cudadrv/driver.py @@ -381,6 +381,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}")