mirror of
				https://github.com/PaddlePaddle/FastDeploy.git
				synced 2025-11-01 04:12:58 +08:00 
			
		
		
		
	
		
			
				
	
	
		
			381 lines
		
	
	
		
			13 KiB
		
	
	
	
		
			CMake
		
	
	
	
	
	
			
		
		
	
	
			381 lines
		
	
	
		
			13 KiB
		
	
	
	
		
			CMake
		
	
	
	
	
	
| ###
 | |
| #
 | |
| # @copyright (c) 2009-2014 The University of Tennessee and The University
 | |
| #                          of Tennessee Research Foundation.
 | |
| #                          All rights reserved.
 | |
| # @copyright (c) 2012-2016 Inria. All rights reserved.
 | |
| # @copyright (c) 2012-2014 Bordeaux INP, CNRS (LaBRI UMR 5800), Inria, Univ. Bordeaux. All rights reserved.
 | |
| #
 | |
| ###
 | |
| #
 | |
| # - Find BLAS EXTENDED for MORSE projects: find include dirs and libraries
 | |
| #
 | |
| # This module allows to find BLAS libraries by calling the official FindBLAS module
 | |
| # and handles the creation of different library lists whether the user wishes to link
 | |
| # with a sequential BLAS or a multihreaded (BLAS_SEQ_LIBRARIES and BLAS_PAR_LIBRARIES).
 | |
| # BLAS is detected with a FindBLAS call then if the BLAS vendor is Intel10_64lp, ACML
 | |
| # or IBMESSLMT then the module attempts to find the corresponding multithreaded libraries.
 | |
| #
 | |
| # The following variables have been added to manage links with sequential or multithreaded
 | |
| # versions:
 | |
| #  BLAS_INCLUDE_DIRS  - BLAS include directories
 | |
| #  BLAS_LIBRARY_DIRS  - Link directories for BLAS libraries
 | |
| #  BLAS_SEQ_LIBRARIES - BLAS component libraries to be linked (sequential)
 | |
| #  BLAS_PAR_LIBRARIES - BLAS component libraries to be linked (multithreaded)
 | |
| 
 | |
| #=============================================================================
 | |
| # Copyright 2012-2013 Inria
 | |
| # Copyright 2012-2013 Emmanuel Agullo
 | |
| # Copyright 2012-2013 Mathieu Faverge
 | |
| # Copyright 2012      Cedric Castagnede
 | |
| # Copyright 2013-2016 Florent Pruvost
 | |
| #
 | |
| # Distributed under the OSI-approved BSD License (the "License");
 | |
| # see accompanying file MORSE-Copyright.txt for details.
 | |
| #
 | |
| # This software is distributed WITHOUT ANY WARRANTY; without even the
 | |
| # implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
 | |
| # See the License for more information.
 | |
| #=============================================================================
 | |
| # (To distribute this file outside of Morse, substitute the full
 | |
| #  License text for the above reference.)
 | |
| 
 | |
| # macro to factorize this call
 | |
| macro(find_package_blas)
 | |
|   if(BLASEXT_FIND_REQUIRED)
 | |
|     if(BLASEXT_FIND_QUIETLY)
 | |
|       find_package(BLAS REQUIRED QUIET)
 | |
|     else()
 | |
|       find_package(BLAS REQUIRED)
 | |
|     endif()
 | |
|   else()
 | |
|     if(BLASEXT_FIND_QUIETLY)
 | |
|       find_package(BLAS QUIET)
 | |
|     else()
 | |
|       find_package(BLAS)
 | |
|     endif()
 | |
|   endif()
 | |
| endmacro()
 | |
| 
 | |
| # add a cache variable to let the user specify the BLAS vendor
 | |
