mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-21 05:34:18 +08:00
642e9305eb
On Linux it failed to detect that there is bin directory as it wasn't looking for armpl-info which is the only file that is in that directory on Linux and also adding link to math library as it is required to link against when checking for LAPACK functions. Fixes #149610 Pull Request resolved: https://github.com/pytorch/pytorch/pull/150031 Approved by: https://github.com/fadara01, https://github.com/malfet
280 lines
9.5 KiB
CMake
280 lines
9.5 KiB
CMake
# - Find LAPACK library
|
|
# This module finds an installed fortran library that implements the LAPACK
|
|
# linear-algebra interface (see http://www.netlib.org/lapack/).
|
|
#
|
|
# The approach follows that taken for the autoconf macro file, acx_lapack.m4
|
|
# (distributed at http://ac-archive.sourceforge.net/ac-archive/acx_lapack.html).
|
|
#
|
|
# This module sets the following variables:
|
|
# LAPACK_FOUND - set to true if a library implementing the LAPACK interface is found
|
|
# LAPACK_LIBRARIES - list of libraries (using full path name) for LAPACK
|
|
|
|
# Note: I do not think it is a good idea to mixup different BLAS/LAPACK versions
|
|
# Hence, this script wants to find a Lapack library matching your Blas library
|
|
|
|
# Do nothing if LAPACK was found before
|
|
IF(NOT LAPACK_FOUND)
|
|
|
|
SET(LAPACK_LIBRARIES)
|
|
SET(LAPACK_INFO)
|
|
|
|
IF(LAPACK_FIND_QUIETLY OR NOT LAPACK_FIND_REQUIRED)
|
|
FIND_PACKAGE(BLAS)
|
|
ELSE(LAPACK_FIND_QUIETLY OR NOT LAPACK_FIND_REQUIRED)
|
|
FIND_PACKAGE(BLAS REQUIRED)
|
|
ENDIF(LAPACK_FIND_QUIETLY OR NOT LAPACK_FIND_REQUIRED)
|
|
|
|
# Old search lapack script
|
|
include(CheckFortranFunctionExists)
|
|
include(CheckFunctionExists)
|
|
|
|
macro(Check_Lapack_Libraries LIBRARIES _prefix _name _flags _list _blas)
|
|
# This macro checks for the existence of the combination of fortran libraries
|
|
# given by _list. If the combination is found, this macro checks (using the
|
|
# Check_Fortran_Function_Exists macro) whether can link against that library
|
|
# combination using the name of a routine given by _name using the linker
|
|
# flags given by _flags. If the combination of libraries is found and passes
|
|
# the link test, LIBRARIES is set to the list of complete library paths that
|
|
# have been found. Otherwise, LIBRARIES is set to FALSE.
|
|
# N.B. _prefix is the prefix applied to the names of all cached variables that
|
|
# are generated internally and marked advanced by this macro.
|
|
set(_libraries_work TRUE)
|
|
set(${LIBRARIES})
|
|
set(_combined_name)
|
|
foreach(_library ${_list})
|
|
set(_combined_name ${_combined_name}_${_library})
|
|
if(_libraries_work)
|
|
if (WIN32)
|
|
find_library(${_prefix}_${_library}_LIBRARY
|
|
NAMES ${_library} PATHS ENV LIB PATHS ENV PATH)
|
|
else (WIN32)
|
|
if(APPLE)
|
|
find_library(${_prefix}_${_library}_LIBRARY
|
|
NAMES ${_library}
|
|
PATHS /usr/local/lib /usr/lib /usr/local/lib64 /usr/lib64 /usr/lib/aarch64-linux-gnu
|
|
ENV DYLD_LIBRARY_PATH)
|
|
else(APPLE)
|
|
find_library(${_prefix}_${_library}_LIBRARY
|
|
NAMES ${_library}
|
|
PATHS /usr/local/lib /usr/lib /usr/local/lib64 /usr/lib64 /usr/lib/aarch64-linux-gnu
|
|
ENV LD_LIBRARY_PATH)
|
|
endif(APPLE)
|
|
endif(WIN32)
|
|
mark_as_advanced(${_prefix}_${_library}_LIBRARY)
|
|
set(${LIBRARIES} ${${LIBRARIES}} ${${_prefix}_${_library}_LIBRARY})
|
|
set(_libraries_work ${${_prefix}_${_library}_LIBRARY})
|
|
endif(_libraries_work)
|
|
endforeach(_library ${_list})
|
|
if(_libraries_work)
|
|
# Test this combination of libraries.
|
|
set(CMAKE_REQUIRED_LIBRARIES ${_flags} ${${LIBRARIES}} ${_blas})
|
|
if (CMAKE_Fortran_COMPILER_WORKS)
|
|
check_fortran_function_exists(${_name} ${_prefix}${_combined_name}_WORKS)
|
|
else (CMAKE_Fortran_COMPILER_WORKS)
|
|
check_function_exists("${_name}_" ${_prefix}${_combined_name}_WORKS)
|
|
endif (CMAKE_Fortran_COMPILER_WORKS)
|
|
set(CMAKE_REQUIRED_LIBRARIES)
|
|
mark_as_advanced(${_prefix}${_combined_name}_WORKS)
|
|
set(_libraries_work ${${_prefix}${_combined_name}_WORKS})
|
|
endif(_libraries_work)
|
|
if(NOT _libraries_work)
|
|
set(${LIBRARIES} FALSE)
|
|
endif(NOT _libraries_work)
|
|
endmacro(Check_Lapack_Libraries)
|
|
|
|
|
|
if(BLAS_FOUND)
|
|
|
|
# Intel MKL
|
|
IF((NOT LAPACK_INFO) AND (BLAS_INFO STREQUAL "mkl"))
|
|
IF(MKL_LAPACK_LIBRARIES)
|
|
SET(LAPACK_LIBRARIES ${MKL_LAPACK_LIBRARIES} ${MKL_LIBRARIES})
|
|
ELSE(MKL_LAPACK_LIBRARIES)
|
|
SET(LAPACK_LIBRARIES ${MKL_LIBRARIES})
|
|
ENDIF(MKL_LAPACK_LIBRARIES)
|
|
SET(LAPACK_INCLUDE_DIR ${MKL_INCLUDE_DIR})
|
|
SET(LAPACK_INFO "mkl")
|
|
ENDIF()
|
|
|
|
# NVPL
|
|
IF((NOT LAPACK_INFO) AND (BLAS_INFO STREQUAL "nvpl"))
|
|
FIND_PACKAGE(NVPL_LAPACK REQUIRED)
|
|
SET(LAPACK_LIBRARIES nvpl::lapack_lp64_omp)
|
|
SET(LAPACK_INFO "nvpl")
|
|
ENDIF()
|
|
|
|
# Accelerate
|
|
IF((NOT LAPACK_INFO) AND (BLAS_INFO STREQUAL "accelerate"))
|
|
SET(CMAKE_REQUIRED_LIBRARIES ${BLAS_LIBRARIES})
|
|
check_function_exists("cheev_" ACCELERATE_LAPACK_WORKS)
|
|
set(CMAKE_REQUIRED_LIBRARIES)
|
|
if(ACCELERATE_LAPACK_WORKS)
|
|
SET(LAPACK_INFO "accelerate")
|
|
else()
|
|
message(STATUS "Strangely, this Accelerate library does not support Lapack?!")
|
|
endif()
|
|
endif()
|
|
|
|
# vecLib
|
|
IF((NOT LAPACK_INFO) AND (BLAS_INFO STREQUAL "veclib"))
|
|
SET(CMAKE_REQUIRED_LIBRARIES ${BLAS_LIBRARIES})
|
|
check_function_exists("cheev_" VECLIB_LAPACK_WORKS)
|
|
set(CMAKE_REQUIRED_LIBRARIES)
|
|
if(VECLIB_LAPACK_WORKS)
|
|
SET(LAPACK_INFO "veclib")
|
|
else()
|
|
message(STATUS "Strangely, this vecLib library does not support Lapack?!")
|
|
endif()
|
|
endif()
|
|
|
|
# FlexiBLAS
|
|
IF((NOT LAPACK_INFO) AND (BLAS_INFO STREQUAL "flexi"))
|
|
SET(CMAKE_REQUIRED_LIBRARIES ${BLAS_LIBRARIES})
|
|
check_function_exists("cheev_" FLEXIBLAS_LAPACK_WORKS)
|
|
set(CMAKE_REQUIRED_LIBRARIES)
|
|
if(FLEXIBLAS_LAPACK_WORKS)
|
|
SET(LAPACK_INFO "flexi")
|
|
else()
|
|
message(STATUS "Strangely, this FlexiBLAS library does not support Lapack?!")
|
|
endif()
|
|
endif()
|
|
|
|
# OpenBlas
|
|
IF((NOT LAPACK_INFO) AND (BLAS_INFO STREQUAL "open"))
|
|
SET(CMAKE_REQUIRED_LIBRARIES ${BLAS_LIBRARIES})
|
|
check_function_exists("cheev_" OPEN_LAPACK_WORKS)
|
|
if(OPEN_LAPACK_WORKS)
|
|
check_function_exists("cgesdd_" LAPACK_CGESDD_WORKS)
|
|
if(NOT LAPACK_CGESDD_WORKS)
|
|
find_library(GFORTRAN_LIBRARY
|
|
NAMES libgfortran.a gfortran
|
|
PATHS ${CMAKE_C_IMPLICIT_LINK_DIRECTORIES})
|
|
list(APPEND CMAKE_REQUIRED_LIBRARIES "${GFORTRAN_LIBRARY}")
|
|
unset(LAPACK_CGESDD_WORKS CACHE)
|
|
check_function_exists("cgesdd_" LAPACK_CGESDD_WORKS)
|
|
if(LAPACK_CGESDD_WORKS)
|
|
list(APPEND LAPACK_LIBRARIES "${GFORTRAN_LIBRARY}")
|
|
else()
|
|
message(WARNING "OpenBlas has been compiled with Lapack support, but cgesdd can not be used")
|
|
set(OPEN_LAPACK_WORKS NO)
|
|
endif()
|
|
endif()
|
|
endif()
|
|
|
|
set(CMAKE_REQUIRED_LIBRARIES)
|
|
if(OPEN_LAPACK_WORKS)
|
|
SET(LAPACK_INFO "open")
|
|
else()
|
|
message(STATUS "It seems OpenBlas has not been compiled with Lapack support")
|
|
endif()
|
|
endif()
|
|
|
|
# GotoBlas
|
|
IF((NOT LAPACK_INFO) AND (BLAS_INFO STREQUAL "goto"))
|
|
SET(CMAKE_REQUIRED_LIBRARIES ${BLAS_LIBRARIES})
|
|
check_function_exists("cheev_" GOTO_LAPACK_WORKS)
|
|
set(CMAKE_REQUIRED_LIBRARIES)
|
|
if(GOTO_LAPACK_WORKS)
|
|
SET(LAPACK_INFO "goto")
|
|
else()
|
|
message(STATUS "It seems GotoBlas has not been compiled with Lapack support")
|
|
endif()
|
|
endif()
|
|
|
|
# FLAME
|
|
IF((NOT LAPACK_INFO) AND (BLAS_INFO STREQUAL "FLAME"))
|
|
check_lapack_libraries(
|
|
LAPACK_LIBRARIES
|
|
LAPACK
|
|
cheev
|
|
""
|
|
"flame"
|
|
"${BLAS_LIBRARIES}"
|
|
)
|
|
if(LAPACK_LIBRARIES)
|
|
SET(LAPACK_INFO "FLAME")
|
|
endif(LAPACK_LIBRARIES)
|
|
endif()
|
|
|
|
# ACML
|
|
IF((NOT LAPACK_INFO) AND (BLAS_INFO STREQUAL "acml"))
|
|
SET(CMAKE_REQUIRED_LIBRARIES ${BLAS_LIBRARIES})
|
|
check_function_exists("cheev_" ACML_LAPACK_WORKS)
|
|
set(CMAKE_REQUIRED_LIBRARIES)
|
|
if(ACML_LAPACK_WORKS)
|
|
SET(LAPACK_INFO "acml")
|
|
else()
|
|
message(STATUS "Strangely, this ACML library does not support Lapack?!")
|
|
endif()
|
|
endif()
|
|
|
|
# Generic LAPACK library?
|
|
IF((NOT LAPACK_INFO) AND (BLAS_INFO STREQUAL "generic"))
|
|
check_lapack_libraries(
|
|
LAPACK_LIBRARIES
|
|
LAPACK
|
|
cheev
|
|
""
|
|
"lapack"
|
|
"${BLAS_LIBRARIES}"
|
|
)
|
|
if(LAPACK_LIBRARIES)
|
|
SET(LAPACK_INFO "generic")
|
|
endif(LAPACK_LIBRARIES)
|
|
endif()
|
|
|
|
#Arm Performance Libraries
|
|
IF((NOT LAPACK_INFO) AND (BLAS_INFO STREQUAL "apl"))
|
|
SET(CMAKE_REQUIRED_LIBRARIES ${BLAS_LIBRARIES})
|
|
if(UNIX)
|
|
list(APPEND CMAKE_REQUIRED_LIBRARIES -lm)
|
|
endif(UNIX)
|
|
check_function_exists("cheev_" APL_LAPACK_WORKS)
|
|
if(APL_LAPACK_WORKS)
|
|
check_function_exists("cgesdd_" LAPACK_CGESDD_WORKS)
|
|
if(NOT LAPACK_CGESDD_WORKS)
|
|
find_library(GFORTRAN_LIBRARY
|
|
NAMES libgfortran.a gfortran
|
|
PATHS ${CMAKE_C_IMPLICIT_LINK_DIRECTORIES})
|
|
list(APPEND CMAKE_REQUIRED_LIBRARIES "${GFORTRAN_LIBRARY}")
|
|
unset(LAPACK_CGESDD_WORKS CACHE)
|
|
check_function_exists("cgesdd_" LAPACK_CGESDD_WORKS)
|
|
if(LAPACK_CGESDD_WORKS)
|
|
list(APPEND LAPACK_LIBRARIES "${GFORTRAN_LIBRARY}")
|
|
else()
|
|
message(WARNING "APL has been compiled with Lapack support, but cgesdd can not be used")
|
|
set(APL_LAPACK_WORKS NO)
|
|
endif()
|
|
endif()
|
|
endif()
|
|
set(CMAKE_REQUIRED_LIBRARIES)
|
|
if(APL_LAPACK_WORKS)
|
|
SET(LAPACK_INFO "apl")
|
|
else()
|
|
message(STATUS "It seems APL has not been compiled with Lapack support")
|
|
endif()
|
|
endif()
|
|
else(BLAS_FOUND)
|
|
message(STATUS "LAPACK requires BLAS")
|
|
endif(BLAS_FOUND)
|
|
|
|
if(LAPACK_INFO)
|
|
set(LAPACK_FOUND TRUE)
|
|
else(LAPACK_INFO)
|
|
set(LAPACK_FOUND FALSE)
|
|
endif(LAPACK_INFO)
|
|
|
|
IF (NOT LAPACK_FOUND AND LAPACK_FIND_REQUIRED)
|
|
message(FATAL_ERROR "Cannot find a library with LAPACK API. Please specify library location.")
|
|
ENDIF (NOT LAPACK_FOUND AND LAPACK_FIND_REQUIRED)
|
|
IF(NOT LAPACK_FIND_QUIETLY)
|
|
IF(LAPACK_FOUND)
|
|
MESSAGE(STATUS "Found a library with LAPACK API (${LAPACK_INFO}).")
|
|
ELSE(LAPACK_FOUND)
|
|
MESSAGE(STATUS "Cannot find a library with LAPACK API. Not using LAPACK.")
|
|
ENDIF(LAPACK_FOUND)
|
|
ENDIF(NOT LAPACK_FIND_QUIETLY)
|
|
|
|
# Do nothing if LAPACK was found before
|
|
ENDIF(NOT LAPACK_FOUND)
|