build: use importlib.metadata to locate cuda-pathfinder instead of scanning sys.path

Made-with: Cursor

Author: rwgk

cuda_bindings/build_hooks.py

@@ -44,18 +44,22 @@ def _import_get_cuda_path_or_home():
     except ModuleNotFoundError as exc:
         if exc.name != "cuda.pathfinder":
             raise
+        from importlib.metadata import PackageNotFoundError, distribution
+        from pathlib import Path
+
         import cuda
 
-        for p in sys.path:
-            sp_cuda = os.path.join(p, "cuda")
-            if os.path.isdir(os.path.join(sp_cuda, "pathfinder")):
-                cuda.__path__ = list(cuda.__path__) + [sp_cuda]
-                break
-        else:
+        try:
+            dist = distribution("cuda-pathfinder")
+        except PackageNotFoundError:
             raise ModuleNotFoundError(
                 "cuda-pathfinder is not installed in the build environment. "
                 "Ensure 'cuda-pathfinder>=1.5' is in build-system.requires."
-            )
+            ) from None
+        site_cuda = str(dist.locate_file(Path("cuda")))
+        cuda_paths = list(cuda.__path__)
+        if site_cuda not in cuda_paths:
+            cuda.__path__ = cuda_paths + [site_cuda]
         import cuda.pathfinder
 
     return cuda.pathfinder.get_cuda_path_or_home

cuda_core/build_hooks.py

@@ -39,18 +39,22 @@ def _import_get_cuda_path_or_home():
     except ModuleNotFoundError as exc:
         if exc.name != "cuda.pathfinder":
             raise
+        from importlib.metadata import PackageNotFoundError, distribution
+        from pathlib import Path
+
         import cuda
 
-        for p in sys.path:
-            sp_cuda = os.path.join(p, "cuda")
-            if os.path.isdir(os.path.join(sp_cuda, "pathfinder")):
-                cuda.__path__ = list(cuda.__path__) + [sp_cuda]
-                break
-        else:
+        try:
+            dist = distribution("cuda-pathfinder")
+        except PackageNotFoundError:
             raise ModuleNotFoundError(
                 "cuda-pathfinder is not installed in the build environment. "
                 "Ensure 'cuda-pathfinder>=1.5' is in build-system.requires."
-            )
+            ) from None
+        site_cuda = str(dist.locate_file(Path("cuda")))
+        cuda_paths = list(cuda.__path__)
+        if site_cuda not in cuda_paths:
+            cuda.__path__ = cuda_paths + [site_cuda]
         import cuda.pathfinder
 
     return cuda.pathfinder.get_cuda_path_or_home