hlrs-software-stack/spack
7
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

py-ucx-py: added new package at v0.16.0 (#20093)

Browse source 
Added also py-pynvml
This commit is contained in:
Andrew W Elble 2020-12-02 13:15:23 -05:00 committed by GitHub
parent 5deeff89f3
commit 3c25ae5130
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 44 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

19
var/spack/repos/builtin/packages/py-pynvml/package.py Normal file
View file

@ -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'))

25
var/spack/repos/builtin/packages/py-ucx-py/package.py Normal file
View file

@ -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')