[CI] Use same image for tests and publishing test execution (#463)

Use same image for tests and publishing test execution
hidet-org · Jul 27, 2024 · 4e33705 · 4e33705
1 parent 3620adc
commit 4e33705
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/.github/workflows/publish-pypi.yaml b/.github/workflows/publish-pypi.yaml
@@ -34,7 +34,7 @@ jobs:
       matrix: ${{ fromJSON(needs.list-test-dirs.outputs.matrix) }}
     runs-on: arc-runner-set
     container:
-      image: nvcr.io/nvidia/pytorch:24.04-py3
+      image: nvidia/cuda:11.8.0-devel-ubuntu20.04
     steps:
       - name: Install dependencies via apt
         run: |
@@ -62,13 +62,14 @@ jobs:
       # Run tests
       - name: Run tests
         run: |
+          rm -rf ~/.config/hidet
           python -m pytest -v --durations=20 --clear-cache ${{ matrix.path }}
 
   build-docs:
     if: github.repository == 'hidet-org/hidet'
     runs-on: arc-runner-set
     container:
-      image: nvcr.io/nvidia/pytorch:24.04-py3
+      image: nvidia/cuda:11.8.0-devel-ubuntu20.04
     steps:
       - name: Install dependencies via apt
         run: |

diff --git a/.github/workflows/tests.yaml b/.github/workflows/tests.yaml
@@ -78,7 +78,6 @@ jobs:
     runs-on: arc-runner-set
     container:
       image: nvidia/cuda:11.8.0-devel-ubuntu20.04
-      options: --gpus all
     steps:
       - name: Install dependencies via apt
         run: |