stencil-kernels.c 23 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638
  1. /* StarPU --- Runtime system for heterogeneous multicore architectures.
  2. *
  3. * Copyright (C) 2010-2015 Université de Bordeaux
  4. * Copyright (C) 2012, 2013, 2016 CNRS
  5. *
  6. * StarPU is free software; you can redistribute it and/or modify
  7. * it under the terms of the GNU Lesser General Public License as published by
  8. * the Free Software Foundation; either version 2.1 of the License, or (at
  9. * your option) any later version.
  10. *
  11. * StarPU is distributed in the hope that it will be useful, but
  12. * WITHOUT ANY WARRANTY; without even the implied warranty of
  13. * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
  14. *
  15. * See the GNU Lesser General Public License in COPYING.LGPL for more details.
  16. */
  17. #include "stencil.h"
  18. /* Computation Kernels */
  19. /*
  20. * There are three codeletets:
  21. *
  22. * - cl_update, which takes a block and the boundaries of its neighbours, loads
  23. * the boundaries into the block and perform some update loops:
  24. *
  25. * comp. buffer save. buffers comp. buffer save. buffers comp. buffer
  26. * | ... |
  27. * | | +------------------+ +------------------+
  28. * | #N+1 | | #N+1 bottom copy====>#N+1 bottom copy |
  29. * +-------------+ +------------------+ +------------------+
  30. * | #N top copy | | #N top copy | | |
  31. * +-------------+ +------------------+ | |
  32. * | #N |
  33. * ...
  34. * | | +----------------+ +----------------------+
  35. * | | | #N bottom copy | | block #N bottom copy |
  36. * ^ +------------------+ +----------------+ +----------------------+
  37. * | | #N-1 top copy <====#N-1 top copy | | block #N-1 |
  38. * | +------------------+ +----------------+ | |
  39. * Z ...
  40. *
  41. * - save_cl_top, which take a block and its top boundary, and saves the top of
  42. * the block into the boundary (to be given as bottom of the neighbour above
  43. * this block).
  44. *
  45. * comp. buffer save. buffers comp. buffer save. buffers comp. buffer
  46. * | ... |
  47. * | | +------------------+ +------------------+
  48. * | #N+1 | | #N+1 bottom copy | | #N+1 bottom copy |
  49. * +-------------+ +------------------+ +------------------+
  50. * | #N top copy | | #N top copy <==== |
  51. * +-------------+ +------------------+ |..................|
  52. * | #N |
  53. * ...
  54. * | | +----------------+ +----------------------+
  55. * | | | #N bottom copy | | block #N bottom copy |
  56. * ^ +------------------+ +----------------+ +----------------------+
  57. * | | #N-1 top copy | | #N-1 top copy | | block #N-1 |
  58. * | +------------------+ +----------------+ | |
  59. * Z ...
  60. *
  61. * - save_cl_bottom, same for the bottom
  62. * comp. buffer save. buffers comp. buffer save. buffers comp. buffer
  63. * | ... |
  64. * | | +------------------+ +------------------+
  65. * | #N+1 | | #N+1 bottom copy | | #N+1 bottom copy |
  66. * +-------------+ +------------------+ +------------------+
  67. * | #N top copy | | #N top copy | | |
  68. * +-------------+ +------------------+ | |
  69. * | #N |
  70. * ...
  71. * |..................| +----------------+ +----------------------+
  72. * | ====>#N bottom copy | | block #N bottom copy |
  73. * ^ +------------------+ +----------------+ +----------------------+
  74. * | | #N-1 top copy | | #N-1 top copy | | block #N-1 |
  75. * | +------------------+ +----------------+ | |
  76. * Z ...
  77. *
  78. * The idea is that the computation buffers thus don't have to move, only their
  79. * boundaries are copied to buffers that do move (be it CPU/GPU, GPU/GPU or via
  80. * MPI)
  81. *
  82. * For each of the buffers above, there are two (0/1) buffers to make new/old switch costless.
  83. */
  84. #if 0
  85. # define DEBUG(fmt, ...) fprintf(stderr,fmt,##__VA_ARGS__)
  86. #else
  87. # define DEBUG(fmt, ...) (void) 0
  88. #endif
  89. /* Record which GPU ran which block, for nice pictures */
  90. int who_runs_what_len;
  91. int *who_runs_what;
  92. int *who_runs_what_index;
  93. double *last_tick;
  94. /* Achieved iterations */
  95. static int achieved_iter;
  96. /* Record how many updates each worker performed */
  97. unsigned update_per_worker[STARPU_NMAXWORKERS];
  98. static void record_who_runs_what(struct block_description *block)
  99. {
  100. double now, now2, diff, delta = get_ticks() * 1000;
  101. int workerid = starpu_worker_get_id_check();
  102. now = starpu_timing_now();
  103. now2 = now - start;
  104. diff = now2 - last_tick[block->bz];
  105. while (diff >= delta)
  106. {
  107. last_tick[block->bz] += delta;
  108. diff = now2 - last_tick[block->bz];
  109. if (who_runs_what_index[block->bz] < who_runs_what_len)
  110. who_runs_what[block->bz + (who_runs_what_index[block->bz]++) * get_nbz()] = -1;
  111. }
  112. if (who_runs_what_index[block->bz] < who_runs_what_len)
  113. who_runs_what[block->bz + (who_runs_what_index[block->bz]++) * get_nbz()] = global_workerid(workerid);
  114. }
  115. static void check_load(struct starpu_block_interface *block, struct starpu_block_interface *boundary)
  116. {
  117. /* Sanity checks */
  118. STARPU_ASSERT(block->nx == boundary->nx);
  119. STARPU_ASSERT(block->ny == boundary->ny);
  120. STARPU_ASSERT(boundary->nz == K);
  121. /* NB: this is not fully garanteed ... but it's *very* likely and that
  122. * makes our life much simpler */
  123. STARPU_ASSERT(block->ldy == boundary->ldy);
  124. STARPU_ASSERT(block->ldz == boundary->ldz);
  125. }
  126. /*
  127. * Load a neighbour's boundary into block, CPU version
  128. */
  129. static void load_subblock_from_buffer_cpu(void *_block,
  130. void *_boundary,
  131. unsigned firstz)
  132. {
  133. struct starpu_block_interface *block = (struct starpu_block_interface *)_block;
  134. struct starpu_block_interface *boundary = (struct starpu_block_interface *)_boundary;
  135. check_load(block, boundary);
  136. /* We do a contiguous memory transfer */
  137. size_t boundary_size = K*block->ldz*block->elemsize;
  138. unsigned offset = firstz*block->ldz;
  139. TYPE *block_data = (TYPE *)block->ptr;
  140. TYPE *boundary_data = (TYPE *)boundary->ptr;
  141. memcpy(&block_data[offset], boundary_data, boundary_size);
  142. }
  143. /*
  144. * Load a neighbour's boundary into block, CUDA version
  145. */
  146. #ifdef STARPU_USE_CUDA
  147. static void load_subblock_from_buffer_cuda(void *_block,
  148. void *_boundary,
  149. unsigned firstz)
  150. {
  151. struct starpu_block_interface *block = (struct starpu_block_interface *)_block;
  152. struct starpu_block_interface *boundary = (struct starpu_block_interface *)_boundary;
  153. check_load(block, boundary);
  154. /* We do a contiguous memory transfer */
  155. size_t boundary_size = K*block->ldz*block->elemsize;
  156. unsigned offset = firstz*block->ldz;
  157. TYPE *block_data = (TYPE *)block->ptr;
  158. TYPE *boundary_data = (TYPE *)boundary->ptr;
  159. cudaMemcpyAsync(&block_data[offset], boundary_data, boundary_size, cudaMemcpyDeviceToDevice, starpu_cuda_get_local_stream());
  160. }
  161. /*
  162. * cl_update (CUDA version)
  163. */
  164. static void update_func_cuda(void *descr[], void *arg)
  165. {
  166. struct block_description *block = arg;
  167. int workerid = starpu_worker_get_id_check();
  168. DEBUG( "!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!\n");
  169. if (block->bz == 0)
  170. FPRINTF(stderr,"!!! DO update_func_cuda z %u CUDA%d !!!\n", block->bz, workerid);
  171. else
  172. DEBUG( "!!! DO update_func_cuda z %u CUDA%d !!!\n", block->bz, workerid);
  173. #if defined(STARPU_USE_MPI) && !defined(STARPU_SIMGRID)
  174. int rank = 0;
  175. MPI_Comm_rank(MPI_COMM_WORLD, &rank);
  176. DEBUG( "!!! RANK %d !!!\n", rank);
  177. #endif
  178. DEBUG( "!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!\n");
  179. unsigned block_size_z = get_block_size(block->bz);
  180. unsigned i;
  181. update_per_worker[workerid]++;
  182. record_who_runs_what(block);
  183. /*
  184. * Load neighbours' boundaries : TOP
  185. */
  186. /* The offset along the z axis is (block_size_z + K) */
  187. load_subblock_from_buffer_cuda(descr[0], descr[2], block_size_z+K);
  188. load_subblock_from_buffer_cuda(descr[1], descr[3], block_size_z+K);
  189. /*
  190. * Load neighbours' boundaries : BOTTOM
  191. */
  192. load_subblock_from_buffer_cuda(descr[0], descr[4], 0);
  193. load_subblock_from_buffer_cuda(descr[1], descr[5], 0);
  194. /*
  195. * Stencils ... do the actual work here :) TODO
  196. */
  197. for (i=1; i<=K; i++)
  198. {
  199. struct starpu_block_interface *oldb = descr[i%2], *newb = descr[(i+1)%2];
  200. TYPE *old = (void*) oldb->ptr, *newer = (void*) newb->ptr;
  201. /* Shadow data */
  202. cuda_shadow_host(block->bz, old, oldb->nx, oldb->ny, oldb->nz, oldb->ldy, oldb->ldz, i);
  203. /* And perform actual computation */
  204. #ifdef LIFE
  205. cuda_life_update_host(block->bz, old, newer, oldb->nx, oldb->ny, oldb->nz, oldb->ldy, oldb->ldz, i);
  206. #else
  207. cudaMemcpyAsync(newer, old, oldb->nx * oldb->ny * oldb->nz * sizeof(*newer), cudaMemcpyDeviceToDevice, starpu_cuda_get_local_stream());
  208. #endif /* LIFE */
  209. }
  210. if (block->bz == 0)
  211. starpu_top_update_data_integer(starpu_top_achieved_loop, ++achieved_iter);
  212. }
  213. #endif /* STARPU_USE_CUDA */
  214. /*
  215. * Load a neighbour's boundary into block, OpenCL version
  216. */
  217. #ifdef STARPU_USE_OPENCL
  218. static void load_subblock_from_buffer_opencl(struct starpu_block_interface *block,
  219. struct starpu_block_interface *boundary,
  220. unsigned firstz)
  221. {
  222. check_load(block, boundary);
  223. /* We do a contiguous memory transfer */
  224. size_t boundary_size = K*block->ldz*block->elemsize;
  225. unsigned offset = firstz*block->ldz;
  226. cl_mem block_data = (cl_mem)block->dev_handle;
  227. cl_mem boundary_data = (cl_mem)boundary->dev_handle;
  228. cl_command_queue cq;
  229. starpu_opencl_get_current_queue(&cq);
  230. cl_int ret = clEnqueueCopyBuffer(cq, boundary_data, block_data, 0, offset, boundary_size, 0, NULL, NULL);
  231. if (ret != CL_SUCCESS) STARPU_OPENCL_REPORT_ERROR(ret);
  232. }
  233. /*
  234. * cl_update (OpenCL version)
  235. */
  236. static void update_func_opencl(void *descr[], void *arg)
  237. {
  238. struct block_description *block = arg;
  239. int workerid = starpu_worker_get_id_check();
  240. DEBUG( "!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!\n");
  241. if (block->bz == 0)
  242. FPRINTF(stderr,"!!! DO update_func_opencl z %u OPENCL%d !!!\n", block->bz, workerid);
  243. else
  244. DEBUG( "!!! DO update_func_opencl z %u OPENCL%d !!!\n", block->bz, workerid);
  245. #if defined(STARPU_USE_MPI) && !defined(STARPU_SIMGRID)
  246. int rank = 0;
  247. MPI_Comm_rank(MPI_COMM_WORLD, &rank);
  248. DEBUG( "!!! RANK %d !!!\n", rank);
  249. #endif
  250. DEBUG( "!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!\n");
  251. unsigned block_size_z = get_block_size(block->bz);
  252. unsigned i;
  253. update_per_worker[workerid]++;
  254. record_who_runs_what(block);
  255. cl_command_queue cq;
  256. starpu_opencl_get_current_queue(&cq);
  257. /*
  258. * Load neighbours' boundaries : TOP
  259. */
  260. /* The offset along the z axis is (block_size_z + K) */
  261. load_subblock_from_buffer_opencl(descr[0], descr[2], block_size_z+K);
  262. load_subblock_from_buffer_opencl(descr[1], descr[3], block_size_z+K);
  263. /*
  264. * Load neighbours' boundaries : BOTTOM
  265. */
  266. load_subblock_from_buffer_opencl(descr[0], descr[4], 0);
  267. load_subblock_from_buffer_opencl(descr[1], descr[5], 0);
  268. /*
  269. * Stencils ... do the actual work here :) TODO
  270. */
  271. for (i=1; i<=K; i++)
  272. {
  273. struct starpu_block_interface *oldb = descr[i%2], *newb = descr[(i+1)%2];
  274. TYPE *old = (void*) oldb->dev_handle, *newer = (void*) newb->dev_handle;
  275. /* Shadow data */
  276. opencl_shadow_host(block->bz, old, oldb->nx, oldb->ny, oldb->nz, oldb->ldy, oldb->ldz, i);
  277. /* And perform actual computation */
  278. #ifdef LIFE
  279. opencl_life_update_host(block->bz, old, newer, oldb->nx, oldb->ny, oldb->nz, oldb->ldy, oldb->ldz, i);
  280. #else
  281. cl_event event;
  282. cl_int ret = clEnqueueCopyBuffer(cq, old, newer, 0, 0, oldb->nx * oldb->ny * oldb->nz * sizeof(*newer), 0, NULL, &event);
  283. if (ret != CL_SUCCESS) STARPU_OPENCL_REPORT_ERROR(ret);
  284. #endif /* LIFE */
  285. }
  286. if (block->bz == 0)
  287. starpu_top_update_data_integer(starpu_top_achieved_loop, ++achieved_iter);
  288. }
  289. #endif /* STARPU_USE_OPENCL */
  290. /*
  291. * cl_update (CPU version)
  292. */
  293. void update_func_cpu(void *descr[], void *arg)
  294. {
  295. struct block_description *block = (struct block_description *) arg;
  296. int workerid = starpu_worker_get_id_check();
  297. DEBUG( "!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!\n");
  298. if (block->bz == 0)
  299. FPRINTF(stderr,"!!! DO update_func_cpu z %u CPU%d !!!\n", block->bz, workerid);
  300. else
  301. DEBUG( "!!! DO update_func_cpu z %u CPU%d !!!\n", block->bz, workerid);
  302. #if defined(STARPU_USE_MPI) && !defined(STARPU_SIMGRID)
  303. int rank = 0;
  304. MPI_Comm_rank(MPI_COMM_WORLD, &rank);
  305. DEBUG( "!!! RANK %d !!!\n", rank);
  306. #endif
  307. DEBUG( "!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!\n");
  308. unsigned block_size_z = get_block_size(block->bz);
  309. unsigned i;
  310. update_per_worker[workerid]++;
  311. record_who_runs_what(block);
  312. /*
  313. * Load neighbours' boundaries : TOP
  314. */
  315. /* The offset along the z axis is (block_size_z + K) */
  316. load_subblock_from_buffer_cpu(descr[0], descr[2], block_size_z+K);
  317. load_subblock_from_buffer_cpu(descr[1], descr[3], block_size_z+K);
  318. /*
  319. * Load neighbours' boundaries : BOTTOM
  320. */
  321. load_subblock_from_buffer_cpu(descr[0], descr[4], 0);
  322. load_subblock_from_buffer_cpu(descr[1], descr[5], 0);
  323. /*
  324. * Stencils ... do the actual work here :) TODO
  325. */
  326. for (i=1; i<=K; i++)
  327. {
  328. struct starpu_block_interface *oldb = (struct starpu_block_interface *) descr[i%2], *newb = (struct starpu_block_interface *) descr[(i+1)%2];
  329. TYPE *old = (TYPE*) oldb->ptr, *newer = (TYPE*) newb->ptr;
  330. /* Shadow data */
  331. unsigned ldy = oldb->ldy, ldz = oldb->ldz;
  332. unsigned nx = oldb->nx, ny = oldb->ny, nz = oldb->nz;
  333. unsigned x, y, z;
  334. unsigned stepx = 1;
  335. unsigned stepy = 1;
  336. unsigned stepz = 1;
  337. unsigned idx = 0;
  338. unsigned idy = 0;
  339. unsigned idz = 0;
  340. TYPE *ptr = old;
  341. # include "shadow.h"
  342. /* And perform actual computation */
  343. #ifdef LIFE
  344. life_update(block->bz, old, newer, oldb->nx, oldb->ny, oldb->nz, oldb->ldy, oldb->ldz, i);
  345. #else
  346. memcpy(newer, old, oldb->nx * oldb->ny * oldb->nz * sizeof(*newer));
  347. #endif /* LIFE */
  348. }
  349. if (block->bz == 0)
  350. starpu_top_update_data_integer(starpu_top_achieved_loop, ++achieved_iter);
  351. }
  352. /* Performance model and codelet structure */
  353. static struct starpu_perfmodel cl_update_model =
  354. {
  355. .type = STARPU_HISTORY_BASED,
  356. .symbol = "cl_update"
  357. };
  358. struct starpu_codelet cl_update =
  359. {
  360. .cpu_funcs = {update_func_cpu},
  361. #ifdef STARPU_USE_CUDA
  362. .cuda_funcs = {update_func_cuda},
  363. .cuda_flags = {STARPU_CUDA_ASYNC},
  364. #endif
  365. #ifdef STARPU_USE_OPENCL
  366. .opencl_funcs = {update_func_opencl},
  367. .opencl_flags = {STARPU_OPENCL_ASYNC},
  368. #endif
  369. .model = &cl_update_model,
  370. .nbuffers = 6,
  371. .modes = {STARPU_RW, STARPU_RW, STARPU_R, STARPU_R, STARPU_R, STARPU_R}
  372. };
  373. /*
  374. * Save the block internal boundaries to give them to our neighbours.
  375. */
  376. /* CPU version */
  377. static void load_subblock_into_buffer_cpu(void *_block,
  378. void *_boundary,
  379. unsigned firstz)
  380. {
  381. struct starpu_block_interface *block = (struct starpu_block_interface *)_block;
  382. struct starpu_block_interface *boundary = (struct starpu_block_interface *)_boundary;
  383. check_load(block, boundary);
  384. /* We do a contiguous memory transfer */
  385. size_t boundary_size = K*block->ldz*block->elemsize;
  386. unsigned offset = firstz*block->ldz;
  387. TYPE *block_data = (TYPE *)block->ptr;
  388. TYPE *boundary_data = (TYPE *)boundary->ptr;
  389. memcpy(boundary_data, &block_data[offset], boundary_size);
  390. }
  391. /* CUDA version */
  392. #ifdef STARPU_USE_CUDA
  393. static void load_subblock_into_buffer_cuda(void *_block,
  394. void *_boundary,
  395. unsigned firstz)
  396. {
  397. struct starpu_block_interface *block = (struct starpu_block_interface *)_block;
  398. struct starpu_block_interface *boundary = (struct starpu_block_interface *)_boundary;
  399. check_load(block, boundary);
  400. /* We do a contiguous memory transfer */
  401. size_t boundary_size = K*block->ldz*block->elemsize;
  402. unsigned offset = firstz*block->ldz;
  403. TYPE *block_data = (TYPE *)block->ptr;
  404. TYPE *boundary_data = (TYPE *)boundary->ptr;
  405. cudaMemcpyAsync(boundary_data, &block_data[offset], boundary_size, cudaMemcpyDeviceToDevice, starpu_cuda_get_local_stream());
  406. }
  407. #endif /* STARPU_USE_CUDA */
  408. /* OPENCL version */
  409. #ifdef STARPU_USE_OPENCL
  410. static void load_subblock_into_buffer_opencl(struct starpu_block_interface *block,
  411. struct starpu_block_interface *boundary,
  412. unsigned firstz)
  413. {
  414. check_load(block, boundary);
  415. /* We do a contiguous memory transfer */
  416. size_t boundary_size = K*block->ldz*block->elemsize;
  417. unsigned offset = firstz*block->ldz;
  418. cl_mem block_data = (cl_mem)block->dev_handle;
  419. cl_mem boundary_data = (cl_mem)boundary->dev_handle;
  420. cl_command_queue cq;
  421. starpu_opencl_get_current_queue(&cq);
  422. cl_int ret = clEnqueueCopyBuffer(cq, block_data, boundary_data, offset, 0, boundary_size, 0, NULL, NULL);
  423. if (ret != CL_SUCCESS) STARPU_OPENCL_REPORT_ERROR(ret);
  424. }
  425. #endif /* STARPU_USE_OPENCL */
  426. /* Record how many top/bottom saves each worker performed */
  427. unsigned top_per_worker[STARPU_NMAXWORKERS];
  428. unsigned bottom_per_worker[STARPU_NMAXWORKERS];
  429. /* top save, CPU version */
  430. void dummy_func_top_cpu(void *descr[] STARPU_ATTRIBUTE_UNUSED, void *arg)
  431. {
  432. struct block_description *block = (struct block_description *) arg;
  433. int workerid = starpu_worker_get_id_check();
  434. top_per_worker[workerid]++;
  435. DEBUG( "DO SAVE Bottom block %d\n", block->bz);
  436. /* The offset along the z axis is (block_size_z + K)- K */
  437. unsigned block_size_z = get_block_size(block->bz);
  438. load_subblock_into_buffer_cpu(descr[0], descr[2], block_size_z);
  439. load_subblock_into_buffer_cpu(descr[1], descr[3], block_size_z);
  440. }
  441. /* bottom save, CPU version */
  442. void dummy_func_bottom_cpu(void *descr[] STARPU_ATTRIBUTE_UNUSED, void *arg)
  443. {
  444. struct block_description *block = (struct block_description *) arg;
  445. int workerid = starpu_worker_get_id_check();
  446. bottom_per_worker[workerid]++;
  447. DEBUG( "DO SAVE Top block %d\n", block->bz);
  448. load_subblock_into_buffer_cpu(descr[0], descr[2], K);
  449. load_subblock_into_buffer_cpu(descr[1], descr[3], K);
  450. }
  451. /* top save, CUDA version */
  452. #ifdef STARPU_USE_CUDA
  453. static void dummy_func_top_cuda(void *descr[] STARPU_ATTRIBUTE_UNUSED, void *arg)
  454. {
  455. struct block_description *block = (struct block_description *) arg;
  456. int workerid = starpu_worker_get_id_check();
  457. top_per_worker[workerid]++;
  458. DEBUG( "DO SAVE Top block %d\n", block->bz);
  459. /* The offset along the z axis is (block_size_z + K)- K */
  460. unsigned block_size_z = get_block_size(block->bz);
  461. load_subblock_into_buffer_cuda(descr[0], descr[2], block_size_z);
  462. load_subblock_into_buffer_cuda(descr[1], descr[3], block_size_z);
  463. }
  464. /* bottom save, CUDA version */
  465. static void dummy_func_bottom_cuda(void *descr[] STARPU_ATTRIBUTE_UNUSED, void *arg)
  466. {
  467. struct block_description *block = (struct block_description *) arg;
  468. int workerid = starpu_worker_get_id_check();
  469. bottom_per_worker[workerid]++;
  470. DEBUG( "DO SAVE Bottom block %d on CUDA\n", block->bz);
  471. load_subblock_into_buffer_cuda(descr[0], descr[2], K);
  472. load_subblock_into_buffer_cuda(descr[1], descr[3], K);
  473. }
  474. #endif /* STARPU_USE_CUDA */
  475. /* top save, OpenCL version */
  476. #ifdef STARPU_USE_OPENCL
  477. static void dummy_func_top_opencl(void *descr[] STARPU_ATTRIBUTE_UNUSED, void *arg)
  478. {
  479. struct block_description *block = (struct block_description *) arg;
  480. int workerid = starpu_worker_get_id_check();
  481. top_per_worker[workerid]++;
  482. DEBUG( "DO SAVE Top block %d\n", block->bz);
  483. /* The offset along the z axis is (block_size_z + K)- K */
  484. unsigned block_size_z = get_block_size(block->bz);
  485. load_subblock_into_buffer_opencl(descr[0], descr[2], block_size_z);
  486. load_subblock_into_buffer_opencl(descr[1], descr[3], block_size_z);
  487. }
  488. /* bottom save, OPENCL version */
  489. static void dummy_func_bottom_opencl(void *descr[] STARPU_ATTRIBUTE_UNUSED, void *arg)
  490. {
  491. struct block_description *block = (struct block_description *) arg;
  492. int workerid = starpu_worker_get_id_check();
  493. bottom_per_worker[workerid]++;
  494. DEBUG( "DO SAVE Bottom block %d on OPENCL\n", block->bz);
  495. load_subblock_into_buffer_opencl(descr[0], descr[2], K);
  496. load_subblock_into_buffer_opencl(descr[1], descr[3], K);
  497. }
  498. #endif /* STARPU_USE_OPENCL */
  499. /* Performance models and codelet for save */
  500. static struct starpu_perfmodel save_cl_bottom_model =
  501. {
  502. .type = STARPU_HISTORY_BASED,
  503. .symbol = "save_cl_bottom"
  504. };
  505. static struct starpu_perfmodel save_cl_top_model =
  506. {
  507. .type = STARPU_HISTORY_BASED,
  508. .symbol = "save_cl_top"
  509. };
  510. struct starpu_codelet save_cl_bottom =
  511. {
  512. .cpu_funcs = {dummy_func_bottom_cpu},
  513. #ifdef STARPU_USE_CUDA
  514. .cuda_funcs = {dummy_func_bottom_cuda},
  515. .cuda_flags = {STARPU_CUDA_ASYNC},
  516. #endif
  517. #ifdef STARPU_USE_OPENCL
  518. .opencl_funcs = {dummy_func_bottom_opencl},
  519. .opencl_flags = {STARPU_OPENCL_ASYNC},
  520. #endif
  521. .model = &save_cl_bottom_model,
  522. .nbuffers = 4,
  523. .modes = {STARPU_R, STARPU_R, STARPU_W, STARPU_W}
  524. };
  525. struct starpu_codelet save_cl_top =
  526. {
  527. .cpu_funcs = {dummy_func_top_cpu},
  528. #ifdef STARPU_USE_CUDA
  529. .cuda_funcs = {dummy_func_top_cuda},
  530. .cuda_flags = {STARPU_CUDA_ASYNC},
  531. #endif
  532. #ifdef STARPU_USE_OPENCL
  533. .opencl_funcs = {dummy_func_top_opencl},
  534. .opencl_flags = {STARPU_OPENCL_ASYNC},
  535. #endif
  536. .model = &save_cl_top_model,
  537. .nbuffers = 4,
  538. .modes = {STARPU_R, STARPU_R, STARPU_W, STARPU_W}
  539. };