add new subtask with new requirements (#100)

megvii-research · Dec 14, 2022 · 12fa56e · 12fa56e
1 parent 3f7b2d3
commit 12fa56e
Show file tree

Hide file tree

Showing 6 changed files with 34 additions and 5 deletions.
diff --git a/.github/workflows/ci_test.yml b/.github/workflows/ci_test.yml
@@ -20,14 +20,41 @@ jobs:
       uses: actions/setup-python@v3
       with:
         python-version: ${{ matrix.python-version }}
+        cache: 'pip'
+        cache-dependency-path: |
+          requirements-ci.txt
     - name: Install dependencies
       run: |
-        python -m pip install --upgrade pip
-        if [ -f requirements.txt ]; then pip install -r requirements-ci.txt; fi
-        python -m pip install black pytest
+        pip install -r requirements-ci.txt
     - name: Lint with Black
       run: |
         black --check .
-    - name: Test with pytest
+    - name: Test regular cases
       run: |
-        cd ci/ && ln -s ../sparsebit && ln -s ../examples && pytest . --ignore=./sparsebit/ --ignore=./examples/
+        cd ci/regular_tests && ln -s ../../sparsebit && ln -s ../../examples && pytest . --ignore=./sparsebit/ --ignore=./examples/
+
+  extra_test:
+    needs: build
+    runs-on: ubuntu-latest
+    strategy:
+      fail-fast: false
+      matrix:
+        python-version: ["3.8"]
+
+    steps:
+    - uses: actions/checkout@v3
+    - name: Set up Python ${{ matrix.python-version }}
+      uses: actions/setup-python@v3
+      with:
+        python-version: ${{ matrix.python-version }}
+        cache: 'pip'
+        cache-dependency-path: |
+          requirements-ci.txt
+          examples/post_training_quantization/GLUE/CoLA/requirements.txt
+    - name: Install dependencies
+      run: |
+        pip install -r requirements-ci.txt
+        pip install -r examples/post_training_quantization/GLUE/CoLA/requirements.txt
+    - name: Test Huggingface cases
+      run: |
+        cd ci/huggingface_tests && ln -s ../../sparsebit && ln -s ../../examples && pytest . --ignore=./sparsebit/ --ignore=./examples/
diff --git a/ci/test_bert_emebddings.py → ci/huggingface_tests/test_bert_emebddings.py b/ci/test_bert_emebddings.py → ci/huggingface_tests/test_bert_emebddings.py
diff --git a/ci/test_MHSA.py → ci/regular_tests/test_MHSA.py b/ci/test_MHSA.py → ci/regular_tests/test_MHSA.py
diff --git a/ci/test_add_extra_info.py → ci/regular_tests/test_add_extra_info.py b/ci/test_add_extra_info.py → ci/regular_tests/test_add_extra_info.py
diff --git a/ci/test_qadd.py → ci/regular_tests/test_qadd.py b/ci/test_qadd.py → ci/regular_tests/test_qadd.py
diff --git a/requirements-ci.txt b/requirements-ci.txt
@@ -6,3 +6,5 @@ tabulate
 onnx>=1.8
 timm
 scipy
+black
+pytest