wip

Author: rparolin

cuda_bindings/setup.py

@@ -13,7 +13,6 @@
 import tempfile
 from warnings import warn
 
-from cuda.pathfinder._utils.env_vars import get_cuda_home_or_path
 from Cython import Tempita
 from Cython.Build import cythonize
 from pyclibrary import CParser
@@ -27,7 +26,13 @@
 # ----------------------------------------------------------------------
 # Fetch configuration options
 
-CUDA_HOME = get_cuda_home_or_path()
+try:
+    from cuda.pathfinder._utils.env_vars import get_cuda_home_or_path
+    CUDA_HOME = get_cuda_home_or_path()
+except ImportError:
+    # Fallback for build environments where cuda-pathfinder may not be available
+    CUDA_HOME = os.environ.get("CUDA_HOME", os.environ.get("CUDA_PATH", None))
+
 if not CUDA_HOME:
     raise RuntimeError("Environment variable CUDA_PATH or CUDA_HOME is not set")
 

cuda_core/build_hooks.py

@@ -31,9 +31,13 @@ def _get_cuda_paths() -> list[str]:
     Supports multiple paths separated by os.pathsep (: on Unix, ; on Windows).
     Returns a list of paths for use in include_dirs and library_dirs.
     """
-    from cuda.pathfinder._utils.env_vars import get_cuda_home_or_path
-
-    CUDA_PATH = get_cuda_home_or_path()
+    try:
+        from cuda.pathfinder._utils.env_vars import get_cuda_home_or_path
+        CUDA_PATH = get_cuda_home_or_path()
+    except ImportError:
+        # Fallback for build environments where cuda-pathfinder may not be available
+        CUDA_PATH = os.environ.get("CUDA_PATH", os.environ.get("CUDA_HOME", None))
+    
     if not CUDA_PATH:
         raise RuntimeError("Environment variable CUDA_PATH or CUDA_HOME is not set")
     CUDA_PATH = CUDA_PATH.split(os.pathsep)