diff options
Diffstat (limited to 'var/spack/repos/builtin/packages/mxnet/package.py')
| -rw-r--r-- | var/spack/repos/builtin/packages/mxnet/package.py | 77 |
1 files changed, 44 insertions, 33 deletions
diff --git a/var/spack/repos/builtin/packages/mxnet/package.py b/var/spack/repos/builtin/packages/mxnet/package.py index 00443ec633..cc1e5861ba 100644 --- a/var/spack/repos/builtin/packages/mxnet/package.py +++ b/var/spack/repos/builtin/packages/mxnet/package.py @@ -1,27 +1,8 @@ -############################################################################## -# Copyright (c) 2013-2016, Lawrence Livermore National Security, LLC. -# Produced at the Lawrence Livermore National Laboratory. +# Copyright 2013-2018 Lawrence Livermore National Security, LLC and other +# Spack Project Developers. See the top-level COPYRIGHT file for details. # -# 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/spack/spack -# Please also see the LICENSE file 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 -############################################################################## +# SPDX-License-Identifier: (Apache-2.0 OR MIT) + from spack import * @@ -32,6 +13,8 @@ class Mxnet(MakefilePackage): homepage = "http://mxnet.io" url = "https://github.com/apache/incubator-mxnet/archive/0.10.0.post2.tar.gz" + version('1.3.0', 'c00d6fbb2947144ce36c835308e603f002c1eb90a9f4c5a62f4d398154eed4d2', + url='https://github.com/apache/incubator-mxnet/releases/download/1.3.0/apache-mxnet-src-1.3.0-incubating.tar.gz') version('0.10.0.post2', '7819d511cf4a6efad681e6662fa966e4', url="https://github.com/apache/incubator-mxnet/archive/0.10.0.post2.tar.gz") version('0.10.0.post1', '16d540f407cd22285555b3ab22040032', @@ -43,6 +26,7 @@ class Mxnet(MakefilePackage): variant('opencv', default=True, description='Enable OpenCV support') variant('openmp', default=False, description='Enable OpenMP support') variant('profiler', default=False, description='Enable Profiler (for verification and debug only).') + variant('python', default=True, description='Install python bindings') depends_on('dmlc-core@20170508') depends_on('dmlc-core+openmp', when='+openmp') @@ -50,29 +34,45 @@ class Mxnet(MakefilePackage): depends_on('mshadow@20170721') depends_on('ps-lite@20170328') depends_on('nnvm~shared@20170418') - depends_on('openblas') + depends_on('blas') depends_on('cudnn', when='+cuda') depends_on('cudnn', when='+cuda') depends_on('cub', when='+cuda') depends_on('opencv+core+imgproc+highgui+jpeg+png+tiff~eigen~ipp@3.0:', when='+opencv') + # python extensions + depends_on('python@2.7:', type=('build', 'run'), when='+python') + depends_on('py-setuptools', type='build', when='+python') + extends('python', when='+python') + patch('makefile.patch', when='@0.10:0.11') def build(self, spec, prefix): - filter_file('export CC = gcc', '', 'make/config.mk', string=True) - filter_file('export CXX = g++', '', 'make/config.mk', string=True) + # copy template configuration file + copy('make/config.mk', 'config.mk') + + # remove compiler overrides + filter_file('export CC = gcc', '', 'config.mk', string=True) + filter_file('export CXX = g++', '', 'config.mk', string=True) + + # add blas prefix to include paths + filter_file( + '-I$(NNVM_PATH)/include', + '-I$(NNVM_PATH)/include -I%s/include' % spec['blas'].prefix, + 'Makefile', string=True + ) + + # mxnet comes with its own version of nnvm and dmlc. + # building it will fail if we use the spack paths args = [ 'CC=%s' % self.compiler.cc, 'CXX=%s' % self.compiler.cxx, - 'DMLC_CORE=%s' % spec['dmlc-core'].prefix, 'MSHADOW_PATH=%s' % spec['mshadow'].prefix, 'PS_PATH=%s' % spec['ps-lite'].prefix, - 'NNVM_PATH=%s' % spec['nnvm'].prefix, 'USE_OPENMP=%s' % ('1' if '+openmp' in spec else '0'), 'USE_CUDA=%s' % ('1' if '+cuda' in spec else '0'), 'USE_CUDNN=%s' % ('1' if '+cuda' in spec else '0'), - 'CUB_INCLUDE=%s' % spec['cub'].prefix.include, 'USE_OPENCV=%s' % ('1' if '+opencv' in spec else '0'), 'USE_PROFILER=%s' % ('1' if '+profiler' in spec else '0'), ] @@ -86,16 +86,27 @@ class Mxnet(MakefilePackage): '-lopencv_core -lopencv_imgproc -lopencv_imgcodecs', 'Makefile', string=True) - # TODO: Add more BLAS support - args.append('USE_BLAS=openblas') + if 'openblas' in spec: + args.extend(['USE_BLAS=openblas']) + elif 'atlas' in spec or 'cblas' in spec: + args.extend(['USE_BLAS=atlas']) + else: + args.extend(['USE_BLAS=blas']) if '+cuda' in spec: args.extend(['USE_CUDA_PATH=%s' % spec['cuda'].prefix, - 'CUDNN_PATH=%s' % spec['cudnn'].prefix]) + 'CUDNN_PATH=%s' % spec['cudnn'].prefix, + 'CUB_INCLUDE=%s' % spec['cub'].prefix.include]) make(*args) def install(self, spec, prefix): + # mxnet is just a shared library -- no need to install a bin tree + install_tree('include', prefix.include) install_tree('lib', prefix.lib) - install_tree('bin', prefix.bin) + + # install python bindings + if '+python' in spec: + python = which('python') + python('python/setup.py', 'install', '--prefix={0}'.format(prefix)) |