xlu_kernels.c 16 KB

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