Skip to content

use cuda128 for build #757

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

Open
hongyx11 wants to merge 2 commits into
base: dev
Choose a base branch
from
Open

use cuda128 for build #757

Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
429da86
use cuda128 for build
Jun 4, 2026
ed04f7f
Update build environment to include 'exouser'
mrava87 Jun 4, 2026
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
46 changes: 19 additions & 27 deletions .github/workflows/buildcupy.yaml
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -10,7 +10,9 @@ on:

jobs:
build:
runs-on: self-hosted
runs-on:
- self-hosted
- exouser
steps:
- name: Check out source repository
uses: actions/checkout@v6
Expand All @@ -22,35 +24,25 @@ jobs:
python-version: 3.11
- name: Install dependencies and pylops
run: |
srun --account=yuxilab -n 1 -N 1 --gres=gpu:L40S:1 bash -c '
uv sync --locked --extra advanced \
--extra stat --extra gpu-cu12 \
--extra deep-cu128 --all-groups
'
uv sync --locked --extra advanced \
--extra stat --extra gpu-cu12 \
--extra deep-cu128 --all-groups
echo "done!"
- name: Install CUDA 12.8 toolkit
- name: Use local CUDA 12.8 toolkit
run: |
srun --account=yuxilab -n 1 -N 1 --gres=gpu:L40S:1 bash -c '
# Currently, the NVIDIA GPU drivers on the self-hosted runner
# and the CUDA installation in CUDA_HOME require different PTX versions
# (8.7 and 8.8, respectively). CUDA and NVCC are temporarely
# installed in the uv environment, and the CUDA paths are set in
# the test command, to work around this issue.
uv add nvidia-cuda-nvcc-cu12==12.8.*
# Numba resolves libNVVM via CUDA_HOME and needs a versioned
# libnvvm.so.N (the pip wheel only ships a bare libnvvm.so), so
# add the symlink once here against the wheel-provided toolkit.
ln -sf libnvvm.so .venv/lib/python3.11/site-packages/nvidia/cuda_nvcc/nvvm/lib64/libnvvm.so.4
'
if [ ! -d "$HOME/local/cuda" ]; then
echo "CUDA 12.8 toolkit not found at $HOME/local/cuda"
exit 1
fi
echo "CUDA_HOME=$HOME/local/cuda" >> "$GITHUB_ENV"
echo "done!"
- name: Test with pytest
run: |
srun --account=yuxilab -n 1 -N 1 --gres=gpu:L40S:1 bash -c '
# Use the pip-installed CUDA 12.8 toolkit (matches the 12.8 driver),
# not any system CUDA on the runner, so numba emits PTX <= 8.7.
export CUDA_HOME="$PWD/.venv/lib/python3.11/site-packages/nvidia/cuda_nvcc"
export CUPY_PYLOPS=1
export TEST_CUPY_PYLOPS=1
uv run pytest --color=yes pytests/
'
# Use the local CUDA 12.8 toolchain from $HOME/local/cuda.
export CUDA_HOME="${CUDA_HOME:-$HOME/local/cuda}"
export CUPY_PYLOPS=1
export TEST_CUPY_PYLOPS=1
export PATH="$CUDA_HOME/bin:$PATH"
export LD_LIBRARY_PATH="$CUDA_HOME/lib64:$LD_LIBRARY_PATH"
uv run pytest --color=yes pytests/
echo "done!"
Loading