Merge pull request #366 from SciML/add-cpu-backend-testing

Add CPU backend testing support via KernelAbstractions

Author: ChrisRackauckas

.github/workflows/CI.yml

@@ -0,0 +1,53 @@
+name: CI
+on:
+  pull_request:
+    branches:
+      - master
+  push:
+    branches:
+      - master
+jobs:
+  test:
+    runs-on: ubuntu-latest
+    continue-on-error: ${{ matrix.group == 'Downstream' }}
+    strategy:
+      fail-fast: false
+      matrix:
+        group:
+          - CPU
+        version:
+          - 'lts'
+          - '1'
+          - 'pre'
+    steps:
+      - uses: actions/checkout@v4
+      - uses: julia-actions/setup-julia@v2
+        with:
+          version: ${{ matrix.version }}
+      - uses: actions/cache@v4
+        env:
+          cache-name: cache-artifacts
+        with:
+          path: ~/.julia/artifacts
+          key: ${{ runner.os }}-test-${{ env.cache-name }}-${{ hashFiles('**/Project.toml') }}
+          restore-keys: |
+            ${{ runner.os }}-test-${{ env.cache-name }}-
+            ${{ runner.os }}-test-
+            ${{ runner.os }}-
+      - uses: julia-actions/julia-runtest@v1
+        with:
+          coverage: false
+          check_bounds: auto
+        env:
+          GROUP: ${{ matrix.group }}
+      - uses: julia-actions/julia-processcoverage@v1
+      - uses: codecov/codecov-action@v5
+        with:
+          token: ${{ secrets.CODECOV_TOKEN }}
+          file: lcov.info
+          fail_ci_if_error: false
+        if: matrix.group == 'CPU'
+        with:
+          files: lcov.info
+          token: ${{ secrets.CODECOV_TOKEN }}
+          fail_ci_if_error: false

.github/workflows/Downgrade.yml

@@ -17,6 +17,7 @@ jobs:
       matrix:
         downgrade_mode: ['alldeps']
         julia-version: ['1.10']
+        group: ['CPU']
     steps:
       - uses: actions/checkout@v4
       - uses: julia-actions/setup-julia@v2
@@ -28,4 +29,6 @@ jobs:
       - uses: julia-actions/julia-buildpkg@v1
       - uses: julia-actions/julia-runtest@v1
         with:
-          ALLOW_RERESOLVE: false
+          ALLOW_RERESOLVE: false
+        env:
+          GROUP: ${{ matrix.group }}

Project.toml

@@ -10,6 +10,7 @@ DiffEqBase = "2b5f629d-d688-5b77-993f-72d75c75574e"
 Distributed = "8ba89e20-285c-5b6f-9357-94700520ee1b"
 DocStringExtensions = "ffbed154-4ef7-542d-bbb7-c09d3a79fcae"
 ForwardDiff = "f6369f11-7733-5829-9624-2563aa707210"
+GPUArraysCore = "46192b85-c4d5-4398-a991-12ede77f4527"
 KernelAbstractions = "63c18a36-062a-441e-b654-da1e3ab1ce7c"
 LinearAlgebra = "37e2e46d-f89d-539d-b4ee-838fcccc9c8e"
 LinearSolve = "7ed4a6bd-45f5-4d41-b270-4a48e9bafcae"
@@ -44,6 +45,7 @@ ChainRulesCore = "1"
 DiffEqBase = "6.122"
 DocStringExtensions = "0.8, 0.9"
 ForwardDiff = "0.10, 1"
+GPUArraysCore = "0.1, 0.2"
 KernelAbstractions = "0.9"
 LinearSolve = "1.15, 2, 3"
 Metal = "0.5, 1"
@@ -62,5 +64,7 @@ oneAPI = "1.2, 2"
 [extras]
 AMDGPU = "21141c5a-9bdb-4563-92ae-f87d6854732e"
 CUDA = "052768ef-5323-5732-b1bb-66c8b64840ba"
+GPUArraysCore = "46192b85-c4d5-4398-a991-12ede77f4527"
 Metal = "dde4c033-4e86-420c-a63e-0dd931031962"
 oneAPI = "8f75cd03-7ff8-4ecb-9b8f-daf728133b1b"
+Test = "8dfed614-e22c-5e08-85e1-65c5234f0b40"

test/utils.jl

@@ -11,6 +11,9 @@ elseif GROUP == "oneAPI"
 elseif GROUP == "Metal"
     using Metal
     Metal.MetalBackend()
+elseif GROUP == "CPU"
+    using KernelAbstractions
+    KernelAbstractions.CPU()
 end
 
 import GPUArraysCore