Skip to content

Commit

Permalink
Update to CUDA 12.4 (#62)
Browse files Browse the repository at this point in the history
This PR updates pynvjitlink to use CUDA 12.4. This depends on
conda-forge/libnvjitlink-feedstock#9.
  • Loading branch information
bdice authored Mar 7, 2024
1 parent 72a76ee commit e1a8ecf
Show file tree
Hide file tree
Showing 3 changed files with 3 additions and 7 deletions.
2 changes: 1 addition & 1 deletion ci/install_latest_cuda_toolkit.sh
Original file line number Diff line number Diff line change
Expand Up @@ -17,4 +17,4 @@ else
exit 1
fi

yum install -y cuda-toolkit-12-3
yum install -y cuda-toolkit-12-4
6 changes: 1 addition & 5 deletions ci/test_wheel.sh
Original file line number Diff line number Diff line change
Expand Up @@ -14,15 +14,11 @@ rapids-logger "Download Wheel"
RAPIDS_PY_CUDA_SUFFIX="$(rapids-wheel-ctk-name-gen ${RAPIDS_CUDA_VERSION})"
RAPIDS_PY_WHEEL_NAME="pynvjitlink_${RAPIDS_PY_CUDA_SUFFIX}" rapids-download-wheels-from-s3 ./dist/

# This is the version of the suffix with a preceding hyphen. It's used
# everywhere except in the final wheel name.
PACKAGE_CUDA_SUFFIX="-${RAPIDS_PY_CUDA_SUFFIX}"

RAPIDS_TESTS_DIR=${RAPIDS_TESTS_DIR:-"${PWD}/test-results"}/
mkdir -p "${RAPIDS_TESTS_DIR}"

rapids-logger "Install wheel"
python -m pip install --find-links ./dist pynvjitlink${PACKAGE_CUDA_SUFFIX}
python -m pip install $(echo ./dist/pynvjitlink_${RAPIDS_PY_CUDA_SUFFIX}*.whl)

rapids-logger "Build Tests"
pushd test_binary_generation
Expand Down
2 changes: 1 addition & 1 deletion conda/recipes/pynvjitlink/conda_build_config.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -8,4 +8,4 @@ cuda_compiler:
- cuda-nvcc

cuda_compiler_version:
- 12.3
- 12.4

0 comments on commit e1a8ecf

Please sign in to comment.