174 lines
7.9 KiB
Python
174 lines
7.9 KiB
Python
##############################################################################
|
|
# Copyright (c) 2013-2016, Lawrence Livermore National Security, LLC.
|
|
# Produced at the Lawrence Livermore National Laboratory.
|
|
#
|
|
# This file is part of Spack.
|
|
# Created by Todd Gamblin, tgamblin@llnl.gov, All rights reserved.
|
|
# LLNL-CODE-647188
|
|
#
|
|
# For details, see https://github.com/llnl/spack
|
|
# Please also see the NOTICE and LICENSE files for our notice and the LGPL.
|
|
#
|
|
# This program is free software; you can redistribute it and/or modify
|
|
# it under the terms of the GNU Lesser General Public License (as
|
|
# published by the Free Software Foundation) version 2.1, February 1999.
|
|
#
|
|
# This program is distributed in the hope that it will be useful, but
|
|
# WITHOUT ANY WARRANTY; without even the IMPLIED WARRANTY OF
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the terms and
|
|
# conditions of the GNU Lesser General Public License for more details.
|
|
#
|
|
# You should have received a copy of the GNU Lesser General Public
|
|
# License along with this program; if not, write to the Free Software
|
|
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
|
|
##############################################################################
|
|
from spack import *
|
|
import platform
|
|
|
|
|
|
class PyNumpy(PythonPackage):
|
|
"""NumPy is the fundamental package for scientific computing with Python.
|
|
It contains among other things: a powerful N-dimensional array object,
|
|
sophisticated (broadcasting) functions, tools for integrating C/C++ and
|
|
Fortran code, and useful linear algebra, Fourier transform, and random
|
|
number capabilities"""
|
|
|
|
homepage = "http://www.numpy.org/"
|
|
url = "https://pypi.io/packages/source/n/numpy/numpy-1.9.1.tar.gz"
|
|
|
|
install_time_test_callbacks = ['install_test', 'import_module_test']
|
|
|
|
import_modules = [
|
|
'numpy', 'numpy.compat', 'numpy.core', 'numpy.distutils', 'numpy.doc',
|
|
'numpy.f2py', 'numpy.fft', 'numpy.lib', 'numpy.linalg', 'numpy.ma',
|
|
'numpy.matrixlib', 'numpy.polynomial', 'numpy.random', 'numpy.testing',
|
|
'numpy.distutils.command', 'numpy.distutils.fcompiler'
|
|
]
|
|
|
|
# FIXME: numpy._build_utils and numpy.core.code_generators failed to import
|
|
# FIXME: Is this expected?
|
|
|
|
version('1.13.0', 'fd044f0b8079abeaf5e6d2e93b2c1d03',
|
|
url="https://pypi.io/packages/source/n/numpy/numpy-1.13.0.zip")
|
|
version('1.12.1', 'c75b072a984028ac746a6a332c209a91',
|
|
url="https://pypi.io/packages/source/n/numpy/numpy-1.12.1.zip")
|
|
version('1.12.0', '33e5a84579f31829bbbba084fe0a4300',
|
|
url="https://pypi.io/packages/source/n/numpy/numpy-1.12.0.zip")
|
|
version('1.11.2', '03bd7927c314c43780271bf1ab795ebc')
|
|
version('1.11.1', '2f44a895a8104ffac140c3a70edbd450')
|
|
version('1.11.0', 'bc56fb9fc2895aa4961802ffbdb31d0b')
|
|
version('1.10.4', 'aed294de0aa1ac7bd3f9745f4f1968ad')
|
|
version('1.9.2', 'a1ed53432dbcd256398898d35bc8e645')
|
|
version('1.9.1', '78842b73560ec378142665e712ae4ad9')
|
|
|
|
variant('blas', default=True, description='Build with BLAS support')
|
|
variant('lapack', default=True, description='Build with LAPACK support')
|
|
|
|
depends_on('python@2.7:2.8,3.4:')
|
|
depends_on('py-setuptools', type='build')
|
|
depends_on('blas', when='+blas')
|
|
depends_on('lapack', when='+lapack')
|
|
|
|
# Tests require:
|
|
# TODO: Add a 'test' deptype
|
|
# depends_on('py-nose@1.0.0:', type='test')
|
|
|
|
def setup_dependent_package(self, module, dependent_spec):
|
|
python_version = self.spec['python'].version.up_to(2)
|
|
arch = '{0}-{1}'.format(platform.system().lower(), platform.machine())
|
|
|
|
self.spec.include = join_path(
|
|
self.prefix.lib,
|
|
'python{0}'.format(python_version),
|
|
'site-packages',
|
|
'numpy-{0}-py{1}-{2}.egg'.format(
|
|
self.spec.version, python_version, arch),
|
|
'numpy/core/include')
|
|
|
|
def patch(self):
|
|
spec = self.spec
|
|
# for build notes see http://www.scipy.org/scipylib/building/linux.html
|
|
lapackblas = LibraryList('')
|
|
if '+lapack' in spec:
|
|
lapackblas += spec['lapack'].libs
|
|
|
|
if '+blas' in spec:
|
|
lapackblas += spec['blas'].libs
|
|
|
|
if '+blas' in spec or '+lapack' in spec:
|
|
# note that one should not use [blas_opt] and [lapack_opt], see
|
|
# https://github.com/numpy/numpy/commit/ffd4332262ee0295cb942c94ed124f043d801eb6
|
|
with open('site.cfg', 'w') as f:
|
|
# Unfortunately, numpy prefers to provide each BLAS/LAPACK
|
|
# differently.
|
|
names = ','.join(lapackblas.names)
|
|
dirs = ':'.join(lapackblas.directories)
|
|
|
|
# Special treatment for some (!) BLAS/LAPACK. Note that
|
|
# in this case library_dirs can not be specified within [ALL].
|
|
if '^openblas' in spec:
|
|
f.write('[openblas]\n')
|
|
f.write('libraries=%s\n' % names)
|
|
elif '^mkl' in spec:
|
|
# numpy does not expect system libraries needed for MKL
|
|
# here.
|
|
# names = [x for x in names if x.startswith('mkl')]
|
|
# FIXME: as of @1.11.2, numpy does not work with separately
|
|
# specified threading and interface layers. A workaround is
|
|
# a terribly bad idea to use mkl_rt. In this case Spack
|
|
# will no longer be able to guarantee that one and the
|
|
# same variant of Blas/Lapack (32/64bit, threaded/serial)
|
|
# is used within the DAG. This may lead to a lot of
|
|
# hard-to-debug segmentation faults on user's side. Users
|
|
# may also break working installation by (unconsciously)
|
|
# setting environment variable to switch between different
|
|
# interface and threading layers dynamically. From this
|
|
# perspective it is no different from throwing away RPATH's
|
|
# and using LD_LIBRARY_PATH throughout Spack.
|
|
f.write('[mkl]\n')
|
|
f.write('mkl_libs=%s\n' % 'mkl_rt')
|
|
elif '^atlas' in spec:
|
|
f.write('[atlas]\n')
|
|
f.write('atlas_libs=%s\n' % names)
|
|
else:
|
|
# The section title for the defaults changed in @1.10, see
|
|
# https://github.com/numpy/numpy/blob/master/site.cfg.example
|
|
if spec.satisfies('@:1.9.2'):
|
|
f.write('[DEFAULT]\n')
|
|
else:
|
|
f.write('[ALL]\n')
|
|
f.write('libraries=%s\n' % names)
|
|
|
|
f.write('library_dirs=%s\n' % dirs)
|
|
if not ((platform.system() == "Darwin") and
|
|
(platform.mac_ver()[0] == '10.12')):
|
|
f.write('rpath=%s\n' % dirs)
|
|
|
|
def build_args(self, spec, prefix):
|
|
args = []
|
|
|
|
# From NumPy 1.10.0 on it's possible to do a parallel build
|
|
if self.version >= Version('1.10.0'):
|
|
args = ['-j', str(make_jobs)]
|
|
|
|
return args
|
|
|
|
def test(self):
|
|
# `setup.py test` is not supported. Use one of the following
|
|
# instead:
|
|
#
|
|
# - `python runtests.py` (to build and test)
|
|
# - `python runtests.py --no-build` (to test installed numpy)
|
|
# - `>>> numpy.test()` (run tests for installed numpy
|
|
# from within an interpreter)
|
|
pass
|
|
|
|
def install_test(self):
|
|
# Change directories due to the following error:
|
|
#
|
|
# ImportError: Error importing numpy: you should not try to import
|
|
# numpy from its source directory; please exit the numpy
|
|
# source tree, and relaunch your python interpreter from there.
|
|
with working_dir('..'):
|
|
python('-c', 'import numpy; numpy.test("full", verbose=2)')
|