123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476147714781479148014811482148314841485148614871488148914901491149214931494149514961497149814991500150115021503150415051506150715081509151015111512151315141515151615171518151915201521152215231524152515261527152815291530153115321533153415351536153715381539154015411542154315441545154615471548154915501551155215531554155515561557155815591560156115621563156415651566156715681569157015711572157315741575157615771578157915801581158215831584158515861587158815891590159115921593159415951596159715981599160016011602160316041605160616071608160916101611161216131614161516161617161816191620162116221623162416251626162716281629163016311632163316341635163616371638163916401641164216431644164516461647164816491650165116521653165416551656165716581659166016611662166316641665166616671668166916701671167216731674167516761677167816791680168116821683168416851686168716881689169016911692169316941695169616971698169917001701170217031704170517061707170817091710171117121713171417151716171717181719172017211722172317241725172617271728172917301731173217331734173517361737173817391740174117421743174417451746174717481749175017511752175317541755175617571758175917601761176217631764176517661767176817691770177117721773177417751776177717781779178017811782178317841785178617871788178917901791179217931794179517961797179817991800180118021803180418051806180718081809181018111812181318141815181618171818181918201821182218231824182518261827182818291830183118321833183418351836183718381839184018411842184318441845184618471848184918501851185218531854185518561857185818591860186118621863186418651866186718681869187018711872187318741875187618771878187918801881188218831884188518861887188818891890189118921893189418951896189718981899190019011902190319041905190619071908190919101911191219131914191519161917191819191920192119221923192419251926192719281929193019311932193319341935193619371938193919401941194219431944194519461947194819491950195119521953195419551956195719581959196019611962196319641965196619671968196919701971197219731974197519761977197819791980198119821983198419851986198719881989199019911992199319941995199619971998199920002001200220032004200520062007200820092010201120122013201420152016201720182019202020212022202320242025202620272028202920302031203220332034203520362037203820392040204120422043204420452046204720482049205020512052205320542055205620572058205920602061206220632064206520662067206820692070207120722073207420752076207720782079208020812082208320842085208620872088208920902091209220932094209520962097209820992100210121022103210421052106210721082109211021112112211321142115211621172118211921202121212221232124212521262127212821292130213121322133213421352136213721382139214021412142214321442145214621472148214921502151215221532154215521562157215821592160216121622163216421652166216721682169217021712172217321742175217621772178217921802181218221832184218521862187218821892190219121922193219421952196219721982199220022012202220322042205220622072208220922102211221222132214221522162217221822192220222122222223222422252226222722282229223022312232223322342235223622372238223922402241224222432244224522462247224822492250225122522253225422552256225722582259226022612262226322642265226622672268226922702271227222732274227522762277227822792280228122822283228422852286228722882289229022912292229322942295229622972298229923002301230223032304230523062307230823092310231123122313231423152316231723182319232023212322232323242325232623272328232923302331233223332334233523362337233823392340234123422343234423452346234723482349235023512352235323542355235623572358235923602361236223632364236523662367236823692370237123722373237423752376237723782379238023812382238323842385238623872388238923902391239223932394239523962397239823992400240124022403240424052406240724082409241024112412241324142415241624172418241924202421242224232424242524262427242824292430243124322433243424352436243724382439244024412442244324442445244624472448244924502451245224532454245524562457245824592460246124622463246424652466246724682469247024712472247324742475247624772478247924802481248224832484248524862487248824892490249124922493249424952496249724982499250025012502250325042505250625072508250925102511251225132514251525162517251825192520252125222523252425252526252725282529253025312532253325342535253625372538 |
- /* StarPU --- Runtime system for heterogeneous multicore architectures.
- *
- * Copyright (C) 2009-2017 Université de Bordeaux
- * Copyright (C) 2010, 2011, 2012, 2013, 2014, 2015, 2016, 2017 CNRS
- * Copyright (C) 2010, 2011 INRIA
- * Copyright (C) 2011 Télécom-SudParis
- * Copyright (C) 2011-2012, 2016, 2017 INRIA
- * Copyright (C) 2016 Uppsala University
- *
- * 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 <stdlib.h>
- #include <stdio.h>
- #ifdef __linux__
- #include <sys/utsname.h>
- #endif
- #include <common/config.h>
- #include <common/utils.h>
- #include <common/graph.h>
- #include <core/progress_hook.h>
- #include <core/workers.h>
- #include <core/debug.h>
- #include <core/disk.h>
- #include <core/task.h>
- #include <datawizard/malloc.h>
- #include <profiling/profiling.h>
- #include <sched_policies/sched_component.h>
- #include <datawizard/memory_nodes.h>
- #include <top/starpu_top_core.h>
- #include <drivers/mp_common/sink_common.h>
- #include <drivers/scc/driver_scc_common.h>
- #include <drivers/mpi/driver_mpi_common.h>
- #include <drivers/cpu/driver_cpu.h>
- #include <drivers/cuda/driver_cuda.h>
- #include <drivers/opencl/driver_opencl.h>
- #ifdef STARPU_SIMGRID
- #include <core/simgrid.h>
- #endif
- #if defined(_WIN32) && !defined(__CYGWIN__)
- #include <windows.h>
- #endif
- /* acquire/release semantic for concurrent initialization/de-initialization */
- static starpu_pthread_mutex_t init_mutex = STARPU_PTHREAD_MUTEX_INITIALIZER;
- static starpu_pthread_cond_t init_cond = STARPU_PTHREAD_COND_INITIALIZER;
- static int init_count = 0;
- static enum initialization initialized = UNINITIALIZED;
- int _starpu_keys_initialized STARPU_ATTRIBUTE_INTERNAL;
- starpu_pthread_key_t _starpu_worker_key STARPU_ATTRIBUTE_INTERNAL;
- starpu_pthread_key_t _starpu_worker_set_key STARPU_ATTRIBUTE_INTERNAL;
- struct _starpu_machine_config _starpu_config STARPU_ATTRIBUTE_INTERNAL;
- static int check_entire_platform;
- int _starpu_worker_parallel_blocks;
- /* Pointers to argc and argv
- */
- static int *my_argc = 0;
- static char ***my_argv = NULL;
- /* Initialize value of static argc and argv, called when the process begins
- */
- void _starpu_set_argc_argv(int *argc_param, char ***argv_param)
- {
- my_argc = argc_param;
- my_argv = argv_param;
- }
- int *_starpu_get_argc()
- {
- return my_argc;
- }
- char ***_starpu_get_argv()
- {
- return my_argv;
- }
- int starpu_is_initialized(void)
- {
- return initialized == INITIALIZED;
- }
- /* Makes sure that at least one of the workers of type <arch> can execute
- * <task>, for at least one of its implementations. */
- static uint32_t _starpu_worker_exists_and_can_execute(struct starpu_task *task,
- enum starpu_worker_archtype arch)
- {
- _starpu_codelet_check_deprecated_fields(task->cl);
- /* make sure there is a worker on the machine able to execute the
- task, independent of the sched_ctx, this latter may receive latter on
- the necessary worker - the user or the hypervisor should take care this happens */
- struct _starpu_sched_ctx *sched_ctx = check_entire_platform == 1 ? _starpu_get_initial_sched_ctx() : _starpu_get_sched_ctx_struct(task->sched_ctx);
- struct starpu_worker_collection *workers = sched_ctx->workers;
- struct starpu_sched_ctx_iterator it;
- workers->init_iterator(workers, &it);
- while(workers->has_next(workers, &it))
- {
- int i = workers->get_next(workers, &it);
- if (starpu_worker_get_type(i) != arch)
- continue;
- unsigned impl;
- for (impl = 0; impl < STARPU_MAXIMPLEMENTATIONS; impl++)
- {
- /* We could call task->cl->can_execute(i, task, impl)
- here, it would definitely work. It is probably
- cheaper to check whether it is necessary in order to
- avoid a useless function call, though. */
- unsigned test_implementation = 0;
- switch (arch)
- {
- case STARPU_CPU_WORKER:
- if (task->cl->cpu_funcs[impl] != NULL)
- test_implementation = 1;
- break;
- case STARPU_CUDA_WORKER:
- if (task->cl->cuda_funcs[impl] != NULL)
- test_implementation = 1;
- break;
- case STARPU_OPENCL_WORKER:
- if (task->cl->opencl_funcs[impl] != NULL)
- test_implementation = 1;
- break;
- case STARPU_MIC_WORKER:
- if (task->cl->cpu_funcs_name[impl] != NULL || task->cl->mic_funcs[impl] != NULL)
- test_implementation = 1;
- break;
- case STARPU_MPI_MS_WORKER:
- if (task->cl->cpu_funcs_name[impl] != NULL || task->cl->mpi_ms_funcs[impl] != NULL)
- test_implementation = 1;
- break;
- case STARPU_SCC_WORKER:
- if (task->cl->cpu_funcs_name[impl] != NULL || task->cl->scc_funcs[impl] != NULL)
- test_implementation = 1;
- break;
- default:
- STARPU_ABORT();
- }
- if (!test_implementation)
- continue;
- if (task->cl->can_execute)
- return task->cl->can_execute(i, task, impl);
- if(test_implementation)
- return 1;
- }
- }
- return 0;
- }
- /* in case a task is submitted, we may check whether there exists a worker
- that may execute the task or not */
- uint32_t _starpu_worker_exists(struct starpu_task *task)
- {
- _starpu_codelet_check_deprecated_fields(task->cl);
- if (task->where == STARPU_NOWHERE)
- return 1;
- /* if the task belongs to the init context we can
- check out all the worker mask of the machine
- if not we should iterate on the workers of the ctx
- and verify if it exists a worker able to exec the task */
- if(task->sched_ctx == 0)
- {
- if (!(task->where & _starpu_config.worker_mask))
- return 0;
- if (!task->cl->can_execute)
- return 1;
- }
- #if defined(STARPU_USE_CPU) || defined(STARPU_SIMGRID)
- if ((task->where & STARPU_CPU) &&
- _starpu_worker_exists_and_can_execute(task, STARPU_CPU_WORKER))
- return 1;
- #endif
- #if defined(STARPU_USE_CUDA) || defined(STARPU_SIMGRID)
- if ((task->where & STARPU_CUDA) &&
- _starpu_worker_exists_and_can_execute(task, STARPU_CUDA_WORKER))
- return 1;
- #endif
- #if defined(STARPU_USE_OPENCL) || defined(STARPU_SIMGRID)
- if ((task->where & STARPU_OPENCL) &&
- _starpu_worker_exists_and_can_execute(task, STARPU_OPENCL_WORKER))
- return 1;
- #endif
- #ifdef STARPU_USE_MIC
- if ((task->where & STARPU_MIC) &&
- _starpu_worker_exists_and_can_execute(task, STARPU_MIC_WORKER))
- return 1;
- #endif
- #ifdef STARPU_USE_MPI_MASTER_SLAVE
- if ((task->where & STARPU_MPI_MS) &&
- _starpu_worker_exists_and_can_execute(task, STARPU_MPI_MS_WORKER))
- return 1;
- #endif
- #ifdef STARPU_USE_SCC
- if ((task->where & STARPU_SCC) &&
- _starpu_worker_exists_and_can_execute(task, STARPU_SCC_WORKER))
- return 1;
- #endif
- return 0;
- }
- uint32_t _starpu_can_submit_cuda_task(void)
- {
- return STARPU_CUDA & _starpu_config.worker_mask;
- }
- uint32_t _starpu_can_submit_cpu_task(void)
- {
- return STARPU_CPU & _starpu_config.worker_mask;
- }
- uint32_t _starpu_can_submit_opencl_task(void)
- {
- return STARPU_OPENCL & _starpu_config.worker_mask;
- }
- uint32_t _starpu_can_submit_scc_task(void)
- {
- return STARPU_SCC & _starpu_config.worker_mask;
- }
- static inline int _starpu_can_use_nth_implementation(enum starpu_worker_archtype arch, struct starpu_codelet *cl, unsigned nimpl)
- {
- switch(arch)
- {
- case STARPU_ANY_WORKER:
- {
- int cpu_func_enabled=1, cuda_func_enabled=1, opencl_func_enabled=1;
- /* TODO: MIC/SCC */
- #if defined(STARPU_USE_CPU) || defined(STARPU_SIMGRID)
- starpu_cpu_func_t cpu_func = _starpu_task_get_cpu_nth_implementation(cl, nimpl);
- cpu_func_enabled = cpu_func != NULL && starpu_cpu_worker_get_count();
- #endif
- #if defined(STARPU_USE_CUDA) || defined(STARPU_SIMGRID)
- starpu_cuda_func_t cuda_func = _starpu_task_get_cuda_nth_implementation(cl, nimpl);
- cuda_func_enabled = cuda_func != NULL && starpu_cuda_worker_get_count();
- #endif
- #if defined(STARPU_USE_OPENCL) || defined(STARPU_SIMGRID)
- starpu_opencl_func_t opencl_func = _starpu_task_get_opencl_nth_implementation(cl, nimpl);
- opencl_func_enabled = opencl_func != NULL && starpu_opencl_worker_get_count();
- #endif
- return cpu_func_enabled && cuda_func_enabled && opencl_func_enabled;
- }
- case STARPU_CPU_WORKER:
- {
- starpu_cpu_func_t func = _starpu_task_get_cpu_nth_implementation(cl, nimpl);
- return func != NULL;
- }
- case STARPU_CUDA_WORKER:
- {
- starpu_cuda_func_t func = _starpu_task_get_cuda_nth_implementation(cl, nimpl);
- return func != NULL;
- }
- case STARPU_OPENCL_WORKER:
- {
- starpu_opencl_func_t func = _starpu_task_get_opencl_nth_implementation(cl, nimpl);
- return func != NULL;
- }
- case STARPU_MIC_WORKER:
- {
- starpu_mic_func_t func = _starpu_task_get_mic_nth_implementation(cl, nimpl);
- const char *func_name = _starpu_task_get_cpu_name_nth_implementation(cl, nimpl);
- return func != NULL || func_name != NULL;
- }
- case STARPU_MPI_MS_WORKER:
- {
- starpu_mpi_ms_func_t func = _starpu_task_get_mpi_ms_nth_implementation(cl, nimpl);
- const char *func_name = _starpu_task_get_cpu_name_nth_implementation(cl, nimpl);
- return func != NULL || func_name != NULL;
- }
- case STARPU_SCC_WORKER:
- {
- starpu_scc_func_t func = _starpu_task_get_scc_nth_implementation(cl, nimpl);
- const char *func_name = _starpu_task_get_cpu_name_nth_implementation(cl, nimpl);
- return func != NULL || func_name != NULL;
- }
- default:
- STARPU_ASSERT_MSG(0, "Unknown arch type %d", arch);
- }
- return 0;
- }
- /* must be called with sched_mutex locked to protect state_blocked_in_parallel */
- int starpu_worker_can_execute_task(unsigned workerid, struct starpu_task *task, unsigned nimpl)
- {
- /* if the worker is blocked in a parallel ctx don't submit tasks on it */
- #ifdef STARPU_DEVEL
- #warning FIXME: this is very expensive, while can_execute is supposed to be not very costly so schedulers can call it a lot
- #endif
- if(starpu_worker_is_blocked_in_parallel(workerid))
- return 0;
- /* TODO: check that the task operand sizes will fit on that device */
- return (task->where & _starpu_config.workers[workerid].worker_mask) &&
- _starpu_can_use_nth_implementation(_starpu_config.workers[workerid].arch, task->cl, nimpl) &&
- (!task->cl->can_execute || task->cl->can_execute(workerid, task, nimpl));
- }
- /* must be called with sched_mutex locked to protect state_blocked_in_parallel */
- int starpu_worker_can_execute_task_impl(unsigned workerid, struct starpu_task *task, unsigned *impl_mask)
- {
- /* if the worker is blocked in a parallel ctx don't submit tasks on it */
- if(starpu_worker_is_blocked_in_parallel(workerid))
- return 0;
- unsigned mask;
- int i;
- enum starpu_worker_archtype arch;
- struct starpu_codelet *cl;
- /* TODO: check that the task operand sizes will fit on that device */
- cl = task->cl;
- if (!(task->where & _starpu_config.workers[workerid].worker_mask))
- return 0;
- if (task->workerids_len)
- {
- size_t div = sizeof(*task->workerids) * 8;
- if (workerid / div >= task->workerids_len || ! (task->workerids[workerid / div] & (1UL << workerid % div)))
- return 0;
- }
- mask = 0;
- arch = _starpu_config.workers[workerid].arch;
- if (!task->cl->can_execute)
- {
- for (i = 0; i < STARPU_MAXIMPLEMENTATIONS; i++)
- if (_starpu_can_use_nth_implementation(arch, cl, i))
- {
- mask |= 1U << i;
- if (!impl_mask)
- break;
- }
- }
- else
- {
- for (i = 0; i < STARPU_MAXIMPLEMENTATIONS; i++)
- if (_starpu_can_use_nth_implementation(arch, cl, i)
- && (!task->cl->can_execute || task->cl->can_execute(workerid, task, i)))
- {
- mask |= 1U << i;
- if (!impl_mask)
- break;
- }
- }
- if (impl_mask)
- *impl_mask = mask;
- return mask != 0;
- }
- /* must be called with sched_mutex locked to protect state_blocked */
- int starpu_worker_can_execute_task_first_impl(unsigned workerid, struct starpu_task *task, unsigned *nimpl)
- {
- /* if the worker is blocked in a parallel ctx don't submit tasks on it */
- if(starpu_worker_is_blocked_in_parallel(workerid))
- return 0;
- int i;
- enum starpu_worker_archtype arch;
- struct starpu_codelet *cl;
- /* TODO: check that the task operand sizes will fit on that device */
- cl = task->cl;
- if (!(task->where & _starpu_config.workers[workerid].worker_mask))
- return 0;
- arch = _starpu_config.workers[workerid].arch;
- if (!task->cl->can_execute)
- {
- for (i = 0; i < STARPU_MAXIMPLEMENTATIONS; i++)
- if (_starpu_can_use_nth_implementation(arch, cl, i))
- {
- if (nimpl)
- *nimpl = i;
- return 1;
- }
- }
- else
- {
- for (i = 0; i < STARPU_MAXIMPLEMENTATIONS; i++)
- if (_starpu_can_use_nth_implementation(arch, cl, i)
- && (task->cl->can_execute(workerid, task, i)))
- {
- if (nimpl)
- *nimpl = i;
- return 1;
- }
- }
- return 0;
- }
- int starpu_combined_worker_can_execute_task(unsigned workerid, struct starpu_task *task, unsigned nimpl)
- {
- /* TODO: check that the task operand sizes will fit on that device */
- struct starpu_codelet *cl = task->cl;
- unsigned nworkers = _starpu_config.topology.nworkers;
- /* Is this a parallel worker ? */
- if (workerid < nworkers)
- {
- return !!((task->where & _starpu_config.workers[workerid].worker_mask) &&
- _starpu_can_use_nth_implementation(_starpu_config.workers[workerid].arch, task->cl, nimpl) &&
- (!task->cl->can_execute || task->cl->can_execute(workerid, task, nimpl)));
- }
- else
- {
- if (cl->type == STARPU_SPMD
- #ifdef STARPU_HAVE_HWLOC
- || cl->type == STARPU_FORKJOIN
- #else
- #ifdef __GLIBC__
- || cl->type == STARPU_FORKJOIN
- #endif
- #endif
- )
- {
- /* TODO we should add other types of constraints */
- /* Is the worker larger than requested ? */
- int worker_size = (int)_starpu_config.combined_workers[workerid - nworkers].worker_size;
- int worker0 = _starpu_config.combined_workers[workerid - nworkers].combined_workerid[0];
- return !!((worker_size <= task->cl->max_parallelism) &&
- _starpu_can_use_nth_implementation(_starpu_config.workers[worker0].arch, task->cl, nimpl) &&
- (!task->cl->can_execute || task->cl->can_execute(workerid, task, nimpl)));
- }
- else
- {
- /* We have a sequential task but a parallel worker */
- return 0;
- }
- }
- }
- /*
- * Runtime initialization methods
- */
- static void _starpu_init_worker_queue(struct _starpu_worker *worker)
- {
- _starpu_memory_node_register_condition(worker, &worker->sched_cond, worker->memory_node);
- }
- /*
- * Returns 0 if the given driver is one of the drivers that must be launched by
- * the application itself, and not by StarPU, 1 otherwise.
- */
- static unsigned _starpu_may_launch_driver(struct starpu_conf *conf,
- struct starpu_driver *d)
- {
- if (conf->n_not_launched_drivers == 0 || conf->not_launched_drivers == NULL)
- return 1;
- /* Is <d> in conf->not_launched_drivers ? */
- unsigned i;
- for (i = 0; i < conf->n_not_launched_drivers; i++)
- {
- if (d->type != conf->not_launched_drivers[i].type)
- continue;
- switch (d->type)
- {
- case STARPU_CPU_WORKER:
- if (d->id.cpu_id == conf->not_launched_drivers[i].id.cpu_id)
- return 0;
- break;
- case STARPU_CUDA_WORKER:
- if (d->id.cuda_id == conf->not_launched_drivers[i].id.cuda_id)
- return 0;
- break;
- #ifdef STARPU_USE_OPENCL
- case STARPU_OPENCL_WORKER:
- if (d->id.opencl_id == conf->not_launched_drivers[i].id.opencl_id)
- return 0;
- break;
- #endif
- default:
- STARPU_ABORT();
- }
- }
- return 1;
- }
- #ifdef STARPU_PERF_DEBUG
- struct itimerval prof_itimer;
- #endif
- static void _starpu_worker_init(struct _starpu_worker *workerarg, struct _starpu_machine_config *pconfig)
- {
- workerarg->config = pconfig;
- STARPU_PTHREAD_MUTEX_INIT(&workerarg->mutex, NULL);
- /* arch initialized by topology.c */
- /* worker_mask initialized by topology.c */
- /* perf_arch initialized by topology.c */
- /* worker_thread initialized by _starpu_launch_drivers */
- /* devid initialized by topology.c */
- /* subworkerid initialized by topology.c */
- /* bindid initialized by topology.c */
- /* workerid initialized by topology.c */
- workerarg->combined_workerid = workerarg->workerid;
- workerarg->current_rank = 0;
- workerarg->worker_size = 1;
- STARPU_PTHREAD_COND_INIT(&workerarg->started_cond, NULL);
- STARPU_PTHREAD_COND_INIT(&workerarg->ready_cond, NULL);
- /* memory_node initialized by topology.c */
- STARPU_PTHREAD_COND_INIT(&workerarg->sched_cond, NULL);
- STARPU_PTHREAD_MUTEX_INIT(&workerarg->sched_mutex, NULL);
- starpu_task_list_init(&workerarg->local_tasks);
- _starpu_ctx_change_list_init(&workerarg->ctx_change_list);
- workerarg->local_ordered_tasks = NULL;
- workerarg->local_ordered_tasks_size = 0;
- workerarg->current_ordered_task = 0;
- workerarg->current_ordered_task_order = 1;
- workerarg->current_task = NULL;
- #ifdef STARPU_SIMGRID
- starpu_pthread_wait_init(&workerarg->wait);
- starpu_pthread_queue_register(&workerarg->wait, &_starpu_simgrid_task_queue[workerarg->workerid]);
- #endif
- workerarg->task_transferring = NULL;
- workerarg->nb_buffers_transferred = 0;
- workerarg->nb_buffers_totransfer = 0;
- workerarg->first_task = 0;
- workerarg->ntasks = 0;
- /* set initialized by topology.c */
- workerarg->pipeline_length = 0;
- workerarg->pipeline_stuck = 0;
- workerarg->worker_is_running = 0;
- workerarg->worker_is_initialized = 0;
- workerarg->status = STATUS_INITIALIZING;
- workerarg->state_keep_awake = 0;
- /* name initialized by driver */
- /* short_name initialized by driver */
- workerarg->run_by_starpu = 1;
- workerarg->driver_ops = NULL;
- workerarg->sched_ctx_list = NULL;
- workerarg->tmp_sched_ctx = -1;
- workerarg->nsched_ctxs = 0;
- _starpu_barrier_counter_init(&workerarg->tasks_barrier, 0);
- workerarg->has_prev_init = 0;
- int ctx;
- for(ctx = 0; ctx < STARPU_NMAX_SCHED_CTXS; ctx++)
- workerarg->removed_from_ctx[ctx] = 0;
- workerarg->spinning_backoff = 1;
- for(ctx = 0; ctx < STARPU_NMAX_SCHED_CTXS; ctx++)
- {
- workerarg->shares_tasks_lists[ctx] = 0;
- workerarg->poped_in_ctx[ctx] = 0;
- }
- workerarg->reverse_phase[0] = 0;
- workerarg->reverse_phase[1] = 0;
- workerarg->pop_ctx_priority = 1;
- workerarg->is_slave_somewhere = 0;
- workerarg->state_relax_refcnt = 1;
- #ifdef STARPU_SPINLOCK_CHECK
- workerarg->relax_on_file = __FILE__;
- workerarg->relax_on_line = __LINE__;
- workerarg->relax_on_func = __starpu_func__;
- workerarg->relax_off_file = NULL;
- workerarg->relax_off_line = 0;
- workerarg->relax_off_func = NULL;
- #endif
- workerarg->state_sched_op_pending = 0;
- workerarg->state_changing_ctx_waiting = 0;
- workerarg->state_changing_ctx_notice = 0;
- workerarg->state_blocked_in_parallel_observed = 0;
- workerarg->state_blocked_in_parallel = 0;
- workerarg->state_block_in_parallel_req = 0;
- workerarg->state_block_in_parallel_ack = 0;
- workerarg->state_unblock_in_parallel_req = 0;
- workerarg->state_unblock_in_parallel_ack = 0;
- workerarg->block_in_parallel_ref_count = 0;
- /* cpu_set/hwloc_cpu_set initialized in topology.c */
- }
- static void _starpu_worker_deinit(struct _starpu_worker *workerarg)
- {
- (void) workerarg;
- #ifdef STARPU_SIMGRID
- starpu_pthread_queue_unregister(&workerarg->wait, &_starpu_simgrid_task_queue[workerarg->workerid]);
- starpu_pthread_wait_destroy(&workerarg->wait);
- #endif
- }
- #ifdef STARPU_USE_FXT
- void _starpu_worker_start(struct _starpu_worker *worker, unsigned fut_key, unsigned sync)
- {
- unsigned devid = worker->devid;
- unsigned memnode = worker->memory_node;
- _STARPU_TRACE_WORKER_INIT_START(fut_key, worker->workerid, devid, memnode, worker->bindid, sync);
- }
- #endif
- void _starpu_driver_start(struct _starpu_worker *worker, unsigned fut_key, unsigned sync STARPU_ATTRIBUTE_UNUSED)
- {
- (void) fut_key;
- int devid = worker->devid;
- (void) devid;
- #ifdef STARPU_USE_FXT
- _starpu_fxt_register_thread(worker->bindid);
- _starpu_worker_start(worker, fut_key, sync);
- #endif
- _starpu_set_local_worker_key(worker);
- STARPU_PTHREAD_MUTEX_LOCK(&worker->mutex);
- worker->worker_is_running = 1;
- STARPU_PTHREAD_COND_SIGNAL(&worker->started_cond);
- STARPU_PTHREAD_MUTEX_UNLOCK(&worker->mutex);
- _starpu_bind_thread_on_cpu(worker->bindid, worker->workerid);
- #if defined(STARPU_PERF_DEBUG) && !defined(STARPU_SIMGRID)
- setitimer(ITIMER_PROF, &prof_itimer, NULL);
- #endif
- _STARPU_DEBUG("worker %p %d for dev %d is ready on logical cpu %d\n", worker, worker->workerid, devid, worker->bindid);
- #ifdef STARPU_HAVE_HWLOC
- _STARPU_DEBUG("worker %p %d cpuset start at %d\n", worker, worker->workerid, hwloc_bitmap_first(worker->hwloc_cpu_set));
- #endif
- }
- static void _starpu_launch_drivers(struct _starpu_machine_config *pconfig)
- {
- pconfig->running = 1;
- pconfig->pause_depth = 0;
- pconfig->submitting = 1;
- STARPU_HG_DISABLE_CHECKING(pconfig->watchdog_ok);
- unsigned nworkers = pconfig->topology.nworkers;
- unsigned worker;
- #if defined(STARPU_PERF_DEBUG) && !defined(STARPU_SIMGRID)
- /* Get itimer of the main thread, to set it for the worker threads */
- getitimer(ITIMER_PROF, &prof_itimer);
- #endif
- STARPU_AYU_INIT();
- /* Launch workers asynchronously */
- for (worker = 0; worker < nworkers; worker++)
- {
- struct _starpu_worker *workerarg = &pconfig->workers[worker];
- unsigned devid = workerarg->devid;
- #if defined(STARPU_USE_MIC) || defined(STARPU_USE_CUDA) || defined(STARPU_SIMGRID) || defined(STARPU_USE_MPI_MASTER_SLAVE)
- struct _starpu_worker_set *worker_set = workerarg->set;
- #endif
- _STARPU_DEBUG("initialising worker %u/%u\n", worker, nworkers);
- _starpu_init_worker_queue(workerarg);
- struct starpu_driver driver;
- driver.type = workerarg->arch;
- switch (workerarg->arch)
- {
- #if defined(STARPU_USE_CPU) || defined(STARPU_SIMGRID)
- case STARPU_CPU_WORKER:
- driver.id.cpu_id = devid;
- workerarg->driver_ops = &_starpu_driver_cpu_ops;
- if (_starpu_may_launch_driver(&pconfig->conf, &driver))
- {
- STARPU_PTHREAD_CREATE_ON(
- "CPU",
- &workerarg->worker_thread,
- NULL,
- _starpu_cpu_worker,
- workerarg,
- _starpu_simgrid_get_host_by_worker(workerarg));
- #ifdef STARPU_USE_FXT
- /* In tracing mode, make sure the
- * thread is really started before
- * starting another one, to make sure
- * they appear in order in the trace.
- */
- STARPU_PTHREAD_MUTEX_LOCK(&workerarg->mutex);
- while (!workerarg->worker_is_running)
- STARPU_PTHREAD_COND_WAIT(&workerarg->started_cond, &workerarg->mutex);
- STARPU_PTHREAD_MUTEX_UNLOCK(&workerarg->mutex);
- #endif
- }
- else
- {
- workerarg->run_by_starpu = 0;
- }
- break;
- #endif
- #if defined(STARPU_USE_CUDA) || defined(STARPU_SIMGRID)
- case STARPU_CUDA_WORKER:
- driver.id.cuda_id = devid;
- workerarg->driver_ops = &_starpu_driver_cuda_ops;
- if (worker_set->workers != workerarg)
- /* We are not the first worker of the
- * set, don't start a thread for it. */
- break;
- worker_set->set_is_initialized = 0;
- if (!_starpu_may_launch_driver(&pconfig->conf, &driver))
- {
- workerarg->run_by_starpu = 0;
- break;
- }
- STARPU_PTHREAD_CREATE_ON(
- "CUDA",
- &worker_set->worker_thread,
- NULL,
- _starpu_cuda_worker,
- worker_set,
- _starpu_simgrid_get_host_by_worker(workerarg));
- #ifdef STARPU_USE_FXT
- STARPU_PTHREAD_MUTEX_LOCK(&workerarg->mutex);
- while (!workerarg->worker_is_running)
- STARPU_PTHREAD_COND_WAIT(&workerarg->started_cond, &workerarg->mutex);
- STARPU_PTHREAD_MUTEX_UNLOCK(&workerarg->mutex);
- #endif
- break;
- #endif
- #if defined(STARPU_USE_OPENCL) || defined(STARPU_SIMGRID)
- case STARPU_OPENCL_WORKER:
- #ifndef STARPU_SIMGRID
- starpu_opencl_get_device(devid, &driver.id.opencl_id);
- workerarg->driver_ops = &_starpu_driver_opencl_ops;
- if (!_starpu_may_launch_driver(&pconfig->conf, &driver))
- {
- workerarg->run_by_starpu = 0;
- break;
- }
- #endif
- STARPU_PTHREAD_CREATE_ON(
- "OpenCL",
- &workerarg->worker_thread,
- NULL,
- _starpu_opencl_worker,
- workerarg,
- _starpu_simgrid_get_host_by_worker(workerarg));
- #ifdef STARPU_USE_FXT
- STARPU_PTHREAD_MUTEX_LOCK(&workerarg->mutex);
- while (!workerarg->worker_is_running)
- STARPU_PTHREAD_COND_WAIT(&workerarg->started_cond, &workerarg->mutex);
- STARPU_PTHREAD_MUTEX_UNLOCK(&workerarg->mutex);
- #endif
- break;
- #endif
- #ifdef STARPU_USE_MIC
- case STARPU_MIC_WORKER:
- /* We spawn only one thread
- * per MIC device, which will control all MIC
- * workers of this device. (by using a worker set). */
- if (worker_set->workers != workerarg)
- break;
- worker_set->set_is_initialized = 0;
- STARPU_PTHREAD_CREATE_ON(
- "MIC",
- &worker_set->worker_thread,
- NULL,
- _starpu_mic_src_worker,
- worker_set,
- _starpu_simgrid_get_host_by_worker(workerarg));
- #ifdef STARPU_USE_FXT
- STARPU_PTHREAD_MUTEX_LOCK(&workerarg->mutex);
- while (!workerarg->worker_is_running)
- STARPU_PTHREAD_COND_WAIT(&workerarg->started_cond, &workerarg->mutex);
- STARPU_PTHREAD_MUTEX_UNLOCK(&workerarg->mutex);
- #endif
- STARPU_PTHREAD_MUTEX_LOCK(&worker_set->mutex);
- while (!worker_set->set_is_initialized)
- STARPU_PTHREAD_COND_WAIT(&worker_set->ready_cond,
- &worker_set->mutex);
- STARPU_PTHREAD_MUTEX_UNLOCK(&worker_set->mutex);
- worker_set->started = 1;
- break;
- #endif /* STARPU_USE_MIC */
- #ifdef STARPU_USE_SCC
- case STARPU_SCC_WORKER:
- workerarg->worker_is_initialized = 0;
- STARPU_PTHREAD_CREATE_ON(
- "SCC",
- &workerarg->worker_thread,
- NULL,
- _starpu_scc_src_worker,
- workerarg,
- _starpu_simgrid_get_host_by_worker(workerarg));
- #ifdef STARPU_USE_FXT
- STARPU_PTHREAD_MUTEX_LOCK(&workerarg->mutex);
- while (!workerarg->worker_is_running)
- STARPU_PTHREAD_COND_WAIT(&workerarg->started_cond, &workerarg->mutex);
- STARPU_PTHREAD_MUTEX_UNLOCK(&workerarg->mutex);
- #endif
- break;
- #endif /* STARPU_USE_SCC */
- #ifdef STARPU_USE_MPI_MASTER_SLAVE
- case STARPU_MPI_MS_WORKER:
- /* We spawn only one thread
- * per MPI device, which will control all MPI
- * workers of this device. (by using a worker set). */
- if (worker_set->workers != workerarg)
- break;
- worker_set->set_is_initialized = 0;
- #ifdef STARPU_MPI_MASTER_SLAVE_MULTIPLE_THREAD
- /* if MPI has multiple threads supports
- * we launch 1 thread per device
- * else
- * we launch one thread for all devices
- */
- STARPU_PTHREAD_CREATE_ON(
- "MPI MS",
- &worker_set->worker_thread,
- NULL,
- _starpu_mpi_src_worker,
- worker_set,
- _starpu_simgrid_get_host_by_worker(workerarg));
- #ifdef STARPU_USE_FXT
- STARPU_PTHREAD_MUTEX_LOCK(&workerarg->mutex);
- while (!workerarg->worker_is_running)
- STARPU_PTHREAD_COND_WAIT(&workerarg->started_cond, &workerarg->mutex);
- STARPU_PTHREAD_MUTEX_UNLOCK(&workerarg->mutex);
- #endif
- STARPU_PTHREAD_MUTEX_LOCK(&worker_set->mutex);
- while (!worker_set->set_is_initialized)
- STARPU_PTHREAD_COND_WAIT(&worker_set->ready_cond,
- &worker_set->mutex);
- STARPU_PTHREAD_MUTEX_UNLOCK(&worker_set->mutex);
- worker_set->started = 1;
- #endif /* STARPU_MPI_MASTER_SLAVE_MULTIPLE_THREAD */
- break;
- #endif /* STARPU_USE_MPI_MASTER_SLAVE */
- default:
- STARPU_ABORT();
- }
- }
- #if defined(STARPU_USE_MPI_MASTER_SLAVE) && !defined(STARPU_MPI_MASTER_SLAVE_MULTIPLE_THREAD)
- if (pconfig->topology.nmpidevices > 0)
- {
- struct _starpu_worker_set * worker_set_zero = &mpi_worker_set[0];
- struct _starpu_worker * worker_zero = &worker_set_zero->workers[0];
- STARPU_PTHREAD_CREATE_ON(
- "zero",
- &worker_set_zero->worker_thread,
- NULL,
- _starpu_mpi_src_worker,
- &mpi_worker_set,
- _starpu_simgrid_get_host_by_worker(worker_zero));
- /* We use the first worker to know if everything are finished */
- #ifdef STARPU_USE_FXT
- STARPU_PTHREAD_MUTEX_LOCK(&worker_zero->mutex);
- while (!worker_zero->worker_is_running)
- STARPU_PTHREAD_COND_WAIT(&worker_zero->started_cond, &worker_zero->mutex);
- STARPU_PTHREAD_MUTEX_UNLOCK(&worker_zero->mutex);
- #endif
- STARPU_PTHREAD_MUTEX_LOCK(&worker_set_zero->mutex);
- while (!worker_set_zero->set_is_initialized)
- STARPU_PTHREAD_COND_WAIT(&worker_set_zero->ready_cond,
- &worker_set_zero->mutex);
- STARPU_PTHREAD_MUTEX_UNLOCK(&worker_set_zero->mutex);
- worker_set_zero->started = 1;
- worker_set_zero->worker_thread = mpi_worker_set[0].worker_thread;
- }
- #endif
- for (worker = 0; worker < nworkers; worker++)
- {
- struct _starpu_worker *workerarg = &pconfig->workers[worker];
- _STARPU_DEBUG("waiting for worker %u initialization\n", worker);
- #ifndef STARPU_SIMGRID
- if (!workerarg->run_by_starpu)
- break;
- #endif
- #if defined(STARPU_USE_CUDA) || defined(STARPU_SIMGRID)
- if (workerarg->arch == STARPU_CUDA_WORKER)
- {
- struct _starpu_worker_set *worker_set = workerarg->set;
- STARPU_PTHREAD_MUTEX_LOCK(&worker_set->mutex);
- while (!worker_set->set_is_initialized)
- STARPU_PTHREAD_COND_WAIT(&worker_set->ready_cond,
- &worker_set->mutex);
- STARPU_PTHREAD_MUTEX_UNLOCK(&worker_set->mutex);
- worker_set->started = 1;
- }
- else
- #endif
- if (workerarg->arch != STARPU_CUDA_WORKER && workerarg->arch != STARPU_MPI_MS_WORKER && workerarg->arch != STARPU_MIC_WORKER)
- {
- STARPU_PTHREAD_MUTEX_LOCK(&workerarg->mutex);
- while (!workerarg->worker_is_initialized)
- STARPU_PTHREAD_COND_WAIT(&workerarg->ready_cond, &workerarg->mutex);
- STARPU_PTHREAD_MUTEX_UNLOCK(&workerarg->mutex);
- }
- }
- _STARPU_DEBUG("finished launching drivers\n");
- }
- /* Initialize the starpu_conf with default values */
- int starpu_conf_init(struct starpu_conf *conf)
- {
- if (!conf)
- return -EINVAL;
- memset(conf, 0, sizeof(*conf));
- conf->magic = 42;
- conf->sched_policy_name = starpu_getenv("STARPU_SCHED");
- conf->sched_policy = NULL;
- conf->global_sched_ctx_min_priority = starpu_get_env_number("STARPU_MIN_PRIO");
- conf->global_sched_ctx_max_priority = starpu_get_env_number("STARPU_MAX_PRIO");
- /* Note that starpu_get_env_number returns -1 in case the variable is
- * not defined */
- /* Backward compatibility: check the value of STARPU_NCPUS if
- * STARPU_NCPU is not set. */
- conf->ncpus = starpu_get_env_number("STARPU_NCPU");
- if (conf->ncpus == -1)
- conf->ncpus = starpu_get_env_number("STARPU_NCPUS");
- conf->ncuda = starpu_get_env_number("STARPU_NCUDA");
- conf->nopencl = starpu_get_env_number("STARPU_NOPENCL");
- conf->nmic = starpu_get_env_number("STARPU_NMIC");
- conf->nscc = starpu_get_env_number("STARPU_NSCC");
- conf->nmpi_ms = starpu_get_env_number("STARPU_NMPI_MS");
- conf->calibrate = starpu_get_env_number("STARPU_CALIBRATE");
- conf->bus_calibrate = starpu_get_env_number("STARPU_BUS_CALIBRATE");
- conf->mic_sink_program_path = starpu_getenv("STARPU_MIC_PROGRAM_PATH");
- if (conf->calibrate == -1)
- conf->calibrate = 0;
- if (conf->bus_calibrate == -1)
- conf->bus_calibrate = 0;
- conf->use_explicit_workers_bindid = 0; /* TODO */
- conf->use_explicit_workers_cuda_gpuid = 0; /* TODO */
- conf->use_explicit_workers_opencl_gpuid = 0; /* TODO */
- conf->use_explicit_workers_mic_deviceid = 0; /* TODO */
- conf->use_explicit_workers_scc_deviceid = 0; /* TODO */
- conf->use_explicit_workers_mpi_ms_deviceid = 0; /* TODO */
- conf->single_combined_worker = starpu_get_env_number("STARPU_SINGLE_COMBINED_WORKER");
- if (conf->single_combined_worker == -1)
- conf->single_combined_worker = 0;
- #if defined(STARPU_DISABLE_ASYNCHRONOUS_COPY)
- conf->disable_asynchronous_copy = 1;
- #else
- conf->disable_asynchronous_copy = starpu_get_env_number("STARPU_DISABLE_ASYNCHRONOUS_COPY");
- if (conf->disable_asynchronous_copy == -1)
- conf->disable_asynchronous_copy = 0;
- #endif
- #if defined(STARPU_DISABLE_ASYNCHRONOUS_CUDA_COPY)
- conf->disable_asynchronous_cuda_copy = 1;
- #else
- conf->disable_asynchronous_cuda_copy = starpu_get_env_number("STARPU_DISABLE_ASYNCHRONOUS_CUDA_COPY");
- if (conf->disable_asynchronous_cuda_copy == -1)
- conf->disable_asynchronous_cuda_copy = 0;
- #endif
- #if defined(STARPU_DISABLE_ASYNCHRONOUS_OPENCL_COPY)
- conf->disable_asynchronous_opencl_copy = 1;
- #else
- conf->disable_asynchronous_opencl_copy = starpu_get_env_number("STARPU_DISABLE_ASYNCHRONOUS_OPENCL_COPY");
- if (conf->disable_asynchronous_opencl_copy == -1)
- conf->disable_asynchronous_opencl_copy = 0;
- #endif
- #if defined(STARPU_DISABLE_ASYNCHRONOUS_MIC_COPY)
- conf->disable_asynchronous_mic_copy = 1;
- #else
- conf->disable_asynchronous_mic_copy = starpu_get_env_number("STARPU_DISABLE_ASYNCHRONOUS_MIC_COPY");
- if (conf->disable_asynchronous_mic_copy == -1)
- conf->disable_asynchronous_mic_copy = 0;
- #endif
- #if defined(STARPU_DISABLE_ASYNCHRONOUS_MPI_MS_COPY)
- conf->disable_asynchronous_mpi_ms_copy = 1;
- #else
- conf->disable_asynchronous_mpi_ms_copy = starpu_get_env_number("STARPU_DISABLE_ASYNCHRONOUS_MPI_MS_COPY");
- if(conf->disable_asynchronous_mpi_ms_copy == -1)
- conf->disable_asynchronous_mpi_ms_copy = 0;
- #endif
- /* 64MiB by default */
- conf->trace_buffer_size = starpu_get_env_number_default("STARPU_TRACE_BUFFER_SIZE", 64) << 20;
- return 0;
- }
- static void _starpu_conf_set_value_against_environment(char *name, int *value)
- {
- int number;
- number = starpu_get_env_number(name);
- if (number != -1)
- {
- *value = number;
- }
- }
- void _starpu_conf_check_environment(struct starpu_conf *conf)
- {
- char *sched = starpu_getenv("STARPU_SCHED");
- if (sched)
- {
- conf->sched_policy_name = sched;
- }
- _starpu_conf_set_value_against_environment("STARPU_NCPUS", &conf->ncpus);
- _starpu_conf_set_value_against_environment("STARPU_NCPU", &conf->ncpus);
- _starpu_conf_set_value_against_environment("STARPU_NCUDA", &conf->ncuda);
- _starpu_conf_set_value_against_environment("STARPU_NOPENCL", &conf->nopencl);
- _starpu_conf_set_value_against_environment("STARPU_CALIBRATE", &conf->calibrate);
- _starpu_conf_set_value_against_environment("STARPU_BUS_CALIBRATE", &conf->bus_calibrate);
- #ifdef STARPU_SIMGRID
- if (conf->calibrate == 2)
- {
- _STARPU_DISP("Warning: History will be cleared due to calibrate or STARPU_CALIBRATE being set to 2. This will prevent simgrid from having task simulation times!");
- }
- if (conf->bus_calibrate)
- {
- _STARPU_DISP("Warning: Bus calibration will be cleared due to bus_calibrate or STARPU_BUS_CALIBRATE being set. This will prevent simgrid from having data transfer simulation times!");
- }
- #endif
- _starpu_conf_set_value_against_environment("STARPU_SINGLE_COMBINED_WORKER", &conf->single_combined_worker);
- _starpu_conf_set_value_against_environment("STARPU_DISABLE_ASYNCHRONOUS_COPY", &conf->disable_asynchronous_copy);
- _starpu_conf_set_value_against_environment("STARPU_DISABLE_ASYNCHRONOUS_CUDA_COPY", &conf->disable_asynchronous_cuda_copy);
- _starpu_conf_set_value_against_environment("STARPU_DISABLE_ASYNCHRONOUS_OPENCL_COPY", &conf->disable_asynchronous_opencl_copy);
- _starpu_conf_set_value_against_environment("STARPU_DISABLE_ASYNCHRONOUS_MIC_COPY", &conf->disable_asynchronous_mic_copy);
- _starpu_conf_set_value_against_environment("STARPU_DISABLE_ASYNCHRONOUS_MPI_MS_COPY", &conf->disable_asynchronous_mpi_ms_copy);
- }
- struct starpu_tree* starpu_workers_get_tree(void)
- {
- return _starpu_config.topology.tree;
- }
- #ifdef STARPU_HAVE_HWLOC
- static void _fill_tree(struct starpu_tree *tree, hwloc_obj_t curr_obj, unsigned depth, hwloc_topology_t topology, struct starpu_tree *father)
- {
- unsigned i, j;
- unsigned arity;
- if (curr_obj->arity == 1)
- {
- /* Nothing interestin here, skip level */
- _fill_tree(tree, curr_obj->children[0], depth+1, topology, father);
- return;
- }
- starpu_tree_insert(tree, curr_obj->logical_index, depth, curr_obj->type == HWLOC_OBJ_PU, curr_obj->arity, father);
- arity = 0;
- for(i = 0; i < curr_obj->arity; i++)
- {
- hwloc_obj_t child = curr_obj->children[i];
- if (child->type == HWLOC_OBJ_BRIDGE && (!child->cpuset || hwloc_bitmap_iszero(child->cpuset)))
- /* I/O stuff, stop caring */
- continue;
- arity++;
- }
- starpu_tree_prepare_children(arity, tree);
- j = 0;
- for(i = 0; i < arity; i++)
- {
- hwloc_obj_t child = curr_obj->children[i];
- if (child->type == HWLOC_OBJ_BRIDGE && (!child->cpuset || hwloc_bitmap_iszero(child->cpuset)))
- /* I/O stuff, stop caring */
- continue;
- #if 0
- char string[128];
- hwloc_obj_snprintf(string, sizeof(string), topology, child, "#", 0);
- printf("%*s%s %d is_pu %d \n", 0, "", string, child->logical_index, child->type == HWLOC_OBJ_PU);
- #endif
- _fill_tree(&tree->nodes[j], child, depth+1, topology, tree);
- j++;
- }
- }
- #endif
- static void _starpu_build_tree(void)
- {
- #ifdef STARPU_HAVE_HWLOC
- struct starpu_tree *tree;
- _STARPU_MALLOC(tree, sizeof(struct starpu_tree));
- _starpu_config.topology.tree = tree;
- hwloc_obj_t root = hwloc_get_root_obj(_starpu_config.topology.hwtopology);
- #if 0
- char string[128];
- hwloc_obj_snprintf(string, sizeof(string), topology, root, "#", 0);
- printf("%*s%s %d is_pu = %d \n", 0, "", string, root->logical_index, root->type == HWLOC_OBJ_PU);
- #endif
- /* level, is_pu, is in the tree (it will be true only after add) */
- _fill_tree(tree, root, 0, _starpu_config.topology.hwtopology, NULL);
- #endif
- }
- static struct sigaction act_sigint;
- static struct sigaction act_sigsegv;
- void _starpu_handler(int sig)
- {
- #ifdef STARPU_VERBOSE
- _STARPU_MSG("Catching signal '%s'\n", sys_siglist[sig]);
- #endif
- #ifdef STARPU_USE_FXT
- _starpu_fxt_dump_file();
- #endif
- if (sig == SIGINT)
- {
- sigaction(SIGINT, &act_sigint, NULL);
- }
- if (sig == SIGSEGV)
- {
- sigaction(SIGSEGV, &act_sigsegv, NULL);
- }
- #ifdef STARPU_VERBOSE
- _STARPU_MSG("Rearming signal '%s'\n", sys_siglist[sig]);
- #endif
- raise(sig);
- }
- void _starpu_catch_signals(void)
- {
- struct sigaction act;
- memset(&act, 0, sizeof(act));
- act.sa_handler = _starpu_handler;
- sigaction(SIGINT, &act, &act_sigint);
- sigaction(SIGSEGV, &act, &act_sigsegv);
- }
- int starpu_init(struct starpu_conf *user_conf)
- {
- return starpu_initialize(user_conf, NULL, NULL);
- }
- int starpu_initialize(struct starpu_conf *user_conf, int *argc, char ***argv)
- {
- int is_a_sink = 0; /* Always defined. If the MP infrastructure is not
- * used, we cannot be a sink. */
- unsigned worker;
- (void)argc;
- (void)argv;
- /* This initializes _starpu_silent, thus needs to be early */
- _starpu_util_init();
- STARPU_HG_DISABLE_CHECKING(_starpu_worker_parallel_blocks);
- #ifdef STARPU_SIMGRID
- /* This initializes the simgrid thread library, thus needs to be early */
- _starpu_simgrid_init_early(argc, argv);
- #endif
- STARPU_PTHREAD_MUTEX_LOCK(&init_mutex);
- while (initialized == CHANGING)
- /* Wait for the other one changing it */
- STARPU_PTHREAD_COND_WAIT(&init_cond, &init_mutex);
- init_count++;
- if (initialized == INITIALIZED)
- {
- /* He initialized it, don't do it again, and let the others get the mutex */
- STARPU_PTHREAD_MUTEX_UNLOCK(&init_mutex);
- return 0;
- }
- /* initialized == UNINITIALIZED */
- initialized = CHANGING;
- STARPU_PTHREAD_MUTEX_UNLOCK(&init_mutex);
- #ifdef STARPU_USE_MP
- _starpu_set_argc_argv(argc, argv);
- #ifdef STARPU_USE_SCC
- /* In SCC case we look at the rank to know if we are a sink */
- if (_starpu_scc_common_mp_init() && !_starpu_scc_common_is_src_node())
- setenv("STARPU_SINK", "STARPU_SCC", 1);
- #endif
- #ifdef STARPU_USE_MPI_MASTER_SLAVE
- if (_starpu_mpi_common_mp_init() == -ENODEV)
- {
- initialized = UNINITIALIZED;
- return -ENODEV;
- }
- /* In MPI case we look at the rank to know if we are a sink */
- if (!_starpu_mpi_common_is_src_node())
- setenv("STARPU_SINK", "STARPU_MPI_MS", 1);
- # endif
- /* If StarPU was configured to use MP sinks, we have to control the
- * kind on node we are running on : host or sink ? */
- if (starpu_getenv("STARPU_SINK"))
- is_a_sink = 1;
- #endif /* STARPU_USE_MP */
- int ret;
- #ifdef STARPU_OPENMP
- _starpu_omp_dummy_init();
- #endif
- #ifdef STARPU_SIMGRID
- /* Warn when the lots of stacks malloc()-ated by simgrid for transfer
- * processes will take a long time to get initialized */
- char *perturb = starpu_getenv("MALLOC_PERTURB_");
- if (perturb && perturb[0] && atoi(perturb) != 0)
- _STARPU_DISP("Warning: MALLOC_PERTURB_ is set to non-zero, this makes simgrid run very slow\n");
- #else
- #ifdef __GNUC__
- #ifndef __OPTIMIZE__
- _STARPU_DISP("Warning: StarPU was configured with --enable-debug (-O0), and is thus not optimized\n");
- #endif
- #endif
- #ifdef STARPU_SPINLOCK_CHECK
- _STARPU_DISP("Warning: StarPU was configured with --enable-spinlock-check, which slows down a bit\n");
- #endif
- #if 0
- #ifndef STARPU_NO_ASSERT
- _STARPU_DISP("Warning: StarPU was configured without --enable-fast\n");
- #endif
- #endif
- #ifdef STARPU_MEMORY_STATS
- _STARPU_DISP("Warning: StarPU was configured with --enable-memory-stats, which slows down a bit\n");
- #endif
- #ifdef STARPU_VERBOSE
- _STARPU_DISP("Warning: StarPU was configured with --enable-verbose, which slows down a bit\n");
- #endif
- #ifdef STARPU_USE_FXT
- _STARPU_DISP("Warning: StarPU was configured with --with-fxt, which slows down a bit, limits scalability and makes worker initialization sequential\n");
- #endif
- #ifdef STARPU_PERF_DEBUG
- _STARPU_DISP("Warning: StarPU was configured with --enable-perf-debug, which slows down a bit\n");
- #endif
- #ifdef STARPU_MODEL_DEBUG
- _STARPU_DISP("Warning: StarPU was configured with --enable-model-debug, which slows down a bit\n");
- #endif
- #ifdef __linux__
- {
- struct utsname buf;
- if (uname(&buf) == 0
- && (!strncmp(buf.release, "4.7.", 4)
- || !strncmp(buf.release, "4.8.", 4)))
- _STARPU_DISP("Warning: This system is running a 4.7 or 4.8 kernel. These have a severe scheduling performance regression issue, please upgrade to at least 4.9.\n");
- }
- #endif
- #endif
- if (starpu_getenv("STARPU_ENABLE_STATS"))
- {
- _STARPU_DISP("Warning: STARPU_ENABLE_STATS is enabled, which slows down a bit\n");
- }
- #if defined(_WIN32) && !defined(__CYGWIN__)
- WSADATA wsadata;
- WSAStartup(MAKEWORD(1,0), &wsadata);
- #endif
- STARPU_AYU_PREINIT();
- /* store the pointer to the user explicit configuration during the
- * initialization */
- if (user_conf == NULL)
- starpu_conf_init(&_starpu_config.conf);
- else
- {
- if (user_conf->magic != 42)
- {
- _STARPU_DISP("starpu_conf structure needs to be initialized with starpu_conf_init\n");
- return -EINVAL;
- }
- _starpu_config.conf = *user_conf;
- }
- _starpu_conf_check_environment(&_starpu_config.conf);
- /* Make a copy of arrays */
- if (_starpu_config.conf.sched_policy_name)
- _starpu_config.conf.sched_policy_name = strdup(_starpu_config.conf.sched_policy_name);
- if (_starpu_config.conf.mic_sink_program_path)
- _starpu_config.conf.mic_sink_program_path = strdup(_starpu_config.conf.mic_sink_program_path);
- if (_starpu_config.conf.n_cuda_opengl_interoperability)
- {
- size_t size = _starpu_config.conf.n_cuda_opengl_interoperability * sizeof(*_starpu_config.conf.cuda_opengl_interoperability);
- unsigned *copy;
- _STARPU_MALLOC(copy, size);
- memcpy(copy, _starpu_config.conf.cuda_opengl_interoperability, size);
- _starpu_config.conf.cuda_opengl_interoperability = copy;
- }
- if (_starpu_config.conf.n_not_launched_drivers)
- {
- size_t size = _starpu_config.conf.n_not_launched_drivers * sizeof(*_starpu_config.conf.not_launched_drivers);
- struct starpu_driver *copy;
- _STARPU_MALLOC(copy, size);
- memcpy(copy, _starpu_config.conf.not_launched_drivers, size);
- _starpu_config.conf.not_launched_drivers = copy;
- }
- _starpu_sched_init();
- _starpu_job_init();
- _starpu_graph_init();
- _starpu_init_all_sched_ctxs(&_starpu_config);
- _starpu_init_progression_hooks();
- _starpu_init_tags();
- #ifdef STARPU_USE_FXT
- _starpu_fxt_init_profiling(_starpu_config.conf.trace_buffer_size);
- #endif
- _starpu_open_debug_logfile();
- _starpu_data_interface_init();
- _starpu_timing_init();
- _starpu_profiling_init();
- _starpu_load_bus_performance_files();
- /* Depending on whether we are a MP sink or not, we must build the
- * topology with MP nodes or not. */
- ret = _starpu_build_topology(&_starpu_config, is_a_sink);
- /* sink doesn't exit even if no worker discorvered */
- if (ret && !is_a_sink)
- {
- starpu_perfmodel_free_sampling_directories();
- STARPU_PTHREAD_MUTEX_LOCK(&init_mutex);
- init_count--;
- _starpu_destroy_machine_config(&_starpu_config);
- #ifdef STARPU_USE_SCC
- if (_starpu_scc_common_is_mp_initialized())
- _starpu_scc_src_mp_deinit();
- #endif
- #ifdef STARPU_USE_MPI_MASTER_SLAVE
- if (_starpu_mpi_common_is_mp_initialized())
- _starpu_mpi_common_mp_deinit();
- #endif
- initialized = UNINITIALIZED;
- /* Let somebody else try to do it */
- STARPU_PTHREAD_COND_SIGNAL(&init_cond);
- STARPU_PTHREAD_MUTEX_UNLOCK(&init_mutex);
- #ifdef STARPU_USE_FXT
- _starpu_stop_fxt_profiling();
- #endif
- return ret;
- }
- _starpu_task_init();
- for (worker = 0; worker < _starpu_config.topology.nworkers; worker++)
- _starpu_worker_init(&_starpu_config.workers[worker], &_starpu_config);
- check_entire_platform = starpu_get_env_number("STARPU_CHECK_ENTIRE_PLATFORM");
- _starpu_config.disable_kernels = starpu_get_env_number("STARPU_DISABLE_KERNELS");
- STARPU_PTHREAD_KEY_CREATE(&_starpu_worker_key, NULL);
- STARPU_PTHREAD_KEY_CREATE(&_starpu_worker_set_key, NULL);
- _starpu_keys_initialized = 1;
- STARPU_WMB();
- _starpu_build_tree();
- if (!is_a_sink)
- {
- struct starpu_sched_policy *selected_policy = _starpu_select_sched_policy(&_starpu_config, _starpu_config.conf.sched_policy_name);
- _starpu_create_sched_ctx(selected_policy, NULL, -1, 1, "init", (_starpu_config.conf.global_sched_ctx_min_priority != -1), _starpu_config.conf.global_sched_ctx_min_priority, (_starpu_config.conf.global_sched_ctx_min_priority != -1), _starpu_config.conf.global_sched_ctx_max_priority, 1, _starpu_config.conf.sched_policy_init, NULL, 0, NULL, 0);
- }
- _starpu_initialize_registered_performance_models();
- #if defined(STARPU_USE_CUDA) || defined(STARPU_SIMGRID)
- _starpu_cuda_init();
- #endif
- #ifdef STARPU_SIMGRID
- _starpu_simgrid_init();
- #endif
- /* Launch "basic" workers (ie. non-combined workers) */
- if (!is_a_sink)
- _starpu_launch_drivers(&_starpu_config);
- /* Allocate swap, if any */
- if (!is_a_sink)
- _starpu_swap_init();
- _starpu_watchdog_init();
- _starpu_profiling_start();
- STARPU_PTHREAD_MUTEX_LOCK(&init_mutex);
- initialized = INITIALIZED;
- /* Tell everybody that we initialized */
- STARPU_PTHREAD_COND_BROADCAST(&init_cond);
- STARPU_PTHREAD_MUTEX_UNLOCK(&init_mutex);
- int main_thread_cpuid = starpu_get_env_number_default("STARPU_MAIN_THREAD_CPUID", -1);
- if (main_thread_cpuid >= 0)
- _starpu_bind_thread_on_cpu(main_thread_cpuid, STARPU_NOWORKERID);
- _STARPU_DEBUG("Initialisation finished\n");
- #ifdef STARPU_USE_MP
- /* Finally, if we are a MP sink, we never leave this function. Else,
- * we enter an infinite event loop which listen for MP commands from
- * the source. */
- if (is_a_sink)
- {
- _starpu_sink_common_worker();
- /* We should normally never leave the loop as we don't want to
- * really initialize STARPU */
- STARPU_ASSERT(0);
- }
- #endif
- _starpu_catch_signals();
- return 0;
- }
- /*
- * Handle runtime termination
- */
- static void _starpu_terminate_workers(struct _starpu_machine_config *pconfig)
- {
- int status = 0;
- unsigned workerid;
- unsigned n;
- starpu_wake_all_blocked_workers();
- for (workerid = 0; workerid < pconfig->topology.nworkers; workerid++)
- {
- _STARPU_DEBUG("wait for worker %u\n", workerid);
- struct _starpu_worker_set *set = pconfig->workers[workerid].set;
- struct _starpu_worker *worker = &pconfig->workers[workerid];
- /* in case StarPU termination code is called from a callback,
- * we have to check if starpu_pthread_self() is the worker itself */
- if (set && set->nworkers > 0)
- {
- if (set->started)
- {
- if (!starpu_pthread_equal(starpu_pthread_self(), set->worker_thread))
- status = starpu_pthread_join(set->worker_thread, NULL);
- if (status)
- {
- #ifdef STARPU_VERBOSE
- _STARPU_DEBUG("starpu_pthread_join -> %d\n", status);
- #endif
- }
- set->started = 0;
- }
- }
- else
- {
- if (!worker->run_by_starpu)
- goto out;
- if (!starpu_pthread_equal(starpu_pthread_self(), worker->worker_thread))
- status = starpu_pthread_join(worker->worker_thread, NULL);
- if (status)
- {
- #ifdef STARPU_VERBOSE
- _STARPU_DEBUG("starpu_pthread_join -> %d\n", status);
- #endif
- }
- }
- out:
- STARPU_ASSERT(starpu_task_list_empty(&worker->local_tasks));
- for (n = 0; n < worker->local_ordered_tasks_size; n++)
- STARPU_ASSERT(worker->local_ordered_tasks[n] == NULL);
- _starpu_sched_ctx_list_delete(&worker->sched_ctx_list);
- free(worker->local_ordered_tasks);
- STARPU_ASSERT(_starpu_ctx_change_list_empty(&worker->ctx_change_list));
- }
- }
- /* Condition variable and mutex used to pause/resume. */
- static starpu_pthread_cond_t pause_cond = STARPU_PTHREAD_COND_INITIALIZER;
- static starpu_pthread_mutex_t pause_mutex = STARPU_PTHREAD_MUTEX_INITIALIZER;
- void _starpu_may_pause(void)
- {
- /* pause_depth is just protected by a memory barrier */
- STARPU_RMB();
- if (STARPU_UNLIKELY(_starpu_config.pause_depth > 0))
- {
- STARPU_PTHREAD_MUTEX_LOCK(&pause_mutex);
- if (_starpu_config.pause_depth > 0)
- {
- STARPU_PTHREAD_COND_WAIT(&pause_cond, &pause_mutex);
- }
- STARPU_PTHREAD_MUTEX_UNLOCK(&pause_mutex);
- }
- }
- void starpu_pause()
- {
- STARPU_HG_DISABLE_CHECKING(_starpu_config.pause_depth);
- _starpu_config.pause_depth += 1;
- }
- void starpu_resume()
- {
- STARPU_PTHREAD_MUTEX_LOCK(&pause_mutex);
- _starpu_config.pause_depth -= 1;
- if (!_starpu_config.pause_depth)
- {
- STARPU_PTHREAD_COND_BROADCAST(&pause_cond);
- }
- STARPU_PTHREAD_MUTEX_UNLOCK(&pause_mutex);
- }
- unsigned _starpu_worker_can_block(unsigned memnode STARPU_ATTRIBUTE_UNUSED, struct _starpu_worker *worker STARPU_ATTRIBUTE_UNUSED)
- {
- #ifdef STARPU_NON_BLOCKING_DRIVERS
- return 0;
- #else
- /* do not block if a sched_ctx change operation is pending */
- if (worker->state_changing_ctx_notice)
- return 0;
- unsigned can_block = 1;
- struct starpu_driver driver;
- driver.type = worker->arch;
- switch (driver.type)
- {
- case STARPU_CPU_WORKER:
- driver.id.cpu_id = worker->devid;
- break;
- case STARPU_CUDA_WORKER:
- driver.id.cuda_id = worker->devid;
- break;
- #ifdef STARPU_USE_OPENCL
- case STARPU_OPENCL_WORKER:
- starpu_opencl_get_device(worker->devid, &driver.id.opencl_id);
- break;
- #endif
- default:
- goto always_launch;
- }
- if (!_starpu_may_launch_driver(&_starpu_config.conf, &driver))
- return 0;
- always_launch:
- #ifndef STARPU_SIMGRID
- if (!_starpu_check_that_no_data_request_exists(memnode))
- can_block = 0;
- #endif
- if (!_starpu_machine_is_running())
- can_block = 0;
- if (!_starpu_execute_registered_progression_hooks())
- can_block = 0;
- return can_block;
- #endif
- }
- static void _starpu_kill_all_workers(struct _starpu_machine_config *pconfig)
- {
- /* set the flag which will tell workers to stop */
- ANNOTATE_HAPPENS_AFTER(&_starpu_config.running);
- pconfig->running = 0;
- /* running is just protected by a memory barrier */
- ANNOTATE_HAPPENS_BEFORE(&_starpu_config.running);
- STARPU_WMB();
- starpu_wake_all_blocked_workers();
- }
- void starpu_display_stats()
- {
- starpu_profiling_bus_helper_display_summary();
- starpu_profiling_worker_helper_display_summary();
- }
- void starpu_shutdown(void)
- {
- unsigned worker;
- STARPU_PTHREAD_MUTEX_LOCK(&init_mutex);
- init_count--;
- STARPU_ASSERT_MSG(init_count >= 0, "Number of calls to starpu_shutdown() can not be higher than the number of calls to starpu_init()\n");
- if (init_count)
- {
- _STARPU_DEBUG("Still somebody needing StarPU, don't deinitialize\n");
- STARPU_PTHREAD_MUTEX_UNLOCK(&init_mutex);
- return;
- }
- /* We're last */
- initialized = CHANGING;
- STARPU_PTHREAD_MUTEX_UNLOCK(&init_mutex);
- /* If the workers are frozen, no progress can be made. */
- STARPU_ASSERT(_starpu_config.pause_depth <= 0);
- starpu_task_wait_for_no_ready();
- /* tell all workers to shutdown */
- _starpu_kill_all_workers(&_starpu_config);
- unsigned i;
- unsigned nb_numa_nodes = starpu_memory_nodes_get_numa_count();
- for (i=0; i<nb_numa_nodes; i++)
- {
- _starpu_free_all_automatically_allocated_buffers(i);
- }
- {
- int stats = starpu_get_env_number("STARPU_STATS");
- if (stats != 0)
- {
- _starpu_display_msi_stats(stderr);
- _starpu_display_alloc_cache_stats(stderr);
- }
- }
- starpu_profiling_bus_helper_display_summary();
- starpu_profiling_worker_helper_display_summary();
- _starpu_deinitialize_registered_performance_models();
- _starpu_watchdog_shutdown();
- /* wait for their termination */
- _starpu_terminate_workers(&_starpu_config);
- {
- int stats = starpu_get_env_number("STARPU_MEMORY_STATS");
- if (stats != 0)
- {
- // Display statistics on data which have not been unregistered
- starpu_data_display_memory_stats();
- }
- }
- _starpu_delete_all_sched_ctxs();
- _starpu_sched_component_workers_destroy();
- _starpu_top_shutdown();
- for (worker = 0; worker < _starpu_config.topology.nworkers; worker++)
- _starpu_worker_deinit(&_starpu_config.workers[worker]);
- _starpu_profiling_terminate();
- _starpu_disk_unregister();
- #ifdef STARPU_HAVE_HWLOC
- starpu_tree_free(_starpu_config.topology.tree);
- free(_starpu_config.topology.tree);
- #endif
- _starpu_destroy_topology(&_starpu_config);
- #ifdef STARPU_USE_FXT
- _starpu_stop_fxt_profiling();
- #endif
- _starpu_data_interface_shutdown();
- _starpu_job_fini();
- /* Drop all remaining tags */
- _starpu_tag_clear();
- #ifdef STARPU_OPENMP
- _starpu_omp_dummy_shutdown();
- #endif
- _starpu_close_debug_logfile();
- _starpu_keys_initialized = 0;
- STARPU_PTHREAD_KEY_DELETE(_starpu_worker_key);
- STARPU_PTHREAD_KEY_DELETE(_starpu_worker_set_key);
- _starpu_task_deinit();
- STARPU_PTHREAD_MUTEX_LOCK(&init_mutex);
- initialized = UNINITIALIZED;
- /* Let someone else that wants to initialize it again do it */
- STARPU_PTHREAD_COND_SIGNAL(&init_cond);
- STARPU_PTHREAD_MUTEX_UNLOCK(&init_mutex);
- /* Clear memory */
- free((char*) _starpu_config.conf.sched_policy_name);
- free(_starpu_config.conf.mic_sink_program_path);
- if (_starpu_config.conf.n_cuda_opengl_interoperability)
- free(_starpu_config.conf.cuda_opengl_interoperability);
- if (_starpu_config.conf.n_not_launched_drivers)
- free(_starpu_config.conf.not_launched_drivers);
- STARPU_AYU_FINISH();
- #ifdef STARPU_USE_SCC
- if (_starpu_scc_common_is_mp_initialized())
- _starpu_scc_src_mp_deinit();
- #endif
- #ifdef STARPU_USE_MPI_MASTER_SLAVE
- if (_starpu_mpi_common_is_mp_initialized())
- _starpu_mpi_common_mp_deinit();
- #endif
- _starpu_print_idle_time();
- _STARPU_DEBUG("Shutdown finished\n");
- #ifdef STARPU_SIMGRID
- /* This finalizes the simgrid thread library, thus needs to be late */
- _starpu_simgrid_deinit();
- #endif
- }
- #undef starpu_worker_get_count
- unsigned starpu_worker_get_count(void)
- {
- return _starpu_config.topology.nworkers;
- }
- unsigned starpu_worker_is_blocked_in_parallel(int workerid)
- {
- if (!_starpu_worker_parallel_blocks)
- return 0;
- int relax_own_observation_state = 0;
- struct _starpu_worker *worker = _starpu_get_worker_struct(workerid);
- STARPU_ASSERT(worker != NULL);
- STARPU_PTHREAD_MUTEX_LOCK_SCHED(&worker->sched_mutex);
- struct _starpu_worker *cur_worker = NULL;
- int cur_workerid = starpu_worker_get_id();
- if (workerid != cur_workerid)
- {
- /* in order to observe the 'blocked' state of a worker from
- * another worker, we must avoid race conditions between
- * 'blocked' state changes and state observations. This is the
- * purpose of this 'if' block. */
- cur_worker = cur_workerid >= 0 ? _starpu_get_worker_struct(cur_workerid) : NULL;
- relax_own_observation_state = (cur_worker != NULL) && (cur_worker->state_relax_refcnt == 0);
- if (relax_own_observation_state && !worker->state_relax_refcnt)
- {
- /* moreover, when a worker (cur_worker != NULL)
- * observes another worker, we need to take special
- * care to avoid live locks, thus the observing worker
- * must enter the relaxed state (if not relaxed
- * already) before doing the observation in mutual
- * exclusion */
- STARPU_PTHREAD_MUTEX_UNLOCK_SCHED(&worker->sched_mutex);
- STARPU_PTHREAD_MUTEX_LOCK_SCHED(&cur_worker->sched_mutex);
- cur_worker->state_relax_refcnt = 1;
- STARPU_PTHREAD_COND_BROADCAST(&cur_worker->sched_cond);
- STARPU_PTHREAD_MUTEX_UNLOCK_SCHED(&cur_worker->sched_mutex);
- STARPU_PTHREAD_MUTEX_LOCK_SCHED(&worker->sched_mutex);
- }
- /* the observer waits for a safe window to observe the state,
- * and also waits for any pending blocking state change
- * requests to be processed, in order to not obtain an
- * ephemeral information */
- while (!worker->state_relax_refcnt
- || worker->state_block_in_parallel_req
- || worker->state_unblock_in_parallel_req)
- {
- STARPU_PTHREAD_COND_WAIT(&worker->sched_cond, &worker->sched_mutex);
- }
- }
- unsigned ret = _starpu_config.workers[workerid].state_blocked_in_parallel;
- /* once a worker state has been observed, the worker is 'tainted' for the next one full sched_op,
- * to avoid changing the observed worker state - on which the observer
- * made a scheduling decision - after the fact. */
- worker->state_blocked_in_parallel_observed = 1;
- STARPU_PTHREAD_MUTEX_UNLOCK_SCHED(&worker->sched_mutex);
- if (relax_own_observation_state)
- {
- STARPU_PTHREAD_MUTEX_LOCK_SCHED(&cur_worker->sched_mutex);
- cur_worker->state_relax_refcnt = 0;
- STARPU_PTHREAD_MUTEX_UNLOCK_SCHED(&cur_worker->sched_mutex);
- }
- return ret;
- }
- unsigned starpu_worker_is_slave_somewhere(int workerid)
- {
- _starpu_worker_lock(workerid);
- unsigned ret = _starpu_config.workers[workerid].is_slave_somewhere;
- _starpu_worker_unlock(workerid);
- return ret;
- }
- int starpu_worker_get_count_by_type(enum starpu_worker_archtype type)
- {
- switch (type)
- {
- case STARPU_CPU_WORKER:
- return _starpu_config.topology.ncpus;
- case STARPU_CUDA_WORKER:
- return _starpu_config.topology.ncudagpus * _starpu_config.topology.nworkerpercuda;
- case STARPU_OPENCL_WORKER:
- return _starpu_config.topology.nopenclgpus;
- case STARPU_MIC_WORKER:
- return _starpu_config.topology.nmicdevices;
- case STARPU_SCC_WORKER:
- return _starpu_config.topology.nsccdevices;
- case STARPU_MPI_MS_WORKER:
- return _starpu_config.topology.nmpidevices;
- case STARPU_ANY_WORKER:
- return _starpu_config.topology.ncpus+
- _starpu_config.topology.ncudagpus * _starpu_config.topology.nworkerpercuda+
- _starpu_config.topology.nopenclgpus+
- _starpu_config.topology.nmicdevices+
- _starpu_config.topology.nsccdevices+
- _starpu_config.topology.nmpidevices;
- default:
- return -EINVAL;
- }
- }
- unsigned starpu_combined_worker_get_count(void)
- {
- return _starpu_config.topology.ncombinedworkers;
- }
- unsigned starpu_cpu_worker_get_count(void)
- {
- return _starpu_config.topology.ncpus;
- }
- unsigned starpu_cuda_worker_get_count(void)
- {
- return _starpu_config.topology.ncudagpus * _starpu_config.topology.nworkerpercuda;
- }
- unsigned starpu_opencl_worker_get_count(void)
- {
- return _starpu_config.topology.nopenclgpus;
- }
- int starpu_asynchronous_copy_disabled(void)
- {
- return _starpu_config.conf.disable_asynchronous_copy;
- }
- int starpu_asynchronous_cuda_copy_disabled(void)
- {
- return _starpu_config.conf.disable_asynchronous_cuda_copy;
- }
- int starpu_asynchronous_opencl_copy_disabled(void)
- {
- return _starpu_config.conf.disable_asynchronous_opencl_copy;
- }
- int starpu_asynchronous_mic_copy_disabled(void)
- {
- return _starpu_config.conf.disable_asynchronous_mic_copy;
- }
- int starpu_asynchronous_mpi_ms_copy_disabled(void)
- {
- return _starpu_config.conf.disable_asynchronous_mpi_ms_copy;
- }
- unsigned starpu_mic_worker_get_count(void)
- {
- int i = 0, count = 0;
- for (i = 0; i < STARPU_MAXMICDEVS; i++)
- count += _starpu_config.topology.nmiccores[i];
- return count;
- }
- unsigned starpu_scc_worker_get_count(void)
- {
- return _starpu_config.topology.nsccdevices;
- }
- unsigned starpu_mpi_ms_worker_get_count(void)
- {
- return _starpu_config.topology.nmpidevices;
- }
- /* When analyzing performance, it is useful to see what is the processing unit
- * that actually performed the task. This function returns the id of the
- * processing unit actually executing it, therefore it makes no sense to use it
- * within the callbacks of SPU functions for instance. If called by some thread
- * that is not controlled by StarPU, starpu_worker_get_id returns -1. */
- #undef starpu_worker_get_id
- int starpu_worker_get_id(void)
- {
- struct _starpu_worker * worker;
- worker = _starpu_get_local_worker_key();
- if (worker)
- {
- return worker->workerid;
- }
- else
- {
- /* there is no worker associated to that thread, perhaps it is
- * a thread from the application or this is some SPU worker */
- return -1;
- }
- }
- #define starpu_worker_get_id _starpu_worker_get_id
- #undef _starpu_worker_get_id_check
- unsigned _starpu_worker_get_id_check(const char *f, int l)
- {
- (void) f;
- (void) l;
- int id = _starpu_worker_get_id();
- STARPU_ASSERT_MSG(id>=0, "%s:%d Cannot be called from outside a worker\n", f, l);
- return id;
- }
- int starpu_combined_worker_get_id(void)
- {
- struct _starpu_worker *worker;
- worker = _starpu_get_local_worker_key();
- if (worker)
- {
- return worker->combined_workerid;
- }
- else
- {
- /* there is no worker associated to that thread, perhaps it is
- * a thread from the application or this is some SPU worker */
- return -1;
- }
- }
- int starpu_combined_worker_get_size(void)
- {
- struct _starpu_worker *worker;
- worker = _starpu_get_local_worker_key();
- if (worker)
- {
- return worker->worker_size;
- }
- else
- {
- /* there is no worker associated to that thread, perhaps it is
- * a thread from the application or this is some SPU worker */
- return -1;
- }
- }
- int starpu_combined_worker_get_rank(void)
- {
- struct _starpu_worker *worker;
- worker = _starpu_get_local_worker_key();
- if (worker)
- {
- return worker->current_rank;
- }
- else
- {
- /* there is no worker associated to that thread, perhaps it is
- * a thread from the application or this is some SPU worker */
- return -1;
- }
- }
- int starpu_worker_get_subworkerid(int id)
- {
- return _starpu_config.workers[id].subworkerid;
- }
- int starpu_worker_get_devid(int id)
- {
- return _starpu_config.workers[id].devid;
- }
- unsigned starpu_worker_is_combined_worker(int id)
- {
- return id >= (int)_starpu_config.topology.nworkers;
- }
- struct _starpu_combined_worker *_starpu_get_combined_worker_struct(unsigned id)
- {
- unsigned basic_worker_count = starpu_worker_get_count();
- //_STARPU_DEBUG("basic_worker_count:%d\n",basic_worker_count);
- STARPU_ASSERT(id >= basic_worker_count);
- return &_starpu_config.combined_workers[id - basic_worker_count];
- }
- enum starpu_worker_archtype starpu_worker_get_type(int id)
- {
- return _starpu_config.workers[id].arch;
- }
- unsigned starpu_worker_get_ids_by_type(enum starpu_worker_archtype type, int *workerids, unsigned maxsize)
- {
- unsigned nworkers = starpu_worker_get_count();
- unsigned cnt = 0;
- unsigned id;
- for (id = 0; id < nworkers; id++)
- {
- if (type == STARPU_ANY_WORKER || starpu_worker_get_type(id) == type)
- {
- /* Perhaps the array is too small ? */
- if (cnt >= maxsize)
- return -ERANGE;
- workerids[cnt++] = id;
- }
- }
- return cnt;
- }
- int starpu_worker_get_by_type(enum starpu_worker_archtype type, int num)
- {
- unsigned nworkers = starpu_worker_get_count();
- int cnt = 0;
- unsigned id;
- for (id = 0; id < nworkers; id++)
- {
- if (type == STARPU_ANY_WORKER || starpu_worker_get_type(id) == type)
- {
- if (num == cnt)
- return id;
- cnt++;
- }
- }
- /* Not found */
- return -1;
- }
- int starpu_worker_get_by_devid(enum starpu_worker_archtype type, int devid)
- {
- unsigned nworkers = starpu_worker_get_count();
- unsigned id;
- for (id = 0; id < nworkers; id++)
- if (starpu_worker_get_type(id) == type && starpu_worker_get_devid(id) == devid)
- return id;
- /* Not found */
- return -1;
- }
- int starpu_worker_get_devids(enum starpu_worker_archtype type, int *devids, int num)
- {
- unsigned nworkers = starpu_worker_get_count();
- int workerids[nworkers];
- unsigned ndevice_workers = starpu_worker_get_ids_by_type(type, workerids, nworkers);
- unsigned ndevids = 0;
- if(ndevice_workers > 0)
- {
- unsigned id, devid;
- int cnt = 0;
- unsigned found = 0;
- for(id = 0; id < ndevice_workers; id++)
- {
- int curr_devid;
- curr_devid = _starpu_config.workers[workerids[id]].devid;
- for(devid = 0; devid < ndevids; devid++)
- {
- if(curr_devid == devids[devid])
- {
- found = 1;
- break;
- }
- }
- if(!found)
- {
- devids[ndevids++] = curr_devid;
- cnt++;
- }
- else
- found = 0;
- if(cnt == num)
- break;
- }
- }
- return ndevids;
- }
- void starpu_worker_get_name(int id, char *dst, size_t maxlen)
- {
- char *name = _starpu_config.workers[id].name;
- snprintf(dst, maxlen, "%s", name);
- }
- int starpu_worker_get_bindid(int workerid)
- {
- return _starpu_config.workers[workerid].bindid;
- }
- int starpu_bindid_get_workerids(int bindid, int **workerids)
- {
- if (bindid >= (int) _starpu_config.nbindid)
- return 0;
- *workerids = _starpu_config.bindid_workers[bindid].workerids;
- return _starpu_config.bindid_workers[bindid].nworkers;
- }
- int starpu_worker_get_stream_workerids(unsigned devid, int *workerids, enum starpu_worker_archtype type)
- {
- unsigned nworkers = starpu_worker_get_count();
- int nw = 0;
- unsigned id;
- for (id = 0; id < nworkers; id++)
- {
- if (_starpu_config.workers[id].devid == devid &&
- (type == STARPU_ANY_WORKER || _starpu_config.workers[id].arch == type))
- workerids[nw++] = id;
- }
- return nw;
- }
- void starpu_worker_get_sched_condition(int workerid, starpu_pthread_mutex_t **sched_mutex, starpu_pthread_cond_t **sched_cond)
- {
- *sched_cond = &_starpu_config.workers[workerid].sched_cond;
- *sched_mutex = &_starpu_config.workers[workerid].sched_mutex;
- }
- /* returns 1 if the call results in initiating a transition of worker WORKERID
- * from sleeping state to awake
- * returns 0 if worker WORKERID is not sleeping or the wake-up transition
- * already has been initiated
- */
- static int starpu_wakeup_worker_locked(int workerid, starpu_pthread_cond_t *sched_cond, starpu_pthread_mutex_t *mutex STARPU_ATTRIBUTE_UNUSED)
- {
- #ifdef STARPU_SIMGRID
- starpu_pthread_queue_broadcast(&_starpu_simgrid_task_queue[workerid]);
- #endif
- if (_starpu_config.workers[workerid].status == STATUS_SCHEDULING)
- {
- _starpu_config.workers[workerid].state_keep_awake = 1;
- return 0;
- }
- else if (_starpu_config.workers[workerid].status == STATUS_SLEEPING)
- {
- int ret = 0;
- if (_starpu_config.workers[workerid].state_keep_awake != 1)
- {
- _starpu_config.workers[workerid].state_keep_awake = 1;
- ret = 1;
- }
- /* cond_broadcast is required over cond_signal since
- * the condition is share for multiple purpose */
- STARPU_PTHREAD_COND_BROADCAST(sched_cond);
- return ret;
- }
- return 0;
- }
- static int starpu_wakeup_worker_no_relax(int workerid, starpu_pthread_cond_t *sched_cond, starpu_pthread_mutex_t *sched_mutex)
- {
- int success;
- STARPU_PTHREAD_MUTEX_LOCK_SCHED(sched_mutex);
- success = starpu_wakeup_worker_locked(workerid, sched_cond, sched_mutex);
- STARPU_PTHREAD_MUTEX_UNLOCK_SCHED(sched_mutex);
- return success;
- }
- int starpu_wake_worker_locked(int workerid)
- {
- starpu_pthread_mutex_t *sched_mutex;
- starpu_pthread_cond_t *sched_cond;
- starpu_worker_get_sched_condition(workerid, &sched_mutex, &sched_cond);
- return starpu_wakeup_worker_locked(workerid, sched_cond, sched_mutex);
- }
- int starpu_wake_worker_no_relax(int workerid)
- {
- starpu_pthread_mutex_t *sched_mutex;
- starpu_pthread_cond_t *sched_cond;
- starpu_worker_get_sched_condition(workerid, &sched_mutex, &sched_cond);
- return starpu_wakeup_worker_no_relax(workerid, sched_cond, sched_mutex);
- }
- int starpu_worker_get_nids_by_type(enum starpu_worker_archtype type, int *workerids, int maxsize)
- {
- unsigned nworkers = starpu_worker_get_count();
- int cnt = 0;
- unsigned id;
- for (id = 0; id < nworkers; id++)
- {
- if (type == STARPU_ANY_WORKER || starpu_worker_get_type(id) == type)
- {
- /* Perhaps the array is too small ? */
- if (cnt >= maxsize)
- return cnt;
- workerids[cnt++] = id;
- }
- }
- return cnt;
- }
- int starpu_worker_get_nids_ctx_free_by_type(enum starpu_worker_archtype type, int *workerids, int maxsize)
- {
- unsigned nworkers = starpu_worker_get_count();
- int cnt = 0;
- unsigned id;
- for (id = 0; id < nworkers; id++)
- {
- if (type == STARPU_ANY_WORKER || starpu_worker_get_type(id) == type)
- {
- /* Perhaps the array is too small ? */
- if (cnt >= maxsize)
- return cnt;
- unsigned found = 0;
- int s;
- for(s = 1; s < STARPU_NMAX_SCHED_CTXS; s++)
- {
- if(_starpu_config.sched_ctxs[s].id != STARPU_NMAX_SCHED_CTXS)
- {
- struct starpu_worker_collection *workers = _starpu_config.sched_ctxs[s].workers;
- struct starpu_sched_ctx_iterator it;
- workers->init_iterator(workers, &it);
- while(workers->has_next(workers, &it))
- {
- unsigned worker = workers->get_next(workers, &it);
- if(worker == id)
- {
- found = 1;
- break;
- }
- }
- if(found)
- break;
- }
- }
- if(!found)
- workerids[cnt++] = id;
- }
- }
- return cnt;
- }
- void starpu_get_version(int *major, int *minor, int *release)
- {
- *major = STARPU_MAJOR_VERSION;
- *minor = STARPU_MINOR_VERSION;
- *release = STARPU_RELEASE_VERSION;
- }
- unsigned starpu_worker_get_sched_ctx_list(int workerid, unsigned **sched_ctxs)
- {
- unsigned s = 0;
- unsigned nsched_ctxs = _starpu_worker_get_nsched_ctxs(workerid);
- _STARPU_MALLOC(*sched_ctxs, nsched_ctxs*sizeof(unsigned));
- struct _starpu_worker *worker = _starpu_get_worker_struct(workerid);
- struct _starpu_sched_ctx_elt *e = NULL;
- struct _starpu_sched_ctx_list_iterator list_it;
- _starpu_sched_ctx_list_iterator_init(worker->sched_ctx_list, &list_it);
- while (_starpu_sched_ctx_list_iterator_has_next(&list_it))
- {
- e = _starpu_sched_ctx_list_iterator_get_next(&list_it);
- (*sched_ctxs)[s++] = e->sched_ctx;
- }
- return nsched_ctxs;
- }
- char *starpu_worker_get_type_as_string(enum starpu_worker_archtype type)
- {
- if (type == STARPU_CPU_WORKER) return "STARPU_CPU_WORKER";
- if (type == STARPU_CUDA_WORKER) return "STARPU_CUDA_WORKER";
- if (type == STARPU_OPENCL_WORKER) return "STARPU_OPENCL_WORKER";
- if (type == STARPU_MIC_WORKER) return "STARPU_MIC_WORKER";
- if (type == STARPU_MPI_MS_WORKER) return "STARPU_MPI_MS_WORKER";
- if (type == STARPU_SCC_WORKER) return "STARPU_SCC_WORKER";
- if (type == STARPU_ANY_WORKER) return "STARPU_ANY_WORKER";
- return "STARPU_unknown_WORKER";
- }
- void _starpu_worker_set_stream_ctx(unsigned workerid, struct _starpu_sched_ctx *sched_ctx)
- {
- STARPU_ASSERT(workerid < starpu_worker_get_count());
- struct _starpu_worker *w = _starpu_get_worker_struct(workerid);
- w->stream_ctx = sched_ctx;
- }
- struct _starpu_sched_ctx* _starpu_worker_get_ctx_stream(unsigned stream_workerid)
- {
- if (stream_workerid >= starpu_worker_get_count())
- return NULL;
- struct _starpu_worker *w = _starpu_get_worker_struct(stream_workerid);
- return w->stream_ctx;
- }
- unsigned starpu_worker_get_sched_ctx_id_stream(unsigned stream_workerid)
- {
- if (stream_workerid >= starpu_worker_get_count())
- return STARPU_NMAX_SCHED_CTXS;
- struct _starpu_worker *w = _starpu_get_worker_struct(stream_workerid);
- return w->stream_ctx != NULL ? w->stream_ctx->id : STARPU_NMAX_SCHED_CTXS;
- }
- void starpu_worker_display_names(FILE *output, enum starpu_worker_archtype type)
- {
- int nworkers = starpu_worker_get_count_by_type(type);
- if (nworkers <= 0)
- {
- fprintf(output, "No %s worker\n", starpu_worker_get_type_as_string(type));
- }
- else
- {
- int i, ids[nworkers];
- starpu_worker_get_ids_by_type(type, ids, nworkers);
- fprintf(output, "%d %s worker%s:\n", nworkers, starpu_worker_get_type_as_string(type), nworkers==1?"":"s");
- for(i = 0; i < nworkers; i++)
- {
- char name[256];
- starpu_worker_get_name(ids[i], name, 256);
- fprintf(output, "\t%s\n", name);
- }
- }
- }
- void _starpu_worker_refuse_task(struct _starpu_worker *worker, struct starpu_task *task)
- {
- if (worker->pipeline_length || worker->arch == STARPU_OPENCL_WORKER)
- {
- int j;
- for (j = 0; j < worker->ntasks; j++)
- {
- const int j_mod = (j+worker->first_task)%STARPU_MAX_PIPELINE;
- if (task == worker->current_tasks[j_mod])
- {
- worker->current_tasks[j_mod] = NULL;
- if (j == 0)
- {
- worker->first_task = (worker->first_task + 1) % STARPU_MAX_PIPELINE;
- worker->current_task = NULL;
- _starpu_set_current_task(NULL);
- }
- break;
- }
- }
- STARPU_ASSERT(j<worker->ntasks);
- }
- else
- {
- worker->current_task = NULL;
- _starpu_set_current_task(NULL);
- }
- worker->ntasks--;
- task->prefetched = 0;
- int res = _starpu_push_task_to_workers(task);
- STARPU_ASSERT_MSG(res == 0, "_starpu_push_task_to_workers() unexpectedly returned = %d\n", res);
- }
- int starpu_worker_sched_op_pending(void)
- {
- return _starpu_worker_sched_op_pending();
- }
- void starpu_worker_relax_on(void)
- {
- _starpu_worker_relax_on();
- }
- void starpu_worker_relax_off(void)
- {
- starpu_worker_relax_off();
- }
- int starpu_worker_get_relax_state(void)
- {
- return _starpu_worker_get_relax_state();
- }
- void starpu_worker_lock(int workerid)
- {
- _starpu_worker_lock(workerid);
- }
- int starpu_worker_trylock(int workerid)
- {
- return _starpu_worker_trylock(workerid);
- }
- void starpu_worker_unlock(int workerid)
- {
- _starpu_worker_unlock(workerid);
- }
- void starpu_worker_lock_self(void)
- {
- _starpu_worker_lock_self();
- }
- void starpu_worker_unlock_self(void)
- {
- _starpu_worker_unlock_self();
- }
- int starpu_wake_worker_relax(int workerid)
- {
- return _starpu_wake_worker_relax(workerid);
- }
- #ifdef STARPU_HAVE_HWLOC
- hwloc_cpuset_t starpu_worker_get_hwloc_cpuset(int workerid)
- {
- struct _starpu_worker *worker = _starpu_get_worker_struct(workerid);
- return hwloc_bitmap_dup(worker->hwloc_cpu_set);
- }
- #endif
- /* Light version of _starpu_wake_worker_relax, which, when possible,
- * speculatively sets keep_awake on the target worker without waiting that
- * worker to enter the relaxed state.
- */
- int _starpu_wake_worker_relax_light(int workerid)
- {
- struct _starpu_worker *worker = _starpu_get_worker_struct(workerid);
- STARPU_ASSERT(worker != NULL);
- int cur_workerid = starpu_worker_get_id();
- if (workerid != cur_workerid)
- {
- _starpu_worker_relax_on();
- STARPU_PTHREAD_MUTEX_LOCK_SCHED(&worker->sched_mutex);
- while (!worker->state_relax_refcnt)
- {
- /* Attempt a fast path if the worker is not really asleep */
- if (_starpu_config.workers[workerid].status == STATUS_SCHEDULING)
- {
- _starpu_config.workers[workerid].state_keep_awake = 1;
- STARPU_PTHREAD_MUTEX_UNLOCK_SCHED(&worker->sched_mutex);
- _starpu_worker_relax_off();
- return 1;
- }
- STARPU_PTHREAD_COND_WAIT(&worker->sched_cond, &worker->sched_mutex);
- }
- }
- else
- {
- STARPU_PTHREAD_MUTEX_LOCK_SCHED(&worker->sched_mutex);
- }
- int ret = starpu_wake_worker_locked(workerid);
- STARPU_PTHREAD_MUTEX_UNLOCK_SCHED(&worker->sched_mutex);
- if (workerid != cur_workerid)
- {
- _starpu_worker_relax_off();
- }
- return ret;
- }
|