mirror of
				https://github.com/PaddlePaddle/FastDeploy.git
				synced 2025-10-31 11:56:44 +08:00 
			
		
		
		
	
		
			
				
	
	
		
			370 lines
		
	
	
		
			12 KiB
		
	
	
	
		
			CMake
		
	
	
	
	
	
			
		
		
	
	
			370 lines
		
	
	
		
			12 KiB
		
	
	
	
		
			CMake
		
	
	
	
	
	
| ###
 | |
| #
 | |
| # @copyright (c) 2009-2014 The University of Tennessee and The University
 | |
| #                          of Tennessee Research Foundation.
 | |
| #                          All rights reserved.
 | |
| # @copyright (c) 2012-2014 Inria. All rights reserved.
 | |
| # @copyright (c) 2012-2014 Bordeaux INP, CNRS (LaBRI UMR 5800), Inria, Univ. Bordeaux. All rights reserved.
 | |
| #
 | |
| ###
 | |
| #
 | |
| # - Find SCOTCH include dirs and libraries
 | |
| # Use this module by invoking find_package with the form:
 | |
| #  find_package(SCOTCH
 | |
| #               [REQUIRED]             # Fail with error if scotch is not found
 | |
| #               [COMPONENTS <comp1> <comp2> ...] # dependencies
 | |
| #              )
 | |
| #
 | |
| #  COMPONENTS can be some of the following:
 | |
| #   - ESMUMPS: to activate detection of Scotch with the esmumps interface
 | |
| #
 | |
| # This module finds headers and scotch library.
 | |
| # Results are reported in variables:
 | |
| #  SCOTCH_FOUND           - True if headers and requested libraries were found
 | |
| #  SCOTCH_INCLUDE_DIRS    - scotch include directories
 | |
| #  SCOTCH_LIBRARY_DIRS    - Link directories for scotch libraries
 | |
| #  SCOTCH_LIBRARIES       - scotch component libraries to be linked
 | |
| #  SCOTCH_INTSIZE         - Number of octets occupied by a SCOTCH_Num
 | |
| #
 | |
| # The user can give specific paths where to find the libraries adding cmake
 | |
| # options at configure (ex: cmake path/to/project -DSCOTCH=path/to/scotch):
 | |
| #  SCOTCH_DIR             - Where to find the base directory of scotch
 | |
| #  SCOTCH_INCDIR          - Where to find the header files
 | |
| #  SCOTCH_LIBDIR          - Where to find the library files
 | |
| # The module can also look for the following environment variables if paths
 | |
| # are not given as cmake variable: SCOTCH_DIR, SCOTCH_INCDIR, SCOTCH_LIBDIR
 | |
| 
 | |
| #=============================================================================
 | |
| # Copyright 2012-2013 Inria
 | |
| # Copyright 2012-2013 Emmanuel Agullo
 | |
| # Copyright 2012-2013 Mathieu Faverge
 | |
| # Copyright 2012      Cedric Castagnede
 | |
| # Copyright 2013      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.)
 | |
| 
 | |
| if (NOT SCOTCH_FOUND)
 | |
|   set(SCOTCH_DIR "" CACHE PATH "Installation directory of SCOTCH library")
 | |
|   if (NOT SCOTCH_FIND_QUIETLY)
 | |
|     message(STATUS "A cache variable, namely SCOTCH_DIR, has been set to specify the install directory of SCOTCH")
 | |
|   endif()
 | |
| endif()
 | |
| 
 | |
| # Set the version to find
 | |
| set(SCOTCH_LOOK_FOR_ESMUMPS OFF)
 | |
| 
 | |
| if( SCOTCH_FIND_COMPONENTS )
 | |
|   foreach( component ${SCOTCH_FIND_COMPONENTS} )
 | |
|     if (${component} STREQUAL "ESMUMPS")
 | |
|       # means we look for esmumps library
 | |
|       set(SCOTCH_LOOK_FOR_ESMUMPS ON)
 | |
|     endif()
 | |
|   endforeach()
 | |
| endif()
 | |
| 
 | |
| # SCOTCH may depend on Threads, try to find it
 | |
| if (NOT THREADS_FOUND)
 | |
|   if (SCOTCH_FIND_REQUIRED)
 | |
