venv changes

.github/workflows/test_e2eshark.yml

@@ -225,7 +225,7 @@ jobs:
           --report \
           --cachedir ${{ matrix.cache-dir }} \
           --mode turbine \
-          -g models \
+          --tests pytorch/models/bart-large \
           --targetbackend ${{ matrix.backend }} \
           --device ${{ matrix.device }} \
           --targetchip ${{ matrix.target-chip }} \
@@ -279,20 +279,14 @@ jobs:
           ref: 'main'
           token: ${{ secrets.E2ESHARK_GITHUB_TOKEN }}
           path: e2eshark-reports
-      - name: "Setup test suite venv"
-        run: |
-          python3.11 -m venv report_venv
-          source report_venv/bin/activate
-          pip install --upgrade pip
-          pip install -r ./test-suite/e2eshark/requirements.txt
       - name: "Setup alt test suite venv"
         run: |
           python3.11 -m venv report_venv_alt
-          source report_venv/bin/activate
+          source report_venv_alt/bin/activate
           pip install --upgrade pip
-          pip install -r ./alt_e2eshark/base_requirements.txt
-          pip install -r ./alt_e2eshark/iree_requirements.txt
-          pip install --no-deps -r ./alt_e2eshark/torch_mlir_requirements.txt
+          pip install -r ./test-suite/alt_e2eshark/base_requirements.txt
+          pip install -r ./test-suite/alt_e2eshark/iree_requirements.txt
+          pip install --no-deps -r ./test-suite/alt_e2eshark/torch_mlir_requirements.txt
       - uses: actions/download-artifact@master
         with:
           name: ci_reports_${{ matrix.backend }}_shark-test-suite_onnx.md

alt_e2eshark/e2e_testing/backends.py

@@ -43,7 +43,7 @@ def __init__(self, *, device="local-task", hal_target_backend="llvm-cpu", extra_
         elif hal_target_backend == "rocm":
             # some extra args for Mi300x - some of these may not work for other chips
             self.extra_args = [
-                "--iree-rocm-target-chip=gfx942",
+                "--iree-hip-target=gfx942",
                 # "--iree-global-opt-propagate-transposes=true",
                 # "--iree-opt-outer-dim-concat=true",
                 # "--iree-opt-const-eval=false",

alt_e2eshark/torch_mlir_requirements.txt

@@ -1,4 +1,4 @@
 --pre
 -f https://github.com/llvm/torch-mlir-release/releases/expanded_assets/dev-wheels
 #install nightly build of torch_mlir, if on Linux (no macOS or Windows nightly builds)
-torch-mlir ; sys_platform == "linux"
+torch-mlir-core ; sys_platform == "linux"