summaryrefslogtreecommitdiff
path: root/numpy/linalg/setup.py
diff options
context:
space:
mode:
authorPauli Virtanen <pav@iki.fi>2019-12-07 14:17:20 +0200
committerPauli Virtanen <pav@iki.fi>2019-12-14 14:14:37 +0200
commita92039db1edf9d85059ddadc10134f434ae82ce7 (patch)
treefbc070996c04be5ac3bae9b88b923d19b4637f2b /numpy/linalg/setup.py
parent2fc10a279edfd132ec27eeac9e72f5a02a8bae5e (diff)
downloadnumpy-a92039db1edf9d85059ddadc10134f434ae82ce7.tar.gz
ENH: distutils: add support for ILP64 OpenBLAS (generic symbol suffix)
Generalize the ILP64 BLAS/LAPACK symbol name handling to deal with arbitrary prefix/suffix. The build-time behavior is changed so that HAVE_BLAS_ILP64 and BLAS_SYMBOL_SUFFIX/PREFIX defines are added to compile options as appropriate. Mainly to make autodetection of BLAS/LAPACK easier for downstream numpy.distutils users, add get_info aliases 'blas_ilp64_opt', 'blas_ilp64_plain_opt', and 'blas64__opt' for any/no/""&"64_" prefix&suffix, and the same for lapack. (Due to the way system_info works, each also gets a separate class.) In addition to openblas64_ which has a fixed suffix, add the (by default suffixless) openblas_ilp64, which correspond to the most likely cases to be present.
Diffstat (limited to 'numpy/linalg/setup.py')
-rw-r--r--numpy/linalg/setup.py6
1 files changed, 2 insertions, 4 deletions
diff --git a/numpy/linalg/setup.py b/numpy/linalg/setup.py
index f5cb04e89..6315a34b4 100644
--- a/numpy/linalg/setup.py
+++ b/numpy/linalg/setup.py
@@ -26,10 +26,8 @@ def configuration(parent_package='', top_path=None):
]
all_sources = config.paths(lapack_lite_src)
- if (os.environ.get('NPY_USE_BLAS64_', "0") != "0"):
- lapack_info = get_info('lapack64__opt', 2)
- lapack_info.setdefault('define_macros', [])
- lapack_info['define_macros'] += [('NPY_UMATH_USE_BLAS64_', None)]
+ if os.environ.get('NPY_USE_BLAS_ILP64', "0") != "0":
+ lapack_info = get_info('lapack_ilp64_opt', 2)
else:
lapack_info = get_info('lapack_opt', 0) # and {}