diff options
| author | rvinaybharadwaj <vinayr@lanl.gov> | 2020-10-08 08:24:50 -0600 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2020-10-08 08:24:50 -0600 |
| commit | cb11a58fdd48717cfbc950b3e87c55a8e5daea81 (patch) | |
| tree | 7b29f73c285fb14a3a6696dbfc97d332def25130 /var | |
| parent | 8b0a9f482f076766f80b2b6bcff39d46bd80ab79 (diff) | |
| download | spack-cb11a58fdd48717cfbc950b3e87c55a8e5daea81.tar.gz spack-cb11a58fdd48717cfbc950b3e87c55a8e5daea81.tar.bz2 spack-cb11a58fdd48717cfbc950b3e87c55a8e5daea81.tar.xz spack-cb11a58fdd48717cfbc950b3e87c55a8e5daea81.zip | |
Add mlperf-deepcam (#19206)
* Add mlperf-deepcam
* Change mlperf-deepcam to reflect updated name of dependency
Diffstat (limited to 'var')
| -rw-r--r-- | var/spack/repos/builtin/packages/mlperf-deepcam/package.py | 43 |
1 files changed, 43 insertions, 0 deletions
diff --git a/var/spack/repos/builtin/packages/mlperf-deepcam/package.py b/var/spack/repos/builtin/packages/mlperf-deepcam/package.py new file mode 100644 index 0000000000..55c74fda45 --- /dev/null +++ b/var/spack/repos/builtin/packages/mlperf-deepcam/package.py @@ -0,0 +1,43 @@ +# Copyright 2013-2020 Lawrence Livermore National Security, LLC and other +# Spack Project Developers. See the top-level COPYRIGHT file for details. +# +# SPDX-License-Identifier: (Apache-2.0 OR MIT) + + +from spack import * + + +class MlperfDeepcam(Package): + """PyTorch implementation for the climate segmentation benchmark, + based on the Exascale Deep Learning for Climate Analytics""" + + homepage = "https://github.com/azrael417/mlperf-deepcam" + git = "https://github.com/azrael417/mlperf-deepcam.git" + + version('master', branch='master') + + tags = ['proxy-app'] + + depends_on('python@3:', type=('build', 'run')) + + depends_on('py-h5py', type=('build', 'run')) + depends_on('py-pycuda', type=('build', 'run')) + depends_on('py-mpi4py', type=('build', 'run')) + depends_on('py-torch+cuda', when='+cuda', type=('build', 'run')) + depends_on('py-torch~cuda~cudnn~nccl', when='~cuda', type=('build', 'run')) + depends_on('py-matplotlib', type=('build', 'run')) + depends_on('py-basemap', type=('build', 'run')) + depends_on('py-pillow', type=('build', 'run')) + depends_on('py-numpy', type=('build', 'run')) + depends_on('py-pillow', type=('build', 'run')) + depends_on('py-argparse', type=('build', 'run')) + depends_on('py-pandas', type=('build', 'run')) + depends_on('py-apex', type=('build', 'run')) + depends_on('py-wandb', type=('build', 'run')) + depends_on('py-apex', type=('build', 'run')) + depends_on('py-mlperf-logging', type=('build', 'run')) + depends_on('py-pytorch-gradual-warmup-lr', type=('build', 'run')) + + def install(self, spec, prefix): + # Mostly about providing an environment so just copy everything + install_tree('.', prefix) |