1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318 |
- /* StarPU --- Runtime system for heterogeneous multicore architectures.
- *
- * Copyright (C) 2011, 2012 INRIA
- *
- * StarPU is free software; you can redistribute it and/or modify
- * it under the terms of the GNU Lesser General Public License as published by
- * the Free Software Foundation; either version 2.1 of the License, or (at
- * your option) any later version.
- *
- * StarPU is distributed in the hope that it will be useful, but
- * WITHOUT ANY WARRANTY; without even the implied warranty of
- * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
- *
- * See the GNU Lesser General Public License in COPYING.LGPL for more details.
- */
- #include <sc_hypervisor_intern.h>
- #include <sc_hypervisor_policy.h>
- #include <starpu_config.h>
- unsigned imposed_resize = 0;
- unsigned type_of_tasks_known = 0;
- struct starpu_sched_ctx_performance_counters* perf_counters = NULL;
- static void notify_idle_cycle(unsigned sched_ctx, int worker, double idle_time);
- static void notify_pushed_task(unsigned sched_ctx, int worker);
- static void notify_post_exec_task(struct starpu_task *task, size_t data_size, uint32_t footprint,
- int hypervisor_tag, double flops);
- static void notify_poped_task(unsigned sched_ctx, int worker);
- static void notify_submitted_job(struct starpu_task *task, unsigned footprint, size_t data_size);
- static void notify_empty_ctx(unsigned sched_ctx, struct starpu_task *task);
- static void notify_delete_context(unsigned sched_ctx);
- extern struct sc_hypervisor_policy idle_policy;
- extern struct sc_hypervisor_policy app_driven_policy;
- extern struct sc_hypervisor_policy gflops_rate_policy;
- #ifdef STARPU_HAVE_GLPK_H
- extern struct sc_hypervisor_policy feft_lp_policy;
- extern struct sc_hypervisor_policy teft_lp_policy;
- extern struct sc_hypervisor_policy ispeed_lp_policy;
- extern struct sc_hypervisor_policy throughput_lp_policy;
- #endif // STARPU_HAVE_GLPK_
- extern struct sc_hypervisor_policy ispeed_policy;
- static struct sc_hypervisor_policy *predefined_policies[] =
- {
- &idle_policy,
- &app_driven_policy,
- #ifdef STARPU_HAVE_GLPK_H
- &feft_lp_policy,
- &teft_lp_policy,
- &ispeed_lp_policy,
- &throughput_lp_policy,
- #endif // STARPU_HAVE_GLPK_H
- &gflops_rate_policy,
- &ispeed_policy
- };
- static void _load_hypervisor_policy(struct sc_hypervisor_policy *policy)
- {
- STARPU_ASSERT(policy);
- hypervisor.policy.name = policy->name;
- hypervisor.policy.size_ctxs = policy->size_ctxs;
- hypervisor.policy.resize_ctxs = policy->resize_ctxs;
- hypervisor.policy.handle_poped_task = policy->handle_poped_task;
- hypervisor.policy.handle_pushed_task = policy->handle_pushed_task;
- hypervisor.policy.handle_idle_cycle = policy->handle_idle_cycle;
- hypervisor.policy.handle_idle_end = policy->handle_idle_end;
- hypervisor.policy.handle_post_exec_hook = policy->handle_post_exec_hook;
- hypervisor.policy.handle_submitted_job = policy->handle_submitted_job;
- hypervisor.policy.end_ctx = policy->end_ctx;
- }
- static struct sc_hypervisor_policy *_find_hypervisor_policy_from_name(const char *policy_name)
- {
- if (!policy_name)
- return NULL;
- unsigned i;
- for (i = 0; i < sizeof(predefined_policies)/sizeof(predefined_policies[0]); i++)
- {
- struct sc_hypervisor_policy *p;
- p = predefined_policies[i];
- if (p->name)
- {
- if (strcmp(policy_name, p->name) == 0) {
- /* we found a policy with the requested name */
- return p;
- }
- }
- }
- fprintf(stderr, "Warning: hypervisor policy \"%s\" was not found, try \"help\" to get a list\n", policy_name);
- /* nothing was found */
- return NULL;
- }
- static void display_sched_help_message(void)
- {
- const char* policy_name = getenv("SC_HYPERVISOR_POLICY");
- if (policy_name && (strcmp(policy_name, "help") == 0))
- {
- fprintf(stderr, "SC_HYPERVISOR_POLICY can be either of\n");
- /* display the description of all predefined policies */
- unsigned i;
- for (i = 0; i < sizeof(predefined_policies)/sizeof(predefined_policies[0]); i++)
- {
- struct sc_hypervisor_policy *p = predefined_policies[i];
- if (p->name)
- {
- fprintf(stderr, "%s\n", p->name);
- }
- }
- }
- }
- static struct sc_hypervisor_policy *_select_hypervisor_policy(struct sc_hypervisor_policy* hypervisor_policy)
- {
- struct sc_hypervisor_policy *selected_policy = NULL;
- if(hypervisor_policy && hypervisor_policy->custom)
- return hypervisor_policy;
- /* we look if the application specified the name of a policy to load */
- const char *policy_name;
- if (hypervisor_policy && hypervisor_policy->name)
- {
- policy_name = hypervisor_policy->name;
- }
- else
- {
- policy_name = getenv("SC_HYPERVISOR_POLICY");
- }
- if (policy_name)
- selected_policy = _find_hypervisor_policy_from_name(policy_name);
- /* Perhaps there was no policy that matched the name */
- if (selected_policy)
- return selected_policy;
- /* If no policy was specified, we use the idle policy as a default */
- return &idle_policy;
- }
- /* initializez the performance counters that starpu will use to retrive hints for resizing */
- void* sc_hypervisor_init(struct sc_hypervisor_policy *hypervisor_policy)
- {
- /* Perhaps we have to display some help */
- display_sched_help_message();
- hypervisor.min_tasks = 0;
- hypervisor.nsched_ctxs = 0;
- char* vel_gap = getenv("SC_HYPERVISOR_MAX_SPEED_GAP");
- hypervisor.max_speed_gap = vel_gap ? atof(vel_gap) : SC_SPEED_MAX_GAP_DEFAULT;
- char* crit = getenv("SC_HYPERVISOR_TRIGGER_RESIZE");
- hypervisor.resize_criteria = !crit ? SC_IDLE : strcmp(crit,"idle") == 0 ? SC_IDLE : (strcmp(crit,"speed") == 0 ? SC_SPEED : SC_NOTHING);
- starpu_pthread_mutex_init(&act_hypervisor_mutex, NULL);
- hypervisor.start_executing_time = starpu_timing_now();
- int i;
- for(i = 0; i < STARPU_NMAX_SCHED_CTXS; i++)
- {
- hypervisor.resize[i] = 0;
- hypervisor.allow_remove[i] = 1;
- hypervisor.configurations[i] = NULL;
- hypervisor.sr = NULL;
- hypervisor.check_min_tasks[i] = 1;
- hypervisor.sched_ctxs[i] = STARPU_NMAX_SCHED_CTXS;
- hypervisor.sched_ctx_w[i].sched_ctx = STARPU_NMAX_SCHED_CTXS;
- hypervisor.sched_ctx_w[i].config = NULL;
- hypervisor.sched_ctx_w[i].total_flops = 0.0;
- hypervisor.sched_ctx_w[i].submitted_flops = 0.0;
- hypervisor.sched_ctx_w[i].remaining_flops = 0.0;
- hypervisor.sched_ctx_w[i].start_time = 0.0;
- hypervisor.sched_ctx_w[i].real_start_time = 0.0;
- hypervisor.sched_ctx_w[i].hyp_react_start_time = 0.0;
- hypervisor.sched_ctx_w[i].resize_ack.receiver_sched_ctx = -1;
- hypervisor.sched_ctx_w[i].resize_ack.moved_workers = NULL;
- hypervisor.sched_ctx_w[i].resize_ack.nmoved_workers = 0;
- hypervisor.sched_ctx_w[i].resize_ack.acked_workers = NULL;
- starpu_pthread_mutex_init(&hypervisor.sched_ctx_w[i].mutex, NULL);
- hypervisor.optimal_v[i] = 0.0;
- hypervisor.sched_ctx_w[i].ref_speed[0] = -1.0;
- hypervisor.sched_ctx_w[i].ref_speed[1] = -1.0;
- hypervisor.sched_ctx_w[i].total_flops_available = 0;
- hypervisor.sched_ctx_w[i].to_be_sized = 0;
- int j;
- for(j = 0; j < STARPU_NMAXWORKERS; j++)
- {
- hypervisor.sched_ctx_w[i].start_time_w[i] = 0.0;
- hypervisor.sched_ctx_w[i].current_idle_time[j] = 0.0;
- hypervisor.sched_ctx_w[i].idle_time[j] = 0.0;
- hypervisor.sched_ctx_w[i].idle_start_time[j] = 0.0;
- hypervisor.sched_ctx_w[i].exec_time[j] = 0.0;
- hypervisor.sched_ctx_w[i].exec_start_time[j] = 0.0;
- hypervisor.sched_ctx_w[i].pushed_tasks[j] = 0;
- hypervisor.sched_ctx_w[i].poped_tasks[j] = 0;
- hypervisor.sched_ctx_w[i].elapsed_flops[j] = 0.0;
- hypervisor.sched_ctx_w[i].elapsed_data[j] = 0;
- hypervisor.sched_ctx_w[i].elapsed_tasks[j] = 0;
- hypervisor.sched_ctx_w[i].total_elapsed_flops[j] = 0.0;
- hypervisor.sched_ctx_w[i].worker_to_be_removed[j] = 0;
- hypervisor.sched_ctx_w[i].compute_idle[j] = 1;
- hypervisor.sched_ctx_w[i].compute_partial_idle[j] = 0;
- }
- }
- struct sc_hypervisor_policy *selected_hypervisor_policy = _select_hypervisor_policy(hypervisor_policy);
- _load_hypervisor_policy(selected_hypervisor_policy);
- perf_counters = (struct starpu_sched_ctx_performance_counters*)malloc(sizeof(struct starpu_sched_ctx_performance_counters));
- perf_counters->notify_idle_cycle = notify_idle_cycle;
- perf_counters->notify_pushed_task = notify_pushed_task;
- perf_counters->notify_poped_task = notify_poped_task;
- perf_counters->notify_post_exec_task = notify_post_exec_task;
- perf_counters->notify_submitted_job = notify_submitted_job;
- perf_counters->notify_empty_ctx = notify_empty_ctx;
- perf_counters->notify_delete_context = notify_delete_context;
- starpu_sched_ctx_notify_hypervisor_exists();
- return (void*)perf_counters;
- }
- const char* sc_hypervisor_get_policy()
- {
- return hypervisor.policy.name;
- }
- /* the user can forbid the resizing process*/
- void sc_hypervisor_stop_resize(unsigned sched_ctx)
- {
- imposed_resize = 1;
- hypervisor.resize[sched_ctx] = 0;
- }
- /* the user can restart the resizing process*/
- void sc_hypervisor_start_resize(unsigned sched_ctx)
- {
- imposed_resize = 1;
- hypervisor.resize[sched_ctx] = 1;
- }
- static void _print_current_time()
- {
- if(!getenv("SC_HYPERVISOR_STOP_PRINT"))
- {
- double curr_time = starpu_timing_now();
- double elapsed_time = (curr_time - hypervisor.start_executing_time) / 1000000.0; /* in seconds */
- fprintf(stdout, "Time: %lf\n", elapsed_time);
- int i;
- for(i = 0; i < STARPU_NMAX_SCHED_CTXS; i++)
- {
- if(hypervisor.sched_ctxs[i] != STARPU_NMAX_SCHED_CTXS)
- {
- struct sc_hypervisor_wrapper *sc_w = &hypervisor.sched_ctx_w[hypervisor.sched_ctxs[i]];
-
- double cpu_speed = sc_hypervisor_get_speed(sc_w, STARPU_CPU_WORKER);
- double cuda_speed = sc_hypervisor_get_speed(sc_w, STARPU_CUDA_WORKER);
- int ncpus = sc_hypervisor_get_nworkers_ctx(sc_w->sched_ctx, STARPU_CPU_WORKER);
- int ncuda = sc_hypervisor_get_nworkers_ctx(sc_w->sched_ctx, STARPU_CUDA_WORKER);
- fprintf(stdout, "%d: cpu_v = %lf cuda_v = %lf ncpus = %d ncuda = %d\n", hypervisor.sched_ctxs[i], cpu_speed, cuda_speed, ncpus, ncuda);
- }
- }
- }
- return;
- }
- void sc_hypervisor_shutdown(void)
- {
- int i;
- for(i = 0; i < STARPU_NMAX_SCHED_CTXS; i++)
- {
- if(hypervisor.sched_ctxs[i] != STARPU_NMAX_SCHED_CTXS && hypervisor.nsched_ctxs > 0)
- {
- sc_hypervisor_stop_resize(hypervisor.sched_ctxs[i]);
- sc_hypervisor_unregister_ctx(hypervisor.sched_ctxs[i]);
- starpu_pthread_mutex_destroy(&hypervisor.sched_ctx_w[i].mutex);
- }
- }
- perf_counters->notify_idle_cycle = NULL;
- perf_counters->notify_pushed_task = NULL;
- perf_counters->notify_poped_task = NULL;
- perf_counters->notify_post_exec_task = NULL;
- perf_counters->notify_delete_context = NULL;
- free(perf_counters);
- perf_counters = NULL;
- starpu_pthread_mutex_destroy(&act_hypervisor_mutex);
- }
- /* the hypervisor is in charge only of the contexts registered to it*/
- void sc_hypervisor_register_ctx(unsigned sched_ctx, double total_flops)
- {
- starpu_pthread_mutex_lock(&act_hypervisor_mutex);
- hypervisor.configurations[sched_ctx] = NULL;
- hypervisor.resize_requests[sched_ctx] = NULL;
- starpu_pthread_mutex_init(&hypervisor.conf_mut[sched_ctx], NULL);
- starpu_pthread_mutex_init(&hypervisor.resize_mut[sched_ctx], NULL);
- _add_config(sched_ctx);
- hypervisor.sched_ctx_w[sched_ctx].sched_ctx = sched_ctx;
- hypervisor.sched_ctxs[hypervisor.nsched_ctxs++] = sched_ctx;
- hypervisor.sched_ctx_w[sched_ctx].total_flops = total_flops;
- hypervisor.sched_ctx_w[sched_ctx].remaining_flops = total_flops;
- hypervisor.resize[sched_ctx] = 1;
- starpu_pthread_mutex_unlock(&act_hypervisor_mutex);
- }
- static int _get_first_free_sched_ctx(unsigned *sched_ctxs, int nsched_ctxs)
- {
- int i;
- for(i = 0; i < nsched_ctxs; i++)
- if(sched_ctxs[i] == STARPU_NMAX_SCHED_CTXS)
- return i;
- return STARPU_NMAX_SCHED_CTXS;
- }
- /* rearange array of sched_ctxs in order not to have {MAXVAL, MAXVAL, 5, MAXVAL, 7}
- and have instead {5, 7, MAXVAL, MAXVAL, MAXVAL}
- it is easier afterwards to iterate the array
- */
- static void _rearange_sched_ctxs(unsigned *sched_ctxs, int old_nsched_ctxs)
- {
- int first_free_id = STARPU_NMAX_SCHED_CTXS;
- int i;
- for(i = 0; i < old_nsched_ctxs; i++)
- {
- if(sched_ctxs[i] != STARPU_NMAX_SCHED_CTXS)
- {
- first_free_id = _get_first_free_sched_ctx(sched_ctxs, old_nsched_ctxs);
- if(first_free_id != STARPU_NMAX_SCHED_CTXS)
- {
- sched_ctxs[first_free_id] = sched_ctxs[i];
- sched_ctxs[i] = STARPU_NMAX_SCHED_CTXS;
- }
- }
- }
- }
- /* unregistered contexts will no longer be resized */
- void sc_hypervisor_unregister_ctx(unsigned sched_ctx)
- {
- if(hypervisor.policy.end_ctx)
- hypervisor.policy.end_ctx(sched_ctx);
- starpu_pthread_mutex_lock(&act_hypervisor_mutex);
- unsigned i;
- for(i = 0; i < hypervisor.nsched_ctxs; i++)
- {
- if(hypervisor.sched_ctxs[i] == sched_ctx)
- {
- hypervisor.sched_ctxs[i] = STARPU_NMAX_SCHED_CTXS;
- break;
- }
- }
- _rearange_sched_ctxs(hypervisor.sched_ctxs, hypervisor.nsched_ctxs);
- hypervisor.nsched_ctxs--;
- hypervisor.sched_ctx_w[sched_ctx].sched_ctx = STARPU_NMAX_SCHED_CTXS;
- _remove_config(sched_ctx);
- starpu_pthread_mutex_destroy(&hypervisor.conf_mut[sched_ctx]);
- starpu_pthread_mutex_destroy(&hypervisor.resize_mut[sched_ctx]);
- if(hypervisor.nsched_ctxs == 1)
- sc_hypervisor_stop_resize(hypervisor.sched_ctxs[0]);
- starpu_pthread_mutex_unlock(&act_hypervisor_mutex);
- }
- double _get_max_speed_gap()
- {
- return hypervisor.max_speed_gap;
- }
- unsigned sc_hypervisor_get_resize_criteria()
- {
- return hypervisor.resize_criteria;
- }
- static int get_ntasks( int *tasks)
- {
- int ntasks = 0;
- int j;
- for(j = 0; j < STARPU_NMAXWORKERS; j++)
- {
- ntasks += tasks[j];
- }
- return ntasks;
- }
- int sc_hypervisor_get_nworkers_ctx(unsigned sched_ctx, enum starpu_worker_archtype arch)
- {
- int nworkers_ctx = 0;
- struct starpu_worker_collection *workers = starpu_sched_ctx_get_worker_collection(sched_ctx);
- int worker;
- struct starpu_sched_ctx_iterator it;
- if(workers->init_iterator)
- workers->init_iterator(workers, &it);
- while(workers->has_next(workers, &it))
- {
- worker = workers->get_next(workers, &it);
- enum starpu_worker_archtype curr_arch = starpu_worker_get_type(worker);
- if(curr_arch == arch || arch == STARPU_ANY_WORKER)
- nworkers_ctx++;
- }
- return nworkers_ctx;
- }
- static void _set_elapsed_flops_per_sched_ctx(unsigned sched_ctx, double val)
- {
- int i;
- for(i = 0; i < STARPU_NMAXWORKERS; i++)
- {
- hypervisor.sched_ctx_w[sched_ctx].elapsed_flops[i] = val;
- if(val == 0)
- {
- hypervisor.sched_ctx_w[sched_ctx].elapsed_data[i] = 0;
- hypervisor.sched_ctx_w[sched_ctx].elapsed_tasks[i] = 0;
- }
- }
- }
- double sc_hypervisor_get_elapsed_flops_per_sched_ctx(struct sc_hypervisor_wrapper* sc_w)
- {
- double ret_val = 0.0;
- struct starpu_worker_collection *workers = starpu_sched_ctx_get_worker_collection(sc_w->sched_ctx);
- int worker;
-
- struct starpu_sched_ctx_iterator it;
- if(workers->init_iterator)
- workers->init_iterator(workers, &it);
-
- while(workers->has_next(workers, &it))
- {
- worker = workers->get_next(workers, &it);
- ret_val += sc_w->elapsed_flops[worker];
- }
- return ret_val;
- }
- double sc_hypervisor_get_total_elapsed_flops_per_sched_ctx(struct sc_hypervisor_wrapper* sc_w)
- {
- double ret_val = 0.0;
- struct starpu_worker_collection *workers = starpu_sched_ctx_get_worker_collection(sc_w->sched_ctx);
- int worker;
-
- struct starpu_sched_ctx_iterator it;
- if(workers->init_iterator)
- workers->init_iterator(workers, &it);
-
- while(workers->has_next(workers, &it))
- {
- worker = workers->get_next(workers, &it);
- ret_val += sc_w->total_elapsed_flops[worker];
- }
- return ret_val;
- }
- void _reset_resize_sample_info(unsigned sender_sched_ctx, unsigned receiver_sched_ctx)
- {
- double start_time = starpu_timing_now();
- if(sender_sched_ctx != STARPU_NMAX_SCHED_CTXS)
- {
- /* info concerning only the gflops_rate strateg */
- struct sc_hypervisor_wrapper *sender_sc_w = &hypervisor.sched_ctx_w[sender_sched_ctx];
-
- sender_sc_w->start_time = start_time;
- _set_elapsed_flops_per_sched_ctx(sender_sched_ctx, 0.0);
- int i;
- for(i = 0; i < STARPU_NMAXWORKERS; i++)
- {
- sender_sc_w->start_time_w[i] = start_time;
- sender_sc_w->idle_time[i] = 0.0;
- sender_sc_w->idle_start_time[i] = 0.0;
- hypervisor.sched_ctx_w[sender_sched_ctx].exec_time[i] = 0.0;
- hypervisor.sched_ctx_w[sender_sched_ctx].exec_start_time[i] = (hypervisor.sched_ctx_w[sender_sched_ctx].exec_start_time[i] != 0.0) ? starpu_timing_now() : 0.0;
- }
-
- }
- if(receiver_sched_ctx != STARPU_NMAX_SCHED_CTXS)
- {
- struct sc_hypervisor_wrapper *receiver_sc_w = &hypervisor.sched_ctx_w[receiver_sched_ctx];
-
- receiver_sc_w->start_time = start_time;
- _set_elapsed_flops_per_sched_ctx(receiver_sched_ctx, 0.0);
- int i;
- for(i = 0; i < STARPU_NMAXWORKERS; i++)
- {
- receiver_sc_w->start_time_w[i] = (receiver_sc_w->start_time_w[i] != 0.0) ? starpu_timing_now() : 0.0;
- receiver_sc_w->idle_time[i] = 0.0;
- receiver_sc_w->idle_start_time[i] = (receiver_sc_w->exec_start_time[i] != 0.0) ? 0.0 : starpu_timing_now();
- hypervisor.sched_ctx_w[receiver_sched_ctx].exec_start_time[i] = (receiver_sc_w->exec_start_time[i] != 0.0) ? starpu_timing_now() : 0.0;
- hypervisor.sched_ctx_w[receiver_sched_ctx].exec_time[i] = 0.0;
- }
- }
- }
- /* actually move the workers: the cpus are moved, gpus are only shared */
- /* forbids another resize request before this one is take into account */
- void sc_hypervisor_move_workers(unsigned sender_sched_ctx, unsigned receiver_sched_ctx, int* workers_to_move, unsigned nworkers_to_move, unsigned now)
- {
- if(nworkers_to_move > 0 && hypervisor.resize[sender_sched_ctx])
- {
- _print_current_time();
- unsigned j;
- printf("resize ctx %d with %d workers", sender_sched_ctx, nworkers_to_move);
- for(j = 0; j < nworkers_to_move; j++)
- printf(" %d", workers_to_move[j]);
- printf("\n");
- starpu_trace_user_event(1);
- hypervisor.allow_remove[receiver_sched_ctx] = 0;
- starpu_sched_ctx_add_workers(workers_to_move, nworkers_to_move, receiver_sched_ctx);
- if(now)
- {
- unsigned j;
- printf("remove now from ctx %d:", sender_sched_ctx);
- for(j = 0; j < nworkers_to_move; j++)
- printf(" %d", workers_to_move[j]);
- printf("\n");
- starpu_sched_ctx_remove_workers(workers_to_move, nworkers_to_move, sender_sched_ctx);
- hypervisor.allow_remove[receiver_sched_ctx] = 1;
- _reset_resize_sample_info(sender_sched_ctx, receiver_sched_ctx);
- }
- else
- {
- int ret = starpu_pthread_mutex_trylock(&hypervisor.sched_ctx_w[sender_sched_ctx].mutex);
- if(ret != EBUSY)
- {
- hypervisor.sched_ctx_w[sender_sched_ctx].resize_ack.receiver_sched_ctx = receiver_sched_ctx;
- hypervisor.sched_ctx_w[sender_sched_ctx].resize_ack.moved_workers = (int*)malloc(nworkers_to_move * sizeof(int));
- hypervisor.sched_ctx_w[sender_sched_ctx].resize_ack.nmoved_workers = nworkers_to_move;
- hypervisor.sched_ctx_w[sender_sched_ctx].resize_ack.acked_workers = (int*)malloc(nworkers_to_move * sizeof(int));
- unsigned i;
- for(i = 0; i < nworkers_to_move; i++)
- {
- hypervisor.sched_ctx_w[sender_sched_ctx].current_idle_time[workers_to_move[i]] = 0.0;
- hypervisor.sched_ctx_w[sender_sched_ctx].resize_ack.moved_workers[i] = workers_to_move[i];
- hypervisor.sched_ctx_w[sender_sched_ctx].resize_ack.acked_workers[i] = 0;
- }
- hypervisor.resize[sender_sched_ctx] = 0;
- if(imposed_resize) imposed_resize = 0;
- starpu_pthread_mutex_unlock(&hypervisor.sched_ctx_w[sender_sched_ctx].mutex);
- }
- }
- struct sc_hypervisor_policy_config *new_config = sc_hypervisor_get_config(receiver_sched_ctx);
- unsigned i;
- for(i = 0; i < nworkers_to_move; i++)
- new_config->max_idle[workers_to_move[i]] = new_config->max_idle[workers_to_move[i]] !=MAX_IDLE_TIME ? new_config->max_idle[workers_to_move[i]] : new_config->new_workers_max_idle;
- }
- return;
- }
- void sc_hypervisor_add_workers_to_sched_ctx(int* workers_to_add, unsigned nworkers_to_add, unsigned sched_ctx)
- {
- if(nworkers_to_add > 0 && hypervisor.resize[sched_ctx])
- {
- _print_current_time();
- unsigned j;
- printf("add to ctx %d:", sched_ctx);
- for(j = 0; j < nworkers_to_add; j++)
- printf(" %d", workers_to_add[j]);
- printf("\n");
- starpu_sched_ctx_add_workers(workers_to_add, nworkers_to_add, sched_ctx);
- starpu_sched_ctx_set_priority(workers_to_add, nworkers_to_add, sched_ctx, 1);
- struct sc_hypervisor_policy_config *new_config = sc_hypervisor_get_config(sched_ctx);
- unsigned i;
- for(i = 0; i < nworkers_to_add; i++)
- new_config->max_idle[workers_to_add[i]] = new_config->max_idle[workers_to_add[i]] != MAX_IDLE_TIME ? new_config->max_idle[workers_to_add[i]] : new_config->new_workers_max_idle;
- _reset_resize_sample_info(STARPU_NMAX_SCHED_CTXS, sched_ctx);
- }
- return;
- }
- unsigned sc_hypervisor_can_resize(unsigned sched_ctx)
- {
- return hypervisor.resize[sched_ctx];
- }
- void sc_hypervisor_remove_workers_from_sched_ctx(int* workers_to_remove, unsigned nworkers_to_remove, unsigned sched_ctx, unsigned now)
- {
- if(nworkers_to_remove > 0 && hypervisor.resize[sched_ctx] && hypervisor.allow_remove[sched_ctx])
- {
- _print_current_time();
- unsigned nworkers = 0;
- int workers[nworkers_to_remove];
- if(now)
- {
- unsigned j;
- printf("remove explicitley now from ctx %d:", sched_ctx);
- for(j = 0; j < nworkers_to_remove; j++)
- printf(" %d", workers_to_remove[j]);
- printf("\n");
-
- starpu_sched_ctx_remove_workers(workers_to_remove, nworkers_to_remove, sched_ctx);
- _reset_resize_sample_info(sched_ctx, STARPU_NMAX_SCHED_CTXS);
- }
- else
- {
- printf("try to remove from ctx %d: ", sched_ctx);
- unsigned j;
- for(j = 0; j < nworkers_to_remove; j++)
- printf(" %d", workers_to_remove[j]);
- printf("\n");
- int ret = starpu_pthread_mutex_trylock(&hypervisor.sched_ctx_w[sched_ctx].mutex);
- if(ret != EBUSY)
- {
- unsigned i;
- for(i = 0; i < nworkers_to_remove; i++)
- if(starpu_sched_ctx_contains_worker(workers_to_remove[i], sched_ctx))
- workers[nworkers++] = workers_to_remove[i];
- hypervisor.sched_ctx_w[sched_ctx].resize_ack.receiver_sched_ctx = -1;
- hypervisor.sched_ctx_w[sched_ctx].resize_ack.moved_workers = (int*)malloc(nworkers_to_remove * sizeof(int));
- hypervisor.sched_ctx_w[sched_ctx].resize_ack.nmoved_workers = (int)nworkers;
- hypervisor.sched_ctx_w[sched_ctx].resize_ack.acked_workers = (int*)malloc(nworkers_to_remove * sizeof(int));
- for(i = 0; i < nworkers; i++)
- {
- hypervisor.sched_ctx_w[sched_ctx].current_idle_time[workers[i]] = 0.0;
- hypervisor.sched_ctx_w[sched_ctx].resize_ack.moved_workers[i] = workers[i];
- hypervisor.sched_ctx_w[sched_ctx].resize_ack.acked_workers[i] = 0;
- }
- hypervisor.resize[sched_ctx] = 0;
- if(imposed_resize) imposed_resize = 0;
- starpu_pthread_mutex_unlock(&hypervisor.sched_ctx_w[sched_ctx].mutex);
- }
- }
- }
- return;
- }
- static unsigned _ack_resize_completed(unsigned sched_ctx, int worker)
- {
- if(worker != -1 && !starpu_sched_ctx_contains_worker(worker, sched_ctx))
- return 0;
-
- struct sc_hypervisor_resize_ack *resize_ack = NULL;
- unsigned sender_sched_ctx = STARPU_NMAX_SCHED_CTXS;
- int i;
- for(i = 0; i < STARPU_NMAX_SCHED_CTXS; i++)
- {
- if(hypervisor.sched_ctxs[i] != STARPU_NMAX_SCHED_CTXS)
- {
- struct sc_hypervisor_wrapper *sc_w = &hypervisor.sched_ctx_w[hypervisor.sched_ctxs[i]];
- starpu_pthread_mutex_lock(&sc_w->mutex);
- unsigned only_remove = 0;
- if(sc_w->resize_ack.receiver_sched_ctx == -1 && hypervisor.sched_ctxs[i] != sched_ctx &&
- sc_w->resize_ack.nmoved_workers > 0 && starpu_sched_ctx_contains_worker(worker, hypervisor.sched_ctxs[i]))
- {
- int j;
- for(j = 0; j < sc_w->resize_ack.nmoved_workers; j++)
- if(sc_w->resize_ack.moved_workers[j] == worker)
- {
- only_remove = 1;
- _reset_resize_sample_info(sched_ctx, STARPU_NMAX_SCHED_CTXS);
- starpu_pthread_mutex_unlock(&sc_w->mutex);
- break;
- }
- }
- if(only_remove ||
- (sc_w->resize_ack.receiver_sched_ctx != -1 && sc_w->resize_ack.receiver_sched_ctx == (int)sched_ctx))
- {
- resize_ack = &sc_w->resize_ack;
- sender_sched_ctx = hypervisor.sched_ctxs[i];
- starpu_pthread_mutex_unlock(&sc_w->mutex);
- break;
- }
- starpu_pthread_mutex_unlock(&sc_w->mutex);
- }
- }
- /* if there is no ctx waiting for its ack return 1*/
- if(resize_ack == NULL)
- {
- return 1;
- }
- int ret = starpu_pthread_mutex_trylock(&hypervisor.sched_ctx_w[sender_sched_ctx].mutex);
- if(ret != EBUSY)
- {
- int *moved_workers = resize_ack->moved_workers;
- int nmoved_workers = resize_ack->nmoved_workers;
- int *acked_workers = resize_ack->acked_workers;
- if(worker != -1)
- {
- for(i = 0; i < nmoved_workers; i++)
- {
- int moved_worker = moved_workers[i];
- if(moved_worker == worker && acked_workers[i] == 0)
- {
- acked_workers[i] = 1;
- }
- }
- }
- int nacked_workers = 0;
- for(i = 0; i < nmoved_workers; i++)
- {
- nacked_workers += (acked_workers[i] == 1);
- }
- unsigned resize_completed = (nacked_workers == nmoved_workers);
- int receiver_sched_ctx = sched_ctx;
- if(resize_completed)
- {
- /* if the permission to resize is not allowed by the user don't do it
- whatever the application says */
- if(!((hypervisor.resize[sender_sched_ctx] == 0 || hypervisor.resize[receiver_sched_ctx] == 0) && imposed_resize))
- {
- /* int j; */
- /* printf("remove after ack from ctx %d:", sender_sched_ctx); */
- /* for(j = 0; j < nmoved_workers; j++) */
- /* printf(" %d", moved_workers[j]); */
- /* printf("\n"); */
- starpu_sched_ctx_remove_workers(moved_workers, nmoved_workers, sender_sched_ctx);
- _reset_resize_sample_info(sender_sched_ctx, receiver_sched_ctx);
- hypervisor.resize[sender_sched_ctx] = 1;
- hypervisor.allow_remove[receiver_sched_ctx] = 1;
- /* if the user allowed resizing leave the decisions to the application */
- if(imposed_resize) imposed_resize = 0;
- resize_ack->receiver_sched_ctx = -1;
- resize_ack->nmoved_workers = 0;
- free(resize_ack->moved_workers);
- free(resize_ack->acked_workers);
- }
- starpu_pthread_mutex_unlock(&hypervisor.sched_ctx_w[sender_sched_ctx].mutex);
- return resize_completed;
- }
- starpu_pthread_mutex_unlock(&hypervisor.sched_ctx_w[sender_sched_ctx].mutex);
- }
- return 0;
- }
- /* Enqueue a resize request for 'sched_ctx', to be executed when the
- * 'task_tag' tasks of 'sched_ctx' complete. */
- void sc_hypervisor_post_resize_request(unsigned sched_ctx, int task_tag)
- {
- struct resize_request_entry *entry;
- entry = malloc(sizeof *entry);
- STARPU_ASSERT(entry != NULL);
- entry->sched_ctx = sched_ctx;
- entry->task_tag = task_tag;
- starpu_pthread_mutex_lock(&hypervisor.resize_mut[sched_ctx]);
- HASH_ADD_INT(hypervisor.resize_requests[sched_ctx], task_tag, entry);
- starpu_pthread_mutex_unlock(&hypervisor.resize_mut[sched_ctx]);
- }
- void sc_hypervisor_resize_ctxs(unsigned *sched_ctxs, int nsched_ctxs , int *workers, int nworkers)
- {
- if(hypervisor.policy.resize_ctxs)
- hypervisor.policy.resize_ctxs(sched_ctxs, nsched_ctxs, workers, nworkers);
- }
- void _sc_hypervisor_allow_compute_idle(unsigned sched_ctx, int worker, unsigned allow)
- {
- hypervisor.sched_ctx_w[sched_ctx].compute_idle[worker] = allow;
- }
- void sc_hypervisor_update_resize_interval(unsigned *sched_ctxs, int nsched_ctxs)
- {
- unsigned sched_ctx;
- int total_max_nworkers = 0;
- int max_cpus = starpu_cpu_worker_get_count();
- double max_workers_idle_time[nsched_ctxs];
- unsigned configured = 0;
- int i;
- for(i = 0; i < nsched_ctxs; i++)
- {
- sched_ctx = sched_ctxs[i];
- if(hypervisor.sched_ctx_w[sched_ctx].to_be_sized) continue;
- struct sc_hypervisor_policy_config *config = sc_hypervisor_get_config(sched_ctx);
- struct starpu_worker_collection *workers = starpu_sched_ctx_get_worker_collection(sched_ctx);
- int worker;
-
- struct starpu_sched_ctx_iterator it;
- if(workers->init_iterator)
- workers->init_iterator(workers, &it);
-
- double elapsed_time_worker[STARPU_NMAXWORKERS];
- double norm_idle_time = 0.0;
- double end_time = starpu_timing_now();
- while(workers->has_next(workers, &it))
- {
- double idle_time = 0.0;
- worker = workers->get_next(workers, &it);
- if(hypervisor.sched_ctx_w[sched_ctx].compute_idle[worker])
- {
- if(hypervisor.sched_ctx_w[sched_ctx].start_time_w[worker] == 0.0)
- elapsed_time_worker[worker] = 0.0;
- else
- elapsed_time_worker[worker] = (end_time - hypervisor.sched_ctx_w[sched_ctx].start_time_w[worker]) / 1000000.0;
- if(hypervisor.sched_ctx_w[sched_ctx].idle_start_time[worker] == 0.0)
- {
- idle_time = hypervisor.sched_ctx_w[sched_ctx].idle_time[worker]; /* in seconds */
- }
- else
- {
- double idle = (end_time - hypervisor.sched_ctx_w[sched_ctx].idle_start_time[worker]) / 1000000.0; /* in seconds */
- idle_time = hypervisor.sched_ctx_w[sched_ctx].idle_time[worker] + idle;
- }
- norm_idle_time += (elapsed_time_worker[worker] == 0.0 ? 0.0 : (idle_time / elapsed_time_worker[worker]));
- /* printf("%d/%d: start time %lf elapsed time %lf idle time %lf norm_idle_time %lf \n", */
- /* worker, sched_ctx, hypervisor.sched_ctx_w[sched_ctx].start_time_w[worker], elapsed_time_worker[worker], idle_time, norm_idle_time); */
- }
- }
- double norm_exec_time = 0.0;
- for(worker = 0; worker < STARPU_NMAXWORKERS; worker++)
- {
- double exec_time = 0.0;
- if(hypervisor.sched_ctx_w[sched_ctx].start_time_w[worker] == 0.0)
- elapsed_time_worker[worker] = 0.0;
- else
- elapsed_time_worker[worker] = (end_time - hypervisor.sched_ctx_w[sched_ctx].start_time_w[worker]) / 1000000.0;
- if(hypervisor.sched_ctx_w[sched_ctx].exec_start_time[worker] == 0.0)
- {
- exec_time = hypervisor.sched_ctx_w[sched_ctx].exec_time[worker];
- }
- else
- {
- double current_exec_time = (end_time - hypervisor.sched_ctx_w[sched_ctx].exec_start_time[worker]) / 1000000.0; /* in seconds */
- exec_time = hypervisor.sched_ctx_w[sched_ctx].exec_time[worker] + current_exec_time;
- }
- norm_exec_time += elapsed_time_worker[worker] == 0.0 ? 0.0 : exec_time / elapsed_time_worker[worker];
- }
- double curr_time = starpu_timing_now();
- double elapsed_time = (curr_time - hypervisor.sched_ctx_w[sched_ctx].start_time) / 1000000.0; /* in seconds */
- int nready_tasks = starpu_get_nready_tasks_of_sched_ctx(sched_ctx);
- if(norm_idle_time >= 0.9)
- {
- config->max_nworkers = lrint(norm_exec_time);
- }
- else
- {
- if(norm_idle_time < 0.1)
- config->max_nworkers = lrint(norm_exec_time) + nready_tasks - 1; //workers->nworkers + hypervisor.sched_ctx_w[sched_ctx].nready_tasks - 1;
- else
- config->max_nworkers = lrint(norm_exec_time);
- }
- // config->max_nworkers = hypervisor.sched_ctx_w[sched_ctx].nready_tasks - 1;
-
- if(config->max_nworkers < 0)
- config->max_nworkers = 0;
- if(config->max_nworkers > max_cpus)
- config->max_nworkers = max_cpus;
-
- printf("%d: ready tasks %d idle for long %lf norm_idle_time %lf elapsed_time %lf norm_exec_time %lf nworker %d max %d \n",
- sched_ctx, nready_tasks, max_workers_idle_time[i], norm_idle_time, elapsed_time, norm_exec_time, workers->nworkers, config->max_nworkers);
- total_max_nworkers += config->max_nworkers;
- configured = 1;
- }
- /*if the sum of the max cpus is smaller than the total cpus available
- increase the max for the ones having more ready tasks to exec */
- if(configured && total_max_nworkers < max_cpus)
- {
- int diff = max_cpus - total_max_nworkers;
- int max_nready = -1;
- unsigned max_nready_sched_ctx = sched_ctxs[0];
- for(i = 0; i < nsched_ctxs; i++)
- {
- int nready_tasks = starpu_get_nready_tasks_of_sched_ctx(sched_ctxs[i]);
- if(max_nready < nready_tasks)
- {
- max_nready = nready_tasks;
- max_nready_sched_ctx = sched_ctxs[i];
- }
- }
- struct sc_hypervisor_policy_config *config = sc_hypervisor_get_config(max_nready_sched_ctx);
- config->max_nworkers += diff;
- printf("%d: redib max_nworkers incr %d \n", max_nready_sched_ctx, config->max_nworkers);
- }
- }
- /* notifies the hypervisor that a new task was pushed on the queue of the worker */
- static void notify_pushed_task(unsigned sched_ctx, int worker)
- {
- hypervisor.sched_ctx_w[sched_ctx].pushed_tasks[worker]++;
- if(hypervisor.sched_ctx_w[sched_ctx].total_flops != 0.0 && hypervisor.sched_ctx_w[sched_ctx].start_time == 0.0)
- hypervisor.sched_ctx_w[sched_ctx].start_time = starpu_timing_now();
- if(hypervisor.sched_ctx_w[sched_ctx].total_flops != 0.0 && hypervisor.sched_ctx_w[sched_ctx].start_time_w[worker] == 0.0)
- hypervisor.sched_ctx_w[sched_ctx].start_time_w[worker] = starpu_timing_now();
- if(hypervisor.sched_ctx_w[sched_ctx].total_flops != 0.0 && hypervisor.sched_ctx_w[sched_ctx].real_start_time == 0.0)
- hypervisor.sched_ctx_w[sched_ctx].real_start_time = starpu_timing_now();
- int ntasks = get_ntasks(hypervisor.sched_ctx_w[sched_ctx].pushed_tasks);
- if((hypervisor.min_tasks == 0 || (!(hypervisor.resize[sched_ctx] == 0 && imposed_resize) && ntasks == hypervisor.min_tasks)) && hypervisor.check_min_tasks[sched_ctx])
- {
- hypervisor.resize[sched_ctx] = 1;
- if(imposed_resize) imposed_resize = 0;
- hypervisor.check_min_tasks[sched_ctx] = 0;
- }
- if(hypervisor.policy.handle_pushed_task)
- hypervisor.policy.handle_pushed_task(sched_ctx, worker);
- }
- unsigned choose_ctx_to_steal(int worker)
- {
- int j;
- int ns = hypervisor.nsched_ctxs;
- int max_ready_tasks = 0;
- unsigned chosen_ctx = STARPU_NMAX_SCHED_CTXS;
- for(j = 0; j < ns; j++)
- {
- unsigned other_ctx = hypervisor.sched_ctxs[j];
- int nready = starpu_get_nready_tasks_of_sched_ctx(other_ctx);
- if(!starpu_sched_ctx_contains_worker(worker, other_ctx) && max_ready_tasks < nready)
- {
- max_ready_tasks = nready;
- chosen_ctx = other_ctx;
- }
- }
- return chosen_ctx;
- }
- /* notifies the hypervisor that the worker spent another cycle in idle time */
- static void notify_idle_cycle(unsigned sched_ctx, int worker, double idle_time)
- {
- struct sc_hypervisor_wrapper *sc_w = &hypervisor.sched_ctx_w[sched_ctx];
- sc_w->current_idle_time[worker] += idle_time;
-
- if(sc_w->idle_start_time[worker] == 0.0 && sc_w->hyp_react_start_time != 0.0)
- sc_w->idle_start_time[worker] = starpu_timing_now();
-
- if(sc_w->idle_start_time[worker] > 0.0)
- {
- double end_time = starpu_timing_now();
- sc_w->idle_time[worker] += (end_time - sc_w->idle_start_time[worker]) / 1000000.0; /* in seconds */
- }
-
- hypervisor.sched_ctx_w[sched_ctx].idle_start_time[worker] = starpu_timing_now();
- if(hypervisor.resize[sched_ctx] && hypervisor.policy.handle_idle_cycle)
- {
- if(sc_w->sched_ctx != STARPU_NMAX_SCHED_CTXS && sc_w->hyp_react_start_time != 0.0)
- {
- if(sc_hypervisor_check_idle(sched_ctx, worker))
- {
- int ret = starpu_pthread_mutex_trylock(&act_hypervisor_mutex);
- if(ret != EBUSY)
- {
- int j;
- int ns = hypervisor.nsched_ctxs;
- unsigned idle_everywhere = 1;
- for(j = 0; j < ns; j++)
- {
- if(starpu_sched_ctx_contains_worker(worker, hypervisor.sched_ctxs[j]))
- {
- if(!sc_hypervisor_check_idle(hypervisor.sched_ctxs[j], worker))
- idle_everywhere = 0;
- }
- }
- if(idle_everywhere)
- {
- unsigned other_ctx = choose_ctx_to_steal(worker);
- if(other_ctx != STARPU_NMAX_SCHED_CTXS)
- {
- sc_hypervisor_add_workers_to_sched_ctx(&worker, 1, other_ctx);
- starpu_sched_ctx_set_priority(&worker, 1, other_ctx, 0);
- _sc_hypervisor_allow_compute_idle(other_ctx, worker, 0);
- }
- }
- starpu_pthread_mutex_unlock(&act_hypervisor_mutex);
- }
- // hypervisor.policy.handle_idle_cycle(sched_ctx, worker);
- }
- }
- }
- return;
- }
- /* notifies the hypervisor that the worker is no longer idle and a new task was pushed on its queue */
- static void notify_poped_task(unsigned sched_ctx, int worker)
- {
- if(hypervisor.sched_ctx_w[sched_ctx].start_time_w[worker] == 0.0)
- hypervisor.sched_ctx_w[sched_ctx].start_time_w[worker] = starpu_timing_now();
- hypervisor.sched_ctx_w[sched_ctx].exec_start_time[worker] = starpu_timing_now();
- if(hypervisor.sched_ctx_w[sched_ctx].idle_start_time[worker] > 0.0)
- {
- int ns = hypervisor.nsched_ctxs;
- int j;
- for(j = 0; j < ns; j++)
- {
- if(hypervisor.sched_ctxs[j] != sched_ctx)
- {
- if(hypervisor.sched_ctx_w[hypervisor.sched_ctxs[j]].idle_start_time[worker] > 0.0)
- hypervisor.sched_ctx_w[hypervisor.sched_ctxs[j]].compute_partial_idle[worker] = 1;
- }
- }
- double end_time = starpu_timing_now();
- double idle = (end_time - hypervisor.sched_ctx_w[sched_ctx].idle_start_time[worker]) / 1000000.0; /* in seconds */
-
- if(hypervisor.sched_ctx_w[sched_ctx].compute_partial_idle[worker])
- hypervisor.sched_ctx_w[sched_ctx].idle_time[worker] += idle / 2.0;
- else
- hypervisor.sched_ctx_w[sched_ctx].idle_time[worker] += idle;
- hypervisor.sched_ctx_w[sched_ctx].compute_partial_idle[worker] = 0;
- hypervisor.sched_ctx_w[sched_ctx].idle_start_time[worker] = 0.0;
- }
- if(hypervisor.resize[sched_ctx])
- hypervisor.sched_ctx_w[sched_ctx].current_idle_time[worker] = 0.0;
-
- if(hypervisor.policy.handle_idle_end)
- hypervisor.policy.handle_idle_end(sched_ctx, worker);
- }
-
- /* notifies the hypervisor that a tagged task has just been executed */
- static void notify_post_exec_task(struct starpu_task *task, size_t data_size, uint32_t footprint, int task_tag, double flops)
- {
- unsigned sched_ctx = task->sched_ctx;
- int worker = starpu_worker_get_id();
- if(hypervisor.sched_ctx_w[sched_ctx].exec_start_time[worker] != 0.0)
- {
- double current_time = starpu_timing_now();
- hypervisor.sched_ctx_w[sched_ctx].exec_time[worker] += (current_time -
- hypervisor.sched_ctx_w[sched_ctx].exec_start_time[worker]) / 1000000.0; /* in seconds */
- hypervisor.sched_ctx_w[sched_ctx].exec_start_time[worker] = 0.0;
- }
- hypervisor.sched_ctx_w[sched_ctx].poped_tasks[worker]++;
- hypervisor.sched_ctx_w[sched_ctx].elapsed_flops[worker] += flops;
- hypervisor.sched_ctx_w[sched_ctx].elapsed_data[worker] += data_size ;
- hypervisor.sched_ctx_w[sched_ctx].elapsed_tasks[worker]++ ;
- hypervisor.sched_ctx_w[sched_ctx].total_elapsed_flops[worker] += flops;
- starpu_pthread_mutex_lock(&hypervisor.sched_ctx_w[sched_ctx].mutex);
- hypervisor.sched_ctx_w[sched_ctx].remaining_flops -= flops;
- if(_sc_hypervisor_use_lazy_resize())
- _ack_resize_completed(sched_ctx, worker);
- starpu_pthread_mutex_unlock(&hypervisor.sched_ctx_w[sched_ctx].mutex);
-
- if(hypervisor.resize[sched_ctx])
- {
- if(hypervisor.policy.handle_poped_task)
- {
- if(hypervisor.sched_ctx_w[sched_ctx].hyp_react_start_time == 0.0)
- hypervisor.sched_ctx_w[sched_ctx].hyp_react_start_time = starpu_timing_now();
- double curr_time = starpu_timing_now();
- double elapsed_time = (curr_time - hypervisor.sched_ctx_w[sched_ctx].hyp_react_start_time) / 1000000.0; /* in seconds */
- if(hypervisor.sched_ctx_w[sched_ctx].sched_ctx != STARPU_NMAX_SCHED_CTXS && elapsed_time > hypervisor.sched_ctx_w[sched_ctx].config->time_sample)
- {
- hypervisor.policy.handle_poped_task(sched_ctx, worker, task, footprint);
- hypervisor.sched_ctx_w[sched_ctx].hyp_react_start_time = starpu_timing_now();
- }
- }
- }
- /* starpu_pthread_mutex_lock(&act_hypervisor_mutex); */
- /* _ack_resize_completed(sched_ctx, worker); */
- /* starpu_pthread_mutex_unlock(&act_hypervisor_mutex); */
- if(hypervisor.sched_ctx_w[sched_ctx].poped_tasks[worker] % 200 == 0)
- _print_current_time();
- if(task_tag <= 0)
- return;
-
- unsigned conf_sched_ctx;
- unsigned i;
- unsigned ns = hypervisor.nsched_ctxs;
- for(i = 0; i < ns; i++)
- {
- struct configuration_entry *entry;
- conf_sched_ctx = hypervisor.sched_ctxs[i];
- starpu_pthread_mutex_lock(&hypervisor.conf_mut[conf_sched_ctx]);
- HASH_FIND_INT(hypervisor.configurations[conf_sched_ctx], &task_tag, entry);
- if (entry != NULL)
- {
- struct sc_hypervisor_policy_config *config = entry->configuration;
- sc_hypervisor_set_config(conf_sched_ctx, config);
- HASH_DEL(hypervisor.configurations[conf_sched_ctx], entry);
- free(config);
- }
- starpu_pthread_mutex_unlock(&hypervisor.conf_mut[conf_sched_ctx]);
- }
- if(hypervisor.resize[sched_ctx])
- {
- starpu_pthread_mutex_lock(&hypervisor.resize_mut[sched_ctx]);
- if(hypervisor.policy.handle_post_exec_hook)
- {
- /* Check whether 'task_tag' is in the 'resize_requests' set. */
- struct resize_request_entry *entry;
- HASH_FIND_INT(hypervisor.resize_requests[sched_ctx], &task_tag, entry);
- if (entry != NULL)
- {
- hypervisor.policy.handle_post_exec_hook(sched_ctx, task_tag);
- HASH_DEL(hypervisor.resize_requests[sched_ctx], entry);
- free(entry);
- }
- }
- starpu_pthread_mutex_unlock(&hypervisor.resize_mut[sched_ctx]);
- }
- return;
- }
- static void notify_submitted_job(struct starpu_task *task, uint32_t footprint, size_t data_size)
- {
- unsigned sched_ctx = task->sched_ctx;
- starpu_pthread_mutex_lock(&hypervisor.sched_ctx_w[sched_ctx].mutex);
- hypervisor.sched_ctx_w[sched_ctx].submitted_flops += task->flops;
- starpu_pthread_mutex_unlock(&hypervisor.sched_ctx_w[sched_ctx].mutex);
- if(hypervisor.policy.handle_submitted_job && !type_of_tasks_known)
- hypervisor.policy.handle_submitted_job(task->cl, task->sched_ctx, footprint, data_size);
- }
- static void notify_empty_ctx(unsigned sched_ctx_id, struct starpu_task *task)
- {
- sc_hypervisor_resize_ctxs(NULL, -1 , NULL, -1);
- }
- void sc_hypervisor_set_type_of_task(struct starpu_codelet *cl, unsigned sched_ctx, uint32_t footprint, size_t data_size)
- {
- type_of_tasks_known = 1;
- if(hypervisor.policy.handle_submitted_job)
- hypervisor.policy.handle_submitted_job(cl, sched_ctx, footprint, data_size);
- }
- static void notify_delete_context(unsigned sched_ctx)
- {
- _print_current_time();
- sc_hypervisor_unregister_ctx(sched_ctx);
- }
- void sc_hypervisor_size_ctxs(unsigned *sched_ctxs, int nsched_ctxs, int *workers, int nworkers)
- {
- // starpu_pthread_mutex_lock(&act_hypervisor_mutex);
- unsigned curr_nsched_ctxs = sched_ctxs == NULL ? hypervisor.nsched_ctxs : (unsigned)nsched_ctxs;
- unsigned *curr_sched_ctxs = sched_ctxs == NULL ? hypervisor.sched_ctxs : sched_ctxs;
- // starpu_pthread_mutex_unlock(&act_hypervisor_mutex);
- unsigned s;
- for(s = 0; s < curr_nsched_ctxs; s++)
- hypervisor.resize[curr_sched_ctxs[s]] = 1;
- if(hypervisor.policy.size_ctxs)
- hypervisor.policy.size_ctxs(curr_sched_ctxs, curr_nsched_ctxs, workers, nworkers);
- }
- struct sc_hypervisor_wrapper* sc_hypervisor_get_wrapper(unsigned sched_ctx)
- {
- return &hypervisor.sched_ctx_w[sched_ctx];
- }
- unsigned* sc_hypervisor_get_sched_ctxs()
- {
- return hypervisor.sched_ctxs;
- }
- int sc_hypervisor_get_nsched_ctxs()
- {
- int ns;
- ns = hypervisor.nsched_ctxs;
- return ns;
- }
- int _sc_hypervisor_use_lazy_resize(void)
- {
- char* lazy = getenv("SC_HYPERVISOR_LAZY_RESIZE");
- return lazy ? atoi(lazy) : 1;
- }
- void sc_hypervisor_save_size_req(unsigned *sched_ctxs, int nsched_ctxs, int *workers, int nworkers)
- {
- hypervisor.sr = (struct size_request*)malloc(sizeof(struct size_request));
- hypervisor.sr->sched_ctxs = sched_ctxs;
- hypervisor.sr->nsched_ctxs = nsched_ctxs;
- hypervisor.sr->workers = workers;
- hypervisor.sr->nworkers = nworkers;
- }
- unsigned sc_hypervisor_get_size_req(unsigned **sched_ctxs, int* nsched_ctxs, int **workers, int *nworkers)
- {
- if(hypervisor.sr != NULL)
- {
- *sched_ctxs = hypervisor.sr->sched_ctxs;
- *nsched_ctxs = hypervisor.sr->nsched_ctxs;
- *workers = hypervisor.sr->workers;
- *nworkers = hypervisor.sr->nworkers;
- return 1;
- }
- return 0;
- }
- void sc_hypervisor_free_size_req(void)
- {
- if(hypervisor.sr != NULL)
- {
- free(hypervisor.sr);
- hypervisor.sr = NULL;
- }
- }
- double _get_optimal_v(unsigned sched_ctx)
- {
- return hypervisor.optimal_v[sched_ctx];
- }
- void _set_optimal_v(unsigned sched_ctx, double optimal_v)
- {
- hypervisor.optimal_v[sched_ctx] = optimal_v;
- }
- static struct types_of_workers* _init_structure_types_of_workers(void)
- {
- struct types_of_workers *tw = (struct types_of_workers*)malloc(sizeof(struct types_of_workers));
- tw->ncpus = 0;
- tw->ncuda = 0;
- tw->nw = 0;
- return tw;
- }
- struct types_of_workers* sc_hypervisor_get_types_of_workers(int *workers, unsigned nworkers)
- {
- struct types_of_workers *tw = _init_structure_types_of_workers();
- unsigned w;
- for(w = 0; w < nworkers; w++)
- {
- enum starpu_worker_archtype arch = workers == NULL ? starpu_worker_get_type((int)w) : starpu_worker_get_type(workers[w]);
- if(arch == STARPU_CPU_WORKER)
- tw->ncpus++;
- if(arch == STARPU_CUDA_WORKER)
- tw->ncuda++;
- }
- if(tw->ncpus > 0) tw->nw++;
- if(tw->ncuda > 0) tw->nw++;
- return tw;
- }
- void sc_hypervisor_update_diff_total_flops(unsigned sched_ctx, double diff_total_flops)
- {
- starpu_pthread_mutex_lock(&hypervisor.sched_ctx_w[sched_ctx].mutex);
- hypervisor.sched_ctx_w[sched_ctx].total_flops += diff_total_flops;
- hypervisor.sched_ctx_w[sched_ctx].remaining_flops += diff_total_flops;
- starpu_pthread_mutex_unlock(&hypervisor.sched_ctx_w[sched_ctx].mutex);
- }
- void sc_hypervisor_update_diff_elapsed_flops(unsigned sched_ctx, double diff_elapsed_flops)
- {
- int workerid = starpu_worker_get_id();
- if(workerid != -1)
- {
- // starpu_pthread_mutex_lock(&hypervisor.sched_ctx_w[sched_ctx].mutex);
- hypervisor.sched_ctx_w[sched_ctx].elapsed_flops[workerid] += diff_elapsed_flops;
- hypervisor.sched_ctx_w[sched_ctx].total_elapsed_flops[workerid] += diff_elapsed_flops;
- // starpu_pthread_mutex_unlock(&hypervisor.sched_ctx_w[sched_ctx].mutex);
- }
- }
|