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

Added also py-pynvml

2 files changed, 44 insertions, 0 deletions

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