Add script to check SSH

Author: xuzhao9

.ci/torchbench/check-ssh.sh

@@ -0,0 +1,13 @@
+#!/bin/bash
+
+# Loop to continuously check the output of `who`
+while true; do
+    # Check if `who` returns non-empty output
+    if who | grep -q '.'; then
+        echo "User(s) logged in. Sleeping for 100 seconds..."
+        sleep 100
+    else
+        echo "No users logged in. Exiting..."
+        exit 0
+    fi
+done

.github/workflows/_linux-test-a10g.yml

@@ -28,9 +28,9 @@ jobs:
       - name: Install TorchBench
         run: |
           bash ./.ci/torchbench/install.sh
-      - name: Test TorchBench
+      - name: Wait for SSH session to end
         run: |
-          bash ./.ci/torchbench/test.sh
+          bash ./.ci/torchbench/check-ssh.sh
       - name: Clean up Conda env
         if: always()
         run: |