diff options
author | Brian Van Essen <vanessen1@llnl.gov> | 2022-06-30 22:37:47 -0700 |
---|---|---|
committer | GitHub <noreply@github.com> | 2022-06-30 22:37:47 -0700 |
commit | 06f4f4391542a46fc7404c1e85a574b464a3a941 (patch) | |
tree | ecd5f6b8d805bdded48bc7c29e397f312fbd0262 /var/spack/repos/builtin/packages/lbann/package.py | |
parent | 4a6b447465bec34d4d0e1f83c1be37f28a3357d4 (diff) | |
download | spack-06f4f4391542a46fc7404c1e85a574b464a3a941.tar.gz spack-06f4f4391542a46fc7404c1e85a574b464a3a941.tar.bz2 spack-06f4f4391542a46fc7404c1e85a574b464a3a941.tar.xz spack-06f4f4391542a46fc7404c1e85a574b464a3a941.zip |
New package: NVIDIA cuTensor (#31397)
Diffstat (limited to 'var/spack/repos/builtin/packages/lbann/package.py')
-rw-r--r-- | var/spack/repos/builtin/packages/lbann/package.py | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/var/spack/repos/builtin/packages/lbann/package.py b/var/spack/repos/builtin/packages/lbann/package.py index 6c80069802..3500d3eb62 100644 --- a/var/spack/repos/builtin/packages/lbann/package.py +++ b/var/spack/repos/builtin/packages/lbann/package.py @@ -172,6 +172,7 @@ class Lbann(CMakePackage, CudaPackage, ROCmPackage): depends_on('cudnn', when='@0.90:0.100 +cuda') depends_on('cudnn@8.0.2:', when='@:0.90,0.101: +cuda') depends_on('cub', when='@0.94:0.98.2 +cuda ^cuda@:10') + depends_on('cutensor', when='@:0.90,0.102: +cuda') depends_on('hipcub', when='+rocm') depends_on('mpi') depends_on('hwloc@1.11:', when='@:0.90,0.102: +hwloc') |