Skip to content

Commit aed1957

Browse files
add tune option for task groups
1 parent 739d6dc commit aed1957

File tree

2 files changed

+12
-2
lines changed

2 files changed

+12
-2
lines changed

taskvine/src/manager/vine_manager.c

Lines changed: 9 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -3928,6 +3928,8 @@ struct vine_manager *vine_ssl_create(int port, const char *key, const char *cert
39283928
// peer transfers enabled by default
39293929
q->peer_transfers_enabled = 1;
39303930

3931+
q->task_groups_enabled = 0;
3932+
39313933
q->load_from_shared_fs_enabled = 0;
39323934

39333935
q->file_source_max_transfers = VINE_FILE_SOURCE_MAX_TRANSFERS;
@@ -4618,8 +4620,10 @@ int vine_submit(struct vine_manager *q, struct vine_task *t)
46184620
/* Ensure category structure is created. */
46194621
vine_category_lookup_or_create(q, t->category);
46204622

4621-
/* Attemp to group this task based on temp dependencies. */
4622-
vine_task_groups_assign_task(q, t);
4623+
/* Attempt to group this task based on temp dependencies. */
4624+
if(q->task_groups_enabled) {
4625+
vine_task_groups_assign_task(q, t);
4626+
}
46234627

46244628
change_task_state(q, t, VINE_TASK_READY);
46254629

@@ -5575,6 +5579,9 @@ int vine_tune(struct vine_manager *q, const char *name, double value)
55755579
} else if (!strcmp(name, "update-interval")) {
55765580
q->update_interval = MAX(1, (int)value);
55775581

5582+
} else if (!strcmp(name, "task-groups")) {
5583+
q->task_groups_enabled = MIN(1, (int)value);
5584+
55785585
} else if (!strcmp(name, "resource-management-interval")) {
55795586
q->resource_management_interval = MAX(1, (int)value);
55805587

taskvine/src/manager/vine_manager.h

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -177,6 +177,9 @@ struct vine_manager {
177177
int file_source_max_transfers;
178178
int worker_source_max_transfers;
179179

180+
/* Task Groups Configuration */
181+
int task_groups_enabled;
182+
180183
/* Various performance knobs that can be tuned. */
181184
int short_timeout; /* Timeout in seconds to send/recv a brief message from worker */
182185
int long_timeout; /* Timeout if in the middle of an incomplete message. */

0 commit comments

Comments
 (0)