apply patch to kornia#2118

johnnv1 · Jan 13, 2023 · bf8a02d · bf8a02d
1 parent 68bab88
commit bf8a02d
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 32 deletions.
diff --git a/.github/workflows/scheduled_test_cpu.yml b/.github/workflows/scheduled_test_cpu.yml
@@ -16,7 +16,7 @@ jobs:
       fail-fast: false
       matrix:
         os: ['Ubuntu-latest', 'Windows-latest', 'MacOS-latest']
-        pytorch-dtype: ['float16', 'float32', 'float64']
+        pytorch-dtype: ['float32', 'float64']
 
     uses: ./.github/workflows/tests.yml
     with:

diff --git a/.github/workflows/scheduled_test_cpu_half.yml b/.github/workflows/scheduled_test_cpu_half.yml
@@ -10,37 +10,14 @@ concurrency:
 
 jobs:
   tests-cpu:
-    name: ${{ matrix.os }} - python-${{ matrix.python-version }}, torch-${{ matrix.pytorch-version }}, ${{ matrix.pytorch-dtype }}
-    runs-on: ${{ matrix.os }}-latest
-
     strategy:
       fail-fast: false
       matrix:
-        os: ['Ubuntu', 'Windows', 'MacOS']
-        python-version: ['3.7', '3.8']
-        pytorch-version: ['1.9.1', '1.10.2', '1.12.1', '1.13.1']
-        pytorch-dtype: ['float16']
-
-    steps:
-      - name: Checkout kornia
-        uses: actions/checkout@v3
-
-      - name: Setting environment on ${{ matrix.os }} with python ${{ matrix.python-version }} and pytorch  ${{ matrix.pytorch-version }}
-        uses: ./.github/actions/env
-        with:
-          python-version: ${{ matrix.python-version }}
-          pytorch-version: ${{ matrix.pytorch-version }}
-
-      - name: Run CPU tests ${{ matrix.pytorch-dtype }}
-        shell: bash -l {0}
-        run: pytest -v --device cpu --dtype ${{ matrix.pytorch-dtype }} --cov=kornia --cov-report xml ./test
-
-
-      - if: always()
-        name: Upload coverage
-        uses: codecov/codecov-action@v3
-        with:
-          file: coverage.xml
-          token: ${{ secrets.CODECOV_TOKEN }} # to not depend on build via GH API
-          flags: cpu,${{ matrix.os }}_py-${{ matrix.python-version }}_pt-${{ matrix.pytorch-version }}_${{ matrix.pytorch-dtype }}
-          name: cpu-coverage
+        os: ['Ubuntu-latest', 'Windows-latest', 'MacOS-latest']
+
+    uses: ./.github/workflows/tests.yml
+    with:
+      os: ${{ matrix.os }}
+      python-version: '["3.7", "3.8", "3.9", "3.10"]'
+      pytorch-version: '["1.9.1", "1.10.2", "1.11.0", "1.12.1", "1.13.1"]'
+      pytorch-dtype: 'float16'