From dd6261491825087e5577d3fdc7444bdbfc3e1924 Mon Sep 17 00:00:00 2001
From: Gertjan van den Burg <gertjanvandenburg@gmail.com>
Date: Wed, 6 Mar 2019 16:10:26 -0500
Subject: Travis (#4)

* add cython to travis

* add blas to travis install

* fix blas dependency

* trying with the atlas version of blas

* add lapack too

* try with lapacke

* attempt to get lapack info

* use correct asserts and lower threshold

* decrease precision for seed test

* add python 2.7 too

* add travis status to readme
---
 setup.py | 32 ++++++++++++++++++++++++++++++--
 1 file changed, 30 insertions(+), 2 deletions(-)

(limited to 'setup.py')

diff --git a/setup.py b/setup.py
index 7f6e5e0..356b393 100644
--- a/setup.py
+++ b/setup.py
@@ -85,6 +85,30 @@ def _skl_get_blas_info():
     return cblas_libs, blas_info
 
 
+def get_lapack_info():
+
+    from numpy.distutils.system_info import get_info
+
+    def atlas_not_found(lapack_info_):
+        def_macros = lapack_info.get("define_macros", [])
+        for x in def_macros:
+            if x[0] == "NO_ATLAS_INFO":
+                return True
+            if x[0] == "ATLAS_INFO":
+                if "None" in x[1]:
+                    return True
+        return False
+
+    lapack_info = get_info("lapack_opt", 0)
+    if (not lapack_info) or atlas_not_found(lapack_info):
+        lapack_libs = ["lapacke"]
+        lapack_info.pop("libraries", None)
+    else:
+        lapack_libs = lapack_info.pop("libraries", [])
+
+    return lapack_libs, lapack_info
+
+
 def configuration():
     from numpy.distutils.misc_util import Configuration
 
@@ -94,6 +118,10 @@ def configuration():
     if os.name == "posix":
         cblas_libs.append("m")
 
+    lapack_libs, lapack_info = get_lapack_info()
+    if os.name == "posix":
+        lapack_libs.append("m") # unsure if necessary
+
     # Wrapper code in Cython uses the .pyx extension if we want to USE_CYTHON,
     # otherwise it ends in .c.
     wrapper_extension = "*.pyx" if USE_CYTHON else "*.c"
@@ -117,7 +145,7 @@ def configuration():
     config.add_extension(
         "cython_wrapper.wrapper",
         sources=gensvm_sources,
-        libraries=cblas_libs,
+        libraries=cblas_libs + lapack_libs,
         include_dirs=[
             os.path.join("src", "gensvm"),
             os.path.join("src", "gensvm", "include"),
@@ -160,7 +188,7 @@ if __name__ == "__main__":
     check_requirements()
 
     version = re.search(
-        "__version__ = \"([^']+)\"", open("gensvm/__init__.py").read()
+        '__version__ = "([^\']+)"', open("gensvm/__init__.py").read()
     ).group(1)
 
     attr = configuration().todict()
-- 
cgit v1.2.3