build: getattr fallback for cuda-pathfinder < 1.5

cuda-pathfinder 1.4.x (currently on PyPI) does not expose
get_cuda_path_or_home yet. Fall back to os.environ.get via getattr
so the namespace fix can be tested on CI now.

Made-with: Cursor

Author: rwgk

cuda_bindings/build_hooks.py

@@ -47,7 +47,11 @@ def _import_get_cuda_path_or_home():
     except ModuleNotFoundError:
         pass
     else:
-        return pathfinder.get_cuda_path_or_home
+        return getattr(
+            pathfinder,
+            "get_cuda_path_or_home",
+            lambda: os.environ.get("CUDA_PATH", os.environ.get("CUDA_HOME")),
+        )
 
     import cuda
 
@@ -57,9 +61,13 @@ def _import_get_cuda_path_or_home():
             cuda.__path__ = list(cuda.__path__) + [sp_cuda]
             break
 
-    from cuda.pathfinder import get_cuda_path_or_home
+    from cuda import pathfinder
 
-    return get_cuda_path_or_home
+    return getattr(
+        pathfinder,
+        "get_cuda_path_or_home",
+        lambda: os.environ.get("CUDA_PATH", os.environ.get("CUDA_HOME")),
+    )
 
 
 @functools.cache

cuda_core/build_hooks.py

@@ -42,7 +42,11 @@ def _import_get_cuda_path_or_home():
     except ModuleNotFoundError:
         pass
     else:
-        return pathfinder.get_cuda_path_or_home
+        return getattr(
+            pathfinder,
+            "get_cuda_path_or_home",
+            lambda: os.environ.get("CUDA_PATH", os.environ.get("CUDA_HOME")),
+        )
 
     import cuda
 
@@ -52,9 +56,13 @@ def _import_get_cuda_path_or_home():
             cuda.__path__ = list(cuda.__path__) + [sp_cuda]
             break
 
-    from cuda.pathfinder import get_cuda_path_or_home
+    from cuda import pathfinder
 
-    return get_cuda_path_or_home
+    return getattr(
+        pathfinder,
+        "get_cuda_path_or_home",
+        lambda: os.environ.get("CUDA_PATH", os.environ.get("CUDA_HOME")),
+    )
 
 
 @functools.cache