|     find_package(Threads REQUIRED)
 | |
|   else()
 | |
|     find_package(Threads)
 | |
|   endif()
 | |
| endif()
 | |
| 
 | |
| # Looking for include
 | |
| # -------------------
 | |
| 
 | |
| # Add system include paths to search include
 | |
| # ------------------------------------------
 | |
| unset(_inc_env)
 | |
| set(ENV_SCOTCH_DIR "$ENV{SCOTCH_DIR}")
 | |
| set(ENV_SCOTCH_INCDIR "$ENV{SCOTCH_INCDIR}")
 | |
| if(ENV_SCOTCH_INCDIR)
 | |
|   list(APPEND _inc_env "${ENV_SCOTCH_INCDIR}")
 | |
| elseif(ENV_SCOTCH_DIR)
 | |
|   list(APPEND _inc_env "${ENV_SCOTCH_DIR}")
 | |
|   list(APPEND _inc_env "${ENV_SCOTCH_DIR}/include")
 | |
|   list(APPEND _inc_env "${ENV_SCOTCH_DIR}/include/scotch")
 | |
| else()
 | |
|   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()
 | |
| endif()
 | |
| list(APPEND _inc_env "${CMAKE_PLATFORM_IMPLICIT_INCLUDE_DIRECTORIES}")
 | |
| list(APPEND _inc_env "${CMAKE_C_IMPLICIT_INCLUDE_DIRECTORIES}")
 | |
| list(REMOVE_DUPLICATES _inc_env)
 | |
| 
 | |
| 
 | |
| # Try to find the scotch header in the given paths
 | |
| # -------------------------------------------------
 | |
| # call cmake macro to find the header path
 | |
| if(SCOTCH_INCDIR)
 | |
|   set(SCOTCH_scotch.h_DIRS "SCOTCH_scotch.h_DIRS-NOTFOUND")
 | |
|   find_path(SCOTCH_scotch.h_DIRS
 | |
|     NAMES scotch.h
 | |
|     HINTS ${SCOTCH_INCDIR})
 | |
| else()
 | |
|   if(SCOTCH_DIR)
 | |
|     set(SCOTCH_scotch.h_DIRS "SCOTCH_scotch.h_DIRS-NOTFOUND")
 | |
|     find_path(SCOTCH_scotch.h_DIRS
 | |
|       NAMES scotch.h
 | |
|       HINTS ${SCOTCH_DIR}
 | |
|       PATH_SUFFIXES "include" "include/scotch")
 | |
|   else()
 | |
|     set(SCOTCH_scotch.h_DIRS "SCOTCH_scotch.h_DIRS-NOTFOUND")
 | |
|     find_path(SCOTCH_scotch.h_DIRS
 | |
|       NAMES scotch.h
 | |
|       HINTS ${_inc_env}
 | |
|       PATH_SUFFIXES "scotch")
 | |
|   endif()
 | |
| endif()
 | |
| mark_as_advanced(SCOTCH_scotch.h_DIRS)
 | |
| 
 | |
| # If found, add path to cmake variable
 | |
| # ------------------------------------
 | |
| if (SCOTCH_scotch.h_DIRS)
 | |
|   set(SCOTCH_INCLUDE_DIRS "${SCOTCH_scotch.h_DIRS}")
 | |
| else ()
 | |
|   set(SCOTCH_INCLUDE_DIRS "SCOTCH_INCLUDE_DIRS-NOTFOUND")
 | |
|   if (NOT SCOTCH_FIND_QUIETLY)
 | |
|     message(STATUS "Looking for scotch -- scotch.h not found")
 | |
|   endif()
 | |
| endif()
 | |
| list(REMOVE_DUPLICATES SCOTCH_INCLUDE_DIRS)
 | |
| 
 | |
| # Looking for lib
 | |
| # ---------------
 | |
| 
 | |
| # Add system library paths to search lib
 | |
| # --------------------------------------
 | |
| unset(_lib_env)
 | |
| set(ENV_SCOTCH_LIBDIR "$ENV{SCOTCH_LIBDIR}")
 | |
| if(ENV_SCOTCH_LIBDIR)
 | |
|   list(APPEND _lib_env "${ENV_SCOTCH_LIBDIR}")
 | |
