diff --git a/var/spack/repos/builtin/packages/py-pynvml/package.py b/var/spack/repos/builtin/packages/py-pynvml/package.py new file mode 100644 index 0000000000..237949e521 --- /dev/null +++ b/var/spack/repos/builtin/packages/py-pynvml/package.py @@ -0,0 +1,19 @@ +# 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) + + +class PyPynvml(PythonPackage): + """Provides a Python interface to GPU management and monitoring + functions. This is a wrapper around the NVML library. For + information about the NVML library, see the NVML developer page + http://developer.nvidia.com/nvidia-management-library-nvml""" + + homepage = "http://www.nvidia.com/" + url = "https://pypi.io/packages/source/p/pynvml/pynvml-8.0.4.tar.gz" + + version('8.0.4', sha256='c8d4eadc648c7e12a3c9182a9750afd8481b76412f83747bcc01e2aa829cde5d') + + depends_on('py-setuptools', type='build') + depends_on('python@3.6:', type=('build', 'run')) diff --git a/var/spack/repos/builtin/packages/py-ucx-py/package.py b/var/spack/repos/builtin/packages/py-ucx-py/package.py new file mode 100644 index 0000000000..973c3a8e99 --- /dev/null +++ b/var/spack/repos/builtin/packages/py-ucx-py/package.py @@ -0,0 +1,25 @@ +# 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) + + +class PyUcxPy(PythonPackage): + """UCX-Py is the Python interface for UCX, a low-level + high-performance networking library. UCX and UCX-Py supports + several transport methods including InfiniBand and NVLink while + still using traditional networking protocols like TCP.""" + + homepage = "https://ucx-py.readthedocs.io/en/latest/" + url = "https://github.com/rapidsai/ucx-py/archive/v0.16.0.tar.gz" + + version('0.16.0', sha256='12c1c982ee337b8dc026d3a6e8e63d96bf021c5c555fe173642908d3c3bec36e') + + depends_on('py-setuptools', type='build') + depends_on('python@3.6:', type=('build', 'run')) + depends_on('py-cython@0.29.14:2.999.999', type='build') + depends_on('py-numpy', type=('build', 'run')) + depends_on('py-psutil', type=('build', 'run')) + depends_on('py-pynvml', type=('build', 'run')) + depends_on('ucx') + depends_on('hwloc')