zhangyiss/spack
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

py-numpy: enable linking with armpl-gcc and acfl for BLAS and LAPACK (#35417)

Browse Source 
Co-authored-by: Adam J. Stewart <ajstewart426@gmail.com>
This commit is contained in:
Annop Wongwathanarat 2023-03-15 11:38:14 +00:00 committed by GitHub
parent 028535030c
commit 5f8f89b9c9
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 11 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
var/spack/repos/builtin/packages/py-numpy/package.py
View File

@ -315,11 +315,17 @@ def write_library_dirs(f, dirs):
f.write("include_dirs = {0}\n".format(lapack_header_dirs)) f.write("include_dirs = {0}\n".format(lapack_header_dirs))
f.write("extra_link_args = {0}\n".format(self.spec["lapack"].libs.ld_flags)) f.write("extra_link_args = {0}\n".format(self.spec["lapack"].libs.ld_flags))
if "^armpl-gcc" in spec: if "^armpl-gcc" in spec or "^acfl" in spec:
f.write("[blas]\n") if spec.satisfies("+blas"):
f.write("libraries = {0}\n".format(lapackblas_lib_names)) f.write("[blas]\n")
write_library_dirs(f, lapackblas_lib_dirs) f.write("libraries = {0}\n".format(blas_lib_names))
f.write("include_dirs = {0}\n".format(lapackblas_header_dirs)) write_library_dirs(f, blas_lib_dirs)
f.write("include_dirs = {0}\n".format(blas_header_dirs))
if spec.satisfies("+lapack"):
f.write("[lapack]\n")
f.write("libraries = {0}\n".format(lapack_lib_names))
write_library_dirs(f, lapack_lib_dirs)
f.write("include_dirs = {0}\n".format(lapack_header_dirs))
def setup_build_environment(self, env): def setup_build_environment(self, env):
# Tell numpy which BLAS/LAPACK libraries we want to use. # Tell numpy which BLAS/LAPACK libraries we want to use.