18324: Add %G for GPU device count
authorPeter Amstutz <peter.amstutz@curii.com>
Tue, 21 Dec 2021 21:57:00 +0000 (16:57 -0500)
committerPeter Amstutz <peter.amstutz@curii.com>
Tue, 21 Dec 2021 21:57:00 +0000 (16:57 -0500)
Arvados-DCO-1.1-Signed-off-by: Peter Amstutz <peter.amstutz@curii.com>

lib/lsf/dispatch.go

index 537d52a072d6a503262b1a228c868afc8f28b151..f34f840e5e131bcc1a6148733b505a0c78c184a6 100644 (file)
@@ -306,6 +306,7 @@ func (disp *dispatcher) bsubArgs(container arvados.Container) ([]string, error)
                "%M": fmt.Sprintf("%d", mem),
                "%T": fmt.Sprintf("%d", tmp),
                "%U": container.UUID,
+               "%G": fmt.Sprintf("%d", container.RuntimeConstraints.CUDA.DeviceCount),
        }
 
        re := regexp.MustCompile(`%.`)