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
This folder contains various custom cmake modules for finding libraries and packages. Details about some of them are listed below.
FindOpenMP.cmake
This is modified from the file included in CMake 3.13 release, with the following changes:
-
Replace
VERSION_GREATER_EQUAL
withNOT ... VERSION_LESS
asVERSION_GREATER_EQUAL
is not supported in CMake 3.5 (our min supported version). -
Update the
separate_arguments
commands to not useNATIVE_COMMAND
which is not supported in CMake 3.5 (our min supported version). -
Make it respect the
QUIET
flag so that, when it is set,try_compile
failures are not reported. -
For
AppleClang
compilers, use-Xpreprocessor
instead of-Xclang
as the later is not documented. -
For
AppleClang
compilers, an extra flag option is tried, which is-Xpreprocessor -openmp -I${DIR_OF_omp_h}
, where${DIR_OF_omp_h}
is a obtained usingfind_path
onomp.h
withbrew
's default include directory as a hint. Without this, the compiler will complain about missing headers as they are not natively included in Apple's LLVM. -
For non-GNU compilers, whenever we try a candidate OpenMP flag, first try it with directly linking MKL's
libomp
if it has one. Otherwise, we may end up linking twolibomp
s and end up with this nasty error:OMP: Error #15: Initializing libomp.dylib, but found libiomp5.dylib already initialized. OMP: Hint This means that multiple copies of the OpenMP runtime have been linked into the program. That is dangerous, since it can degrade performance or cause incorrect results. The best thing to do is to ensure that only a single OpenMP runtime is linked into the process, e.g. by avoiding static linking of the OpenMP runtime in any library. As an unsafe, unsupported, undocumented workaround you can set the environment variable KMP_DUPLICATE_LIB_OK=TRUE to allow the program to continue to execute, but that may cause crashes or silently produce incorrect results. For more information, please see http://openmp.llvm.org/
See NOTE [ Linking both MKL and OpenMP ] for details.