| set(BLA_VENDOR "" CACHE STRING "list of possible BLAS vendor:
 | |
|     Open, Eigen, Goto, ATLAS PhiPACK, CXML, DXML, SunPerf, SCSL, SGIMATH, IBMESSL, IBMESSLMT,
 | |
|     Intel10_32 (intel mkl v10 32 bit),
 | |
|     Intel10_64lp (intel mkl v10 64 bit, lp thread model, lp64 model),
 | |
|     Intel10_64lp_seq (intel mkl v10 64 bit, sequential code, lp64 model),
 | |
|     Intel( older versions of mkl 32 and 64 bit),
 | |
|     ACML, ACML_MP, ACML_GPU, Apple, NAS, Generic")
 | |
| 
 | |
| if(NOT BLASEXT_FIND_QUIETLY)
 | |
|   message(STATUS "In FindBLASEXT")
 | |
|   message(STATUS "If you want to force the use of one specific library, "
 | |
|     "\n   please specify the BLAS vendor by setting -DBLA_VENDOR=blas_vendor_name"
 | |
|     "\n   at cmake configure.")
 | |
|   message(STATUS "List of possible BLAS vendor: Goto, ATLAS PhiPACK, CXML, "
 | |
|     "\n   DXML, SunPerf, SCSL, SGIMATH, IBMESSL, IBMESSLMT, Intel10_32 (intel mkl v10 32 bit),"
 | |
|     "\n   Intel10_64lp (intel mkl v10 64 bit, lp thread model, lp64 model),"
 | |
|     "\n   Intel10_64lp_seq (intel mkl v10 64 bit, sequential code, lp64 model),"
 | |
|     "\n   Intel( older versions of mkl 32 and 64 bit),"
 | |
|     "\n   ACML, ACML_MP, ACML_GPU, Apple, NAS, Generic")
 | |
| endif()
 | |
| 
 | |
| if (NOT BLAS_FOUND)
 | |
|   # First try to detect two cases:
 | |
|   # 1: only SEQ libs are handled
 | |
|   # 2: both SEQ and PAR libs are handled
 | |
|   find_package_blas()
 | |
| endif ()
 | |
| 
 | |
| # detect the cases where SEQ and PAR libs are handled
 | |
| if(BLA_VENDOR STREQUAL "All" AND
 | |
|     (BLAS_mkl_core_LIBRARY OR BLAS_mkl_core_dll_LIBRARY)
 | |
|     )
 | |
|   set(BLA_VENDOR "Intel")
 | |
|   if(BLAS_mkl_intel_LIBRARY)
 | |
|     set(BLA_VENDOR "Intel10_32")
 | |
|   endif()
 | |
|   if(BLAS_mkl_intel_lp64_LIBRARY)
 | |
|     set(BLA_VENDOR "Intel10_64lp")
 | |
|   endif()
 | |
|   if(NOT BLASEXT_FIND_QUIETLY)
 | |
|     message(STATUS "A BLAS library has been found (${BLAS_LIBRARIES}) but we"
 | |
|       "\n   have also potentially detected some multithreaded BLAS libraries from the MKL."
 | |
|       "\n   We try to find both libraries lists (Sequential/Multithreaded).")
 | |
|   endif()
 | |
|   set(BLAS_FOUND "")
 | |
| elseif(BLA_VENDOR STREQUAL "All" AND BLAS_acml_LIBRARY)
 | |
|   set(BLA_VENDOR "ACML")
 | |
|   if(NOT BLASEXT_FIND_QUIETLY)
 | |
|     message(STATUS "A BLAS library has been found (${BLAS_LIBRARIES}) but we"
 | |
|       "\n   have also potentially detected some multithreaded BLAS libraries from the ACML."
 | |
|       "\n   We try to find both libraries lists (Sequential/Multithreaded).")
 | |
|   endif()
 | |
|   set(BLAS_FOUND "")
 | |
| elseif(BLA_VENDOR STREQUAL "All" AND BLAS_essl_LIBRARY)
 | |
|   set(BLA_VENDOR "IBMESSL")
 | |
|   if(NOT BLASEXT_FIND_QUIETLY)
 | |
|     message(STATUS "A BLAS library has been found (${BLAS_LIBRARIES}) but we"
 | |
|       "\n   have also potentially detected some multithreaded BLAS libraries from the ESSL."
 | |
|       "\n   We try to find both libraries lists (Sequential/Multithreaded).")
 | |
|   endif()
 | |
|   set(BLAS_FOUND "")
 | |
| endif()
 | |
| 
 | |
| # Intel case
 | |
| if(BLA_VENDOR MATCHES "Intel*")
 | |
| 
 | |
|   ###
 | |
|   # look for include path if the BLAS vendor is Intel
 | |
|   ###
 | |
| 
 | |
|   # gather system include paths
 | |
|   unset(_inc_env)
 | |
|   if(WIN32)
 | |
|     string(REPLACE ":" ";" _inc_env "$ENV{INCLUDE}")
 | |
|   else()
 | |
|     string(REPLACE ":" ";" _path_env "$ENV{INCLUDE}")
 | |
|     list(APPEND _inc_env "${_path_env}")
 | |
|     string(REPLACE ":" ";" _path_env "$ENV{C_INCLUDE_PATH}")
 | |
|     list(APPEND _inc_env "${_path_env}")
 | |
|     string(REPLACE ":" ";" _path_env "$ENV{CPATH}")
 | |
|     list(APPEND _inc_env "${_path_env}")
 | |
|     string(REPLACE ":" ";" _path_env "$ENV{INCLUDE_PATH}")
 | |
|     list(APPEND _inc_env "${_path_env}")
 | |
|   endif()
 | |
|   list(APPEND _inc_env "${CMAKE_PLATFORM_IMPLICIT_INCLUDE_DIRECTORIES}")
 | |
|   list(APPEND _inc_env "${CMAKE_C_IMPLICIT_INCLUDE_DIRECTORIES}")
 | |
|   set(ENV_MKLROOT "$ENV{MKLROOT}")
 | |
|   if (ENV_MKLROOT)
 | |
|     list(APPEND _inc_env "${ENV_MKLROOT}/include")
 | |
|   endif()
 | |
|   list(REMOVE_DUPLICATES _inc_env)
 | |
| 
 | |
|   # find mkl.h inside known include paths
 | |
|   set(BLAS_mkl.h_INCLUDE_DIRS "BLAS_mkl.h_INCLUDE_DIRS-NOTFOUND")
 | |
|   if(BLAS_INCDIR)
 | |
|     set(BLAS_mkl.h_INCLUDE_DIRS "BLAS_mkl.h_INCLUDE_DIRS-NOTFOUND")
 | |
|     find_path(BLAS_mkl.h_INCLUDE_DIRS
 | |
|       NAMES mkl.h
 | |
|       HINTS ${BLAS_INCDIR})
 | |
|   else()
 | |
|     if(BLAS_DIR)
 | |
|       set(BLAS_mkl.h_INCLUDE_DIRS "BLAS_mkl.h_INCLUDE_DIRS-NOTFOUND")
 | |
|       find_path(BLAS_mkl.h_INCLUDE_DIRS
 | |
| 	NAMES mkl.h
 | |
| 	HINTS ${BLAS_DIR}
 | |
| 	PATH_SUFFIXES include)
 | |
|     else()
 | |
|       set(BLAS_mkl.h_INCLUDE_DIRS "BLAS_mkl.h_INCLUDE_DIRS-NOTFOUND")
 | |
|       find_path(BLAS_mkl.h_INCLUDE_DIRS
 | |
| 	NAMES mkl.h
 | |
| 	HINTS ${_inc_env})
 | |
|     endif()
 | |
|   endif()
 | |
|   mark_as_advanced(BLAS_mkl.h_INCLUDE_DIRS)
 | |
|   ## Print status if not found
 | |
|   ## -------------------------
 | |
|   #if (NOT BLAS_mkl.h_INCLUDE_DIRS AND MORSE_VERBOSE)
 | |
|   #    Print_Find_Header_Status(blas mkl.h)
 | |
|   #endif ()
 | |
|   set(BLAS_INCLUDE_DIRS "")
 | |
|   if(BLAS_mkl.h_INCLUDE_DIRS)
 | |
|     list(APPEND BLAS_INCLUDE_DIRS "${BLAS_mkl.h_INCLUDE_DIRS}" )
 | |
|   endif()
 | |
| 
 | |
|   ###
 | |
|   # look for libs
 | |
|   ###
 | |
|   # if Intel 10 64 bit -> look for sequential and multithreaded versions
 | |
|   if(BLA_VENDOR MATCHES "Intel10_64lp*")
 | |
| 
 | |
|     ## look for the sequential version
 | |
|     set(BLA_VENDOR "Intel10_64lp_seq")
 | |
|     if(NOT BLASEXT_FIND_QUIETLY)
 | |
|       message(STATUS "Look for the sequential version Intel10_64lp_seq")
 | |
|     endif()
 | |
|     find_package_blas()
 | |
|     if(BLAS_FOUND)
 | |
|       set(BLAS_SEQ_LIBRARIES "${BLAS_LIBRARIES}")
 | |
|     else()
 | |
|       set(BLAS_SEQ_LIBRARIES "${BLAS_SEQ_LIBRARIES-NOTFOUND}")
 | |
|     endif()
 | |
| 
 | |
|     ## look for the multithreaded version
 | |
|     set(BLA_VENDOR "Intel10_64lp")
 | |
|     if(NOT BLASEXT_FIND_QUIETLY)
 | |
|       message(STATUS "Look for the multithreaded version Intel10_64lp")
 | |
|     endif()
 | |
|     find_package_blas()
 | |
|     if(BLAS_FOUND)
 | |
|       set(BLAS_PAR_LIBRARIES "${BLAS_LIBRARIES}")
 | |
|     else()
 | |
|       set(BLAS_PAR_LIBRARIES "${BLAS_PAR_LIBRARIES-NOTFOUND}")
 | |
|     endif()
 | |
| 
 | |
|   else()
 | |
| 
 | |
|     if(BLAS_FOUND)
 | |
|       set(BLAS_SEQ_LIBRARIES "${BLAS_LIBRARIES}")
 | |
|     else()
 | |
|       set(BLAS_SEQ_LIBRARIES "${BLAS_SEQ_LIBRARIES-NOTFOUND}")
 | |
|     endif()
 | |
| 
 | |
|   endif()
 | |
| 
 | |
|   # ACML case
 | |
| elseif(BLA_VENDOR MATCHES "ACML*")
 | |
| 
 | |
|   ## look for the sequential version
 | |
|   set(BLA_VENDOR "ACML")
 | |
|   find_package_blas()
 | |
|   if(BLAS_FOUND)
 | |
|     set(BLAS_SEQ_LIBRARIES "${BLAS_LIBRARIES}")
 | |
|   else()
 | |
|     set(BLAS_SEQ_LIBRARIES "${BLAS_SEQ_LIBRARIES-NOTFOUND}")
 | |
|   endif()
 | |
| 
 | |
|   ## look for the multithreaded version
 | |
|   set(BLA_VENDOR "ACML_MP")
 | |
|   find_package_blas()
 | |
|   if(BLAS_FOUND)
 | |
|     set(BLAS_PAR_LIBRARIES "${BLAS_LIBRARIES}")
 | |
|   else()
 | |
|     set(BLAS_PAR_LIBRARIES "${BLAS_PAR_LIBRARIES-NOTFOUND}")
 | |
|   endif()
 | |
| 
 | |
|   # IBMESSL case
 | |
| elseif(BLA_VENDOR MATCHES "IBMESSL*")
 | |
| 
 | |
|   ## look for the sequential version
 | |
|   set(BLA_VENDOR "IBMESSL")
 | |
|   find_package_blas()
 | |
|   if(BLAS_FOUND)
 | |
|     set(BLAS_SEQ_LIBRARIES "${BLAS_LIBRARIES}")
 | |
|   else()
 | |
|     set(BLAS_SEQ_LIBRARIES "${BLAS_SEQ_LIBRARIES-NOTFOUND}")
 | |
|   endif()
 | |
| 
 | |
|   ## look for the multithreaded version
 | |
|   set(BLA_VENDOR "IBMESSLMT")
 | |
|   find_package_blas()
 | |
|   if(BLAS_FOUND)
 | |
|     set(BLAS_PAR_LIBRARIES "${BLAS_LIBRARIES}")
 | |
|   else()
 | |
|     set(BLAS_PAR_LIBRARIES "${BLAS_PAR_LIBRARIES-NOTFOUND}")
 | |
|   endif()
 | |
| 
 | |
| else()
 | |
| 
 | |
|   if(BLAS_FOUND)
 | |
|     # define the SEQ libs as the BLAS_LIBRARIES
 | |
|     set(BLAS_SEQ_LIBRARIES "${BLAS_LIBRARIES}")
 | |
|   else()
 | |
|     set(BLAS_SEQ_LIBRARIES "${BLAS_SEQ_LIBRARIES-NOTFOUND}")
 | |
|   endif()
 | |
|   set(BLAS_PAR_LIBRARIES "${BLAS_PAR_LIBRARIES-NOTFOUND}")
 | |
| 
 | |
| endif()
 | |
| 
 | |
| 
 | |
| if(BLAS_SEQ_LIBRARIES)
 | |
|   set(BLAS_LIBRARIES "${BLAS_SEQ_LIBRARIES}")
 | |
| endif()
 | |
| 
 | |
| # extract libs paths
 | |
| # remark: because it is not given by find_package(BLAS)
 | |
| set(BLAS_LIBRARY_DIRS "")
 | |
| string(REPLACE " " ";" BLAS_LIBRARIES "${BLAS_LIBRARIES}")
 | |
| foreach(blas_lib ${BLAS_LIBRARIES})
 | |
|   if (EXISTS "${blas_lib}")
 | |
|     get_filename_component(a_blas_lib_dir "${blas_lib}" PATH)
 | |
|     list(APPEND BLAS_LIBRARY_DIRS "${a_blas_lib_dir}" )
 | |
|   else()
 | |
|     string(REPLACE "-L" "" blas_lib "${blas_lib}")
 | |
|     if (EXISTS "${blas_lib}")
 | |
|       list(APPEND BLAS_LIBRARY_DIRS "${blas_lib}" )
 | |
|     else()
 | |
|       get_filename_component(a_blas_lib_dir "${blas_lib}" PATH)
 | |
|       if (EXISTS "${a_blas_lib_dir}")
 | |
| 	list(APPEND BLAS_LIBRARY_DIRS "${a_blas_lib_dir}" )
 | |
|       endif()
 | |
|     endif()
 | |
|   endif()
 | |
| endforeach()
 | |
| if (BLAS_LIBRARY_DIRS)
 | |
|   list(REMOVE_DUPLICATES BLAS_LIBRARY_DIRS)
 | |
| endif ()
 | |
| 
 | |
| # check that BLAS has been found
 | |
| # ---------------------------------
 | |
| include(FindPackageHandleStandardArgs)
 | |
| if(BLA_VENDOR MATCHES "Intel*")
 | |
|   if(BLA_VENDOR MATCHES "Intel10_64lp*")
 | |
|     if(NOT BLASEXT_FIND_QUIETLY)
 | |
|       message(STATUS "BLAS found is Intel MKL:"
 | |
| 	"\n   we manage two lists of libs, one sequential and one parallel if found"
 | |
| 	"\n   (see BLAS_SEQ_LIBRARIES and BLAS_PAR_LIBRARIES)")
 | |
|       message(STATUS "BLAS sequential libraries stored in BLAS_SEQ_LIBRARIES")
 | |
|     endif()
 | |
|     find_package_handle_standard_args(BLAS DEFAULT_MSG
 | |
|       BLAS_SEQ_LIBRARIES
 | |
|       BLAS_LIBRARY_DIRS
 | |
|       BLAS_INCLUDE_DIRS)
 | |
|     if(BLAS_PAR_LIBRARIES)
 | |
|       if(NOT BLASEXT_FIND_QUIETLY)
 | |
| 	message(STATUS "BLAS parallel libraries stored in BLAS_PAR_LIBRARIES")
 | |
|       endif()
 | |
|       find_package_handle_standard_args(BLAS DEFAULT_MSG
 | |
| 	BLAS_PAR_LIBRARIES)
 | |
|     endif()
 | |
|   else()
 | |
|     if(NOT BLASEXT_FIND_QUIETLY)
 | |
|       message(STATUS "BLAS sequential libraries stored in BLAS_SEQ_LIBRARIES")
 | |
|     endif()
 | |
|     find_package_handle_standard_args(BLAS DEFAULT_MSG
 | |
|       BLAS_SEQ_LIBRARIES
 | |
|       BLAS_LIBRARY_DIRS
 | |
|       BLAS_INCLUDE_DIRS)
 | |
|   endif()
 | |
| elseif(BLA_VENDOR MATCHES "ACML*")
 | |
|   if(NOT BLASEXT_FIND_QUIETLY)
 | |
|     message(STATUS "BLAS found is ACML:"
 | |
|       "\n   we manage two lists of libs, one sequential and one parallel if found"
 | |
|       "\n   (see BLAS_SEQ_LIBRARIES and BLAS_PAR_LIBRARIES)")
 | |
|     message(STATUS "BLAS sequential libraries stored in BLAS_SEQ_LIBRARIES")
 | |
|   endif()
 | |
|   find_package_handle_standard_args(BLAS DEFAULT_MSG
 | |
|     BLAS_SEQ_LIBRARIES
 | |
|     BLAS_LIBRARY_DIRS)
 | |
|   if(BLAS_PAR_LIBRARIES)
 | |
|     if(NOT BLASEXT_FIND_QUIETLY)
 | |
|       message(STATUS "BLAS parallel libraries stored in BLAS_PAR_LIBRARIES")
 | |
|     endif()
 | |
|     find_package_handle_standard_args(BLAS DEFAULT_MSG
 | |
|       BLAS_PAR_LIBRARIES)
 | |
|   endif()
 | |
| elseif(BLA_VENDOR MATCHES "IBMESSL*")
 | |
|   if(NOT BLASEXT_FIND_QUIETLY)
 | |
|     message(STATUS "BLAS found is ESSL:"
 | |
|       "\n   we manage two lists of libs, one sequential and one parallel if found"
 | |
|       "\n   (see BLAS_SEQ_LIBRARIES and BLAS_PAR_LIBRARIES)")
 | |
|     message(STATUS "BLAS sequential libraries stored in BLAS_SEQ_LIBRARIES")
 | |
|   endif()
 | |
|   find_package_handle_standard_args(BLAS DEFAULT_MSG
 | |
|     BLAS_SEQ_LIBRARIES
 | |
|     BLAS_LIBRARY_DIRS)
 | |
|   if(BLAS_PAR_LIBRARIES)
 | |
|     if(NOT BLASEXT_FIND_QUIETLY)
 | |
|       message(STATUS "BLAS parallel libraries stored in BLAS_PAR_LIBRARIES")
 | |
|     endif()
 | |
|     find_package_handle_standard_args(BLAS DEFAULT_MSG
 | |
|       BLAS_PAR_LIBRARIES)
 | |
|   endif()
 | |
| else()
 | |
|   if(NOT BLASEXT_FIND_QUIETLY)
 | |
|     message(STATUS "BLAS sequential libraries stored in BLAS_SEQ_LIBRARIES")
 | |
|   endif()
 | |
|   find_package_handle_standard_args(BLAS DEFAULT_MSG
 | |
|     BLAS_SEQ_LIBRARIES
 | |
|     BLAS_LIBRARY_DIRS)
 | |
| endif()
 | 