| elseif(ENV_SCOTCH_DIR)
 | |
|   list(APPEND _lib_env "${ENV_SCOTCH_DIR}")
 | |
|   list(APPEND _lib_env "${ENV_SCOTCH_DIR}/lib")
 | |
| else()
 | |
|   if(WIN32)
 | |
|     string(REPLACE ":" ";" _lib_env "$ENV{LIB}")
 | |
|   else()
 | |
|     if(APPLE)
 | |
|       string(REPLACE ":" ";" _lib_env "$ENV{DYLD_LIBRARY_PATH}")
 | |
|     else()
 | |
|       string(REPLACE ":" ";" _lib_env "$ENV{LD_LIBRARY_PATH}")
 | |
|     endif()
 | |
|     list(APPEND _lib_env "${CMAKE_PLATFORM_IMPLICIT_LINK_DIRECTORIES}")
 | |
|     list(APPEND _lib_env "${CMAKE_C_IMPLICIT_LINK_DIRECTORIES}")
 | |
|   endif()
 | |
| endif()
 | |
| list(REMOVE_DUPLICATES _lib_env)
 | |
| 
 | |
| # Try to find the scotch lib in the given paths
 | |
| # ----------------------------------------------
 | |
| 
 | |
| set(SCOTCH_libs_to_find "scotch;scotcherrexit")
 | |
| if (SCOTCH_LOOK_FOR_ESMUMPS)
 | |
|   list(INSERT SCOTCH_libs_to_find 0 "esmumps")
 | |
| endif()
 | |
| 
 | |
| # call cmake macro to find the lib path
 | |
| if(SCOTCH_LIBDIR)
 | |
|   foreach(scotch_lib ${SCOTCH_libs_to_find})
 | |
|     set(SCOTCH_${scotch_lib}_LIBRARY "SCOTCH_${scotch_lib}_LIBRARY-NOTFOUND")
 | |
|     find_library(SCOTCH_${scotch_lib}_LIBRARY
 | |
|       NAMES ${scotch_lib}
 | |
|       HINTS ${SCOTCH_LIBDIR})
 | |
|   endforeach()
 | |
| else()
 | |
|   if(SCOTCH_DIR)
 | |
|     foreach(scotch_lib ${SCOTCH_libs_to_find})
 | |
|       set(SCOTCH_${scotch_lib}_LIBRARY "SCOTCH_${scotch_lib}_LIBRARY-NOTFOUND")
 | |
|       find_library(SCOTCH_${scotch_lib}_LIBRARY
 | |
| 	NAMES ${scotch_lib}
 | |
| 	HINTS ${SCOTCH_DIR}
 | |
| 	PATH_SUFFIXES lib lib32 lib64)
 | |
|     endforeach()
 | |
|   else()
 | |
|     foreach(scotch_lib ${SCOTCH_libs_to_find})
 | |
|       set(SCOTCH_${scotch_lib}_LIBRARY "SCOTCH_${scotch_lib}_LIBRARY-NOTFOUND")
 | |
|       find_library(SCOTCH_${scotch_lib}_LIBRARY
 | |
| 	NAMES ${scotch_lib}
 | |
| 	HINTS ${_lib_env})
 | |
|     endforeach()
 | |
|   endif()
 | |
| endif()
 | |
| 
 | |
| set(SCOTCH_LIBRARIES "")
 | |
| set(SCOTCH_LIBRARY_DIRS "")
 | |
| # If found, add path to cmake variable
 | |
| # ------------------------------------
 | |
| foreach(scotch_lib ${SCOTCH_libs_to_find})
 | |
| 
 | |
|   if (SCOTCH_${scotch_lib}_LIBRARY)
 | |
|     get_filename_component(${scotch_lib}_lib_path "${SCOTCH_${scotch_lib}_LIBRARY}" PATH)
 | |
|     # set cmake variables
 | |
|     list(APPEND SCOTCH_LIBRARIES "${SCOTCH_${scotch_lib}_LIBRARY}")
 | |
|     list(APPEND SCOTCH_LIBRARY_DIRS "${${scotch_lib}_lib_path}")
 | |
|   else ()
 | |
