Skip to content

Commit 3863bbb

Browse files
Skylion007pytorchmergebot
Skylion007
authored and
pytorchmergebot
committed
[BE]: Update cusparselt to 0.7.1 (#155232)
Needed to support sparse operations on Blackwell, and implements new features for the library. Also optimizes library sizes vs 0.7 Pull Request resolved: #155232 Approved by: https://github.com/nWEIdia, https://github.com/malfet
1 parent 79bdafe commit 3863bbb

7 files changed

+26
-25
lines changed

‎.ci/docker/common/install_cuda.sh

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -56,7 +56,7 @@ function install_118 {
5656

5757
function install_126 {
5858
CUDNN_VERSION=9.5.1.17
59-
echo "Installing CUDA 12.6.3 and cuDNN ${CUDNN_VERSION} and NCCL and cuSparseLt-0.6.3"
59+
echo "Installing CUDA 12.6.3 and cuDNN ${CUDNN_VERSION} and NCCL and cuSparseLt-0.7.1"
6060
install_cuda 12.6.3 cuda_12.6.3_560.35.05_linux
6161

6262
install_cudnn 12 $CUDNN_VERSION
@@ -135,7 +135,7 @@ function prune_126 {
135135

136136
function install_128 {
137137
CUDNN_VERSION=9.8.0.87
138-
echo "Installing CUDA 12.8.1 and cuDNN ${CUDNN_VERSION} and NCCL and cuSparseLt-0.6.3"
138+
echo "Installing CUDA 12.8.1 and cuDNN ${CUDNN_VERSION} and NCCL and cuSparseLt-0.7.1"
139139
# install CUDA 12.8.1 in the same container
140140
install_cuda 12.8.1 cuda_12.8.1_570.124.06_linux
141141

‎.ci/docker/common/install_cusparselt.sh

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -11,7 +11,7 @@ if [[ ${CUDA_VERSION:0:4} =~ ^12\.[5-8]$ ]]; then
1111
if [ ${TARGETARCH} = 'amd64' ] || [ "${TARGETARCH}" = 'x86_64' ]; then
1212
arch_path='x86_64'
1313
fi
14-
CUSPARSELT_NAME="libcusparse_lt-linux-${arch_path}-0.6.3.2-archive"
14+
CUSPARSELT_NAME="libcusparse_lt-linux-${arch_path}-0.7.1.0-archive"
1515
curl --retry 3 -OLs https://developer.download.nvidia.com/compute/cusparselt/redist/libcusparse_lt/linux-${arch_path}/${CUSPARSELT_NAME}.tar.xz
1616
elif [[ ${CUDA_VERSION:0:4} == "11.8" ]]; then
1717
CUSPARSELT_NAME="libcusparse_lt-linux-x86_64-0.4.0.7-archive"

‎.ci/manywheel/build_cuda.sh

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -165,6 +165,7 @@ if [[ $CUDA_VERSION == 12* ]]; then
165165
'$ORIGIN/../../nvidia/curand/lib'
166166
'$ORIGIN/../../nvidia/cusolver/lib'
167167
'$ORIGIN/../../nvidia/cusparse/lib'
168+
'$ORIGIN/../../nvidia/cusparselt/lib'
168169
'$ORIGIN/../../cusparselt/lib'
169170
'$ORIGIN/../../nvidia/nccl/lib'
170171
'$ORIGIN/../../nvidia/nvshmem/lib'

‎.github/scripts/generate_binary_build_matrix.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -65,7 +65,7 @@
6565
"nvidia-curand-cu12==10.3.7.77; platform_system == 'Linux' and platform_machine == 'x86_64' | "
6666
"nvidia-cusolver-cu12==11.7.1.2; platform_system == 'Linux' and platform_machine == 'x86_64' | "
6767
"nvidia-cusparse-cu12==12.5.4.2; platform_system == 'Linux' and platform_machine == 'x86_64' | "
68-
"nvidia-cusparselt-cu12==0.6.3; platform_system == 'Linux' and platform_machine == 'x86_64' | "
68+
"nvidia-cusparselt-cu12==0.7.1; platform_system == 'Linux' and platform_machine == 'x86_64' | "
6969
"nvidia-nccl-cu12==2.26.5; platform_system == 'Linux' and platform_machine == 'x86_64' | "
7070
"nvidia-nvshmem-cu12==3.2.5; platform_system == 'Linux' and platform_machine == 'x86_64' | "
7171
"nvidia-nvtx-cu12==12.6.77; platform_system == 'Linux' and platform_machine == 'x86_64' | "
@@ -82,7 +82,7 @@
8282
"nvidia-curand-cu12==10.3.9.90; platform_system == 'Linux' and platform_machine == 'x86_64' | "
8383
"nvidia-cusolver-cu12==11.7.3.90; platform_system == 'Linux' and platform_machine == 'x86_64' | "
8484
"nvidia-cusparse-cu12==12.5.8.93; platform_system == 'Linux' and platform_machine == 'x86_64' | "
85-
"nvidia-cusparselt-cu12==0.6.3; platform_system == 'Linux' and platform_machine == 'x86_64' | "
85+
"nvidia-cusparselt-cu12==0.7.1; platform_system == 'Linux' and platform_machine == 'x86_64' | "
8686
"nvidia-nccl-cu12==2.26.5; platform_system == 'Linux' and platform_machine == 'x86_64' | "
8787
"nvidia-nvshmem-cu12==3.2.5; platform_system == 'Linux' and platform_machine == 'x86_64' | "
8888
"nvidia-nvtx-cu12==12.8.90; platform_system == 'Linux' and platform_machine == 'x86_64' | "

‎.github/workflows/generated-linux-aarch64-binary-manywheel-nightly.yml

Lines changed: 6 additions & 6 deletions
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

‎.github/workflows/generated-linux-binary-manywheel-main.yml

Lines changed: 2 additions & 2 deletions
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

0 commit comments

Comments
 (0)