From 88b374080b764867c647b6bd3df974164583f1b5 Mon Sep 17 00:00:00 2001
From: Dmitry Shmulevich <dshmulevich@nvidia.com>
Date: Mon, 20 May 2024 16:09:50 -0700
Subject: [PATCH] add 'Configure' task to task configs

Signed-off-by: Dmitry Shmulevich <dshmulevich@nvidia.com>
---
 resources/tests/k8s/test-failed-job.yml         | 9 +++++++++
 resources/tests/k8s/test-jobset-with-driver.yml | 9 +++++++++
 resources/tests/k8s/test-jobset.yml             | 9 +++++++++
 resources/tests/kueue/test-job.yml              | 9 +++++++++
 resources/tests/volcano/test-job.yml            | 9 +++++++++
 5 files changed, 45 insertions(+)

diff --git a/resources/tests/k8s/test-failed-job.yml b/resources/tests/k8s/test-failed-job.yml
index 9a114cb..d0051af 100644
--- a/resources/tests/k8s/test-failed-job.yml
+++ b/resources/tests/k8s/test-failed-job.yml
@@ -8,6 +8,15 @@ tasks:
     nameFormat: "job{{._ENUM_}}"
     podNameFormat: "{{._NAME_}}-[0-9]-.*"
     podCount: "{{.parallelism}}"
+- id: configure
+  type: Configure
+  params:
+    nodes:
+    - type: dgxa100.80g
+      count: 2
+      labels:
+        nvidia.com/gpu.count: "8"
+    timeout: 1m
 - id: job
   type: SubmitObj
   params:
diff --git a/resources/tests/k8s/test-jobset-with-driver.yml b/resources/tests/k8s/test-jobset-with-driver.yml
index 89e5514..4125648 100644
--- a/resources/tests/k8s/test-jobset-with-driver.yml
+++ b/resources/tests/k8s/test-jobset-with-driver.yml
@@ -8,6 +8,15 @@ tasks:
     nameFormat: "jobset{{._ENUM_}}"
     podNameFormat: "{{._NAME_}}-(worker|driver)-[0-9]+-[0-9]+-.+"
     podCount: "{{.replicas}}"
+- id: configure
+  type: Configure
+  params:
+    nodes:
+    - type: dgxa100.80g
+      count: 4
+      labels:
+        nvidia.com/gpu.count: "8"
+    timeout: 1m
 - id: jobset
   type: SubmitObj
   params:
diff --git a/resources/tests/k8s/test-jobset.yml b/resources/tests/k8s/test-jobset.yml
index 564600f..d637e02 100644
--- a/resources/tests/k8s/test-jobset.yml
+++ b/resources/tests/k8s/test-jobset.yml
@@ -8,6 +8,15 @@ tasks:
     nameFormat: "jobset{{._ENUM_}}"
     podNameFormat: "{{._NAME_}}-workers-[0-9]+-[0-9]+-.+"
     podCount: "{{.replicas}}"
+- id: configure
+  type: Configure
+  params:
+    nodes:
+    - type: dgxa100.80g
+      count: 2
+      labels:
+        nvidia.com/gpu.count: "8"
+    timeout: 1m
 - id: jobset
   type: SubmitObj
   params:
diff --git a/resources/tests/kueue/test-job.yml b/resources/tests/kueue/test-job.yml
index c29316a..4b01902 100644
--- a/resources/tests/kueue/test-job.yml
+++ b/resources/tests/kueue/test-job.yml
@@ -8,6 +8,15 @@ tasks:
     nameFormat: "job{{._ENUM_}}"
     podNameFormat: "{{._NAME_}}-[0-9]-.*"
     podCount: "{{.parallelism}}"
+- id: configure
+  type: Configure
+  params:
+    nodes:
+    - type: dgxa100.80g
+      count: 4
+      labels:
+        nvidia.com/gpu.count: "8"
+    timeout: 1m
 - id: job
   type: SubmitObj
   params:
diff --git a/resources/tests/volcano/test-job.yml b/resources/tests/volcano/test-job.yml
index 4906313..00e4403 100644
--- a/resources/tests/volcano/test-job.yml
+++ b/resources/tests/volcano/test-job.yml
@@ -8,6 +8,15 @@ tasks:
     nameFormat: "j{{._ENUM_}}"
     podNameFormat: "{{._NAME_}}-test-[0-9]+"
     podCount: "{{.replicas}}"
+- id: configure
+  type: Configure
+  params:
+    nodes:
+    - type: dgxa100.80g
+      count: 4
+      labels:
+        nvidia.com/gpu.count: "8"
+    timeout: 1m
 - id: job
   type: SubmitObj
   params: