diff --git a/test/gpu-consumer-job.yaml b/test/gpu-consumer-job.yaml new file mode 100644 index 0000000..845969f --- /dev/null +++ b/test/gpu-consumer-job.yaml @@ -0,0 +1,21 @@ +apiVersion: batch/v1 +kind: Job +metadata: + name: gpu-consumer-job +spec: + template: + metadata: + name: gpu-consumer-pod + labels: + susql.label/1: gpu + spec: + restartPolicy: Never + containers: + - name: gpu-consumer-container + image: "quay.io/trent_s/gpu-burn" + env: + - name: RUNSECONDS + value: "900" + resources: + limits: + nvidia.com/gpu: 1 diff --git a/test/labelgroups.yaml b/test/labelgroups.yaml index 55321a2..3728013 100644 --- a/test/labelgroups.yaml +++ b/test/labelgroups.yaml @@ -1,6 +1,15 @@ --- apiVersion: susql.ibm.com/v1 kind: LabelGroup +metadata: + name: gpu + namespace: default +spec: + labels: + - gpu +--- +apiVersion: susql.ibm.com/v1 +kind: LabelGroup metadata: name: only-model-1 namespace: default