frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 21:14:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

Set USE_NVSHMEM only if USE_DISTRIBUTED is set (#161451)

Browse Source 
Pull Request resolved: https://github.com/pytorch/pytorch/pull/161451
Approved by: https://github.com/eqy
This commit is contained in:
lakshayg
2025-08-27 17:11:19 +00:00
committed by PyTorch MergeBot
parent 620d52e882
commit a3fa1b8c2a
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
CMakeLists.txt
View File

@ -272,7 +272,7 @@ cmake_dependent_option(USE_STATIC_NCCL "Use static NCCL" OFF "USE_NCCL" OFF)
cmake_dependent_option(USE_SYSTEM_NCCL "Use system-wide NCCL" OFF "USE_NCCL" cmake_dependent_option(USE_SYSTEM_NCCL "Use system-wide NCCL" OFF "USE_NCCL"
OFF) OFF)
cmake_dependent_option(USE_NVSHMEM "Use NVSHMEM" ON cmake_dependent_option(USE_NVSHMEM "Use NVSHMEM" ON
"USE_CUDA OR USE_ROCM;UNIX;NOT APPLE" OFF) "USE_DISTRIBUTED;USE_CUDA OR USE_ROCM;UNIX;NOT APPLE" OFF)
option(USE_NNAPI "Use NNAPI" OFF) option(USE_NNAPI "Use NNAPI" OFF)
option(USE_NNPACK "Use NNPACK" ON) option(USE_NNPACK "Use NNPACK" ON)
cmake_dependent_option(USE_NUMA "Use NUMA. Only available on Linux." ON "LINUX" cmake_dependent_option(USE_NUMA "Use NUMA. Only available on Linux." ON "LINUX"