Add CheckPod tasks to k8s job and jobset tests

Signed-off-by: Yuan Chen <[email protected]>
NVIDIA · May 10, 2024 · 9ef26ee · 9ef26ee
1 parent c132709
commit 9ef26ee
Show file tree

Hide file tree

Showing 3 changed files with 24 additions and 0 deletions.
diff --git a/resources/tests/k8s/test-job.yml b/resources/tests/k8s/test-job.yml
@@ -22,3 +22,11 @@ tasks:
       cpu: 100m
       memory: 512M
       gpu: 8
+- id: status
+  type: CheckPod
+  params:
+    refTaskId: job
+    status: Completed
+    nodeLabels:
+      nodeType: gpu
+    timeout: 5s
diff --git a/resources/tests/k8s/test-jobset-with-driver.yml b/resources/tests/k8s/test-jobset-with-driver.yml
@@ -23,3 +23,11 @@ tasks:
       cpu: 100m
       memory: 512M
       gpu: 8
+- id: status
+  type: CheckPod
+  params:
+    refTaskId: jobset
+    status: Completed
+    nodeLabels:
+      nodeType: gpu
+    timeout: 5s
diff --git a/resources/tests/k8s/test-jobset.yml b/resources/tests/k8s/test-jobset.yml
@@ -23,3 +23,11 @@ tasks:
       cpu: 100m
       memory: 512M
       gpu: 8
+- id: status
+  type: CheckPod
+  params:
+    refTaskId: jobset
+    status: Completed
+    nodeLabels:
+      nodeType: gpu
+    timeout: 5s