diff options
author | Andrew W Elble <aweits@rit.edu> | 2021-04-30 10:55:17 -0400 |
---|---|---|
committer | GitHub <noreply@github.com> | 2021-04-30 09:55:17 -0500 |
commit | 0a259ab934b2fc68245efd828b2f8dd7d1bbf275 (patch) | |
tree | d0c6d02b373eebdda19611db53e669b0717bb1c8 | |
parent | bb038c3b6c4448a151e47551155cec8010d0c03c (diff) | |
download | spack-0a259ab934b2fc68245efd828b2f8dd7d1bbf275.tar.gz spack-0a259ab934b2fc68245efd828b2f8dd7d1bbf275.tar.bz2 spack-0a259ab934b2fc68245efd828b2f8dd7d1bbf275.tar.xz spack-0a259ab934b2fc68245efd828b2f8dd7d1bbf275.zip |
mxnet: new version 1.8.0 (#23358)
* mxnet: new version 1.8.0
use submodules on master
introduce constraints on cuda versions supported
handle USE_MKLDNN->USE_ONEDNN conversion
* * use define for USE_CUTENSOR
* fix up dependencies for 2.0.0+
-rw-r--r-- | var/spack/repos/builtin/packages/mxnet/package.py | 20 |
1 files changed, 15 insertions, 5 deletions
diff --git a/var/spack/repos/builtin/packages/mxnet/package.py b/var/spack/repos/builtin/packages/mxnet/package.py index 9da93e0154..72cf4ec3ac 100644 --- a/var/spack/repos/builtin/packages/mxnet/package.py +++ b/var/spack/repos/builtin/packages/mxnet/package.py @@ -35,8 +35,9 @@ class Mxnet(CMakePackage, CudaPackage): 'mxnet._ffi._cy3', 'mxnet._ffi._ctypes' ] - version('master', branch='master') - version('1.master', branch='v1.x') + version('master', branch='master', submodules=True) + version('1.master', branch='v1.x', submodules=True) + version('1.8.0', sha256='95aff985895aba409c08d5514510ae38b88490cfb6281ab3a5ff0f5826c8db54') version('1.7.0', sha256='1d20c9be7d16ccb4e830e9ee3406796efaf96b0d93414d676337b64bc59ced18') version('1.6.0', sha256='01eb06069c90f33469c7354946261b0a94824bbaf819fd5d5a7318e8ee596def') version('1.3.0', sha256='c00d6fbb2947144ce36c835308e603f002c1eb90a9f4c5a62f4d398154eed4d2', deprecated=True) @@ -58,6 +59,8 @@ class Mxnet(CMakePackage, CudaPackage): depends_on('ninja', type='build') depends_on('pkgconfig', when='@1.6.0', type='build') depends_on('blas') + depends_on('cuda@:10.2.999', when='@:1.8.0') + depends_on('cuda@:11.3', when='@2.0.0:') depends_on('cudnn', when='+cudnn') depends_on('nccl', when='+nccl') depends_on('opencv+core+highgui+imgproc+imgcodecs', when='+opencv') @@ -66,10 +69,13 @@ class Mxnet(CMakePackage, CudaPackage): # python/setup.py extends('python', when='+python') - depends_on('python@2.7:2.8,3.4:', when='+python', type=('build', 'run')) + depends_on('python@2.7:2.8,3.4:', when='@:1.8.0+python', type=('build', 'run')) + depends_on('python@3.6:', when='@2.0.0:+python', type=('build', 'run')) + depends_on('py-contextvars', when='@2.0.0:+python ^python@3.6.0:3.6.999', type=('build', 'run')) depends_on('py-setuptools', when='+python', type='build') depends_on('py-cython', when='+python', type='build') - depends_on('py-numpy@1.16.1:1.999', when='@1.6:+python', type=('build', 'run')) + depends_on('py-numpy@1.17:', when='@2.0.0:+python', type=('build', 'run')) + depends_on('py-numpy@1.16.1:1.999', when='@1.6:1.8.0+python', type=('build', 'run')) depends_on('py-numpy@1.8.2:1.15.0', when='@1.3.0+python', type=('build', 'run')) depends_on('py-requests@2.20.0:2.999', when='@1.6:+python', type=('build', 'run')) depends_on('py-requests@2.18.4:2.18.999', when='@1.3.0+python', type=('build', 'run')) @@ -101,8 +107,12 @@ class Mxnet(CMakePackage, CudaPackage): self.define_from_variant('USE_OPENCV', 'opencv'), self.define_from_variant('USE_OPENMP', 'openmp'), self.define_from_variant('USE_LAPACK', 'lapack'), - self.define_from_variant('USE_MKLDNN', 'mkldnn'), ] + if self.spec.satisfies('@:1.8.0'): + args.append(self.define_from_variant('USE_MKLDNN', 'mkldnn')) + if self.spec.satisfies('@2.0.0:'): + args.append(self.define_from_variant('USE_ONEDNN', 'mkldnn')) + args.append(self.define('USE_CUTENSOR', False)) if '+cuda' in self.spec: if 'cuda_arch=none' not in self.spec: |