From 7bd5397aca5f003168f4b99d06471542bb70215f Mon Sep 17 00:00:00 2001 From: Jen Herting Date: Tue, 27 Oct 2020 21:10:40 -0400 Subject: New Package: py-rmm (#19104) * [py-rmm] created template * [py-rmm] added dependencies * [py-rmm] setting build_directory * [py-rmm] added dependencies cuda and spdlog * [py-rmm] added homepage and description. removed fixmes * [py-rmm] added for loop for dependency versions * [py-rmm] flake8 * [py-rmm] updating upper range of python version * [py-rmm] removing python upper limit --- var/spack/repos/builtin/packages/py-rmm/package.py | 30 ++++++++++++++++++++++ 1 file changed, 30 insertions(+) create mode 100644 var/spack/repos/builtin/packages/py-rmm/package.py diff --git a/var/spack/repos/builtin/packages/py-rmm/package.py b/var/spack/repos/builtin/packages/py-rmm/package.py new file mode 100644 index 0000000000..96631d3fbb --- /dev/null +++ b/var/spack/repos/builtin/packages/py-rmm/package.py @@ -0,0 +1,30 @@ +# 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 PyRmm(PythonPackage): + """RMM: RAPIDS Memory Manager. Achieving optimal + performance in GPU-centric workflows frequently requires + customizing how host and device memory are allocated.""" + + homepage = "https://github.com/rapidsai/rmm" + url = "https://github.com/rapidsai/rmm/archive/v0.15.0.tar.gz" + + version('0.15.0', sha256='599f97b95d169a90d11296814763f7e151a8a1e060ba10bc6c8f4684a5cd7972') + + depends_on('python@3.6:', type=('build', 'run')) + depends_on('py-setuptools', type='build') + depends_on('py-cython', type='build') + depends_on('py-numba', type=('build', 'run')) + + for v in ('@0.15.0',): + depends_on('librmm' + v, when=v) + + depends_on('cuda@9:') + depends_on('spdlog') + + build_directory = 'python' -- cgit v1.2.3-70-g09d2