cmake/gmxDetectAcceleration.cmake

   1 #
   2 # This file is part of the GROMACS molecular simulation package.
   3 #
   4 # Copyright (c) 2012, by the GROMACS development team, led by
   5 # David van der Spoel, Berk Hess, Erik Lindahl, and including many
   6 # others, as listed in the AUTHORS file in the top-level source
   7 # directory and at http://www.gromacs.org.
   8 #
   9 # GROMACS is free software; you can redistribute it and/or
  10 # modify it under the terms of the GNU Lesser General Public License
  11 # as published by the Free Software Foundation; either version 2.1
  12 # of the License, or (at your option) any later version.
  13 #
  14 # GROMACS is distributed in the hope that it will be useful,
  15 # but WITHOUT ANY WARRANTY; without even the implied warranty of
  16 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  17 # Lesser General Public License for more details.
  18 #
  19 # You should have received a copy of the GNU Lesser General Public
  20 # License along with GROMACS; if not, see
  21 # http://www.gnu.org/licenses, or write to the Free Software Foundation,
  22 # Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301  USA.
  23 #
  24 # If you want to redistribute modifications to GROMACS, please
  25 # consider that scientific software is very special. Version
  26 # control is crucial - bugs must be traceable. We will be happy to
  27 # consider code for inclusion in the official distribution, but
  28 # derived work must not be called official GROMACS. Details are found
  29 # in the README & COPYING files - if they are missing, get the
  30 # official version at http://www.gromacs.org.
  31 #
  32 # To help us fund GROMACS development, we humbly ask that you cite
  33 # the research papers on the package. Check out http://www.gromacs.org.
  34 #
  35 # - Check the username performing the build, as well as date and time
  36 #
  37 # GMX_DETECT_ACCELERATION(GMX_SUGGESTED_ACCELERATION)
  38 #
  39 # Try to detect CPU information and suggest an acceleration option
  40 # (such as SSE/AVX) that fits the current CPU.
  41 #
  42 # GMX_SUGGESTED_ACCELERATION
  43 #
  44
  45 # we rely on inline asm support for GNU!
  46 include(gmxTestInlineASM)
  47
  48 macro(gmx_detect_acceleration GMX_SUGGESTED_ACCELERATION)
  49     IF(NOT DEFINED ${GMX_SUGGESTED_ACCELERATION})
  50
  51     gmx_test_inline_asm_gcc_x86(GMX_X86_GCC_INLINE_ASM)
  52
  53     if(GMX_X86_GCC_INLINE_ASM)
  54         set(GCC_INLINE_ASM_DEFINE "-DGMX_X86_GCC_INLINE_ASM")
  55     else(GMX_X86_GCC_INLINE_ASM)
  56         set(GCC_INLINE_ASM_DEFINE "")
  57     endif(GMX_X86_GCC_INLINE_ASM)
  58
  59     message(STATUS "Detecting best acceleration for this CPU")
  60
  61     # Get CPU acceleration information
  62     try_run(GMX_CPUID_RUN_ACC GMX_CPUID_COMPILED
  63             ${CMAKE_BINARY_DIR}
  64             ${CMAKE_SOURCE_DIR}/src/gmxlib/gmx_cpuid.c
  65             COMPILE_DEFINITIONS "@GCC_INLINE_ASM_DEFINE@ -I${CMAKE_SOURCE_DIR}/include -DGMX_CPUID_STANDALONE"
  66             RUN_OUTPUT_VARIABLE OUTPUT_TMP
  67             COMPILE_OUTPUT_VARIABLE GMX_CPUID_COMPILE_OUTPUT
  68             ARGS "-acceleration")
  69
  70     if(NOT GMX_CPUID_COMPILED)
  71         message(WARNING "Cannot compile CPUID code, which means no CPU-specific acceleration.")
  72         message(STATUS "Compile output: ${GMX_CPUID_COMPILE_OUTPUT}")
  73         set(OUTPUT_TMP "None")
  74     elseif(NOT GMX_CPUID_RUN_ACC EQUAL 0)
  75         message(WARNING "Cannot run CPUID code, which means no CPU-specific optimization.")
  76         message(STATUS "Run output: ${OUTPUT_TMP}")
  77         set(OUTPUT_TMP "None")
  78     endif(NOT GMX_CPUID_COMPILED)
  79
  80     string(STRIP "@OUTPUT_TMP@" OUTPUT_ACC)
  81
  82     message(STATUS "Detecting best acceleration for this CPU - @OUTPUT_ACC@")
  83
  84     set(${GMX_SUGGESTED_ACCELERATION}    "@OUTPUT_ACC@" CACHE INTERNAL "GROMACS CPU-specific acceleration")
  85
  86     ENDIF(NOT DEFINED ${GMX_SUGGESTED_ACCELERATION})
  87 endmacro(gmx_detect_acceleration GMX_SUGGESTED_ACCELERATION)
  88