Run CI on aarch64 (#350)

* Run CI on aarch64 * use julia pipeline for aarch64-linux * fix var * exclude aarch64-linux jobs from github ci
EnzymeAD · Dec 13, 2024 · 9f96c09 · 9f96c09
1 parent b56e661
commit 9f96c09
Show file tree

Hide file tree

Showing 2 changed files with 76 additions and 32 deletions.
diff --git a/.buildkite/pipeline.yml b/.buildkite/pipeline.yml
@@ -1,37 +1,76 @@
 steps:
-  - label: "CUDA Julia v{{matrix.version}} -- {{matrix.group}}"
-    matrix:
-      setup:
-        version:
-          - "1.10"
-        group:
-          - core
-          - neural_networks
-          - integration
-    plugins:
-      - JuliaCI/julia#v1:
-          version: "{{matrix.version}}"
-      - JuliaCI/julia-coverage#v1:
-          codecov: true
-          dirs:
-            - src
-            - ext
-            - lib/ReactantCore/src
-    commands: |
-      julia --project=. -e 'println("--- :julia: Instantiating project")
-          using Pkg
-          Pkg.develop([PackageSpec(path="lib/ReactantCore")])'
+  - group: ":test_tube: Tests"
+    steps:
+      - label: "CUDA Julia v{{matrix.version}} -- {{matrix.group}}"
+        matrix:
+          setup:
+            version:
+              - "1.10"
+            group:
+              - core
+              - neural_networks
+              - integration
+        plugins:
+          - JuliaCI/julia#v1:
+              version: "{{matrix.version}}"
+          - JuliaCI/julia-coverage#v1:
+              codecov: true
+              dirs:
+                - src
+                - ext
+                - lib/ReactantCore/src
+        commands: |
+          julia --project=. -e 'println("--- :julia: Instantiating project")
+              using Pkg
+              Pkg.develop([PackageSpec(path="lib/ReactantCore")])'
+
+          julia --project=. -e 'println("--- :julia: Run Tests")
+              using Pkg
+              Pkg.test(; coverage="user")'
+        agents:
+          queue: "juliagpu"
+          cuda: "*"
+        env:
+          REACTANT_TEST_GROUP: "{{matrix.group}}"
+        if: build.message !~ /\[skip tests\]/
+        timeout_in_minutes: 60
+
+      - label: ":julia: :linux: aarch64 - Julia v{{matrix.version}} -- {{matrix.group}}"
+        matrix:
+          setup:
+            version:
+              - "1.10"
+              - "1.11"
+            group:
+              - core
+              - neural_networks
+              - integration
+        plugins:
+          - JuliaCI/julia#v1:
+              version: "{{matrix.version}}"
+          - JuliaCI/julia-coverage#v1:
+              codecov: true
+              dirs:
+                - src
+                - ext
+                - lib/ReactantCore/src
+        commands: |
+          julia --project=. -e 'println("--- :julia: Instantiating project")
+              using Pkg
+              Pkg.develop([PackageSpec(path="lib/ReactantCore")])'
 
-      julia --project=. -e 'println("--- :julia: Run Tests")
-          using Pkg
-          Pkg.test(; coverage="user")'
-    agents:
-      queue: "juliagpu"
-      cuda: "*"
-    env:
-      REACTANT_TEST_GROUP: "{{matrix.group}}"
-    if: build.message !~ /\[skip tests\]/
-    timeout_in_minutes: 60
+          julia --project=. -e 'println("--- :julia: Run Tests")
+              using Pkg
+              Pkg.test(; coverage="user")'
+        agents:
+          queue: "juliaecosystem"
+          os: "linux"
+          sandbox_capable: "true"
+          arch: "aarch64"
+        env:
+          REACTANT_TEST_GROUP: "{{matrix.group}}"
+        if: build.message !~ /\[skip tests\]/
+        timeout_in_minutes: 60
 
   - group: ":racehorse: Benchmarks"
     steps:

diff --git a/.github/workflows/CI.yml b/.github/workflows/CI.yml
@@ -33,6 +33,7 @@ jobs:
           - integration
         arch:
           - x64
+          - aarch64
         assertions:
           - false
         libReactant: [packaged]
@@ -64,6 +65,10 @@ jobs:
             libReactant: packaged
             version: '1.10'
             test_group: integration
+        exclude:
+          # these are run on Buildkite
+          - os: ubuntu-20.04
+            arch: aarch64
     steps:
       - uses: actions/checkout@v4
       - uses: julia-actions/setup-julia@v2