OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-06 12:20:52 +01:00
Code Issues Actions 38 Packages Projects Releases Wiki Activity

[BE] Replace lib with TORCH_INSTALL_LIB_DIR (#158235)

Browse Source 
Their values are actually the same. Just staying in line with other `INSTALL` commands.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/158235
Approved by: https://github.com/Skylion007
ghstack dependencies: #158234
This commit is contained in:
Ke Wen 2025-07-14 08:05:18 -07:00 committed by PyTorch MergeBot
parent 2043f6911e
commit 5763ec5f8d
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
caffe2/CMakeLists.txt
View File

@ -1044,7 +1044,7 @@ elseif(USE_CUDA)
target_compile_definitions(torch_cuda PUBLIC USE_NVSHMEM)
target_compile_definitions(torch_nvshmem PUBLIC USE_NVSHMEM)
target_link_libraries(torch_cuda PRIVATE torch_nvshmem)
install(TARGETS torch_nvshmem EXPORT Caffe2Targets DESTINATION lib)
install(TARGETS torch_nvshmem EXPORT Caffe2Targets DESTINATION "${TORCH_INSTALL_LIB_DIR}")
else()
message(STATUS "NVSHMEM not found, not building with NVSHMEM support.")
endif()