test without

huggingface · Aug 30, 2024 · ae5d606 · ae5d606
1 parent ed111df
commit ae5d606
Showing 1 changed file with 28 additions and 25 deletions.
diff --git a/.github/workflows/test_api_rocm.yaml b/.github/workflows/test_api_rocm.yaml
@@ -26,37 +26,37 @@ concurrency:
   group: ${{ github.workflow }}-${{ github.event.pull_request.number || github.ref }}
 
 jobs:
-  setup_rocm_docker_devices:
-    runs-on: [self-hosted, amd-gpu, single-gpu, mi250]
+  # setup_rocm_docker_devices:
+  #   runs-on: [self-hosted, amd-gpu, single-gpu, mi250]
 
-    container:
-      image: ghcr.io/huggingface/optimum-benchmark:latest-rocm
-      options: --ipc host
-        --shm-size "16gb"
-        --group-add video
-        --device /dev/kfd
-        --device /dev/dri
-        --env ROCR_VISIBLE_DEVICES
-        --volume /mnt/cache/.cache/huggingface:/mnt/cache/
+  #   container:
+  #     image: ghcr.io/huggingface/optimum-benchmark:latest-rocm
+  #     options: --ipc host
+  #       --shm-size "16gb"
+  #       --group-add video
+  #       --device /dev/kfd
+  #       --device /dev/dri
+  #       --env ROCR_VISIBLE_DEVICES
+  #       --volume /mnt/cache/.cache/huggingface:/mnt/cache/
 
-    outputs:
-      docker_devices: ${{ steps.docker_devices.outputs.docker_devices }}
+  #   outputs:
+  #     docker_devices: ${{ steps.docker_devices.outputs.docker_devices }}
 
-    steps:
-      - name: Checkout
-        uses: actions/checkout@v4
+  #   steps:
+  #     - name: Checkout
+  #       uses: actions/checkout@v4
 
-      - name: Run script to get devices
-        run: |
-          chmod +x scripts/rocm_docker_devices.sh
-          ./scripts/rocm_docker_devices.sh
+  #     - name: Run script to get devices
+  #       run: |
+  #         chmod +x scripts/rocm_docker_devices.sh
+  #         ./scripts/rocm_docker_devices.sh
 
-      - name: Set outputs
-        id: docker_devices
-        run: echo "docker_devices=$(./scripts/rocm_docker_devices.sh)" >> "$GITHUB_OUTPUT"
+  #     - name: Set outputs
+  #       id: docker_devices
+  #       run: echo "docker_devices=$(./scripts/rocm_docker_devices.sh)" >> "$GITHUB_OUTPUT"
 
   run_api_rocm_tests:
-    needs: setup_rocm_docker_devices
+    # needs: setup_rocm_docker_devices
 
     runs-on: [self-hosted, amd-gpu, single-gpu, mi250]
 
@@ -66,7 +66,10 @@ jobs:
         --shm-size "16gb"
         --group-add video
         --volume /mnt/cache/.cache/huggingface:/mnt/cache/
-        ${{ needs.setup_rocm_docker_devices.outputs.docker_devices }}
+        --device /dev/kfd
+        --device /dev/dri
+        --env ROCR_VISIBLE_DEVICES
+        # ${{ needs.setup_rocm_docker_devices.outputs.docker_devices }}
 
     steps:
       - name: Checkout