add rocminfo in GPU tests

huggingface · Nov 16, 2023 · 83f24ca · 83f24ca
1 parent 3961c52
commit 83f24ca
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/.github/workflows/self-scheduled-amd.yml b/.github/workflows/self-scheduled-amd.yml
@@ -131,6 +131,10 @@ jobs:
         working-directory: /transformers
         run: |
           python3 utils/print_env.py
+
+      - name: ROCMINFO
+        run: |
+          rocminfo  | grep "Agent" -A 14
           
       - name: Update clone
         working-directory: /transformers