Skip to content

Adds path for the CUDA static library based on CUDA_HOME #837

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
Andy-Jost merged 3 commits into from
Aug 19, 2025
Merged

Adds path for the CUDA static library based on CUDA_HOME #837

Changes from 1 commit
Commits
Show all changes
3 commits
Select commit Hold shift + click to select a range
b400e34
Adds paths for the CUDA static library based on CUDA_HOME (#608).
Andy-Jost Aug 14, 2025
82dcb06
Removes LIB and LIBRARY_PATH environment variables from the build-whe…
Andy-Jost Aug 19, 2025
73acddf
Updates Linux install to search both lib and lib64 directories for CU…
Andy-Jost Aug 19, 2025
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
3 changes: 3 additions & 0 deletions cuda_bindings/setup.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -31,6 +31,7 @@
raise RuntimeError("Environment variable CUDA_HOME or CUDA_PATH is not set")

CUDA_HOME = CUDA_HOME.split(os.pathsep)
Comment thread
leofang marked this conversation as resolved.
Outdated

if os.environ.get("PARALLEL_LEVEL") is not None:
warn(
"Environment variable PARALLEL_LEVEL is deprecated. Use CUDA_PYTHON_PARALLEL_LEVEL instead",
Expand Down Expand Up @@ -238,6 +239,8 @@ def generate_output(infile, local):
os.path.dirname(sysconfig.get_path("include")),
] + include_path_list
library_dirs = [sysconfig.get_path("platlib"), os.path.join(os.sys.prefix, "lib")]
cudalib_subdir = r"lib\x64" if sys.platform == "win32" else "lib64"

@leofang leofang Aug 19, 2025

Copy link
Copy Markdown
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

On the Linux side, one annoying thing is that depending on how CUDA is installed, this could be lib or lib64:

$ ls -l /usr/local/cuda-13.0/lib64/libcudart_static.a 
-rw-r--r-- 1 root root 1361104 Aug  5 09:32 /usr/local/cuda-13.0/lib64/libcudart_static.a
$ ls -l /usr/local/cuda-13.0/targets/x86_64-linux/lib/libcudart_static.a 
-rw-r--r-- 1 root root 1361104 Aug  5 09:32 /usr/local/cuda-13.0/targets/x86_64-linux/lib/libcudart_static.a

https://github.com/conda-forge/cuda-cudart-feedstock/blob/38fb08899691fa53fce33960ef37d48ab58584b0/recipe/meta.yaml#L220

https://github.com/conda-forge/cuda-cudart-feedstock/blob/38fb08899691fa53fce33960ef37d48ab58584b0/recipe/meta.yaml#L251

We probably should support both.

library_dirs.extend(os.path.join(prefix, cudalib_subdir) for prefix in CUDA_HOME)

extra_compile_args = []
extra_cythonize_kwargs = {}
Expand Down
Loading