|     list(APPEND SCOTCH_LIBRARIES "${SCOTCH_${scotch_lib}_LIBRARY}")
 | |
|     if (NOT SCOTCH_FIND_QUIETLY)
 | |
|       message(STATUS "Looking for scotch -- lib ${scotch_lib} not found")
 | |
|     endif()
 | |
|   endif ()
 | |
| 
 | |
|   mark_as_advanced(SCOTCH_${scotch_lib}_LIBRARY)
 | |
| 
 | |
| endforeach()
 | |
| list(REMOVE_DUPLICATES SCOTCH_LIBRARY_DIRS)
 | |
| 
 | |
| # check a function to validate the find
 | |
| if(SCOTCH_LIBRARIES)
 | |
| 
 | |
|   set(REQUIRED_INCDIRS)
 | |
|   set(REQUIRED_LIBDIRS)
 | |
|   set(REQUIRED_LIBS)
 | |
| 
 | |
|   # SCOTCH
 | |
|   if (SCOTCH_INCLUDE_DIRS)
 | |
|     set(REQUIRED_INCDIRS  "${SCOTCH_INCLUDE_DIRS}")
 | |
|   endif()
 | |
|   if (SCOTCH_LIBRARY_DIRS)
 | |
|     set(REQUIRED_LIBDIRS "${SCOTCH_LIBRARY_DIRS}")
 | |
|   endif()
 | |
|   set(REQUIRED_LIBS "${SCOTCH_LIBRARIES}")
 | |
|   # THREADS
 | |
|   if(CMAKE_THREAD_LIBS_INIT)
 | |
|     list(APPEND REQUIRED_LIBS "${CMAKE_THREAD_LIBS_INIT}")
 | |
|   endif()
 | |
|   set(Z_LIBRARY "Z_LIBRARY-NOTFOUND")
 | |
|   find_library(Z_LIBRARY NAMES z)
 | |
|   mark_as_advanced(Z_LIBRARY)
 | |
|   if(Z_LIBRARY)
 | |
|     list(APPEND REQUIRED_LIBS "-lz")
 | |
|   endif()
 | |
|   set(M_LIBRARY "M_LIBRARY-NOTFOUND")
 | |
|   find_library(M_LIBRARY NAMES m)
 | |
|   mark_as_advanced(M_LIBRARY)
 | |
|   if(M_LIBRARY)
 | |
|     list(APPEND REQUIRED_LIBS "-lm")
 | |
|   endif()
 | |
|   set(RT_LIBRARY "RT_LIBRARY-NOTFOUND")
 | |
|   find_library(RT_LIBRARY NAMES rt)
 | |
|   mark_as_advanced(RT_LIBRARY)
 | |
|   if(RT_LIBRARY)
 | |
|     list(APPEND REQUIRED_LIBS "-lrt")
 | |
|   endif()
 | |
| 
 | |
|   # set required libraries for link
 | |
|   set(CMAKE_REQUIRED_INCLUDES "${REQUIRED_INCDIRS}")
 | |
|   set(CMAKE_REQUIRED_LIBRARIES)
 | |
|   foreach(lib_dir ${REQUIRED_LIBDIRS})
 | |
|     list(APPEND CMAKE_REQUIRED_LIBRARIES "-L${lib_dir}")
 | |
|   endforeach()
 | |
|   list(APPEND CMAKE_REQUIRED_LIBRARIES "${REQUIRED_LIBS}")
 | |
|   string(REGEX REPLACE "^ -" "-" CMAKE_REQUIRED_LIBRARIES "${CMAKE_REQUIRED_LIBRARIES}")
 | |
| 
 | |
|   # test link
 | |
|   unset(SCOTCH_WORKS CACHE)
 | |
|   include(CheckFunctionExists)
 | |
|   check_function_exists(SCOTCH_graphInit SCOTCH_WORKS)
 | |
|   mark_as_advanced(SCOTCH_WORKS)
 | |
| 
 | |
|   if(SCOTCH_WORKS)
 | |
|     # save link with dependencies
 | |
|     set(SCOTCH_LIBRARIES "${REQUIRED_LIBS}")
 | |
|   else()
 | |
|     if(NOT SCOTCH_FIND_QUIETLY)
 | |
