test : add libnvml for mig devices

This commit is contained in:
Yann Dupont 2024-05-06 12:01:20 +02:00
parent 7f19a10086
commit 6eaead1912

View file

@ -24,3 +24,21 @@
#$flags))))))
(define-public slurm-glicid+nvml
(package
(inherit slurm-glicid)
(name "slurm-glicid+nvml")
(inputs (modify-inputs (package-inputs slurm-glicid)
(prepend cuda nvidia-nvml)))
(arguments
(substitute-keyword-arguments (package-arguments slurm-glicid)
;; Necessary because libnvidia-ml.so.1 does not actually exist.
((#:validate-runpath? _ #f)
#false)
((#:configure-flags flags '())
#~(append (list (string-append "--with-nvml="
#$(this-package-input "nvidia-nvml")))
#$flags))))))