fix

huggingface · Mar 29, 2024 · 675410c · 675410c
1 parent 2026fa5
commit 675410c
Show file tree

Hide file tree

Showing 9 changed files with 21 additions and 12 deletions.
diff --git a/.github/workflows/test_api_cuda.yaml b/.github/workflows/test_api_cuda.yaml
@@ -31,7 +31,7 @@ jobs:
             --shm-size 64G 
             --env USE_CUDA="1" 
             --env PROCESS_SPECIFIC_VRAM="0" 
-            --volume $WORKSPACE:/workspace
+            --volume ${{ env.WORKSPACE }}:/workspace
             --workdir /workspace
           run: |
             pip install -e .[testing,timm,diffusers]

diff --git a/.github/workflows/test_api_rocm.yaml b/.github/workflows/test_api_rocm.yaml
@@ -48,8 +48,8 @@ jobs:
             --shm-size 64G
             --env USE_ROCM="1"
             --device /dev/kfd
-            --device /dev/dri/$DEVICE
-            --volume $WORKSPACE:/workspace
+            --device /dev/dri/${{ env.DEVICE }}
+            --volume ${{ env.WORKSPACE }}:/workspace
             --workdir /workspace
           run: |
             pip install -e .[testing,timm,diffusers]

diff --git a/.github/workflows/test_cli_cuda_pytorch_multi_gpu.yaml b/.github/workflows/test_cli_cuda_pytorch_multi_gpu.yaml
@@ -19,6 +19,9 @@ jobs:
       - name: Checkout
         uses: actions/checkout@v3
 
+      - name: Pull image
+        run: docker pull pytorch/pytorch:2.2.2-cuda12.1-cudnn8-devel
+
       - name: Run tests
         uses: addnab/docker-run-action@v3
         env:

diff --git a/.github/workflows/test_cli_cuda_pytorch_single_gpu.yaml b/.github/workflows/test_cli_cuda_pytorch_single_gpu.yaml
@@ -19,6 +19,9 @@ jobs:
       - name: Checkout
         uses: actions/checkout@v3
 
+      - name: Pull image
+        run: docker pull pytorch/pytorch:2.2.2-cuda12.1-cudnn8-runtime
+
       - name: Run tests
         uses: addnab/docker-run-action@v3
         env:
@@ -31,7 +34,7 @@ jobs:
             --shm-size 64G
             --env USE_CUDA="1"
             --env PROCESS_SPECIFIC_VRAM="0"
-            --volume $WORKSPACE:/workspace
+            --volume ${{ env.WORKSPACE }}:/workspace
             --workdir /workspace
           run: |
             pip install -e .[testing,diffusers,timm,peft,bitsandbytes,autoawq,auto-gptq]

diff --git a/.github/workflows/test_cli_cuda_tensorrt_llm.yaml b/.github/workflows/test_cli_cuda_tensorrt_llm.yaml
@@ -19,6 +19,9 @@ jobs:
       - name: Checkout
         uses: actions/checkout@v3
 
+      - name: Pull image
+        run: docker pull huggingface/optimum-nvidia:latest
+
       - name: Run tests
         uses: addnab/docker-run-action@v3
         env:
@@ -31,7 +34,7 @@ jobs:
             --shm-size 64G 
             --env USE_CUDA="1" 
             --env PROCESS_SPECIFIC_VRAM="0" 
-            --volume ${{ github.workspace }}:/workspace
+            --volume ${{ env.WORKSPACE }}:/workspace
             --workdir /workspace
           run: |
             pip install -e .[testing] &&

diff --git a/.github/workflows/test_cli_cuda_torch_ort_multi_gpu.yaml b/.github/workflows/test_cli_cuda_torch_ort_multi_gpu.yaml
@@ -45,7 +45,7 @@ jobs:
             --shm-size 64G
             --env USE_CUDA="1"
             --env PROCESS_SPECIFIC_VRAM="0"
-            --volume $WORKSPACE:/workspace
+            --volume ${{ env.WORKSPACE }}:/workspace
             --workdir /workspace
           run: |
             pip install -e .[testing,torch-ort,deepspeed,peft] &&

diff --git a/.github/workflows/test_cli_cuda_torch_ort_single_gpu.yaml b/.github/workflows/test_cli_cuda_torch_ort_single_gpu.yaml
@@ -45,7 +45,7 @@ jobs:
             --shm-size 64G
             --env USE_CUDA="1"
             --env PROCESS_SPECIFIC_VRAM="0"
-            --volume $WORKSPACE:/workspace
+            --volume ${{ env.WORKSPACE }}:/workspace
             --workdir /workspace
           run: |
             pip install -e .[testing,torch-ort,peft] &&

diff --git a/.github/workflows/test_cli_rocm_pytorch_multi_gpu.yaml b/.github/workflows/test_cli_rocm_pytorch_multi_gpu.yaml
@@ -49,9 +49,9 @@ jobs:
             --shm-size 64G
             --env USE_ROCM="1"
             --device /dev/kfd
-            --device /dev/dri/$DEVICE0
-            --device /dev/dri/$DEVICE1
-            --volume $WORKSPACE:/workspace
+            --device /dev/dri/${{ env.DEVICE0 }}
+            --device /dev/dri/${{ env.DEVICE1 }}
+            --volume ${{ env.WORKSPACE }}:/workspace
             --workdir /workspace
           run: |
             pip install -e .[testing,diffusers,timm,deepspeed,peft,autoawq-rocm,auto-gptq-rocm]

diff --git a/.github/workflows/test_cli_rocm_pytorch_single_gpu.yaml b/.github/workflows/test_cli_rocm_pytorch_single_gpu.yaml
@@ -48,8 +48,8 @@ jobs:
             --shm-size 64G
             --env USE_ROCM="1"
             --device /dev/kfd
-            --device /dev/dri/$DEVICE
-            --volume $WORKSPACE:/workspace
+            --device /dev/dri/${{ env.DEVICE }}
+            --volume ${{ env.WORKSPACE }}:/workspace
             --workdir /workspace
           run: |
             pip install -e .[testing,diffusers,timm,peft,autoawq-rocm,auto-gptq-rocm]