Skip to content

Commit 51f8222

Browse files
kwen2501kwen2501
committed
Update on "Add NVSHMEM to PYTORCH_EXTRA_INSTALL_REQUIREMENTS"
 NVSHMEM 3.2.5 (released Mar 2025) have both cu11 and cu12 builds. See: https://pypi.nvidia.com/nvidia-nvshmem-cu12/ https://pypi.nvidia.com/nvidia-nvshmem-cu11/ [ghstack-poisoned]
2 parents 266d9ad + 0b35dbb commit 51f8222

File tree

1 file changed

+0
-2
lines changed

1 file changed

+0
-2
lines changed

‎caffe2/CMakeLists.txt‎

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -991,8 +991,6 @@ elseif(USE_CUDA)
991991
# Otherwise, NVSHMEM_HOME is auto detected in tools/setup_helpers/cmake.py
992992
if($ENV{NVSHMEM_HOME})
993993
set(NVSHMEM_HOME $ENV{NVSHMEM_HOME} CACHE PATH "Path to NVSHMEM build dir")
994-
else()
995-
message(WARNING "NVSHMEM_HOME not found. Building without NVSHMEM.")
996994
endif()
997995

998996
if(USE_NVSHMEM AND DEFINED NVSHMEM_HOME)

0 commit comments

Comments
 (0)