trigger test

huggingface · Nov 21, 2023 · 7c1a41e · 7c1a41e
1 parent 7857978
commit 7c1a41e
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 10 deletions.
diff --git a/.github/workflows/self-scheduled-amd-caller.yml b/.github/workflows/self-scheduled-amd-caller.yml
@@ -6,19 +6,20 @@ on:
   push:
     branches:
       - run_amd_scheduled_ci_caller*
+      - rocm-deepspeed-scheduled
 
 jobs:
   run_amd_ci_mi210:
     name: AMD mi210
-    if: (cancelled() != true) && ((github.event_name == 'schedule') || ((github.event_name == 'push') && startsWith(github.ref_name, 'run_amd_scheduled_ci_caller')))
+    if: (cancelled() != true) && ((github.event_name == 'schedule') || ((github.event_name == 'push') && startsWith(github.ref_name, 'run_amd_scheduled_ci_caller'))) || ((github.event_name == 'push') && startsWith(github.ref_name, 'rocm-deepspeed-scheduled')))
     uses: ./.github/workflows/self-scheduled-amd.yml
     with:
       gpu_flavor: mi210
     secrets: inherit
 
   run_amd_ci_mi250:
     name: AMD mi250
-    if: (cancelled() != true) && ((github.event_name == 'schedule') || ((github.event_name == 'push') && startsWith(github.ref_name, 'run_amd_scheduled_ci_caller')))
+    if: (cancelled() != true) && ((github.event_name == 'schedule') || ((github.event_name == 'push') && startsWith(github.ref_name, 'run_amd_scheduled_ci_caller'))) || ((github.event_name == 'push') && startsWith(github.ref_name, 'rocm-deepspeed-scheduled')))
     uses: ./.github/workflows/self-scheduled-amd.yml
     with:
       gpu_flavor: mi250

diff --git a/.github/workflows/self-scheduled-amd.yml b/.github/workflows/self-scheduled-amd.yml
@@ -357,7 +357,7 @@ jobs:
           path: /transformers/reports/${{ matrix.machine_type }}_tests_torch_pipeline_gpu
 
   run_all_tests_torch_rocm_deepspeed_gpu:
-    name: Torch ROCm extension tests
+    name: Torch ROCm deepspeed tests
     strategy:
       fail-fast: false
       matrix:
@@ -380,34 +380,37 @@ jobs:
       - name: ROCM-SMI
         run: |
           rocm-smi
+      - name: ROCM-INFO
+        run: |
+          rocminfo  | grep "Agent" -A 14
       - name: Show ROCR environment
         run: |
           echo "ROCR: $ROCR_VISIBLE_DEVICES"
 
       - name: Environment
-        working-directory: /workspace/transformers
+        working-directory: /transformers
         run: |
           python utils/print_env.py
 
       - name: Show installed libraries and their versions
-        working-directory: /workspace/transformers
+        working-directory: /transformers
         run: pip freeze
 
       - name: Run all tests on GPU
-        working-directory: /workspace/transformers
-        run: python -m pytest -v --make-reports=${{ matrix.machine_type }}_tests_torch_rocm_extensions_gpu tests/deepspeed tests/extended
+        working-directory: /transformers
+        run: python -m pytest -v --make-reports=${{ matrix.machine_type }}_tests_torch_rocm_deepspeed_gpu tests/deepspeed tests/extended
 
       - name: Failure short reports
         if: ${{ failure() }}
         continue-on-error: true
-        run: cat /workspace/transformers/reports/${{ matrix.machine_type }}_tests_torch_rocm_extensions_gpu/failures_short.txt
+        run: cat /transformers/reports/${{ matrix.machine_type }}_tests_torch_rocm_deepspeed_gpu/failures_short.txt
 
       - name: Test suite reports artifacts
         if: ${{ always() }}
         uses: actions/upload-artifact@v3
         with:
-          name: ${{ matrix.machine_type }}_run_tests_torch_rocm_extensions_gpu_test_reports
-          path: /workspace/transformers/reports/${{ matrix.machine_type }}_tests_torch_rocm_extensions_gpu
+          name: ${{ matrix.machine_type }}_run_tests_torch_rocm_deepspeed_gpu_test_reports
+          path: /transformers/reports/${{ matrix.machine_type }}_tests_torch_rocm_deepspeed_gpu
 
   run_extract_warnings:
     name: Extract warnings in CI artifacts