From c457442dbe0d55fbed2d48547f5d98675e558915 Mon Sep 17 00:00:00 2001
From: Vibhu Jawa <vibhujawa@gmail.com>
Date: Tue, 16 Jul 2024 18:26:01 -0700
Subject: [PATCH] Test CI

Signed-off-by: Vibhu Jawa <vibhujawa@gmail.com>
---
 .github/workflows/gpu-ci.yml | 41 +++++++++++++++++++++---------------
 ci/test_gpu.sh               |  8 +++++++
 2 files changed, 32 insertions(+), 17 deletions(-)
 create mode 100644 ci/test_gpu.sh

diff --git a/.github/workflows/gpu-ci.yml b/.github/workflows/gpu-ci.yml
index 09363186..72cb9814 100644
--- a/.github/workflows/gpu-ci.yml
+++ b/.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/pr-builder.yaml@branch-24.06
+
+  checks:
+    secrets: inherit
+    uses: rapidsai/shared-workflows/.github/workflows/checks.yaml@branch-24.06
+    with:
+      enable_check_generated_files: false
+
+  python-gpu-tests:
+    needs:
+      - checks
+    secrets: inherit
+    uses: rapidsai/shared-workflows/.github/workflows/custom-job.yaml@branch-24.06
+    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
diff --git a/ci/test_gpu.sh b/ci/test_gpu.sh
new file mode 100644
index 00000000..0a860690
--- /dev/null
+++ b/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}