Test CI

Signed-off-by: Vibhu Jawa <[email protected]>

.github/workflows/gpu-ci.yml

@@ -10,23 +10,30 @@ on:
       - "v[0-9]+.[0-9]+.[0-9]+"
 
 jobs:
-  gpu-ci:
-    runs-on: linux-amd64-gpu-p100-latest-1
-    container:
-      image: nvcr.io/nvidia/rapidsai/base:24.06-cuda12.2-py3.11
-      options: >-
-        --shm-size=1G
-        --ulimit memlock=-1
-        --ulimit stack=67108864
-
-    steps:
-      - uses: actions/checkout@v3
-        with:
-          fetch-depth: 0
-      - name: Run tests
-        run: |
-          bash -c "echo 'Running tests'"
-
+  pr-builder:
+    needs:
+      - checks
+      - python-gpu-tests
+    secrets: inherit
+    uses: rapidsai/shared-workflows/.github/workflows/[email protected]
+
+  checks:
+    secrets: inherit
+    uses: rapidsai/shared-workflows/.github/workflows/[email protected]
+    with:
+      enable_check_generated_files: false
+
+  python-gpu-tests:
+    needs:
+      - checks
+    secrets: inherit
+    uses: rapidsai/shared-workflows/.github/workflows/[email protected]
+    with:
+      build_type: pull-request
+      node_type: "gpu-v100-latest-1"
+      arch: "amd64"
+      container_image: "nvcr.io/nvidia/rapidsai/base:24.06-cuda12.2-py3.11"
+      run_script: "ci/test_gpu.sh"
 
   # benchmark:
   #   runs-on: linux-amd64-gpu-p100-latest-1

ci/test_gpu.sh

@@ -0,0 +1,8 @@
+
+set -Eeuo pipefail
+
+echo "Generate conda env for crossfit tests"
+EXITCODE=0
+
+echo "Crossfit test script exiting with value: ${EXITCODE}"
+exit ${EXITCODE}