mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 21:14:14 +08:00
93da9952a7
Fix incorrect linking of Gloo's libraries when building with system Gloo. Previously, either Gloo's native library or Gloo's CUDA library were linked. However, Gloo had changed such that all users of Gloo must link the native library, and can optionally link the CUDA or HIP library for Gloo + CUDA/HIP support. This had been updated when building/linking with vendored Gloo, but not when using system Gloo. Fixes: #146239 Reported-by: Adam J Stewart <ajstewart426@gmail.com> Pull Request resolved: https://github.com/pytorch/pytorch/pull/146637 Approved by: https://github.com/malfet
43 lines
1.1 KiB
CMake
43 lines
1.1 KiB
CMake
# Try to find the Gloo library and headers.
|
|
# Gloo_FOUND - system has Gloo lib
|
|
# Gloo_INCLUDE_DIRS - the Gloo include directory
|
|
# Gloo_NATIVE_LIBRARY - base gloo library, needs to be linked
|
|
# Gloo_CUDA_LIBRARY/Gloo_HIP_LIBRARY - CUDA/HIP support library in Gloo
|
|
|
|
find_path(Gloo_INCLUDE_DIR
|
|
NAMES gloo/common/common.h
|
|
DOC "The directory where Gloo includes reside"
|
|
)
|
|
|
|
find_library(Gloo_NATIVE_LIBRARY
|
|
NAMES gloo
|
|
DOC "The Gloo library"
|
|
)
|
|
|
|
# Gloo has optional CUDA support
|
|
# if Gloo + CUDA is desired, Gloo_CUDA_LIBRARY
|
|
# needs to be linked into desired target
|
|
find_library(Gloo_CUDA_LIBRARY
|
|
NAMES gloo_cuda
|
|
DOC "Gloo's CUDA support/code"
|
|
)
|
|
|
|
# Gloo has optional HIP support
|
|
# if Gloo + HIP is desired, Gloo_HIP_LIBRARY
|
|
# needs to be linked to desired target
|
|
find_library(Gloo_HIP_LIBRARY
|
|
NAMES gloo_hiop
|
|
DOC "Gloo's HIP support/code"
|
|
)
|
|
|
|
set(Gloo_INCLUDE_DIRS ${Gloo_INCLUDE_DIR})
|
|
|
|
|
|
include(FindPackageHandleStandardArgs)
|
|
find_package_handle_standard_args(Gloo
|
|
FOUND_VAR Gloo_FOUND
|
|
REQUIRED_VARS Gloo_INCLUDE_DIR Gloo_NATIVE_LIBRARY
|
|
)
|
|
|
|
mark_as_advanced(Gloo_FOUND)
|