Skip to content

Commit

Permalink
a synthetic gpu energy consuming job based on gpu-burn
Browse files Browse the repository at this point in the history
Signed-off-by: Scott Trent <[email protected]>
  • Loading branch information
trent-s committed Jun 25, 2024
1 parent 1247829 commit 65b847b
Show file tree
Hide file tree
Showing 2 changed files with 30 additions and 0 deletions.
21 changes: 21 additions & 0 deletions test/gpu-consumer-job.yaml
Original file line number Diff line number Diff line change
@@ -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
9 changes: 9 additions & 0 deletions test/labelgroups.yaml
Original file line number Diff line number Diff line change
@@ -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
Expand Down

0 comments on commit 65b847b

Please sign in to comment.