From cb11a58fdd48717cfbc950b3e87c55a8e5daea81 Mon Sep 17 00:00:00 2001 From: rvinaybharadwaj Date: Thu, 8 Oct 2020 08:24:50 -0600 Subject: Add mlperf-deepcam (#19206) * Add mlperf-deepcam * Change mlperf-deepcam to reflect updated name of dependency --- .../builtin/packages/mlperf-deepcam/package.py | 43 ++++++++++++++++++++++ 1 file changed, 43 insertions(+) create mode 100644 var/spack/repos/builtin/packages/mlperf-deepcam/package.py 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) -- cgit v1.2.3-60-g2f50