From d9fc042199ec9eb6a1728ff0dbf1d00e8f0c5a06 Mon Sep 17 00:00:00 2001
From: Guillaume Dalle <22795598+gdalle@users.noreply.github.com>
Date: Thu, 7 Nov 2024 10:45:43 +0100
Subject: [PATCH] test: set up buildkite for GPU testing (#617)

* test: set up buildkite for GPU testing

* Don't run buildkite on drafts

* Remove diff

* Trigger CI

* Condition on "gpu" label

* Trigger CI
---
 .buildkite/pipeline.yml                          | 14 ++++++++++++++
 DifferentiationInterface/test/GPU/CUDA/simple.jl |  5 +++++
 2 files changed, 19 insertions(+)
 create mode 100644 .buildkite/pipeline.yml
 create mode 100644 DifferentiationInterface/test/GPU/CUDA/simple.jl

diff --git a/.buildkite/pipeline.yml b/.buildkite/pipeline.yml
new file mode 100644
index 000000000..68aa085ee
--- /dev/null
+++ b/.buildkite/pipeline.yml
@@ -0,0 +1,14 @@
+steps:
+  - label: "DI GPU tests"
+    if: |
+      !build.pull_request.draft &&
+        build.pull_request.labels includes "gpu"
+    plugins:
+      - JuliaCI/julia#v1:
+          version: "1"
+    command: |
+      julia ./DifferentiationInterface/test/GPU/CUDA/simple.jl
+    agents:
+      queue: "juliagpu"
+      cuda: "*"
+    timeout_in_minutes: 60
\ No newline at end of file
diff --git a/DifferentiationInterface/test/GPU/CUDA/simple.jl b/DifferentiationInterface/test/GPU/CUDA/simple.jl
new file mode 100644
index 000000000..efc38541f
--- /dev/null
+++ b/DifferentiationInterface/test/GPU/CUDA/simple.jl
@@ -0,0 +1,5 @@
+@info "Testing on CUDA"
+using Pkg
+Pkg.add("CUDA")
+using CUDA
+CUDA.versioninfo()