Comment out gpu model for easier testing and fix github/home path perms

amd-chrissosa · Aug 26, 2024 · 0e4ae4a · 0e4ae4a
1 parent 62a626d
commit 0e4ae4a
Showing 1 changed file with 40 additions and 37 deletions.
diff --git a/.github/workflows/test_models.yml b/.github/workflows/test_models.yml
@@ -29,6 +29,9 @@ jobs:
     runs-on:
       - arc-test-runners
     steps:
+      # Hack to fix running some of this stuff in a container, see: https://github.com/actions/setup-python/issues/816"
+      - name: "Change ownership of /github/home"
+        run: sudo chown -R $(whoami) /github/home
       #- name: "Setting up Python"
       #  uses: actions/setup-python@75f3110429a8c05be0e1bf360334e4cced2b63fa # v2.3.3
       #  with:
@@ -71,40 +74,40 @@ jobs:
           source turbine_venv/bin/activate
           pytest -v ${{ github.workspace }}/SHARK-Turbine/models/turbine_models/tests/sd_test.py
           pytest -v ${{ github.workspace }}/SHARK-Turbine/models/turbine_models/tests/sdxl_test.py --device cpu --rt_device local-task --iree_target_triple x86_64-linux-gnu
-  test-turbine-models-gpu:
-    runs-on: nodai-amdgpu-mi250-x86-64
-    steps:
-      - name: "Checkout This Repo"
-        uses: actions/checkout@v4
-      - name: "Checkout SHARK-Turbine repo"
-        uses: actions/checkout@v4
-        with:
-          repository: nod-ai/SHARK-Turbine
-          # TODO: Let the ref be passed as a parameter to run integration tests.
-          path: SHARK-Turbine
-      - name: Sync source deps
-        # build IREE from source with -DIREE_BUILD_TRACY=ON if getting tracy profile
-        run: |
-          python3 -m venv turbine_venv
-          source turbine_venv/bin/activate
-          python3 -m pip install --upgrade pip
-          # Note: We install in three steps in order to satisfy requirements
-          # from non default locations first. Installing the PyTorch CPU
-          # wheels saves multiple minutes and a lot of bandwidth on runner setup.
-          pip install --no-compile -r pytorch-cpu-requirements.txt
-          pip install --no-compile --pre --upgrade -r requirements.txt
-          pip install --no-compile --pre -e .[testing]
-          pip install --no-compile --pre --upgrade -e ${{ github.workspace }}/SHARK-Turbine/models -r ${{ github.workspace }}/SHARK-Turbine/models/requirements.txt
-          pip install \
-            --find-links https://iree.dev/pip-release-links.html \
-            --upgrade \
-            iree-compiler \
-            iree-runtime
-      - name: Show current free memory
-        run: |
-          free -mh
-      - name: Run sdxl gpu tests
-        run: |
-          source turbine_venv/bin/activate
-          pytest -v ${{ github.workspace }}/SHARK-Turbine/models/turbine_models/tests/sdxl_test.py --device vulkan --rt_device vulkan --iree_target_triple rdna3-unknown-linux
-          pytest -v ${{ github.workspace }}/SHARK-Turbine/models/turbine_models/tests/sdxl_test.py --device rocm --rt_device hip --iree_target_triple gfx90a --precision fp16
+#  test-turbine-models-gpu:
+#     runs-on: nodai-amdgpu-mi250-x86-64
+#     steps:
+#       - name: "Checkout This Repo"
+#         uses: actions/checkout@v4
+#       - name: "Checkout SHARK-Turbine repo"
+#         uses: actions/checkout@v4
+#         with:
+#           repository: nod-ai/SHARK-Turbine
+#           # TODO: Let the ref be passed as a parameter to run integration tests.
+#           path: SHARK-Turbine
+#       - name: Sync source deps
+#         # build IREE from source with -DIREE_BUILD_TRACY=ON if getting tracy profile
+#         run: |
+#           python3 -m venv turbine_venv
+#           source turbine_venv/bin/activate
+#           python3 -m pip install --upgrade pip
+#           # Note: We install in three steps in order to satisfy requirements
+#           # from non default locations first. Installing the PyTorch CPU
+#           # wheels saves multiple minutes and a lot of bandwidth on runner setup.
+#           pip install --no-compile -r pytorch-cpu-requirements.txt
+#           pip install --no-compile --pre --upgrade -r requirements.txt
+#           pip install --no-compile --pre -e .[testing]
+#           pip install --no-compile --pre --upgrade -e ${{ github.workspace }}/SHARK-Turbine/models -r ${{ github.workspace }}/SHARK-Turbine/models/requirements.txt
+#           pip install \
+#             --find-links https://iree.dev/pip-release-links.html \
+#             --upgrade \
+#             iree-compiler \
+#             iree-runtime
+#       - name: Show current free memory
+#         run: |
+#           free -mh
+#       - name: Run sdxl gpu tests
+#         run: |
+#           source turbine_venv/bin/activate
+#           pytest -v ${{ github.workspace }}/SHARK-Turbine/models/turbine_models/tests/sdxl_test.py --device vulkan --rt_device vulkan --iree_target_triple rdna3-unknown-linux
+#           pytest -v ${{ github.workspace }}/SHARK-Turbine/models/turbine_models/tests/sdxl_test.py --device rocm --rt_device hip --iree_target_triple gfx90a --precision fp16