From 8543545ce6f82315fe77459b9a05d48d53eb1187 Mon Sep 17 00:00:00 2001
From: Swati Gupta <swatig@nvidia.com>
Date: Wed, 23 Apr 2025 21:22:44 +0000
Subject: [PATCH] Add more gpu examples

Signed-off-by: Swati Gupta <swatig@nvidia.com>
---
 demo/specs/quickstart/gpu-test7.yaml | 54 ++++++++++++++++++++++++++++
 demo/specs/quickstart/gpu-test8.yaml | 43 ++++++++++++++++++++++
 2 files changed, 97 insertions(+)
 create mode 100644 demo/specs/quickstart/gpu-test7.yaml
 create mode 100644 demo/specs/quickstart/gpu-test8.yaml

diff --git a/demo/specs/quickstart/gpu-test7.yaml b/demo/specs/quickstart/gpu-test7.yaml
new file mode 100644
index 000000000..e468c4f2a
--- /dev/null
+++ b/demo/specs/quickstart/gpu-test7.yaml
@@ -0,0 +1,54 @@
+# One pod, two containers
+# Each container asking for 1 distinct GPU
+
+---
+apiVersion: v1
+kind: Namespace
+metadata:
+  name: gpu-test7
+
+---
+apiVersion: resource.k8s.io/v1beta1
+kind: ResourceClaimTemplate
+metadata:
+  namespace: gpu-test7
+  name: single-gpu
+spec:
+  spec:
+    devices:
+      requests:
+      - name: gpu
+        deviceClassName: gpu.nvidia.com
+---
+apiVersion: v1
+kind: Pod
+metadata:
+  namespace: gpu-test7
+  name: gpu-pod
+  labels:
+    app: pod
+spec:
+  containers:
+  - name: ctr0
+    image: ubuntu:22.04
+    command: ["bash", "-c"]
+    args: ["nvidia-smi -L; trap 'exit 0' TERM; sleep 9999 & wait"]
+    resources:
+      claims:
+      - name: gpu0
+  - name: ctr1
+    image: ubuntu:22.04
+    command: ["bash", "-c"]
+    args: ["nvidia-smi -L; trap 'exit 0' TERM; sleep 9999 & wait"]
+    resources:
+      claims:
+      - name: gpu1
+  resourceClaims:
+  - name: gpu0
+    resourceClaimTemplateName: single-gpu
+  - name: gpu1
+    resourceClaimTemplateName: single-gpu
+  tolerations:
+  - key: "nvidia.com/gpu"
+    operator: "Exists"
+    effect: "NoSchedule"
diff --git a/demo/specs/quickstart/gpu-test8.yaml b/demo/specs/quickstart/gpu-test8.yaml
new file mode 100644
index 000000000..7c2b523f9
--- /dev/null
+++ b/demo/specs/quickstart/gpu-test8.yaml
@@ -0,0 +1,43 @@
+# One pod, one container asking 2 GPUs
+---
+apiVersion: v1
+kind: Namespace
+metadata:
+  name: gpu-test8
+---
+apiVersion: resource.k8s.io/v1beta1
+kind: ResourceClaimTemplate
+metadata:
+  namespace: gpu-test8
+  name: double-gpu
+spec:
+  spec:
+    devices:
+      requests:
+      - name: gpu
+        deviceClassName: gpu.nvidia.com
+        count: 2
+---
+apiVersion: v1
+kind: Pod
+metadata:
+  namespace: gpu-test8
+  name: gpu-pod
+  labels:
+    app: pod
+spec:
+  containers:
+  - name: ctr0
+    image: ubuntu:22.04
+    command: ["bash", "-c"]
+    args: ["nvidia-smi; trap 'exit 0' TERM; sleep 9999 & wait"]
+    resources:
+      claims:
+      - name: twogpus
+  resourceClaims:
+  - name: twogpus
+    resourceClaimTemplateName: double-gpu
+  tolerations:
+  - key: "nvidia.com/gpu"
+    operator: "Exists"
+    effect: "NoSchedule"