cmake: search for BLIS and libFLAME

Author: mgates3

CMakeLists.txt

@@ -106,7 +106,7 @@ set( blas "auto" CACHE STRING
      "BLAS library to search for" )
 set_property(
     CACHE blas PROPERTY STRINGS
-    "auto" "Apple Accelerate" "Cray LibSci" "IBM ESSL"
+    "auto" "AMD AOCL" "Apple Accelerate" "BLIS" "Cray LibSci" "IBM ESSL"
     "Intel MKL" "OpenBLAS" "generic" )
 
 set( blas_fortran "auto" CACHE STRING

cmake/BLASConfig.cmake

@@ -70,6 +70,37 @@ if (NOT found)
     endif()
 endif()
 
+#---------------------------------------- BLIS
+if (NOT found)
+    try_run(
+        run_result compile_result ${CMAKE_CURRENT_BINARY_DIR}
+        SOURCES
+            "${CMAKE_CURRENT_SOURCE_DIR}/config/blis_version.cc"
+        LINK_LIBRARIES
+            ${BLAS_LIBRARIES} ${openmp_lib} # not "..." quoted; screws up OpenMP
+        COMPILE_DEFINITIONS
+            ${blaspp_defs_}
+        COMPILE_OUTPUT_VARIABLE
+            compile_output
+        RUN_OUTPUT_VARIABLE
+            run_output
+    )
+    # For cross-compiling, if it links, assume the run is okay.
+    if (CMAKE_CROSSCOMPILING AND compile_result)
+        message( DEBUG "cross: blis" )
+        set( run_result "0"  CACHE STRING "" FORCE )
+        set( run_output "BLIS_VERSION=unknown" CACHE STRING "" FORCE )
+    endif()
+    debug_try_run( "blis_version.cc" "${compile_result}" "${compile_output}"
+                                     "${run_result}" "${run_output}" )
+
+    if (compile_result AND "${run_output}" MATCHES "BLIS_VERSION")
+        message( "${blue}   ${run_output}${plain}" )
+        list( APPEND blaspp_defs_ "-DBLAS_HAVE_BLIS" )
+        set( found true )
+    endif()
+endif()
+
 #---------------------------------------- IBM ESSL
 if (NOT found)
     try_run(

cmake/BLASFinder.cmake

@@ -125,19 +125,21 @@ endif()
 #---------------------------------------- blas
 string( TOLOWER "${blas}" blas_ )
 
+string( REGEX MATCH "auto|apple|accelerate"    test_accelerate "${blas_}" )
+string( REGEX MATCH "auto|aocl|blis"           test_blis       "${blas_}" )
+string( REGEX MATCH "auto|cray|libsci|default" test_default    "${blas_}" )
 string( REGEX MATCH "auto|ibm|essl"  test_essl     "${blas_}" )
 string( REGEX MATCH "auto|intel|mkl" test_mkl      "${blas_}" )
 string( REGEX MATCH "auto|openblas"  test_openblas "${blas_}" )
 string( REGEX MATCH "auto|generic"   test_generic  "${blas_}" )
-string( REGEX MATCH "auto|apple|accelerate"    test_accelerate "${blas_}" )
-string( REGEX MATCH "auto|cray|libsci|default" test_default    "${blas_}" )
 
 message( DEBUG "
 BLAS_LIBRARIES      = '${BLAS_LIBRARIES}'
 blas                = '${blas}'
 blas_               = '${blas_}'
 test_blas_libraries = '${test_blas_libraries}'
 test_accelerate     = '${test_accelerate}'
+test_blis           = '${test_blis}'
 test_default        = '${test_default}'
 test_essl           = '${test_essl}'
 test_mkl            = '${test_mkl}'
@@ -360,6 +362,13 @@ if (test_openblas)
     debug_print_list( "openblas" )
 endif()
 
+#---------------------------------------- BLIS (also used by AMD AOCL)
+if (test_blis)
+    list( APPEND blas_name_list "BLIS" )
+    list( APPEND blas_libs_list "-lflame -lblis" )
+    debug_print_list( "blis" )
+endif()
+
 #---------------------------------------- Apple Accelerate
 if (test_accelerate)
     list( APPEND blas_name_list "Apple Accelerate" )