expose input compute api

cooperative-computing-lab · Dec 19, 2024 · ccbca01 · ccbca01
1 parent 03ca4f6
commit ccbca01
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/taskvine/src/manager/vine_manager.c b/taskvine/src/manager/vine_manager.c
@@ -2564,7 +2564,7 @@ static int build_poll_table(struct vine_manager *q)
 	return n;
 }
 
-static void vine_manager_compute_input_size(struct vine_manager *q, struct vine_task *t)
+void vine_manager_compute_input_size(struct vine_manager *q, struct vine_task *t)
 {
 	t->input_files_size = -1;
 

diff --git a/taskvine/src/manager/vine_manager.h b/taskvine/src/manager/vine_manager.h
@@ -272,6 +272,8 @@ struct rmsummary *vine_manager_choose_resources_for_task( struct vine_manager *q
 
 int64_t overcommitted_resource_total(struct vine_manager *q, int64_t total);
 
+void vine_manager_compute_input_size(struct vine_manager *q, struct vine_task *t);
+
 /* Internal: Shut down a specific worker. */
 int vine_manager_shut_down_worker(struct vine_manager *q, struct vine_worker_info *w);