ci: fix

Author: avik-pal

.buildkite/pipeline.yml

@@ -1,15 +1,11 @@
 steps:
   - group: ":test_tube: Tests"
     steps:
-      - label: ":julia: :linux: CUDA Julia v{{matrix.version}} -- {{matrix.group}} -- {{matrix.runtime}}"
+      - label: ":julia: :linux: CUDA Julia v{{matrix.version}} -- {{matrix.runtime}}"
         matrix:
           setup:
             version:
               - "1.10"
-            group:
-              - core
-              - neural_networks
-              - integration
             runtime:
               - "PJRT"
               - "IFRT"
@@ -41,7 +37,6 @@ steps:
           queue: "juliagpu"
           cuda: "*"
         env:
-          REACTANT_TEST_GROUP: "{{matrix.group}}"
           JULIA_DEBUG: "Reactant,Reactant_jll"
           CUDA_VISIBLE_DEVICES: 0
           REACTANT_BACKEND_GROUP: "GPU"

.github/workflows/CI.yml

@@ -83,3 +83,35 @@ jobs:
       os: "ubuntu-24.04"
       runtime: ${{ matrix.runtime }}
       downgrade_testing: true
+
+  integration:
+    strategy:
+      fail-fast: false
+      matrix:
+        project:
+          - FancyArrayTypes
+          - FFT
+          - Flux
+          - Lux
+          - NNlib
+          - Python
+        version:
+          - "1.10"
+          - "1.11"
+        os:
+          - ubuntu-latest
+        # TODO: enable these tests
+        #   - linux-x86-ct6e-180-4tpu
+        include:
+          - project: CUDA
+            os: linux-x86-a2-48-a100-4gpu
+            version: "1.10"
+          - project: CUDA
+            os: linux-x86-a2-48-a100-4gpu
+            version: "1.11"
+    uses: ./.github/workflows/CommonCI.yml
+    with:
+      julia_version: ${{ matrix.version }}
+      os: ${{ matrix.os }}
+      runtime: "both"
+      project: "test/integration/${{ matrix.project }}"

.github/workflows/CommonCI.yml

@@ -187,7 +187,7 @@ jobs:
       - name: "Run Integration Tests (PJRT)"
         if: ${{ (inputs.runtime == 'pjrt' || inputs.runtime == 'both') && inputs.project != '.' }}
         timeout-minutes: 30
-        run: julia --color=yes --project=test/integration/${{ inputs.project }} --threads=auto --check-bounds=yes -O1 test/integration/${{ inputs.project }}/runtests.jl
+        run: julia --color=yes --project=${{ inputs.project }} --threads=auto --check-bounds=yes -O1 ${{ inputs.project }}/runtests.jl
         shell: bash
         env:
           JULIA_PKG_SERVER_REGISTRY_PREFERENCE: eager
@@ -223,7 +223,7 @@ jobs:
       - name: "Run Integration Tests (IFRT)"
         if: ${{ (inputs.runtime == 'ifrt' || inputs.runtime == 'both') && inputs.project != '.' }}
         timeout-minutes: 30
-        run: julia --color=yes --project=test/integration/${{ inputs.project }} --threads=auto --check-bounds=yes -O1 test/integration/${{ inputs.project }}/runtests.jl
+        run: julia --color=yes --project=${{ inputs.project }} --threads=auto --check-bounds=yes -O1 ${{ inputs.project }}/runtests.jl
         shell: bash
         env:
           JULIA_PKG_SERVER_REGISTRY_PREFERENCE: eager