Update test-models.yml

add a100

.github/workflows/test-models.yml

@@ -71,24 +71,38 @@ jobs:
         restore-keys: |
           ${{ runner.os }}-pip-
     
-    - name: Install dependencies
+    - name: Validate Models dependencies
       run: |
         cd $GITHUB_WORKSPACE
         PYTHON=python3.10 IMPORTER=1 ./setup_venv.sh
-        source shark.venv/bin/activate
-        python -m pip install flake8 pytest yapf toml
-
-    - name: Lint with flake8
-      run: |
-        source shark.venv/bin/activate
-        # stop the build if there are Python syntax errors or undefined names
-        flake8 . --count --select=E9,F63,F7,F82 --show-source --statistics --exclude lit.cfg.py,shark.venv
-        # exit-zero treats all errors as warnings. The GitHub editor is 127 chars wide
-        flake8 . --count --exit-zero --max-complexity=10 --max-line-length=127 --statistics --exclude lit.cfg.py,shark.venv
-        yapf -i --style .style.yapf shark/*.py
-        
-    - name: Validate Models
-      run: |
         source shark.venv/bin/activate
         pytest -k 'not benchmark' --ignore=shark/tests/test_hf_benchmark.py --ignore=shark/tests/test_shark_importer.py
         
+  perf-linux:
+    runs-on: a100
+    strategy:
+      fail-fast: false
+      matrix:
+        python-version: ["3.10"]
+
+    steps:
+    - uses: actions/checkout@v3
+    - name: Set up Python ${{ matrix.python-version }}
+      uses: actions/setup-python@v3
+      with:
+        python-version: ${{ matrix.python-version }}
+    
+    - name: Setup pip cache
+      uses: actions/cache@v3
+      with:
+        path: ~/.cache/pip
+        key: ${{ runner.os }}-pip-${{ hashFiles('**/requirements.txt') }}
+        restore-keys: |
+          ${{ runner.os }}-pip-
+
+    - name: Validate Models
+      run: |
+        cd $GITHUB_WORKSPACE
+        IMPORTER=1 ./setup_venv.sh
+        source shark.venv/bin/activate
+        pytest --ignore=shark/tests/test_hf_benchmark.py --ignore=shark/tests/test_shark_importer.py