summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorBrian Van Essen <vanessen1@llnl.gov>2019-10-28 13:24:50 -0700
committerPeter Scheibel <scheibel1@llnl.gov>2019-10-28 13:24:50 -0700
commit802c3c0c5133c44bd037cac2f798b0fe3ea444fb (patch)
tree5ba3db4076b726ca5341565a9527f0b53e6582c2
parentf4e6f28e6dc0f1b16ed81fe0e14b2ce653ff63a0 (diff)
downloadspack-802c3c0c5133c44bd037cac2f798b0fe3ea444fb.tar.gz
spack-802c3c0c5133c44bd037cac2f798b0fe3ea444fb.tar.bz2
spack-802c3c0c5133c44bd037cac2f798b0fe3ea444fb.tar.xz
spack-802c3c0c5133c44bd037cac2f798b0fe3ea444fb.zip
Packages: Add cuDNN versions; update LBANN dependencies (#13412)
* Add cuDNN version 7.6.3-10.1 (for x86_64 and ppc64le) * Add py-pytest and py-protobuf depenencies to LBANN
-rw-r--r--var/spack/repos/builtin/packages/cudnn/package.py6
-rw-r--r--var/spack/repos/builtin/packages/lbann/package.py2
2 files changed, 7 insertions, 1 deletions
diff --git a/var/spack/repos/builtin/packages/cudnn/package.py b/var/spack/repos/builtin/packages/cudnn/package.py
index b56aacab4f..8a872ce949 100644
--- a/var/spack/repos/builtin/packages/cudnn/package.py
+++ b/var/spack/repos/builtin/packages/cudnn/package.py
@@ -13,6 +13,10 @@ class Cudnn(Package):
homepage = "https://developer.nvidia.com/cudnn"
+ version('7.6.3-10.1-x86_64', sha256='352557346d8111e2f954c494be1a90207103d316b8777c33e62b3a7f7b708961',
+ url='https://developer.download.nvidia.com/compute/redist/cudnn/v7.6.3/cudnn-10.1-linux-x64-v7.6.3.30.tgz')
+ version('7.6.3-10.1-ppc64le', sha256='f274735a8fc31923d3623b1c3d2b1d0d35bb176687077c6a4d4353c6b900d8ee',
+ url='https://developer.download.nvidia.com/compute/redist/cudnn/v7.6.3/cudnn-10.1-linux-ppc64le-v7.6.3.30.tgz')
version('7.5.1-10.1-x86_64', sha256='2c833f43c9147d9a25a20947a4c5a5f5c33b2443240fd767f63b330c482e68e0',
url='https://developer.download.nvidia.com/compute/redist/cudnn/v7.5.1/cudnn-10.1-linux-x64-v7.5.1.10.tgz')
version('7.5.1-10.1-ppc64le', sha256='a9e23bc83c970daec20874ccd1d8d80b648adf15440ecd0164818b330b1e2663',
@@ -41,7 +45,7 @@ class Cudnn(Package):
depends_on('cuda@8:', when='@5.1:@7')
depends_on('cuda@9:', when='@7.2:@7.4')
depends_on('cuda@10:', when='@7.5.0-10.0-ppc64le,7.5.0-10.0-x86_64,7.5.1-10.0-ppc64le,7.5.1-10.0-x86_64')
- depends_on('cuda@10.1:', when='@7.5.0-10.1-ppc64le,7.5.0-10.1-x86_64,7.5.1-10.1-ppc64le,7.5.1-10.1-x86_64')
+ depends_on('cuda@10.1:', when='@7.5.0-10.1-ppc64le,7.5.0-10.1-x86_64,7.5.1-10.1-ppc64le,7.5.1-10.1-x86_64,7.6.3-10.1-ppc64le,7.6.3-10.1-x86_64')
def install(self, spec, prefix):
install_tree('.', prefix)
diff --git a/var/spack/repos/builtin/packages/lbann/package.py b/var/spack/repos/builtin/packages/lbann/package.py
index cf39b3e5c1..79fd5f0035 100644
--- a/var/spack/repos/builtin/packages/lbann/package.py
+++ b/var/spack/repos/builtin/packages/lbann/package.py
@@ -112,6 +112,8 @@ class Lbann(CMakePackage):
depends_on('py-onnx@1.3.0:', type='run', when='@:0.90,0.99:')
depends_on('py-pandas@0.24.1:', type='run', when='@:0.90,0.99:')
depends_on('py-texttable@1.4.0:', type='run', when='@:0.90,0.99:')
+ depends_on('py-pytest', type='test', when='@:0.90,0.99:')
+ depends_on('py-protobuf+cpp', type='run', when='@:0.90,0.99:')
depends_on('py-breathe', type='build', when='+docs')
depends_on('py-m2r', type='build', when='+docs')