From 0a259ab934b2fc68245efd828b2f8dd7d1bbf275 Mon Sep 17 00:00:00 2001 From: Andrew W Elble Date: Fri, 30 Apr 2021 10:55:17 -0400 Subject: 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+ --- var/spack/repos/builtin/packages/mxnet/package.py | 20 +++++++++++++++----- 1 file changed, 15 insertions(+), 5 deletions(-) (limited to 'var') 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: -- cgit v1.2.3-70-g09d2