Merge pull request #55 from stackhpc/feature/vgpu

nvidia vgpu related elements

Author: jovial

elements/extra-repos/README.rst

@@ -0,0 +1,7 @@
+===========
+extra-repos
+===========
+
+Installs extra repos prior to package install
+
+* ``DIB_EXTRA_REPOS`` A space separated list of repos to install

elements/extra-repos/pre-install.d/01-extra-repos

@@ -0,0 +1,22 @@
+#!/bin/bash
+
+if [ ${DIB_DEBUG_TRACE:-0} -gt 0 ]; then
+    set -x
+fi
+
+set -euo pipefail
+
+case "$DISTRO_NAME" in
+    centos* | rhel* | rocky*)
+    	rpm -q dnf-plugins-core || dnf install -y dnf-plugins-core
+	for repo in ${DIB_EXTRA_REPOS:-}; do
+	  dnf config-manager --add-repo "$repo"
+	done
+	;;
+    *)
+	echo "Distro \"$DISTRO_NAME\" is not supported"
+	exit 1
+	;;
+esac
+
+exit 0

elements/nvidia-cuda-samples/README.rst

@@ -0,0 +1,14 @@
+===================
+nvidia-cuda-samples
+===================
+
+* Downloads and compiles NVIDIA's collection of `CUDA samples <https://github.com/NVIDIA/cuda-samples>`_ for demo purposes.
+
+`DIB_NVIDIA_CUDA_SAMPLES_VERSION`: Git ref to checkout.
+
+For example:
+
+  .. code-block::
+
+  # GIT TAG
+  export DIB_NVIDIA_CUDA_SAMPLES_VERSION:v12.1

elements/nvidia-cuda-samples/element-deps

@@ -0,0 +1 @@
+package-installs

elements/nvidia-cuda-samples/package-installs.yaml

@@ -0,0 +1,3 @@
+make:
+gcc:
+git:

elements/nvidia-cuda-samples/post-install.d/99-cuda-samples

@@ -0,0 +1,13 @@
+#!/bin/bash
+
+if [ ${DIB_DEBUG_TRACE:-1} -gt 0 ]; then
+    set -x
+fi
+set -eu
+set -o pipefail
+
+cd /opt
+git clone https://github.com/NVIDIA/cuda-samples
+cd cuda-samples
+git checkout "${DIB_NVIDIA_CUDA_SAMPLES_VERSION:-master}"
+make

elements/nvidia-vgpu/README.rst

@@ -0,0 +1,15 @@
+===========
+nvidia-vgpu
+===========
+
+* Configures client token for license server
+* Sets license type to VGPU
+
+`DIB_NVIDIA_VGPU_CLIENT_TOKEN`: Contents of license file. This can be exported from the
+license server.
+
+For example:
+
+  .. code-block::
+
+   export DIB_NVIDIA_VGPU_CLIENT_TOKEN="{{ lookup('file', 'path/to/token/downloaded/from/license/server') }}"

elements/nvidia-vgpu/environment.d/nvidia-license

@@ -0,0 +1,3 @@
+#!/bin/bash
+
+export NVIDIA_VGPU_CLIENT_TOKEN=${DIB_NVIDIA_VGPU_CLIENT_TOKEN:?"You must define DIB_NVIDIA_VGPU_CLIENT_TOKEN"}

elements/nvidia-vgpu/post-install.d/99-nvidia-license

@@ -0,0 +1,12 @@
+#!/bin/bash
+
+if [ ${DIB_DEBUG_TRACE:-1} -gt 0 ]; then
+    set -x
+fi
+set -eu
+set -o pipefail
+
+mkdir -p /etc/nvidia/ClientConfigToken/ || echo "Directory: /etc/nvidia/ClientConfigToken/ already exists, skipping"
+printf "$NVIDIA_VGPU_CLIENT_TOKEN" >> /etc/nvidia/ClientConfigToken/client.tok
+
+sed -i 's/^FeatureType.*$/FeatureType=1/' /etc/nvidia/gridd.conf