Skip to content

Commit b33b809

Browse files
authored
Use RAFT 23.12 (#703)
* Use RAFT 23.12 * use cuda11.4 for building wheels
1 parent cb2a66d commit b33b809

2 files changed

Lines changed: 9 additions & 9 deletions

File tree

ci/install_cuda.sh

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -1,9 +1,9 @@
11
yum-config-manager --add-repo https://developer.download.nvidia.com/compute/cuda/repos/rhel7/x86_64/cuda-rhel7.repo
22
yum install --setopt=obsoletes=0 -y \
3-
cuda-nvcc-11-2 \
4-
cuda-cudart-devel-11-2 \
5-
libcurand-devel-11-2 \
6-
libcublas-devel-11-2 \
7-
cuda-nvprof-11-2 \
3+
cuda-nvcc-11-4 \
4+
cuda-cudart-devel-11-4 \
5+
libcurand-devel-11-4 \
6+
libcublas-devel-11-4 \
7+
cuda-nvprof-11-4 \
88
ninja-build
9-
ln -s cuda-11.2 /usr/local/cuda
9+
ln -s cuda-11.4 /usr/local/cuda

implicit/gpu/CMakeLists.txt

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -14,7 +14,7 @@ else()
1414
add_cython_target(_cuda CXX)
1515

1616
# use rapids-cmake to install dependencies
17-
file(DOWNLOAD https://raw.githubusercontent.com/rapidsai/rapids-cmake/branch-23.08/RAPIDS.cmake
17+
file(DOWNLOAD https://raw.githubusercontent.com/rapidsai/rapids-cmake/branch-23.12/RAPIDS.cmake
1818
${CMAKE_BINARY_DIR}/RAPIDS.cmake)
1919
include(${CMAKE_BINARY_DIR}/RAPIDS.cmake)
2020
include(rapids-cmake)
@@ -57,10 +57,10 @@ else()
5757
# get raft
5858
# note: we're using RAFT in header only mode right now - mainly to reduce binary
5959
# size of the compiled wheels
60-
rapids_cpm_find(raft 23.08
60+
rapids_cpm_find(raft 23.12
6161
CPM_ARGS
6262
GIT_REPOSITORY https://github.com/rapidsai/raft.git
63-
GIT_TAG branch-23.08
63+
GIT_TAG branch-23.12
6464
DOWNLOAD_ONLY YES
6565
)
6666
include_directories(${raft_SOURCE_DIR}/cpp/include)

0 commit comments

Comments
 (0)