MAINT: core: optionally use symbol-suffixed 64-bit cblas

This is enabled by setting NPY_USE_BLAS64_=1 environment variable at
build time.  If set, Numpy will be linked both against a 32-bit BLAS and
a symbol-suffixed 64-bit BLAS.  (This is safe and does not cause symbol
clashes thanks to the suffixing.)

1 files changed, 15 insertions, 2 deletions

diff --git a/numpy/core/setup.py b/numpy/core/setup.py
index a4b5cfe5f..2ecd2616f 100644
--- a/numpy/core/setup.py
+++ b/numpy/core/setup.py
@@ -91,6 +91,9 @@ def is_npy_no_smp():
     #  block.
     return 'NPY_NOSMP' in os.environ
 
+def is_npy_use_blas64_():
+    return (os.environ.get('NPY_USE_BLAS64_', "0") != "0")
+
 def win32_checks(deflist):
     from numpy.distutils.misc_util import get_build_architecture
     a = get_build_architecture()
@@ -394,7 +397,7 @@ def visibility_define(config):
 
 def configuration(parent_package='',top_path=None):
     from numpy.distutils.misc_util import Configuration, dot_join
-    from numpy.distutils.system_info import get_info
+    from numpy.distutils.system_info import get_info, dict_append
 
     config = Configuration('core', parent_package, top_path)
     local_dir = config.local_path
@@ -754,8 +757,18 @@ def configuration(parent_package='',top_path=None):
             ]
 
     blas_info = get_info('blas_opt', 0)
-    if blas_info and ('HAVE_CBLAS', None) in blas_info.get('define_macros', []):
+
+    if is_npy_use_blas64_():
+        blas64__info = get_info('blas64__opt', 2)
+    else:
+        blas64__info = {}
+
+    have_blas = blas_info and ('HAVE_CBLAS', None) in blas_info.get('define_macros', [])
+    have_blas64_ = blas64__info and ('HAVE_CBLAS64_', None) in blas64__info.get('define_macros', [])
+
+    if have_blas or have_blas64_:
         extra_info = blas_info
+        dict_append(blas_info, **blas64__info)
         # These files are also in MANIFEST.in so that they are always in
         # the source distribution independently of HAVE_CBLAS.
         common_src.extend([join('src', 'common', 'cblasfuncs.c'),