tmp

Author: S1ro1

.github/workflows/nvidia-smoke.yml

@@ -1,24 +1,18 @@
 name: gpu-smoke
 on:
-  workflow_dispatch: {}
   push:
-    branches: [ main ]
-
+    branches: [ nvidia-gpu-runners ]
+  workflow_dispatch: {}
 jobs:
   smoke:
     runs-on: [self-hosted, gpu]
-    timeout-minutes: 30
     steps:
       - uses: actions/checkout@v4
-      - name: Show GPU
-        run: |
-          nvidia-smi || true
-      - name: Torch sanity
-        run: |
+      - run: nvidia-smi || true
+      - run: |
           python - <<'PY'
           import torch
-          print("CUDA available:", torch.cuda.is_available())
-          print("device_count:", torch.cuda.device_count())
+          print("cuda?", torch.cuda.is_available(), "count:", torch.cuda.device_count())
           if torch.cuda.is_available():
-              print("device_0:", torch.cuda.get_device_name(0))
+              print("name:", torch.cuda.get_device_name(0))
           PY