Skip to content

Commit 96b6184

Browse files
Skylion007pytorchmergebot
authored andcommitted
[BE]: Update nvshmem to 3.4.5 (pytorch#164046)
Release notes can be found here: https://docs.nvidia.com/nvshmem/release-notes-install-guide/release-notes/release-3405.html main difference is the addition of a CPU assisted IBGDA fallback which should allow NVSHMEM IBGDA to work on way more systems without admin intervention and without using GDRCopy. Pull Request resolved: pytorch#164046 Approved by: https://github.com/ezyang, https://github.com/kwen2501
1 parent 1b655a8 commit 96b6184

File tree

4 files changed

+61
-61
lines changed

4 files changed

+61
-61
lines changed

.ci/docker/common/install_cuda.sh

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -10,7 +10,7 @@ else
1010
arch_path='sbsa'
1111
fi
1212

13-
NVSHMEM_VERSION=3.3.24
13+
NVSHMEM_VERSION=3.4.5
1414

1515
function install_cuda {
1616
version=$1

.github/scripts/generate_binary_build_matrix.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -56,7 +56,7 @@
5656
"nvidia-cusparse-cu12==12.5.4.2; platform_system == 'Linux' | "
5757
"nvidia-cusparselt-cu12==0.7.1; platform_system == 'Linux' | "
5858
"nvidia-nccl-cu12==2.27.5; platform_system == 'Linux' | "
59-
"nvidia-nvshmem-cu12==3.3.24; platform_system == 'Linux' | "
59+
"nvidia-nvshmem-cu12==3.4.5; platform_system == 'Linux' | "
6060
"nvidia-nvtx-cu12==12.6.77; platform_system == 'Linux' | "
6161
"nvidia-nvjitlink-cu12==12.6.85; platform_system == 'Linux' | "
6262
"nvidia-cufile-cu12==1.11.1.6; platform_system == 'Linux'"
@@ -73,7 +73,7 @@
7373
"nvidia-cusparse-cu12==12.5.8.93; platform_system == 'Linux' | "
7474
"nvidia-cusparselt-cu12==0.7.1; platform_system == 'Linux' | "
7575
"nvidia-nccl-cu12==2.27.5; platform_system == 'Linux' | "
76-
"nvidia-nvshmem-cu12==3.3.24; platform_system == 'Linux' | "
76+
"nvidia-nvshmem-cu12==3.4.5; platform_system == 'Linux' | "
7777
"nvidia-nvtx-cu12==12.8.90; platform_system == 'Linux' | "
7878
"nvidia-nvjitlink-cu12==12.8.93; platform_system == 'Linux' | "
7979
"nvidia-cufile-cu12==1.13.1.3; platform_system == 'Linux'"
@@ -90,7 +90,7 @@
9090
"nvidia-cusparse-cu12==12.5.10.65; platform_system == 'Linux' | "
9191
"nvidia-cusparselt-cu12==0.7.1; platform_system == 'Linux' | "
9292
"nvidia-nccl-cu12==2.27.5; platform_system == 'Linux' | "
93-
"nvidia-nvshmem-cu12==3.3.20; platform_system == 'Linux' | "
93+
"nvidia-nvshmem-cu12==3.4.5; platform_system == 'Linux' | "
9494
"nvidia-nvtx-cu12==12.9.79; platform_system == 'Linux' | "
9595
"nvidia-nvjitlink-cu12==12.9.86; platform_system == 'Linux' | "
9696
"nvidia-cufile-cu12==1.14.1.1; platform_system == 'Linux'"
@@ -107,7 +107,7 @@
107107
"nvidia-cusparse==12.6.3.3; platform_system == 'Linux' | "
108108
"nvidia-cusparselt-cu13==0.8.0; platform_system == 'Linux' | "
109109
"nvidia-nccl-cu13==2.27.7; platform_system == 'Linux' | "
110-
"nvidia-nvshmem-cu13==3.3.24; platform_system == 'Linux' | "
110+
"nvidia-nvshmem-cu13==3.4.5; platform_system == 'Linux' | "
111111
"nvidia-nvtx==13.0.85; platform_system == 'Linux' | "
112112
"nvidia-nvjitlink==13.0.88; platform_system == 'Linux' | "
113113
"nvidia-cufile==1.15.1.6; platform_system == 'Linux'"

0 commit comments

Comments
 (0)