|       message(STATUS "Looking for SCOTCH : test of SCOTCH_graphInit with SCOTCH library fails")
 | |
|       message(STATUS "CMAKE_REQUIRED_LIBRARIES: ${CMAKE_REQUIRED_LIBRARIES}")
 | |
|       message(STATUS "CMAKE_REQUIRED_INCLUDES: ${CMAKE_REQUIRED_INCLUDES}")
 | |
|       message(STATUS "Check in CMakeFiles/CMakeError.log to figure out why it fails")
 | |
|     endif()
 | |
|   endif()
 | |
|   set(CMAKE_REQUIRED_INCLUDES)
 | |
|   set(CMAKE_REQUIRED_FLAGS)
 | |
|   set(CMAKE_REQUIRED_LIBRARIES)
 | |
| endif()
 | |
| 
 | |
| if (SCOTCH_LIBRARIES)
 | |
|   list(GET SCOTCH_LIBRARIES 0 first_lib)
 | |
|   get_filename_component(first_lib_path "${first_lib}" PATH)
 | |
|   if (${first_lib_path} MATCHES "/lib(32|64)?$")
 | |
|     string(REGEX REPLACE "/lib(32|64)?$" "" not_cached_dir "${first_lib_path}")
 | |
|     set(SCOTCH_DIR_FOUND "${not_cached_dir}" CACHE PATH "Installation directory of SCOTCH library" FORCE)
 | |
|   else()
 | |
|     set(SCOTCH_DIR_FOUND "${first_lib_path}" CACHE PATH "Installation directory of SCOTCH library" FORCE)
 | |
|   endif()
 | |
| endif()
 | |
| mark_as_advanced(SCOTCH_DIR)
 | |
| mark_as_advanced(SCOTCH_DIR_FOUND)
 | |
| 
 | |
| # Check the size of SCOTCH_Num
 | |
| # ---------------------------------
 | |
| set(CMAKE_REQUIRED_INCLUDES ${SCOTCH_INCLUDE_DIRS})
 | |
| 
 | |
| include(CheckCSourceRuns)
 | |
| #stdio.h and stdint.h should be included by scotch.h directly
 | |
| set(SCOTCH_C_TEST_SCOTCH_Num_4 "
 | |
| #include <stdio.h>
 | |
| #include <stdint.h>
 | |
| #include <scotch.h>
 | |
| int main(int argc, char **argv) {
 | |
|   if (sizeof(SCOTCH_Num) == 4)
 | |
|     return 0;
 | |
|   else
 | |
|     return 1;
 | |
| }
 | |
| ")
 | |
| 
 | |
| set(SCOTCH_C_TEST_SCOTCH_Num_8 "
 | |
| #include <stdio.h>
 | |
| #include <stdint.h>
 | |
| #include <scotch.h>
 | |
| int main(int argc, char **argv) {
 | |
|   if (sizeof(SCOTCH_Num) == 8)
 | |
|     return 0;
 | |
|   else
 | |
|     return 1;
 | |
| }
 | |
| ")
 | |
| check_c_source_runs("${SCOTCH_C_TEST_SCOTCH_Num_4}" SCOTCH_Num_4)
 | |
| if(NOT SCOTCH_Num_4)
 | |
|   check_c_source_runs("${SCOTCH_C_TEST_SCOTCH_Num_8}" SCOTCH_Num_8)
 | |
|   if(NOT SCOTCH_Num_8)
 | |
|     set(SCOTCH_INTSIZE -1)
 | |
|   else()
 | |
|     set(SCOTCH_INTSIZE 8)
 | |
|   endif()
 | |
| else()
 | |
|   set(SCOTCH_INTSIZE 4)
 | |
| endif()
 | |
| set(CMAKE_REQUIRED_INCLUDES "")
 | |
| 
 | |
| # check that SCOTCH has been found
 | |
| # ---------------------------------
 | |
| include(FindPackageHandleStandardArgs)
 | |
| find_package_handle_standard_args(SCOTCH DEFAULT_MSG
 | |
|   SCOTCH_LIBRARIES
 | |
|   SCOTCH_WORKS)
 | |
| #
 | |
| # TODO: Add possibility to check for specific functions in the library
 | |
| #
 | 
