Add GPU path, fast path, parallelism and 1.6

FluxML · Dec 12, 2023 · 2eac7f4 · 2eac7f4
1 parent 7b67124
commit 2eac7f4
Show file tree

Hide file tree

Showing 7 changed files with 162 additions and 136 deletions.
diff --git a/.buildkite/pipeline.yml b/.buildkite/pipeline.yml
@@ -19,7 +19,9 @@ steps:
             Pkg.rm("ReTestItems") # not compatible with 1.6
           end'
     env:
-      GROUP: "CUDA" # TODO there are zero tests under this group
+      GROUP: "All"
+      TEST_FAST: true
+      TEST_WORKERS: 4
     if: build.message !~ /\[skip tests\]/
     timeout_in_minutes: 180
     matrix:

diff --git a/.github/workflows/CI.yml b/.github/workflows/CI.yml
@@ -26,11 +26,13 @@ jobs:
     runs-on: ${{ matrix.os }}
     env:
       GROUP: ${{ matrix.suite }}
+      TEST_FAST: ${{ matrix.version != '1' || matrix.os != 'ubuntu-latest' }}
     strategy:
       fail-fast: false
       matrix:
         version:
-          - '1' # Replace this with the minimum Julia version that your package supports.
+          - '1.6' # Replace this with the minimum Julia version that your package supports.
+          - '1'
           - 'nightly'
         os:
           - ubuntu-latest
@@ -61,13 +63,15 @@ jobs:
       - uses: julia-actions/cache@v1
       - uses: julia-actions/julia-buildpkg@v1
 
+      - name: Setup test env for 1.6
+        if: ${{ matrix.version == '1.6' }}
+        run: |
+          julia --color=yes --depwarn=yes --project=./test -e 'using Pkg; Pkg.rm("ReTestItems")'
       - name: Run tests
         uses: julia-actions/julia-runtest@v1
         continue-on-error: ${{ !(matrix.version == '1' && matrix.os == 'ubuntu-latest') && matrix.version == 'nightly' }}
         with:
           coverage: ${{ matrix.version == '1' && matrix.os == 'ubuntu-latest' }}
-        # run: |
-        #   julia --color=yes --depwarn=yes --project=./test -e 'include("test/retest.jl"); retest(${{ matrix.suite }})'
       - uses: actions/upload-artifact@v3
         with:
           name: coverage-${{ hashFiles('**/*.cov') }}