Add custom BLAS selector to support MKL/OpenBLAS

dftd3 · Jan 12, 2021 · 49547e0 · 49547e0
1 parent a75641f
commit 49547e0
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 2 deletions.
diff --git a/config/meson.build b/config/meson.build
@@ -54,13 +54,19 @@ if get_option('openmp')
   lib_deps += omp_dep
 endif
 
+if get_option('blas') != 'none'
+  add_project_arguments('-DWITH_BLAS', language: 'fortran')
+endif
 if get_option('blas') == 'netlib'
   blas_dep = dependency('blas', required: false)
   if not blas_dep.found()
     blas_dep = fc.find_library('blas', required: true)
   endif
   lib_deps += blas_dep
-  add_project_arguments('-DWITH_BLAS', language: 'fortran')
+elif get_option('blas') == 'custom'
+  foreach lib: get_option('blas_libs')
+    lib_deps += fc.find_library(lib)
+  endforeach
 endif
 
 # Create the tool chain library as subproject

diff --git a/meson_options.txt b/meson_options.txt
@@ -18,9 +18,15 @@ option(
   'blas',
   type: 'combo',
   value: 'none',
-  choices: ['none', 'netlib'],
+  choices: ['none', 'netlib', 'custom'],
   description: 'Use basic linear algebra subprograms',
 )
+option(
+  'blas_libs',
+  type: 'array',
+  value: [],
+  description: 'Libraries to link against as custom BLAS backend',
+)
 option(
   'openmp',
   type: 'boolean',