set resources in factories (cooperative-computing-lab#3924)

colinthomas-z80 · Oct 15, 2024 · 29d70ca · 29d70ca
1 parent 1555a40
commit 29d70ca
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/batch_job/src/vine_factory.c b/batch_job/src/vine_factory.c
@@ -494,6 +494,10 @@ static int submit_worker( struct batch_queue *queue )
 
 	batch_job_set_command(task,cmd);
 	batch_job_add_input_file(task,worker_command,0);
+
+	if(resources) {
+		batch_job_set_resources(task, resources);
+	}
 
 	if(password_file) {
 		batch_job_add_input_file(task,"pwfile",0);

diff --git a/batch_job/src/work_queue_factory.c b/batch_job/src/work_queue_factory.c
@@ -492,6 +492,10 @@ static int submit_worker( struct batch_queue *queue )
 
 	batch_job_set_command(task,cmd);
 
+	if(resources) {
+		batch_job_set_resources(task, resources);
+	}
+
 	if(!runos_os && !k8s_worker_image) {
 		batch_job_add_input_file(task,worker_command,0);
 	} else {