From 6aa2a76a9047be0be806cfa3869759d14ebd00e3 Mon Sep 17 00:00:00 2001
From: daniel <bauer@cbs.tu-darmstadt.de>
Date: Sun, 2 Aug 2020 12:37:13 +0200
Subject: [PATCH] add multi gpu support, closes #1

---
 scripts/gpu.sh | 12 ++++++++----
 1 file changed, 8 insertions(+), 4 deletions(-)

diff --git a/scripts/gpu.sh b/scripts/gpu.sh
index 81bc22d..31e2e04 100755
--- a/scripts/gpu.sh
+++ b/scripts/gpu.sh
@@ -2,13 +2,17 @@
 
 print_gpu_usage() {
     function getGPUUsage() {
-        gpuUsage=$(nvidia-smi -q -d UTILIZATION | grep Gpu | awk '{print $3}')
+	    gpuUsage=($(nvidia-smi -q -d UTILIZATION | grep Gpu | awk '{print $3}'))
     }
     getGPUUsage
-    if [ -z "$GpuUsage" ]; then
-        echo "$gpuUsage%"
-    else
+    gpuUsage=("${gpuUsage[@]/%/%}")
+    
+    function join_by { local IFS="$1"; shift; echo "$*"; }
+    gpuUsage=`join_by " " ${gpuUsage[@]}`
+    if [ -z "$gpuUsage" ]; then
 	echo "-"
+    else
+	echo "$gpuUsage"
     fi
 }