xlu_kernels.c 16 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725
  1. /* StarPU --- Runtime system for heterogeneous multicore architectures.
  2. *
  3. * Copyright (C) 2009, 2010-2012, 2014-2017 Université de Bordeaux
  4. * Copyright (C) 2010, 2011, 2012, 2015, 2017 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. /* LU Kernels */
  18. #include "xlu.h"
  19. #include <math.h>
  20. #include <complex.h>
  21. #ifdef STARPU_USE_CUDA
  22. #include <cublas.h>
  23. #include <starpu_cublas_v2.h>
  24. #endif
  25. #define str(s) #s
  26. #define xstr(s) str(s)
  27. #define STARPU_LU_STR(name) xstr(STARPU_LU(name))
  28. #ifdef STARPU_USE_CUDA
  29. static const TYPE p1 = 1.0f;
  30. static const TYPE m1 = -1.0f;
  31. #endif
  32. /*
  33. * U22
  34. */
  35. static inline void STARPU_LU(common_u22)(void *descr[], int s, void *_args)
  36. {
  37. (void)_args;
  38. TYPE *right = (TYPE *)STARPU_MATRIX_GET_PTR(descr[0]);
  39. TYPE *left = (TYPE *)STARPU_MATRIX_GET_PTR(descr[1]);
  40. TYPE *center = (TYPE *)STARPU_MATRIX_GET_PTR(descr[2]);
  41. unsigned dx = STARPU_MATRIX_GET_NX(descr[2]);
  42. unsigned dy = STARPU_MATRIX_GET_NY(descr[2]);
  43. unsigned dz = STARPU_MATRIX_GET_NY(descr[0]);
  44. unsigned ld12 = STARPU_MATRIX_GET_LD(descr[0]);
  45. unsigned ld21 = STARPU_MATRIX_GET_LD(descr[1]);
  46. unsigned ld22 = STARPU_MATRIX_GET_LD(descr[2]);
  47. #ifdef STARPU_USE_CUDA
  48. cublasStatus status;
  49. cudaError_t cures;
  50. #endif
  51. switch (s)
  52. {
  53. case 0:
  54. CPU_GEMM("N", "N", dy, dx, dz,
  55. (TYPE)-1.0, right, ld21, left, ld12,
  56. (TYPE)1.0, center, ld22);
  57. break;
  58. #ifdef STARPU_USE_CUDA
  59. case 1:
  60. {
  61. status = CUBLAS_GEMM(starpu_cublas_get_local_handle(),
  62. CUBLAS_OP_N, CUBLAS_OP_N, dx, dy, dz,
  63. (CUBLAS_TYPE *)&m1, (CUBLAS_TYPE *)right, ld21, (CUBLAS_TYPE *)left, ld12,
  64. (CUBLAS_TYPE *)&p1, (CUBLAS_TYPE *)center, ld22);
  65. if (STARPU_UNLIKELY(status != CUBLAS_STATUS_SUCCESS))
  66. STARPU_CUBLAS_REPORT_ERROR(status);
  67. break;
  68. }
  69. #endif
  70. default:
  71. STARPU_ABORT();
  72. break;
  73. }
  74. }
  75. void STARPU_LU(cpu_u22)(void *descr[], void *_args)
  76. {
  77. STARPU_LU(common_u22)(descr, 0, _args);
  78. }
  79. #ifdef STARPU_USE_CUDA
  80. void STARPU_LU(cublas_u22)(void *descr[], void *_args)
  81. {
  82. STARPU_LU(common_u22)(descr, 1, _args);
  83. }
  84. #endif /* STARPU_USE_CUDA */
  85. static struct starpu_perfmodel STARPU_LU(model_22) =
  86. {
  87. .type = STARPU_HISTORY_BASED,
  88. #ifdef STARPU_ATLAS
  89. .symbol = STARPU_LU_STR(lu_model_22_atlas)
  90. #elif defined(STARPU_GOTO)
  91. .symbol = STARPU_LU_STR(lu_model_22_goto)
  92. #else
  93. .symbol = STARPU_LU_STR(lu_model_22)
  94. #endif
  95. };
  96. #ifdef STARPU_USE_CUDA
  97. static int can_execute(unsigned workerid, struct starpu_task *task, unsigned nimpl)
  98. {
  99. (void)task;
  100. (void)nimpl;
  101. enum starpu_worker_archtype type = starpu_worker_get_type(workerid);
  102. if (type == STARPU_CPU_WORKER || type == STARPU_MIC_WORKER || type == STARPU_SCC_WORKER)
  103. return 1;
  104. #ifdef STARPU_SIMGRID
  105. /* We don't know, let's assume it can */
  106. return 1;
  107. #else
  108. /* Cuda device */
  109. const struct cudaDeviceProp *props;
  110. props = starpu_cuda_get_device_properties(workerid);
  111. if (props->major >= 2 || props->minor >= 3)
  112. {
  113. /* At least compute capability 1.3, supports doubles */
  114. return 1;
  115. }
  116. else
  117. {
  118. /* Old card does not support doubles */
  119. return 0;
  120. }
  121. #endif
  122. }
  123. #endif
  124. #define STRINGIFY_(x) #x
  125. #define STRINGIFY(x) STRINGIFY_(x)
  126. struct starpu_codelet cl22 =
  127. {
  128. .cpu_funcs = {STARPU_LU(cpu_u22)},
  129. .cpu_funcs_name = {STRINGIFY(STARPU_LU(cpu_u22))},
  130. #ifdef STARPU_USE_CUDA
  131. .cuda_funcs = {STARPU_LU(cublas_u22)},
  132. CAN_EXECUTE
  133. #elif defined(STARPU_SIMGRID)
  134. .cuda_funcs = {(void*)1},
  135. #endif
  136. .cuda_flags = {STARPU_CUDA_ASYNC},
  137. .nbuffers = 3,
  138. .modes = {STARPU_R, STARPU_R, STARPU_RW},
  139. .model = &STARPU_LU(model_22)
  140. };
  141. /*
  142. * U12
  143. */
  144. static inline void STARPU_LU(common_u12)(void *descr[], int s, void *_args)
  145. {
  146. (void)_args;
  147. TYPE *sub11;
  148. TYPE *sub12;
  149. sub11 = (TYPE *)STARPU_MATRIX_GET_PTR(descr[0]);
  150. sub12 = (TYPE *)STARPU_MATRIX_GET_PTR(descr[1]);
  151. unsigned ld11 = STARPU_MATRIX_GET_LD(descr[0]);
  152. unsigned ld12 = STARPU_MATRIX_GET_LD(descr[1]);
  153. unsigned nx12 = STARPU_MATRIX_GET_NX(descr[1]);
  154. unsigned ny12 = STARPU_MATRIX_GET_NY(descr[1]);
  155. #ifdef STARPU_USE_CUDA
  156. cublasStatus status;
  157. cudaError_t cures;
  158. #endif
  159. /* solve L11 U12 = A12 (find U12) */
  160. switch (s)
  161. {
  162. case 0:
  163. CPU_TRSM("L", "L", "N", "N", nx12, ny12,
  164. (TYPE)1.0, sub11, ld11, sub12, ld12);
  165. break;
  166. #ifdef STARPU_USE_CUDA
  167. case 1:
  168. status = CUBLAS_TRSM(starpu_cublas_get_local_handle(),
  169. CUBLAS_SIDE_LEFT, CUBLAS_FILL_MODE_LOWER, CUBLAS_OP_N, CUBLAS_DIAG_NON_UNIT,
  170. ny12, nx12,
  171. (CUBLAS_TYPE*)&p1, (CUBLAS_TYPE*)sub11, ld11, (CUBLAS_TYPE*)sub12, ld12);
  172. if (STARPU_UNLIKELY(status != CUBLAS_STATUS_SUCCESS))
  173. STARPU_CUBLAS_REPORT_ERROR(status);
  174. break;
  175. #endif
  176. default:
  177. STARPU_ABORT();
  178. break;
  179. }
  180. }
  181. void STARPU_LU(cpu_u12)(void *descr[], void *_args)
  182. {
  183. STARPU_LU(common_u12)(descr, 0, _args);
  184. }
  185. #ifdef STARPU_USE_CUDA
  186. void STARPU_LU(cublas_u12)(void *descr[], void *_args)
  187. {
  188. STARPU_LU(common_u12)(descr, 1, _args);
  189. }
  190. #endif /* STARPU_USE_CUDA */
  191. static struct starpu_perfmodel STARPU_LU(model_12) =
  192. {
  193. .type = STARPU_HISTORY_BASED,
  194. #ifdef STARPU_ATLAS
  195. .symbol = STARPU_LU_STR(lu_model_12_atlas)
  196. #elif defined(STARPU_GOTO)
  197. .symbol = STARPU_LU_STR(lu_model_12_goto)
  198. #else
  199. .symbol = STARPU_LU_STR(lu_model_12)
  200. #endif
  201. };
  202. struct starpu_codelet cl12 =
  203. {
  204. .cpu_funcs = {STARPU_LU(cpu_u12)},
  205. .cpu_funcs_name = {STRINGIFY(STARPU_LU(cpu_u12))},
  206. #ifdef STARPU_USE_CUDA
  207. .cuda_funcs = {STARPU_LU(cublas_u12)},
  208. CAN_EXECUTE
  209. #elif defined(STARPU_SIMGRID)
  210. .cuda_funcs = {(void*)1},
  211. #endif
  212. .cuda_flags = {STARPU_CUDA_ASYNC},
  213. .nbuffers = 2,
  214. .modes = {STARPU_R, STARPU_RW},
  215. .model = &STARPU_LU(model_12)
  216. };
  217. /*
  218. * U21
  219. */
  220. static inline void STARPU_LU(common_u21)(void *descr[], int s, void *_args)
  221. {
  222. (void)_args;
  223. TYPE *sub11;
  224. TYPE *sub21;
  225. sub11 = (TYPE *)STARPU_MATRIX_GET_PTR(descr[0]);
  226. sub21 = (TYPE *)STARPU_MATRIX_GET_PTR(descr[1]);
  227. unsigned ld11 = STARPU_MATRIX_GET_LD(descr[0]);
  228. unsigned ld21 = STARPU_MATRIX_GET_LD(descr[1]);
  229. unsigned nx21 = STARPU_MATRIX_GET_NX(descr[1]);
  230. unsigned ny21 = STARPU_MATRIX_GET_NY(descr[1]);
  231. #ifdef STARPU_USE_CUDA
  232. cublasStatus status;
  233. #endif
  234. switch (s)
  235. {
  236. case 0:
  237. CPU_TRSM("R", "U", "N", "U", nx21, ny21,
  238. (TYPE)1.0, sub11, ld11, sub21, ld21);
  239. break;
  240. #ifdef STARPU_USE_CUDA
  241. case 1:
  242. status = CUBLAS_TRSM(starpu_cublas_get_local_handle(),
  243. CUBLAS_SIDE_RIGHT, CUBLAS_FILL_MODE_UPPER, CUBLAS_OP_N, CUBLAS_DIAG_UNIT,
  244. ny21, nx21,
  245. (CUBLAS_TYPE*)&p1, (CUBLAS_TYPE*)sub11, ld11, (CUBLAS_TYPE*)sub21, ld21);
  246. if (status != CUBLAS_STATUS_SUCCESS)
  247. STARPU_CUBLAS_REPORT_ERROR(status);
  248. break;
  249. #endif
  250. default:
  251. STARPU_ABORT();
  252. break;
  253. }
  254. }
  255. void STARPU_LU(cpu_u21)(void *descr[], void *_args)
  256. {
  257. STARPU_LU(common_u21)(descr, 0, _args);
  258. }
  259. #ifdef STARPU_USE_CUDA
  260. void STARPU_LU(cublas_u21)(void *descr[], void *_args)
  261. {
  262. STARPU_LU(common_u21)(descr, 1, _args);
  263. }
  264. #endif
  265. static struct starpu_perfmodel STARPU_LU(model_21) =
  266. {
  267. .type = STARPU_HISTORY_BASED,
  268. #ifdef STARPU_ATLAS
  269. .symbol = STARPU_LU_STR(lu_model_21_atlas)
  270. #elif defined(STARPU_GOTO)
  271. .symbol = STARPU_LU_STR(lu_model_21_goto)
  272. #else
  273. .symbol = STARPU_LU_STR(lu_model_21)
  274. #endif
  275. };
  276. struct starpu_codelet cl21 =
  277. {
  278. .cpu_funcs = {STARPU_LU(cpu_u21)},
  279. .cpu_funcs_name = {STRINGIFY(STARPU_LU(cpu_u21))},
  280. #ifdef STARPU_USE_CUDA
  281. .cuda_funcs = {STARPU_LU(cublas_u21)},
  282. CAN_EXECUTE
  283. #elif defined(STARPU_SIMGRID)
  284. .cuda_funcs = {(void*)1},
  285. #endif
  286. .cuda_flags = {STARPU_CUDA_ASYNC},
  287. .nbuffers = 2,
  288. .modes = {STARPU_R, STARPU_RW},
  289. .model = &STARPU_LU(model_21)
  290. };
  291. /*
  292. * U11
  293. */
  294. static inline void STARPU_LU(common_u11)(void *descr[], int s, void *_args)
  295. {
  296. (void)_args;
  297. TYPE *sub11;
  298. sub11 = (TYPE *)STARPU_MATRIX_GET_PTR(descr[0]);
  299. unsigned long nx = STARPU_MATRIX_GET_NX(descr[0]);
  300. unsigned long ld = STARPU_MATRIX_GET_LD(descr[0]);
  301. unsigned long z;
  302. #ifdef STARPU_USE_CUDA
  303. cublasStatus status;
  304. cublasHandle_t handle;
  305. cudaStream_t stream;
  306. #endif
  307. switch (s)
  308. {
  309. case 0:
  310. for (z = 0; z < nx; z++)
  311. {
  312. TYPE pivot;
  313. pivot = sub11[z+z*ld];
  314. STARPU_ASSERT(!ISZERO(pivot));
  315. CPU_SCAL(nx - z - 1, (1.0/pivot), &sub11[z+(z+1)*ld], ld);
  316. CPU_GER(nx - z - 1, nx - z - 1, -1.0,
  317. &sub11[(z+1)+z*ld], 1,
  318. &sub11[z+(z+1)*ld], ld,
  319. &sub11[(z+1) + (z+1)*ld],ld);
  320. }
  321. break;
  322. #ifdef STARPU_USE_CUDA
  323. case 1:
  324. handle = starpu_cublas_get_local_handle();
  325. stream = starpu_cuda_get_local_stream();
  326. for (z = 0; z < nx; z++)
  327. {
  328. TYPE pivot;
  329. TYPE inv_pivot;
  330. cudaMemcpyAsync(&pivot, &sub11[z+z*ld], sizeof(TYPE), cudaMemcpyDeviceToHost, stream);
  331. cudaStreamSynchronize(stream);
  332. STARPU_ASSERT(!ISZERO(pivot));
  333. inv_pivot = 1.0/pivot;
  334. status = CUBLAS_SCAL(handle,
  335. nx - z - 1,
  336. (CUBLAS_TYPE*)&inv_pivot, (CUBLAS_TYPE*)&sub11[z+(z+1)*ld], ld);
  337. if (status != CUBLAS_STATUS_SUCCESS)
  338. STARPU_CUBLAS_REPORT_ERROR(status);
  339. status = CUBLAS_GER(handle,
  340. nx - z - 1, nx - z - 1,
  341. (CUBLAS_TYPE*)&m1,
  342. (CUBLAS_TYPE*)&sub11[(z+1)+z*ld], 1,
  343. (CUBLAS_TYPE*)&sub11[z+(z+1)*ld], ld,
  344. (CUBLAS_TYPE*)&sub11[(z+1) + (z+1)*ld],ld);
  345. if (status != CUBLAS_STATUS_SUCCESS)
  346. STARPU_CUBLAS_REPORT_ERROR(status);
  347. }
  348. cudaStreamSynchronize(stream);
  349. break;
  350. #endif
  351. default:
  352. STARPU_ABORT();
  353. break;
  354. }
  355. }
  356. void STARPU_LU(cpu_u11)(void *descr[], void *_args)
  357. {
  358. STARPU_LU(common_u11)(descr, 0, _args);
  359. }
  360. #ifdef STARPU_USE_CUDA
  361. void STARPU_LU(cublas_u11)(void *descr[], void *_args)
  362. {
  363. STARPU_LU(common_u11)(descr, 1, _args);
  364. }
  365. #endif /* STARPU_USE_CUDA */
  366. static struct starpu_perfmodel STARPU_LU(model_11) =
  367. {
  368. .type = STARPU_HISTORY_BASED,
  369. #ifdef STARPU_ATLAS
  370. .symbol = STARPU_LU_STR(lu_model_11_atlas)
  371. #elif defined(STARPU_GOTO)
  372. .symbol = STARPU_LU_STR(lu_model_11_goto)
  373. #else
  374. .symbol = STARPU_LU_STR(lu_model_11)
  375. #endif
  376. };
  377. struct starpu_codelet cl11 =
  378. {
  379. .cpu_funcs = {STARPU_LU(cpu_u11)},
  380. .cpu_funcs_name = {STRINGIFY(STARPU_LU(cpu_u11))},
  381. #ifdef STARPU_USE_CUDA
  382. .cuda_funcs = {STARPU_LU(cublas_u11)},
  383. CAN_EXECUTE
  384. #elif defined(STARPU_SIMGRID)
  385. .cuda_funcs = {(void*)1},
  386. #endif
  387. .nbuffers = 1,
  388. .modes = {STARPU_RW},
  389. .model = &STARPU_LU(model_11)
  390. };
  391. /*
  392. * U11 with pivoting
  393. */
  394. static inline void STARPU_LU(common_u11_pivot)(void *descr[],
  395. int s, void *_args)
  396. {
  397. TYPE *sub11;
  398. sub11 = (TYPE *)STARPU_MATRIX_GET_PTR(descr[0]);
  399. unsigned long nx = STARPU_MATRIX_GET_NX(descr[0]);
  400. unsigned long ld = STARPU_MATRIX_GET_LD(descr[0]);
  401. unsigned long z;
  402. struct piv_s *piv = _args;
  403. unsigned *ipiv = piv->piv;
  404. unsigned first = piv->first;
  405. #ifdef STARPU_USE_CUDA
  406. cublasStatus status;
  407. cublasHandle_t handle;
  408. cudaStream_t stream;
  409. #endif
  410. switch (s)
  411. {
  412. case 0:
  413. for (z = 0; z < nx; z++)
  414. {
  415. TYPE pivot;
  416. pivot = sub11[z+z*ld];
  417. if (fabs((double)(pivot)) < PIVOT_THRESHHOLD)
  418. {
  419. /* find the pivot */
  420. int piv_ind = CPU_IAMAX(nx - z, &sub11[z*(ld+1)], ld);
  421. ipiv[z + first] = piv_ind + z + first;
  422. /* swap if needed */
  423. if (piv_ind != 0)
  424. {
  425. CPU_SWAP(nx, &sub11[z*ld], 1, &sub11[(z+piv_ind)*ld], 1);
  426. }
  427. pivot = sub11[z+z*ld];
  428. }
  429. STARPU_ASSERT(!ISZERO(pivot));
  430. CPU_SCAL(nx - z - 1, (1.0/pivot), &sub11[z+(z+1)*ld], ld);
  431. CPU_GER(nx - z - 1, nx - z - 1, -1.0,
  432. &sub11[(z+1)+z*ld], 1,
  433. &sub11[z+(z+1)*ld], ld,
  434. &sub11[(z+1) + (z+1)*ld],ld);
  435. }
  436. break;
  437. #ifdef STARPU_USE_CUDA
  438. case 1:
  439. handle = starpu_cublas_get_local_handle();
  440. stream = starpu_cuda_get_local_stream();
  441. for (z = 0; z < nx; z++)
  442. {
  443. TYPE pivot;
  444. TYPE inv_pivot;
  445. cudaMemcpyAsync(&pivot, &sub11[z+z*ld], sizeof(TYPE), cudaMemcpyDeviceToHost, stream);
  446. cudaStreamSynchronize(stream);
  447. if (fabs((double)(pivot)) < PIVOT_THRESHHOLD)
  448. {
  449. /* find the pivot */
  450. int piv_ind;
  451. status = CUBLAS_IAMAX(handle,
  452. nx - z, (CUBLAS_TYPE*)&sub11[z*(ld+1)], ld, &piv_ind);
  453. piv_ind -= 1;
  454. if (status != CUBLAS_STATUS_SUCCESS)
  455. STARPU_CUBLAS_REPORT_ERROR(status);
  456. ipiv[z + first] = piv_ind + z + first;
  457. /* swap if needed */
  458. if (piv_ind != 0)
  459. {
  460. status = CUBLAS_SWAP(handle,
  461. nx,
  462. (CUBLAS_TYPE*)&sub11[z*ld], 1,
  463. (CUBLAS_TYPE*)&sub11[(z+piv_ind)*ld], 1);
  464. if (status != CUBLAS_STATUS_SUCCESS)
  465. STARPU_CUBLAS_REPORT_ERROR(status);
  466. }
  467. cudaMemcpyAsync(&pivot, &sub11[z+z*ld], sizeof(TYPE), cudaMemcpyDeviceToHost, stream);
  468. cudaStreamSynchronize(stream);
  469. }
  470. STARPU_ASSERT(!ISZERO(pivot));
  471. inv_pivot = 1.0/pivot;
  472. status = CUBLAS_SCAL(handle,
  473. nx - z - 1,
  474. (CUBLAS_TYPE*)&inv_pivot,
  475. (CUBLAS_TYPE*)&sub11[z+(z+1)*ld], ld);
  476. if (status != CUBLAS_STATUS_SUCCESS)
  477. STARPU_CUBLAS_REPORT_ERROR(status);
  478. status = CUBLAS_GER(handle,
  479. nx - z - 1, nx - z - 1,
  480. (CUBLAS_TYPE*)&m1,
  481. (CUBLAS_TYPE*)&sub11[(z+1)+z*ld], 1,
  482. (CUBLAS_TYPE*)&sub11[z+(z+1)*ld], ld,
  483. (CUBLAS_TYPE*)&sub11[(z+1) + (z+1)*ld],ld);
  484. if (status != CUBLAS_STATUS_SUCCESS)
  485. STARPU_CUBLAS_REPORT_ERROR(status);
  486. }
  487. cudaStreamSynchronize(stream);
  488. break;
  489. #endif
  490. default:
  491. STARPU_ABORT();
  492. break;
  493. }
  494. }
  495. void STARPU_LU(cpu_u11_pivot)(void *descr[], void *_args)
  496. {
  497. STARPU_LU(common_u11_pivot)(descr, 0, _args);
  498. }
  499. #ifdef STARPU_USE_CUDA
  500. void STARPU_LU(cublas_u11_pivot)(void *descr[], void *_args)
  501. {
  502. STARPU_LU(common_u11_pivot)(descr, 1, _args);
  503. }
  504. #endif /* STARPU_USE_CUDA */
  505. static struct starpu_perfmodel STARPU_LU(model_11_pivot) =
  506. {
  507. .type = STARPU_HISTORY_BASED,
  508. #ifdef STARPU_ATLAS
  509. .symbol = STARPU_LU_STR(lu_model_11_pivot_atlas)
  510. #elif defined(STARPU_GOTO)
  511. .symbol = STARPU_LU_STR(lu_model_11_pivot_goto)
  512. #else
  513. .symbol = STARPU_LU_STR(lu_model_11_pivot)
  514. #endif
  515. };
  516. struct starpu_codelet cl11_pivot =
  517. {
  518. .cpu_funcs = {STARPU_LU(cpu_u11_pivot)},
  519. .cpu_funcs_name = {STRINGIFY(STARPU_LU(cpu_u11_pivot))},
  520. #ifdef STARPU_USE_CUDA
  521. .cuda_funcs = {STARPU_LU(cublas_u11_pivot)},
  522. CAN_EXECUTE
  523. #elif defined(STARPU_SIMGRID)
  524. .cuda_funcs = {(void*)1},
  525. #endif
  526. .nbuffers = 1,
  527. .modes = {STARPU_RW},
  528. .model = &STARPU_LU(model_11_pivot)
  529. };
  530. /*
  531. * Pivoting
  532. */
  533. static inline void STARPU_LU(common_pivot)(void *descr[],
  534. int s, void *_args)
  535. {
  536. TYPE *matrix;
  537. matrix = (TYPE *)STARPU_MATRIX_GET_PTR(descr[0]);
  538. unsigned long nx = STARPU_MATRIX_GET_NX(descr[0]);
  539. unsigned long ld = STARPU_MATRIX_GET_LD(descr[0]);
  540. unsigned row;
  541. struct piv_s *piv = _args;
  542. unsigned *ipiv = piv->piv;
  543. unsigned first = piv->first;
  544. #ifdef STARPU_USE_CUDA
  545. cublasStatus status;
  546. cublasHandle_t handle;
  547. #endif
  548. switch (s)
  549. {
  550. case 0:
  551. for (row = 0; row < nx; row++)
  552. {
  553. unsigned rowpiv = ipiv[row+first] - first;
  554. if (rowpiv != row)
  555. {
  556. CPU_SWAP(nx, &matrix[row*ld], 1, &matrix[rowpiv*ld], 1);
  557. }
  558. }
  559. break;
  560. #ifdef STARPU_USE_CUDA
  561. case 1:
  562. handle = starpu_cublas_get_local_handle();
  563. for (row = 0; row < nx; row++)
  564. {
  565. unsigned rowpiv = ipiv[row+first] - first;
  566. if (rowpiv != row)
  567. {
  568. status = CUBLAS_SWAP(handle,
  569. nx,
  570. (CUBLAS_TYPE*)&matrix[row*ld], 1,
  571. (CUBLAS_TYPE*)&matrix[rowpiv*ld], 1);
  572. if (status != CUBLAS_STATUS_SUCCESS)
  573. STARPU_CUBLAS_REPORT_ERROR(status);
  574. }
  575. }
  576. break;
  577. #endif
  578. default:
  579. STARPU_ABORT();
  580. break;
  581. }
  582. }
  583. void STARPU_LU(cpu_pivot)(void *descr[], void *_args)
  584. {
  585. STARPU_LU(common_pivot)(descr, 0, _args);
  586. }
  587. #ifdef STARPU_USE_CUDA
  588. void STARPU_LU(cublas_pivot)(void *descr[], void *_args)
  589. {
  590. STARPU_LU(common_pivot)(descr, 1, _args);
  591. }
  592. #endif /* STARPU_USE_CUDA */
  593. static struct starpu_perfmodel STARPU_LU(model_pivot) =
  594. {
  595. .type = STARPU_HISTORY_BASED,
  596. #ifdef STARPU_ATLAS
  597. .symbol = STARPU_LU_STR(lu_model_pivot_atlas)
  598. #elif defined(STARPU_GOTO)
  599. .symbol = STARPU_LU_STR(lu_model_pivot_goto)
  600. #else
  601. .symbol = STARPU_LU_STR(lu_model_pivot)
  602. #endif
  603. };
  604. struct starpu_codelet cl_pivot =
  605. {
  606. .cpu_funcs = {STARPU_LU(cpu_pivot)},
  607. .cpu_funcs_name = {STRINGIFY(STARPU_LU(cpu_pivot))},
  608. #ifdef STARPU_USE_CUDA
  609. .cuda_funcs = {STARPU_LU(cublas_pivot)},
  610. CAN_EXECUTE
  611. #elif defined(STARPU_SIMGRID)
  612. .cuda_funcs = {(void*)1},
  613. #endif
  614. .cuda_flags = {STARPU_CUDA_ASYNC},
  615. .nbuffers = 1,
  616. .modes = {STARPU_RW},
  617. .model = &STARPU_LU(model_pivot)
  618. };