Skip to content

Commit 8b20f58

Browse files
kwen2501kwen2501
committed
Add NVSHMEM to PYTORCH_EXTRA_INSTALL_REQUIREMENTS
ghstack-source-id: c81bf59 Pull Request resolved: #154568
1 parent 841c93b commit 8b20f58

File tree

5 files changed

+23
-20
lines changed

5 files changed

+23
-20
lines changed

‎.ci/manywheel/build_cuda.sh‎

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -164,6 +164,7 @@ if [[ $CUDA_VERSION == 12* ]]; then
164164
'$ORIGIN/../../nvidia/cusparse/lib'
165165
'$ORIGIN/../../cusparselt/lib'
166166
'$ORIGIN/../../nvidia/nccl/lib'
167+
'$ORIGIN/../../nvidia/nvshmem/lib'
167168
'$ORIGIN/../../nvidia/nvtx/lib'
168169
'$ORIGIN/../../nvidia/cufile/lib'
169170
)

‎.github/scripts/generate_binary_build_matrix.py‎

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -67,6 +67,7 @@
6767
"nvidia-cusparse-cu12==12.5.4.2; platform_system == 'Linux' and platform_machine == 'x86_64' | "
6868
"nvidia-cusparselt-cu12==0.6.3; platform_system == 'Linux' and platform_machine == 'x86_64' | "
6969
"nvidia-nccl-cu12==2.26.5; platform_system == 'Linux' and platform_machine == 'x86_64' | "
70+
"nvidia-nvshmem-cu12==3.2.5; platform_system == 'Linux' and platform_machine == 'x86_64' | "
7071
"nvidia-nvtx-cu12==12.6.77; platform_system == 'Linux' and platform_machine == 'x86_64' | "
7172
"nvidia-nvjitlink-cu12==12.6.85; platform_system == 'Linux' and platform_machine == 'x86_64' | "
7273
"nvidia-cufile-cu12==1.11.1.6; platform_system == 'Linux' and platform_machine == 'x86_64'"
@@ -83,6 +84,7 @@
8384
"nvidia-cusparse-cu12==12.5.8.93; platform_system == 'Linux' and platform_machine == 'x86_64' | "
8485
"nvidia-cusparselt-cu12==0.6.3; platform_system == 'Linux' and platform_machine == 'x86_64' | "
8586
"nvidia-nccl-cu12==2.26.5; platform_system == 'Linux' and platform_machine == 'x86_64' | "
87+
"nvidia-nvshmem-cu12==3.2.5; platform_system == 'Linux' and platform_machine == 'x86_64' | "
8688
"nvidia-nvtx-cu12==12.8.90; platform_system == 'Linux' and platform_machine == 'x86_64' | "
8789
"nvidia-nvjitlink-cu12==12.8.93; platform_system == 'Linux' and platform_machine == 'x86_64' | "
8890
"nvidia-cufile-cu12==1.13.1.3; 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.

0 commit comments

Comments
 (0)