advanced-examples.texi 42 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069
  1. @c -*-texinfo-*-
  2. @c This file is part of the StarPU Handbook.
  3. @c Copyright (C) 2009--2011 Universit@'e de Bordeaux 1
  4. @c Copyright (C) 2010, 2011, 2012 Centre National de la Recherche Scientifique
  5. @c Copyright (C) 2011 Institut National de Recherche en Informatique et Automatique
  6. @c See the file starpu.texi for copying conditions.
  7. @menu
  8. * Using multiple implementations of a codelet::
  9. * Enabling implementation according to capabilities::
  10. * Task and Worker Profiling::
  11. * Partitioning Data::
  12. * Performance model example::
  13. * Theoretical lower bound on execution time::
  14. * Insert Task Utility::
  15. * Data reduction::
  16. * Temporary buffers::
  17. * Parallel Tasks::
  18. * Debugging::
  19. * The multiformat interface::
  20. * On-GPU rendering::
  21. * More examples:: More examples shipped with StarPU
  22. @end menu
  23. @node Using multiple implementations of a codelet
  24. @section Using multiple implementations of a codelet
  25. One may want to write multiple implementations of a codelet for a single type of
  26. device and let StarPU choose which one to run. As an example, we will show how
  27. to use SSE to scale a vector. The codelet can be written as follows:
  28. @cartouche
  29. @smallexample
  30. #include <xmmintrin.h>
  31. void scal_sse_func(void *buffers[], void *cl_arg)
  32. @{
  33. float *vector = (float *) STARPU_VECTOR_GET_PTR(buffers[0]);
  34. unsigned int n = STARPU_VECTOR_GET_NX(buffers[0]);
  35. unsigned int n_iterations = n/4;
  36. if (n % 4 != 0)
  37. n_iterations++;
  38. __m128 *VECTOR = (__m128*) vector;
  39. __m128 factor __attribute__((aligned(16)));
  40. factor = _mm_set1_ps(*(float *) cl_arg);
  41. unsigned int i;
  42. for (i = 0; i < n_iterations; i++)
  43. VECTOR[i] = _mm_mul_ps(factor, VECTOR[i]);
  44. @}
  45. @end smallexample
  46. @end cartouche
  47. @cartouche
  48. @smallexample
  49. struct starpu_codelet cl = @{
  50. .where = STARPU_CPU,
  51. .cpu_funcs = @{ scal_cpu_func, scal_sse_func, NULL @},
  52. .nbuffers = 1,
  53. .modes = @{ STARPU_RW @}
  54. @};
  55. @end smallexample
  56. @end cartouche
  57. Schedulers which are multi-implementation aware (only @code{dmda}, @code{heft}
  58. and @code{pheft} for now) will use the performance models of all the
  59. implementations it was given, and pick the one that seems to be the fastest.
  60. @node Enabling implementation according to capabilities
  61. @section Enabling implementation according to capabilities
  62. Some implementations may not run on some devices. For instance, some CUDA
  63. devices do not support double floating point precision, and thus the kernel
  64. execution would just fail; or the device may not have enough shared memory for
  65. the implementation being used. The @code{can_execute} field of the @code{struct
  66. starpu_codelet} structure permits to express this. For instance:
  67. @cartouche
  68. @smallexample
  69. static int can_execute(unsigned workerid, struct starpu_task *task, unsigned nimpl)
  70. @{
  71. const struct cudaDeviceProp *props;
  72. if (starpu_worker_get_type(workerid) == STARPU_CPU_WORKER)
  73. return 1;
  74. /* Cuda device */
  75. props = starpu_cuda_get_device_properties(workerid);
  76. if (props->major >= 2 || props->minor >= 3)
  77. /* At least compute capability 1.3, supports doubles */
  78. return 1;
  79. /* Old card, does not support doubles */
  80. return 0;
  81. @}
  82. struct starpu_codelet cl = @{
  83. .where = STARPU_CPU|STARPU_CUDA,
  84. .can_execute = can_execute,
  85. .cpu_funcs = @{ cpu_func, NULL @},
  86. .cuda_funcs = @{ gpu_func, NULL @}
  87. .nbuffers = 1,
  88. .modes = @{ STARPU_RW @}
  89. @};
  90. @end smallexample
  91. @end cartouche
  92. This can be essential e.g. when running on a machine which mixes various models
  93. of CUDA devices, to take benefit from the new models without crashing on old models.
  94. Note: the @code{can_execute} function is called by the scheduler each time it
  95. tries to match a task with a worker, and should thus be very fast. The
  96. @code{starpu_cuda_get_device_properties} provides a quick access to CUDA
  97. properties of CUDA devices to achieve such efficiency.
  98. Another example is compiling CUDA code for various compute capabilities,
  99. resulting with two CUDA functions, e.g. @code{scal_gpu_13} for compute capability
  100. 1.3, and @code{scal_gpu_20} for compute capability 2.0. Both functions can be
  101. provided to StarPU by using @code{cuda_funcs}, and @code{can_execute} can then be
  102. used to rule out the @code{scal_gpu_20} variant on a CUDA device which
  103. will not be able to execute it:
  104. @cartouche
  105. @smallexample
  106. static int can_execute(unsigned workerid, struct starpu_task *task, unsigned nimpl)
  107. @{
  108. const struct cudaDeviceProp *props;
  109. if (starpu_worker_get_type(workerid) == STARPU_CPU_WORKER)
  110. return 1;
  111. /* Cuda device */
  112. if (nimpl == 0)
  113. /* Trying to execute the 1.3 capability variant, we assume it is ok in all cases. */
  114. return 1;
  115. /* Trying to execute the 2.0 capability variant, check that the card can do it. */
  116. props = starpu_cuda_get_device_properties(workerid);
  117. if (props->major >= 2 || props->minor >= 0)
  118. /* At least compute capability 2.0, can run it */
  119. return 1;
  120. /* Old card, does not support 2.0, will not be able to execute the 2.0 variant. */
  121. return 0;
  122. @}
  123. struct starpu_codelet cl = @{
  124. .where = STARPU_CPU|STARPU_CUDA,
  125. .can_execute = can_execute,
  126. .cpu_funcs = @{ cpu_func, NULL @},
  127. .cuda_funcs = @{ scal_gpu_13, scal_gpu_20, NULL @},
  128. .nbuffers = 1,
  129. .modes = @{ STARPU_RW @}
  130. @};
  131. @end smallexample
  132. @end cartouche
  133. Note: the most generic variant should be provided first, as some schedulers are
  134. not able to try the different variants.
  135. @node Task and Worker Profiling
  136. @section Task and Worker Profiling
  137. A full example showing how to use the profiling API is available in
  138. the StarPU sources in the directory @code{examples/profiling/}.
  139. @cartouche
  140. @smallexample
  141. struct starpu_task *task = starpu_task_create();
  142. task->cl = &cl;
  143. task->synchronous = 1;
  144. /* We will destroy the task structure by hand so that we can
  145. * query the profiling info before the task is destroyed. */
  146. task->destroy = 0;
  147. /* Submit and wait for completion (since synchronous was set to 1) */
  148. starpu_task_submit(task);
  149. /* The task is finished, get profiling information */
  150. struct starpu_task_profiling_info *info = task->profiling_info;
  151. /* How much time did it take before the task started ? */
  152. double delay += starpu_timing_timespec_delay_us(&info->submit_time, &info->start_time);
  153. /* How long was the task execution ? */
  154. double length += starpu_timing_timespec_delay_us(&info->start_time, &info->end_time);
  155. /* We don't need the task structure anymore */
  156. starpu_task_destroy(task);
  157. @end smallexample
  158. @end cartouche
  159. @cartouche
  160. @smallexample
  161. /* Display the occupancy of all workers during the test */
  162. int worker;
  163. for (worker = 0; worker < starpu_worker_get_count(); worker++)
  164. @{
  165. struct starpu_worker_profiling_info worker_info;
  166. int ret = starpu_worker_get_profiling_info(worker, &worker_info);
  167. STARPU_ASSERT(!ret);
  168. double total_time = starpu_timing_timespec_to_us(&worker_info.total_time);
  169. double executing_time = starpu_timing_timespec_to_us(&worker_info.executing_time);
  170. double sleeping_time = starpu_timing_timespec_to_us(&worker_info.sleeping_time);
  171. float executing_ratio = 100.0*executing_time/total_time;
  172. float sleeping_ratio = 100.0*sleeping_time/total_time;
  173. char workername[128];
  174. starpu_worker_get_name(worker, workername, 128);
  175. fprintf(stderr, "Worker %s:\n", workername);
  176. fprintf(stderr, "\ttotal time: %.2lf ms\n", total_time*1e-3);
  177. fprintf(stderr, "\texec time: %.2lf ms (%.2f %%)\n", executing_time*1e-3,
  178. executing_ratio);
  179. fprintf(stderr, "\tblocked time: %.2lf ms (%.2f %%)\n", sleeping_time*1e-3,
  180. sleeping_ratio);
  181. @}
  182. @end smallexample
  183. @end cartouche
  184. @node Partitioning Data
  185. @section Partitioning Data
  186. An existing piece of data can be partitioned in sub parts to be used by different tasks, for instance:
  187. @cartouche
  188. @smallexample
  189. int vector[NX];
  190. starpu_data_handle_t handle;
  191. /* Declare data to StarPU */
  192. starpu_vector_data_register(&handle, 0, (uintptr_t)vector, NX, sizeof(vector[0]));
  193. /* Partition the vector in PARTS sub-vectors */
  194. starpu_filter f =
  195. @{
  196. .filter_func = starpu_block_filter_func_vector,
  197. .nchildren = PARTS
  198. @};
  199. starpu_data_partition(handle, &f);
  200. @end smallexample
  201. @end cartouche
  202. The task submission then uses @code{starpu_data_get_sub_data} to retrieve the
  203. sub-handles to be passed as tasks parameters.
  204. @cartouche
  205. @smallexample
  206. /* Submit a task on each sub-vector */
  207. for (i=0; i<starpu_data_get_nb_children(handle); i++) @{
  208. /* Get subdata number i (there is only 1 dimension) */
  209. starpu_data_handle_t sub_handle = starpu_data_get_sub_data(handle, 1, i);
  210. struct starpu_task *task = starpu_task_create();
  211. task->handles[0] = sub_handle;
  212. task->cl = &cl;
  213. task->synchronous = 1;
  214. task->cl_arg = &factor;
  215. task->cl_arg_size = sizeof(factor);
  216. starpu_task_submit(task);
  217. @}
  218. @end smallexample
  219. @end cartouche
  220. Partitioning can be applied several times, see
  221. @code{examples/basic_examples/mult.c} and @code{examples/filters/}.
  222. Wherever the whole piece of data is already available, the partitioning will
  223. be done in-place, i.e. without allocating new buffers but just using pointers
  224. inside the existing copy. This is particularly important to be aware of when
  225. using OpenCL, where the kernel parameters are not pointers, but handles. The
  226. kernel thus needs to be also passed the offset within the OpenCL buffer:
  227. @cartouche
  228. @smallexample
  229. void opencl_func(void *buffers[], void *cl_arg)
  230. @{
  231. cl_mem vector = (cl_mem) STARPU_VECTOR_GET_DEV_HANDLE(buffers[0]);
  232. unsigned offset = STARPU_BLOCK_GET_OFFSET(buffers[0]);
  233. ...
  234. clSetKernelArg(kernel, 0, sizeof(vector), &vector);
  235. clSetKernelArg(kernel, 1, sizeof(offset), &offset);
  236. ...
  237. @}
  238. @end smallexample
  239. @end cartouche
  240. And the kernel has to shift from the pointer passed by the OpenCL driver:
  241. @cartouche
  242. @smallexample
  243. __kernel void opencl_kernel(__global int *vector, unsigned offset)
  244. @{
  245. block = (__global void *)block + offset;
  246. ...
  247. @}
  248. @end smallexample
  249. @end cartouche
  250. @node Performance model example
  251. @section Performance model example
  252. To achieve good scheduling, StarPU scheduling policies need to be able to
  253. estimate in advance the duration of a task. This is done by giving to codelets
  254. a performance model, by defining a @code{starpu_perfmodel} structure and
  255. providing its address in the @code{model} field of the @code{struct starpu_codelet}
  256. structure. The @code{symbol} and @code{type} fields of @code{starpu_perfmodel}
  257. are mandatory, to give a name to the model, and the type of the model, since
  258. there are several kinds of performance models. For compatibility, make sure to
  259. initialize the whole structure to zero, either by using explicit memset, or by
  260. letting the compiler implicitly do it as examplified below.
  261. @itemize
  262. @item
  263. Measured at runtime (@code{STARPU_HISTORY_BASED} model type). This assumes that for a
  264. given set of data input/output sizes, the performance will always be about the
  265. same. This is very true for regular kernels on GPUs for instance (<0.1% error),
  266. and just a bit less true on CPUs (~=1% error). This also assumes that there are
  267. few different sets of data input/output sizes. StarPU will then keep record of
  268. the average time of previous executions on the various processing units, and use
  269. it as an estimation. History is done per task size, by using a hash of the input
  270. and ouput sizes as an index.
  271. It will also save it in @code{~/.starpu/sampling/codelets}
  272. for further executions, and can be observed by using the
  273. @code{starpu_perfmodel_display} command, or drawn by using
  274. the @code{starpu_perfmodel_plot} (@pxref{Performance model calibration}). The
  275. models are indexed by machine name. To
  276. share the models between machines (e.g. for a homogeneous cluster), use
  277. @code{export STARPU_HOSTNAME=some_global_name}. Measurements are only done when using a task scheduler which makes use of it, such as @code{heft} or @code{dmda}.
  278. The following is a small code example.
  279. If e.g. the code is recompiled with other compilation options, or several
  280. variants of the code are used, the symbol string should be changed to reflect
  281. that, in order to recalibrate a new model from zero. The symbol string can even
  282. be constructed dynamically at execution time, as long as this is done before
  283. submitting any task using it.
  284. @cartouche
  285. @smallexample
  286. static struct starpu_perfmodel mult_perf_model = @{
  287. .type = STARPU_HISTORY_BASED,
  288. .symbol = "mult_perf_model"
  289. @};
  290. struct starpu_codelet cl = @{
  291. .where = STARPU_CPU,
  292. .cpu_funcs = @{ cpu_mult, NULL @},
  293. .nbuffers = 3,
  294. .modes = @{ STARPU_R, STARPU_R, STARPU_W @},
  295. /* for the scheduling policy to be able to use performance models */
  296. .model = &mult_perf_model
  297. @};
  298. @end smallexample
  299. @end cartouche
  300. @item
  301. Measured at runtime and refined by regression (@code{STARPU_*REGRESSION_BASED}
  302. model type). This still assumes performance regularity, but works
  303. with various data input sizes, by applying regression over observed
  304. execution times. STARPU_REGRESSION_BASED uses an a*n^b regression
  305. form, STARPU_NL_REGRESSION_BASED uses an a*n^b+c (more precise than
  306. STARPU_REGRESSION_BASED, but costs a lot more to compute).
  307. For instance,
  308. @code{tests/perfmodels/regression_based.c} uses a regression-based performance
  309. model for the @code{memset} operation.
  310. Of course, the application has to issue
  311. tasks with varying size so that the regression can be computed. StarPU will not
  312. trust the regression unless there is at least 10% difference between the minimum
  313. and maximum observed input size. It can be useful to set the
  314. @code{STARPU_CALIBRATE} environment variable to @code{1} and run the application
  315. on varying input sizes, so as to feed the performance model for a variety of
  316. inputs. The @code{starpu_perfmodel_display} and @code{starpu_perfmodel_plot}
  317. tools can be used to observe how much the performance model is calibrated (@pxref{Performance model calibration}); when
  318. their output look good, @code{STARPU_CALIBRATE} can be reset to @code{0} to let
  319. StarPU use the resulting performance model without recording new measures. If
  320. the data input sizes vary a lot, it is really important to set
  321. @code{STARPU_CALIBRATE} to @code{0}, otherwise StarPU will continue adding the
  322. measures, and result with a very big performance model, which will take time a
  323. lot of time to load and save.
  324. For non-linear regression, since computing it
  325. is quite expensive, it is only done at termination of the application. This
  326. means that the first execution of the application will use only history-based
  327. performance model to perform scheduling, without using regression.
  328. @item
  329. Provided as an estimation from the application itself (@code{STARPU_COMMON} model type and @code{cost_function} field),
  330. see for instance
  331. @code{examples/common/blas_model.h} and @code{examples/common/blas_model.c}.
  332. @item
  333. Provided explicitly by the application (@code{STARPU_PER_ARCH} model type): the
  334. @code{.per_arch[arch][nimpl].cost_function} fields have to be filled with pointers to
  335. functions which return the expected duration of the task in micro-seconds, one
  336. per architecture.
  337. @end itemize
  338. For the @code{STARPU_HISTORY_BASED} and @code{STARPU_*REGRESSION_BASE},
  339. the total size of task data (both input and output) is used as an index by
  340. default. The @code{size_base} field of @code{struct starpu_perfmodel} however
  341. permits the application to override that, when for instance some of the data
  342. do not matter for task cost (e.g. mere reference table), or when using sparse
  343. structures (in which case it is the number of non-zeros which matter), or when
  344. there is some hidden parameter such as the number of iterations, etc.
  345. How to use schedulers which can benefit from such performance model is explained
  346. in @ref{Task scheduling policy}.
  347. The same can be done for task power consumption estimation, by setting the
  348. @code{power_model} field the same way as the @code{model} field. Note: for
  349. now, the application has to give to the power consumption performance model
  350. a name which is different from the execution time performance model.
  351. The application can request time estimations from the StarPU performance
  352. models by filling a task structure as usual without actually submitting
  353. it. The data handles can be created by calling @code{starpu_data_register}
  354. functions with a @code{NULL} pointer (and need to be unregistered as usual)
  355. and the desired data sizes. The @code{starpu_task_expected_length} and
  356. @code{starpu_task_expected_power} functions can then be called to get an
  357. estimation of the task duration on a given arch. @code{starpu_task_destroy}
  358. needs to be called to destroy the dummy task afterwards. See
  359. @code{tests/perfmodels/regression_based.c} for an example.
  360. @node Theoretical lower bound on execution time
  361. @section Theoretical lower bound on execution time
  362. For kernels with history-based performance models, StarPU can very easily provide a theoretical lower
  363. bound for the execution time of a whole set of tasks. See for
  364. instance @code{examples/lu/lu_example.c}: before submitting tasks,
  365. call @code{starpu_bound_start}, and after complete execution, call
  366. @code{starpu_bound_stop}. @code{starpu_bound_print_lp} or
  367. @code{starpu_bound_print_mps} can then be used to output a Linear Programming
  368. problem corresponding to the schedule of your tasks. Run it through
  369. @code{lp_solve} or any other linear programming solver, and that will give you a
  370. lower bound for the total execution time of your tasks. If StarPU was compiled
  371. with the glpk library installed, @code{starpu_bound_compute} can be used to
  372. solve it immediately and get the optimized minimum, in ms. Its @code{integer}
  373. parameter allows to decide whether integer resolution should be computed
  374. and returned too.
  375. The @code{deps} parameter tells StarPU whether to take tasks and implicit data
  376. dependencies into account. It must be understood that the linear programming
  377. problem size is quadratic with the number of tasks and thus the time to solve it
  378. will be very long, it could be minutes for just a few dozen tasks. You should
  379. probably use @code{lp_solve -timeout 1 test.pl -wmps test.mps} to convert the
  380. problem to MPS format and then use a better solver, @code{glpsol} might be
  381. better than @code{lp_solve} for instance (the @code{--pcost} option may be
  382. useful), but sometimes doesn't manage to converge. @code{cbc} might look
  383. slower, but it is parallel. Be sure to try at least all the @code{-B} options
  384. of @code{lp_solve}. For instance, we often just use
  385. @code{lp_solve -cc -B1 -Bb -Bg -Bp -Bf -Br -BG -Bd -Bs -BB -Bo -Bc -Bi} , and
  386. the @code{-gr} option can also be quite useful.
  387. Setting @code{deps} to 0 will only take into account the actual computations
  388. on processing units. It however still properly takes into account the varying
  389. performances of kernels and processing units, which is quite more accurate than
  390. just comparing StarPU performances with the fastest of the kernels being used.
  391. The @code{prio} parameter tells StarPU whether to simulate taking into account
  392. the priorities as the StarPU scheduler would, i.e. schedule prioritized
  393. tasks before less prioritized tasks, to check to which extend this results
  394. to a less optimal solution. This increases even more computation time.
  395. Note that for simplicity, all this however doesn't take into account data
  396. transfers, which are assumed to be completely overlapped.
  397. @node Insert Task Utility
  398. @section Insert Task Utility
  399. StarPU provides the wrapper function @code{starpu_insert_task} to ease
  400. the creation and submission of tasks.
  401. @deftypefun int starpu_insert_task (struct starpu_codelet *@var{cl}, ...)
  402. Create and submit a task corresponding to @var{cl} with the following
  403. arguments. The argument list must be zero-terminated.
  404. The arguments following the codelets can be of the following types:
  405. @itemize
  406. @item
  407. @code{STARPU_R}, @code{STARPU_W}, @code{STARPU_RW}, @code{STARPU_SCRATCH}, @code{STARPU_REDUX} an access mode followed by a data handle;
  408. @item
  409. the specific values @code{STARPU_VALUE}, @code{STARPU_CALLBACK},
  410. @code{STARPU_CALLBACK_ARG}, @code{STARPU_CALLBACK_WITH_ARG},
  411. @code{STARPU_PRIORITY}, followed by the appropriated objects as
  412. defined below.
  413. @end itemize
  414. Parameters to be passed to the codelet implementation are defined
  415. through the type @code{STARPU_VALUE}. The function
  416. @code{starpu_codelet_unpack_args} must be called within the codelet
  417. implementation to retrieve them.
  418. @end deftypefun
  419. @defmac STARPU_VALUE
  420. this macro is used when calling @code{starpu_insert_task}, and must be
  421. followed by a pointer to a constant value and the size of the constant
  422. @end defmac
  423. @defmac STARPU_CALLBACK
  424. this macro is used when calling @code{starpu_insert_task}, and must be
  425. followed by a pointer to a callback function
  426. @end defmac
  427. @defmac STARPU_CALLBACK_ARG
  428. this macro is used when calling @code{starpu_insert_task}, and must be
  429. followed by a pointer to be given as an argument to the callback
  430. function
  431. @end defmac
  432. @defmac STARPU_CALLBACK_WITH_ARG
  433. this macro is used when calling @code{starpu_insert_task}, and must be
  434. followed by two pointers: one to a callback function, and the other to
  435. be given as an argument to the callback function; this is equivalent
  436. to using both @code{STARPU_CALLBACK} and
  437. @code{STARPU_CALLBACK_WITH_ARG}
  438. @end defmac
  439. @defmac STARPU_PRIORITY
  440. this macro is used when calling @code{starpu_insert_task}, and must be
  441. followed by a integer defining a priority level
  442. @end defmac
  443. @deftypefun void starpu_codelet_pack_args ({char **}@var{arg_buffer}, {size_t *}@var{arg_buffer_size}, ...)
  444. Pack arguments of type @code{STARPU_VALUE} into a buffer which can be
  445. given to a codelet and later unpacked with the function
  446. @code{starpu_codelet_unpack_args} defined below.
  447. @end deftypefun
  448. @deftypefun void starpu_codelet_unpack_args ({void *}@var{cl_arg}, ...)
  449. Retrieve the arguments of type @code{STARPU_VALUE} associated to a
  450. task automatically created using the function
  451. @code{starpu_insert_task} defined above.
  452. @end deftypefun
  453. Here the implementation of the codelet:
  454. @smallexample
  455. void func_cpu(void *descr[], void *_args)
  456. @{
  457. int *x0 = (int *)STARPU_VARIABLE_GET_PTR(descr[0]);
  458. float *x1 = (float *)STARPU_VARIABLE_GET_PTR(descr[1]);
  459. int ifactor;
  460. float ffactor;
  461. starpu_codelet_unpack_args(_args, &ifactor, &ffactor);
  462. *x0 = *x0 * ifactor;
  463. *x1 = *x1 * ffactor;
  464. @}
  465. struct starpu_codelet mycodelet = @{
  466. .where = STARPU_CPU,
  467. .cpu_funcs = @{ func_cpu, NULL @},
  468. .nbuffers = 2,
  469. .modes = @{ STARPU_RW, STARPU_RW @}
  470. @};
  471. @end smallexample
  472. And the call to the @code{starpu_insert_task} wrapper:
  473. @smallexample
  474. starpu_insert_task(&mycodelet,
  475. STARPU_VALUE, &ifactor, sizeof(ifactor),
  476. STARPU_VALUE, &ffactor, sizeof(ffactor),
  477. STARPU_RW, data_handles[0], STARPU_RW, data_handles[1],
  478. 0);
  479. @end smallexample
  480. The call to @code{starpu_insert_task} is equivalent to the following
  481. code:
  482. @smallexample
  483. struct starpu_task *task = starpu_task_create();
  484. task->cl = &mycodelet;
  485. task->handles[0] = data_handles[0];
  486. task->handles[1] = data_handles[1];
  487. char *arg_buffer;
  488. size_t arg_buffer_size;
  489. starpu_codelet_pack_args(&arg_buffer, &arg_buffer_size,
  490. STARPU_VALUE, &ifactor, sizeof(ifactor),
  491. STARPU_VALUE, &ffactor, sizeof(ffactor),
  492. 0);
  493. task->cl_arg = arg_buffer;
  494. task->cl_arg_size = arg_buffer_size;
  495. int ret = starpu_task_submit(task);
  496. @end smallexample
  497. If some part of the task insertion depends on the value of some computation,
  498. the @code{STARPU_DATA_ACQUIRE_CB} macro can be very convenient. For
  499. instance, assuming that the index variable @code{i} was registered as handle
  500. @code{i_handle}:
  501. @smallexample
  502. /* Compute which portion we will work on, e.g. pivot */
  503. starpu_insert_task(&which_index, STARPU_W, i_handle, 0);
  504. /* And submit the corresponding task */
  505. STARPU_DATA_ACQUIRE_CB(i_handle, STARPU_R, starpu_insert_task(&work, STARPU_RW, A_handle[i], 0));
  506. @end smallexample
  507. The @code{STARPU_DATA_ACQUIRE_CB} macro submits an asynchronous request for
  508. acquiring data @code{i} for the main application, and will execute the code
  509. given as third parameter when it is acquired. In other words, as soon as the
  510. value of @code{i} computed by the @code{which_index} codelet can be read, the
  511. portion of code passed as third parameter of @code{STARPU_DATA_ACQUIRE_CB} will
  512. be executed, and is allowed to read from @code{i} to use it e.g. as an
  513. index. Note that this macro is only avaible when compiling StarPU with
  514. the compiler @code{gcc}.
  515. @node Data reduction
  516. @section Data reduction
  517. In various cases, some piece of data is used to accumulate intermediate
  518. results. For instances, the dot product of a vector, maximum/minimum finding,
  519. the histogram of a photograph, etc. When these results are produced along the
  520. whole machine, it would not be efficient to accumulate them in only one place,
  521. incurring data transmission each and access concurrency.
  522. StarPU provides a @code{STARPU_REDUX} mode, which permits to optimize
  523. that case: it will allocate a buffer on each memory node, and accumulate
  524. intermediate results there. When the data is eventually accessed in the normal
  525. @code{STARPU_R} mode, StarPU will collect the intermediate results in just one
  526. buffer.
  527. For this to work, the user has to use the
  528. @code{starpu_data_set_reduction_methods} to declare how to initialize these
  529. buffers, and how to assemble partial results.
  530. For instance, @code{cg} uses that to optimize its dot product: it first defines
  531. the codelets for initialization and reduction:
  532. @smallexample
  533. struct starpu_codelet bzero_variable_cl =
  534. @{
  535. .cpu_funcs = @{ bzero_variable_cpu, NULL @},
  536. .cuda_funcs = @{ bzero_variable_cuda, NULL @},
  537. .nbuffers = 1,
  538. @}
  539. static void accumulate_variable_cpu(void *descr[], void *cl_arg)
  540. @{
  541. double *v_dst = (double *)STARPU_VARIABLE_GET_PTR(descr[0]);
  542. double *v_src = (double *)STARPU_VARIABLE_GET_PTR(descr[1]);
  543. *v_dst = *v_dst + *v_src;
  544. @}
  545. static void accumulate_variable_cuda(void *descr[], void *cl_arg)
  546. @{
  547. double *v_dst = (double *)STARPU_VARIABLE_GET_PTR(descr[0]);
  548. double *v_src = (double *)STARPU_VARIABLE_GET_PTR(descr[1]);
  549. cublasaxpy(1, (double)1.0, v_src, 1, v_dst, 1);
  550. cudaStreamSynchronize(starpu_cuda_get_local_stream());
  551. @}
  552. struct starpu_codelet accumulate_variable_cl =
  553. @{
  554. .cpu_funcs = @{ accumulate_variable_cpu, NULL @},
  555. .cuda_funcs = @{ accumulate_variable_cuda, NULL @},
  556. .nbuffers = 1,
  557. @}
  558. @end smallexample
  559. and attaches them as reduction methods for its dtq handle:
  560. @smallexample
  561. starpu_data_set_reduction_methods(dtq_handle,
  562. &accumulate_variable_cl, &bzero_variable_cl);
  563. @end smallexample
  564. and dtq_handle can now be used in @code{STARPU_REDUX} mode for the dot products
  565. with partitioned vectors:
  566. @smallexample
  567. int dots(starpu_data_handle_t v1, starpu_data_handle_t v2,
  568. starpu_data_handle_t s, unsigned nblocks)
  569. @{
  570. starpu_insert_task(&bzero_variable_cl, STARPU_W, s, 0);
  571. for (b = 0; b < nblocks; b++)
  572. starpu_insert_task(&dot_kernel_cl,
  573. STARPU_RW, s,
  574. STARPU_R, starpu_data_get_sub_data(v1, 1, b),
  575. STARPU_R, starpu_data_get_sub_data(v2, 1, b),
  576. 0);
  577. @}
  578. @end smallexample
  579. The @code{cg} example also uses reduction for the blocked gemv kernel, leading
  580. to yet more relaxed dependencies and more parallelism.
  581. @node Temporary buffers
  582. @section Temporary buffers
  583. There are two kinds of temporary buffers: temporary data which just pass results
  584. from a task to another, and scratch data which are needed only internally by
  585. tasks.
  586. @subsection Temporary data
  587. Data can sometimes be entirely produced by a task, and entirely consumed by
  588. another task, without the need for other parts of the application to access
  589. it. In such case, registration can be done without prior allocation, by using
  590. the special -1 memory node number, and passing a zero pointer. StarPU will
  591. actually allocate memory only when the task creating the content gets scheduled,
  592. and destroy it on unregistration.
  593. In addition to that, it can be tedious for the application to have to unregister
  594. the data, since it will not use its content anyway. The unregistration can be
  595. done lazily by using the @code{starpu_data_unregister_lazy(handle)} function,
  596. which will record that no more tasks accessing the handle will be submitted, so
  597. that it can be freed as soon as the last task accessing it is over.
  598. The following code examplifies both points: it registers the temporary
  599. data, submits three tasks accessing it, and records the data for automatic
  600. unregistration.
  601. @smallexample
  602. starpu_vector_data_register(&handle, -1, 0, n, sizeof(float));
  603. starpu_insert_task(&produce_data, STARPU_W, handle, 0);
  604. starpu_insert_task(&compute_data, STARPU_RW, handle, 0);
  605. starpu_insert_task(&summarize_data, STARPU_R, handle, STARPU_W, result_handle, 0);
  606. starpu_data_unregister_lazy(handle);
  607. @end smallexample
  608. @subsection Scratch data
  609. Some kernels sometimes need temporary data to achieve the computations, i.e. a
  610. workspace. The application could allocate it at the start of the codelet
  611. function, and free it at the end, but that would be costly. It could also
  612. allocate one buffer per worker (similarly to @ref{Per-worker library
  613. initialization }), but that would make them systematic and permanent. A more
  614. optimized way is to use the SCRATCH data access mode, as examplified below,
  615. which provides per-worker buffers without content consistency.
  616. @smallexample
  617. starpu_vector_data_register(&workspace, -1, 0, sizeof(float));
  618. for (i = 0; i < N; i++)
  619. starpu_insert_task(&compute, STARPU_R, input[i], STARPU_SCRATCH, workspace, STARPU_W, output[i], 0);
  620. @end smallexample
  621. StarPU will make sure that the buffer is allocated before executing the task,
  622. and make this allocation per-worker: for CPU workers, notably, each worker has
  623. its own buffer. This means that each task submitted above will actually have its
  624. own workspace, which will actually be the same for all tasks running one after
  625. the other on the same worker. Also, if for instance GPU memory becomes scarce,
  626. StarPU will notice that it can free such buffers easily, since the content does
  627. not matter.
  628. @node Parallel Tasks
  629. @section Parallel Tasks
  630. StarPU can leverage existing parallel computation libraries by the means of
  631. parallel tasks. A parallel task is a task which gets worked on by a set of CPUs
  632. (called a parallel or combined worker) at the same time, by using an existing
  633. parallel CPU implementation of the computation to be achieved. This can also be
  634. useful to improve the load balance between slow CPUs and fast GPUs: since CPUs
  635. work collectively on a single task, the completion time of tasks on CPUs become
  636. comparable to the completion time on GPUs, thus relieving from granularity
  637. discrepancy concerns. Hwloc support needs to be enabled to get good performance,
  638. otherwise StarPU will not know how to better group cores.
  639. Two modes of execution exist to accomodate with existing usages.
  640. @subsection Fork-mode parallel tasks
  641. In the Fork mode, StarPU will call the codelet function on one
  642. of the CPUs of the combined worker. The codelet function can use
  643. @code{starpu_combined_worker_get_size()} to get the number of threads it is
  644. allowed to start to achieve the computation. The CPU binding mask for the whole
  645. set of CPUs is already enforced, so that threads created by the function will
  646. inherit the mask, and thus execute where StarPU expected, the OS being in charge
  647. of choosing how to schedule threads on the corresponding CPUs. The application
  648. can also choose to bind threads by hand, using e.g. sched_getaffinity to know
  649. the CPU binding mask that StarPU chose.
  650. For instance, using OpenMP (full source is available in
  651. @code{examples/openmp/vector_scal.c}):
  652. @example
  653. void scal_cpu_func(void *buffers[], void *_args)
  654. @{
  655. unsigned i;
  656. float *factor = _args;
  657. struct starpu_vector_interface *vector = buffers[0];
  658. unsigned n = STARPU_VECTOR_GET_NX(vector);
  659. float *val = (float *)STARPU_VECTOR_GET_PTR(vector);
  660. #pragma omp parallel for num_threads(starpu_combined_worker_get_size())
  661. for (i = 0; i < n; i++)
  662. val[i] *= *factor;
  663. @}
  664. static struct starpu_codelet cl =
  665. @{
  666. .modes = @{ STARPU_RW @},
  667. .where = STARPU_CPU,
  668. .type = STARPU_FORKJOIN,
  669. .max_parallelism = INT_MAX,
  670. .cpu_funcs = @{scal_cpu_func, NULL@},
  671. .nbuffers = 1,
  672. @};
  673. @end example
  674. Other examples include for instance calling a BLAS parallel CPU implementation
  675. (see @code{examples/mult/xgemm.c}).
  676. @subsection SPMD-mode parallel tasks
  677. In the SPMD mode, StarPU will call the codelet function on
  678. each CPU of the combined worker. The codelet function can use
  679. @code{starpu_combined_worker_get_size()} to get the total number of CPUs
  680. involved in the combined worker, and thus the number of calls that are made in
  681. parallel to the function, and @code{starpu_combined_worker_get_rank()} to get
  682. the rank of the current CPU within the combined worker. For instance:
  683. @example
  684. static void func(void *buffers[], void *args)
  685. @{
  686. unsigned i;
  687. float *factor = _args;
  688. struct starpu_vector_interface *vector = buffers[0];
  689. unsigned n = STARPU_VECTOR_GET_NX(vector);
  690. float *val = (float *)STARPU_VECTOR_GET_PTR(vector);
  691. /* Compute slice to compute */
  692. unsigned m = starpu_combined_worker_get_size();
  693. unsigned j = starpu_combined_worker_get_rank();
  694. unsigned slice = (n+m-1)/m;
  695. for (i = j * slice; i < (j+1) * slice && i < n; i++)
  696. val[i] *= *factor;
  697. @}
  698. static struct starpu_codelet cl =
  699. @{
  700. .modes = @{ STARPU_RW @},
  701. .where = STARP_CPU,
  702. .type = STARPU_SPMD,
  703. .max_parallelism = INT_MAX,
  704. .cpu_funcs = @{ func, NULL @},
  705. .nbuffers = 1,
  706. @}
  707. @end example
  708. Of course, this trivial example will not really benefit from parallel task
  709. execution, and was only meant to be simple to understand. The benefit comes
  710. when the computation to be done is so that threads have to e.g. exchange
  711. intermediate results, or write to the data in a complex but safe way in the same
  712. buffer.
  713. @subsection Parallel tasks performance
  714. To benefit from parallel tasks, a parallel-task-aware StarPU scheduler has to
  715. be used. When exposed to codelets with a Fork or SPMD flag, the @code{pheft}
  716. (parallel-heft) and @code{pgreedy} (parallel greedy) schedulers will indeed also
  717. try to execute tasks with several CPUs. It will automatically try the various
  718. available combined worker sizes and thus be able to avoid choosing a large
  719. combined worker if the codelet does not actually scale so much.
  720. @subsection Combined worker sizes
  721. By default, StarPU creates combined workers according to the architecture
  722. structure as detected by hwloc. It means that for each object of the hwloc
  723. topology (NUMA node, socket, cache, ...) a combined worker will be created. If
  724. some nodes of the hierarchy have a big arity (e.g. many cores in a socket
  725. without a hierarchy of shared caches), StarPU will create combined workers of
  726. intermediate sizes.
  727. The user can give some hints to StarPU about combined workers sizes to favor.
  728. This can be done by using the environment variables @code{STARPU_MIN_WORKERSIZE}
  729. and @code{STARPU_MAX_WORKERSIZE}. When set, they will force StarPU to create the
  730. biggest combined workers possible without overstepping the defined boundaries.
  731. However, StarPU will create the remaining combined workers without abiding by
  732. the rules if not possible.
  733. For example : if the user specifies a minimum and maximum combined workers size
  734. of 3 on a machine containing 8 CPUs, StarPU will create a combined worker of
  735. size 2 beside the combined workers of size 3.
  736. @subsection Concurrent parallel tasks
  737. Unfortunately, many environments and librairies do not support concurrent
  738. calls.
  739. For instance, most OpenMP implementations (including the main ones) do not
  740. support concurrent @code{pragma omp parallel} statements without nesting them in
  741. another @code{pragma omp parallel} statement, but StarPU does not yet support
  742. creating its CPU workers by using such pragma.
  743. Other parallel libraries are also not safe when being invoked concurrently
  744. from different threads, due to the use of global variables in their sequential
  745. sections for instance.
  746. The solution is then to use only one combined worker at a time. This can be
  747. done by setting @code{single_combined_worker} to 1 in the @code{starpu_conf}
  748. structure, or setting the @code{STARPU_SINGLE_COMBINED_WORKER} environment
  749. variable to 1. StarPU will then run only one parallel task at a time.
  750. @node Debugging
  751. @section Debugging
  752. StarPU provides several tools to help debugging aplications. Execution traces
  753. can be generated and displayed graphically, see @ref{Generating traces}. Some
  754. gdb helpers are also provided to show the whole StarPU state:
  755. @smallexample
  756. (gdb) source tools/gdbinit
  757. (gdb) help starpu
  758. @end smallexample
  759. @node The multiformat interface
  760. @section The multiformat interface
  761. It may be interesting to represent the same piece of data using two different
  762. data structures: one that would only be used on CPUs, and one that would only
  763. be used on GPUs. This can be done by using the multiformat interface. StarPU
  764. will be able to convert data from one data structure to the other when needed.
  765. Note that the heft scheduler is the only one optimized for this interface. The
  766. user must provide StarPU with conversion codelets:
  767. @cartouche
  768. @smallexample
  769. #define NX 1024
  770. struct point array_of_structs[NX];
  771. starpu_data_handle_t handle;
  772. /*
  773. * The conversion of a piece of data is itself a task, though it is created,
  774. * submitted and destroyed by StarPU internals and not by the user. Therefore,
  775. * we have to define two codelets.
  776. * Note that for now the conversion from the CPU format to the GPU format has to
  777. * be executed on the GPU, and the conversion from the GPU to the CPU has to be
  778. * executed on the CPU.
  779. */
  780. #ifdef STARPU_USE_OPENCL
  781. void cpu_to_opencl_opencl_func(void *buffers[], void *args);
  782. struct starpu_codelet cpu_to_opencl_cl = @{
  783. .where = STARPU_OPENCL,
  784. .opencl_funcs = @{ cpu_to_opencl_opencl_func, NULL @},
  785. .nbuffers = 1,
  786. .modes = @{ STARPU_RW @}
  787. @};
  788. void opencl_to_cpu_func(void *buffers[], void *args);
  789. struct starpu_codelet opencl_to_cpu_cl = @{
  790. .where = STARPU_CPU,
  791. .cpu_funcs = @{ opencl_to_cpu_func, NULL @},
  792. .nbuffers = 1,
  793. .modes = @{ STARPU_RW @}
  794. @};
  795. #endif
  796. struct starpu_multiformat_data_interface_ops format_ops = @{
  797. #ifdef STARPU_USE_OPENCL
  798. .opencl_elemsize = 2 * sizeof(float),
  799. .cpu_to_opencl_cl = &cpu_to_opencl_cl,
  800. .opencl_to_cpu_cl = &opencl_to_cpu_cl,
  801. #endif
  802. .cpu_elemsize = 2 * sizeof(float),
  803. ...
  804. @};
  805. starpu_multiformat_data_register(handle, 0, &array_of_structs, NX, &format_ops);
  806. @end smallexample
  807. @end cartouche
  808. Kernels can be written almost as for any other interface. Note that
  809. STARPU_MULTIFORMAT_GET_CPU_PTR shall only be used for CPU kernels. CUDA kernels
  810. must use STARPU_MULTIFORMAT_GET_CUDA_PTR, and OpenCL kernels must use
  811. STARPU_MULTIFORMAT_GET_OPENCL_PTR. STARPU_MULTIFORMAT_GET_NX may be used in any
  812. kind of kernel.
  813. @cartouche
  814. @smallexample
  815. static void
  816. multiformat_scal_cpu_func(void *buffers[], void *args)
  817. @{
  818. struct point *aos;
  819. unsigned int n;
  820. aos = STARPU_MULTIFORMAT_GET_CPU_PTR(buffers[0]);
  821. n = STARPU_MULTIFORMAT_GET_NX(buffers[0]);
  822. ...
  823. @}
  824. extern "C" void multiformat_scal_cuda_func(void *buffers[], void *_args)
  825. @{
  826. unsigned int n;
  827. struct struct_of_arrays *soa;
  828. soa = (struct struct_of_arrays *) STARPU_MULTIFORMAT_GET_CUDA_PTR(buffers[0]);
  829. n = STARPU_MULTIFORMAT_GET_NX(buffers[0]);
  830. ...
  831. @}
  832. @end smallexample
  833. @end cartouche
  834. A full example may be found in @code{examples/basic_examples/multiformat.c}.
  835. @node On-GPU rendering
  836. @section On-GPU rendering
  837. Graphical-oriented applications need to draw the result of their computations,
  838. typically on the very GPU where these happened. Technologies such as OpenGL/CUDA
  839. interoperability permit to let CUDA directly work on the OpenGL buffers, making
  840. them thus immediately ready for drawing, by mapping OpenGL buffer, textures or
  841. renderbuffer objects into CUDA. CUDA however imposes some technical
  842. constraints: peer memcpy has to be disabled, and the thread that runs OpenGL has
  843. to be the one that runs CUDA computations for that GPU.
  844. To achieve this with StarPU, pass the @code{--disable-cuda-memcpy-peer} option
  845. to @code{./configure} (TODO: make it dynamic), the interoperability mode has to
  846. be enabled by using the @code{cuda_opengl_interoperability} field of the
  847. @code{starpu_conf} structure, and the driver loop has to be run by
  848. the application, by using the @code{not_launched_drivers} field of
  849. @code{starpu_conf} to prevent StarPU from running it in a separate thread, and
  850. by using @code{starpu_driver_run} to run the loop. The @code{gl_interop} example
  851. shows how it articulates in a simple case, where rendering is done in task
  852. callbacks. TODO: provide glutIdleFunc alternative.
  853. Then, to use an OpenGL buffer as a CUDA data, StarPU simply needs to be given
  854. the CUDA pointer at registration, for instance:
  855. @cartouche
  856. @smallexample
  857. for (workerid = 0; workerid < starpu_worker_get_count(); workerid++)
  858. if (starpu_worker_get_type(workerid) == STARPU_CUDA_WORKER)
  859. break;
  860. cudaGraphicsResourceGetMappedPointer((void**)&output, &num_bytes, resource);
  861. starpu_vector_data_register(&handle, starpu_worker_get_memory_node(workerid), output, num_bytes / sizeof(float4), sizeof(float4));
  862. starpu_insert_task(&cl, STARPU_RW, handle, 0);
  863. @end smallexample
  864. @end cartouche
  865. and display it e.g. in the callback function.
  866. @node More examples
  867. @section More examples
  868. More examples are available in the StarPU sources in the @code{examples/}
  869. directory. Simple examples include:
  870. @table @asis
  871. @item @code{incrementer/}:
  872. Trivial incrementation test.
  873. @item @code{basic_examples/}:
  874. Simple documented Hello world (as shown in @ref{Hello World}), vector/scalar product (as shown
  875. in @ref{Vector Scaling on an Hybrid CPU/GPU Machine}), matrix
  876. product examples (as shown in @ref{Performance model example}), an example using the blocked matrix data
  877. interface, an example using the variable data interface, and an example
  878. using different formats on CPUs and GPUs.
  879. @item @code{matvecmult/}:
  880. OpenCL example from NVidia, adapted to StarPU.
  881. @item @code{axpy/}:
  882. AXPY CUBLAS operation adapted to StarPU.
  883. @item @code{fortran/}:
  884. Example of Fortran bindings.
  885. @end table
  886. More advanced examples include:
  887. @table @asis
  888. @item @code{filters/}:
  889. Examples using filters, as shown in @ref{Partitioning Data}.
  890. @item @code{lu/}:
  891. LU matrix factorization, see for instance @code{xlu_implicit.c}
  892. @item @code{cholesky/}:
  893. Cholesky matrix factorization, see for instance @code{cholesky_implicit.c}.
  894. @end table