You can not select more than 25 topics Topics must start with a chinese character,a letter or number, can include dashes ('-') and can be up to 35 characters long.

dgemm_kernel_power10.c 23 kB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864
  1. /*********************************************************************************
  2. Copyright (c) 2020, The OpenBLAS Project
  3. All rights reserved.
  4. Redistribution and use in source and binary forms, with or without
  5. modification, are permitted provided that the following conditions are
  6. met:
  7. 1. Redistributions of source code must retain the above copyright
  8. notice, this list of conditions and the following disclaimer.
  9. 2. Redistributions in binary form must reproduce the above copyright
  10. notice, this list of conditions and the following disclaimer in
  11. the documentation and/or other materials provided with the
  12. distribution.
  13. 3. Neither the name of the OpenBLAS project nor the names of
  14. its contributors may be used to endorse or promote products
  15. derived from this software without specific prior written permission.
  16. THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
  17. AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  18. IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  19. ARE DISCLAIMED. IN NO EVENT SHALL THE OPENBLAS PROJECT OR CONTRIBUTORS BE
  20. LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
  21. DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
  22. SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
  23. CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
  24. OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE
  25. USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  26. **********************************************************************************/
  27. #include "common.h"
  28. #include <altivec.h>
  29. typedef unsigned char vec_t __attribute__ ((vector_size (16)));
  30. typedef FLOAT v4sf_t __attribute__ ((vector_size (16)));
  31. typedef FLOAT v2sf_t __attribute__ ((vector_size (8)));
  32. #ifdef TRMMKERNEL
  33. #define SAVE_ACC(ACC, J) \
  34. __builtin_mma_disassemble_acc (result, ACC); \
  35. rowC = (v4sf_t *) &CO[0* ldc+J]; \
  36. rowC[0] = result[3] * alpha; \
  37. rowC = (v4sf_t *) &CO[1*ldc+J]; \
  38. rowC[0] = result[2] * alpha; \
  39. rowC = (v4sf_t *) &CO[2*ldc+J]; \
  40. rowC[0] = result[1] * alpha; \
  41. rowC = (v4sf_t *) &CO[3*ldc+J]; \
  42. rowC[0] = result[0] * alpha;
  43. #define SAVE_ACC1(ACC, J) \
  44. __builtin_mma_disassemble_acc (result, ACC); \
  45. rowC = (v4sf_t *) &CO[4* ldc+J]; \
  46. rowC[0] = result[3] * alpha; \
  47. rowC = (v4sf_t *) &CO[5*ldc+J]; \
  48. rowC[0] = result[2] * alpha; \
  49. rowC = (v4sf_t *) &CO[6*ldc+J]; \
  50. rowC[0] = result[1] * alpha; \
  51. rowC = (v4sf_t *) &CO[7*ldc+J]; \
  52. rowC[0] = result[0] * alpha;
  53. #define SAVE2x4_ACC(ACC, J) \
  54. __builtin_mma_disassemble_acc (result, ACC); \
  55. rowC = (v4sf_t *) &CO[0* ldc+J]; \
  56. rowC[0] = result[3] * alpha; \
  57. rowC = (v4sf_t *) &CO[1* ldc+J]; \
  58. rowC[0] = result[2] * alpha;
  59. #else
  60. #define SAVE_ACC(ACC, J) \
  61. __builtin_mma_disassemble_acc (result, ACC); \
  62. rowC = (v4sf_t *) &CO[0* ldc+J]; \
  63. rowC[0] += result[3] * alpha; \
  64. rowC = (v4sf_t *) &CO[1*ldc+J]; \
  65. rowC[0] += result[2] * alpha; \
  66. rowC = (v4sf_t *) &CO[2*ldc+J]; \
  67. rowC[0] += result[1] * alpha; \
  68. rowC = (v4sf_t *) &CO[3*ldc+J]; \
  69. rowC[0] += result[0] * alpha;
  70. #define SAVE_ACC1(ACC, J) \
  71. __builtin_mma_disassemble_acc (result, ACC); \
  72. rowC = (v4sf_t *) &CO[4* ldc+J]; \
  73. rowC[0] += result[3] * alpha; \
  74. rowC = (v4sf_t *) &CO[5*ldc+J]; \
  75. rowC[0] += result[2] * alpha; \
  76. rowC = (v4sf_t *) &CO[6*ldc+J]; \
  77. rowC[0] += result[1] * alpha; \
  78. rowC = (v4sf_t *) &CO[7*ldc+J]; \
  79. rowC[0] += result[0] * alpha;
  80. #define SAVE2x4_ACC(ACC, J) \
  81. __builtin_mma_disassemble_acc (result, ACC); \
  82. rowC = (v4sf_t *) &CO[0* ldc+J]; \
  83. rowC[0] += result[3] * alpha; \
  84. rowC = (v4sf_t *) &CO[1* ldc+J]; \
  85. rowC[0] += result[2] * alpha;
  86. #endif
  87. #define SET_ACC_ZERO4() \
  88. __builtin_mma_xxsetaccz (&acc0); \
  89. __builtin_mma_xxsetaccz (&acc1); \
  90. __builtin_mma_xxsetaccz (&acc2); \
  91. __builtin_mma_xxsetaccz (&acc3);
  92. #define SET_ACC_ZERO8() \
  93. __builtin_mma_xxsetaccz (&acc0); \
  94. __builtin_mma_xxsetaccz (&acc1); \
  95. __builtin_mma_xxsetaccz (&acc2); \
  96. __builtin_mma_xxsetaccz (&acc3); \
  97. __builtin_mma_xxsetaccz (&acc4); \
  98. __builtin_mma_xxsetaccz (&acc5); \
  99. __builtin_mma_xxsetaccz (&acc6); \
  100. __builtin_mma_xxsetaccz (&acc7);
  101. #define PREFETCH1(x, y) asm volatile ("dcbt %0, %1" : : "r" (x), "b" (y) : "memory");
  102. #if (defined(LEFT) && !defined(TRANSA)) || (!defined(LEFT) && defined(TRANSA))
  103. #define REFRESH_TEMP_BK(x, y) \
  104. temp = k - off;
  105. #elif defined(LEFT)
  106. #define REFRESH_TEMP_BK(x, y) \
  107. temp = off + x;
  108. #else
  109. #define REFRESH_TEMP_BK(x, y) \
  110. temp = off + y;
  111. #endif
  112. #if (defined(LEFT) && defined(TRANSA)) || (!defined(LEFT) && !defined(TRANSA))
  113. #define REFRESH_POINTERS(x, y) \
  114. BO = B; \
  115. REFRESH_TEMP_BK(x, y)
  116. #else
  117. #define REFRESH_POINTERS(x, y) \
  118. AO += off * x; \
  119. BO = B + off * y; \
  120. REFRESH_TEMP_BK(x, y)
  121. #endif
  122. #ifdef LEFT
  123. #define REFRESH_OFF(x) \
  124. off += x;
  125. #else
  126. #define REFRESH_OFF(x)
  127. #endif
  128. #ifdef LEFT
  129. #define UPDATE_TEMP(x, y) \
  130. temp -= x;
  131. #else
  132. #define UPDATE_TEMP(x, y) \
  133. temp -= y;
  134. #endif
  135. #if (defined(LEFT) && defined(TRANSA)) || (!defined(LEFT) && !defined(TRANSA))
  136. #define REFRESH_TMP_AFTER_SAVE(x, y) \
  137. temp = k - off; \
  138. UPDATE_TEMP(x, y) \
  139. AO += temp * x; \
  140. BO += temp * y;
  141. #else
  142. #define REFRESH_TMP_AFTER_SAVE(x, y)
  143. #endif
  144. #define REFRESH_AFTER_SAVE(x,y) \
  145. REFRESH_TMP_AFTER_SAVE(x, y) \
  146. REFRESH_OFF(x)
  147. /*************************************************************************************
  148. * GEMM Kernel
  149. *************************************************************************************/
  150. int
  151. CNAME (BLASLONG m, BLASLONG n, BLASLONG k, FLOAT alpha, FLOAT * A, FLOAT * B,
  152. FLOAT * C, BLASLONG ldc
  153. #ifdef TRMMKERNEL
  154. , BLASLONG offset
  155. #endif
  156. )
  157. {
  158. BLASLONG N = n;
  159. BLASLONG i1;
  160. #if defined(TRMMKERNEL)
  161. BLASLONG off;
  162. #endif
  163. #if defined(TRMMKERNEL) && !defined(LEFT)
  164. off = -offset;
  165. #endif
  166. v4sf_t valpha = { alpha, alpha };
  167. N = n >> 2;
  168. for (i1 = 0; i1 < N; i1++)
  169. {
  170. BLASLONG i, j, temp;
  171. FLOAT *CO;
  172. FLOAT *AO;
  173. #if defined(TRMMKERNEL) && defined(LEFT)
  174. off = offset;
  175. #endif
  176. CO = C;
  177. C += ldc << 2;
  178. AO = A;
  179. PREFETCH1 (A, 128);
  180. PREFETCH1 (A, 256);
  181. i = m >> 4;
  182. for (j = 0; j < i; j++)
  183. {
  184. FLOAT *BO;
  185. #if defined(TRMMKERNEL)
  186. REFRESH_POINTERS (16, 4);
  187. #else
  188. BO = B;
  189. temp = k;
  190. #endif
  191. v4sf_t *rowC;
  192. v4sf_t result[4];
  193. BLASLONG l = 0;
  194. PREFETCH1 (CO, 0);
  195. PREFETCH1 (CO + ldc, 0);
  196. PREFETCH1 (CO + ldc + ldc, 0);
  197. PREFETCH1 (CO + ldc + ldc + ldc, 0);
  198. PREFETCH1 (CO, 128);
  199. PREFETCH1 (CO + ldc, 128);
  200. PREFETCH1 (CO + ldc + ldc, 128);
  201. PREFETCH1 (CO + ldc + ldc + ldc, 128);
  202. __vector_quad acc0, acc1, acc2, acc3, acc4, acc5, acc6, acc7;
  203. SET_ACC_ZERO8 ();
  204. for (l = 0; l < temp; l++)
  205. {
  206. vec_t *rowA = (vec_t *) & AO[l << 4];
  207. __vector_pair rowB;
  208. vec_t *rb = (vec_t *) & BO[l << 2];
  209. __builtin_mma_assemble_pair (&rowB, rb[1], rb[0]);
  210. __builtin_mma_xvf64gerpp (&acc0, rowB, rowA[0]);
  211. __builtin_mma_xvf64gerpp (&acc1, rowB, rowA[1]);
  212. __builtin_mma_xvf64gerpp (&acc2, rowB, rowA[2]);
  213. __builtin_mma_xvf64gerpp (&acc3, rowB, rowA[3]);
  214. __builtin_mma_xvf64gerpp (&acc4, rowB, rowA[4]);
  215. __builtin_mma_xvf64gerpp (&acc5, rowB, rowA[5]);
  216. __builtin_mma_xvf64gerpp (&acc6, rowB, rowA[6]);
  217. __builtin_mma_xvf64gerpp (&acc7, rowB, rowA[7]);
  218. }
  219. SAVE_ACC (&acc0, 0);
  220. SAVE_ACC (&acc2, 4);
  221. SAVE_ACC (&acc1, 2);
  222. SAVE_ACC (&acc3, 6);
  223. SAVE_ACC (&acc4, 8);
  224. SAVE_ACC (&acc6, 12);
  225. SAVE_ACC (&acc5, 10);
  226. SAVE_ACC (&acc7, 14);
  227. AO += temp << 4;
  228. BO += temp << 2;
  229. #if defined(TRMMKERNEL)
  230. REFRESH_AFTER_SAVE (16, 4)
  231. #endif
  232. CO += 16;
  233. }
  234. i = (m & 15) >> 3;
  235. for (j = 0; j < i; j++)
  236. {
  237. FLOAT *BO;
  238. #if defined(TRMMKERNEL)
  239. REFRESH_POINTERS (8, 4);
  240. #else
  241. BO = B;
  242. temp = k;
  243. #endif
  244. v4sf_t *rowC;
  245. v4sf_t result[4];
  246. __vector_quad acc0, acc1, acc2, acc3;
  247. SET_ACC_ZERO4 ();
  248. BLASLONG l = 0;
  249. for (l = 0; l < temp; l++)
  250. {
  251. vec_t *rowA = (vec_t *) & AO[l << 3];
  252. __vector_pair rowB;
  253. vec_t *rb = (vec_t *) & BO[l << 2];
  254. __builtin_mma_assemble_pair (&rowB, rb[1], rb[0]);
  255. __builtin_mma_xvf64gerpp (&acc0, rowB, rowA[0]);
  256. __builtin_mma_xvf64gerpp (&acc1, rowB, rowA[1]);
  257. __builtin_mma_xvf64gerpp (&acc2, rowB, rowA[2]);
  258. __builtin_mma_xvf64gerpp (&acc3, rowB, rowA[3]);
  259. }
  260. SAVE_ACC (&acc0, 0);
  261. SAVE_ACC (&acc2, 4);
  262. SAVE_ACC (&acc1, 2);
  263. SAVE_ACC (&acc3, 6);
  264. CO += 8;
  265. AO += temp << 3;
  266. BO += temp << 2;
  267. #if defined(TRMMKERNEL)
  268. REFRESH_AFTER_SAVE (8, 4)
  269. #endif
  270. }
  271. i = (m & 7) >> 2;
  272. for (j = 0; j < i; j++)
  273. {
  274. FLOAT *BO;
  275. #if defined(TRMMKERNEL)
  276. REFRESH_POINTERS (4, 4);
  277. #else
  278. BO = B;
  279. temp = k;
  280. #endif
  281. v4sf_t *rowC;
  282. v4sf_t result[4];
  283. __vector_quad acc0, acc1;
  284. __builtin_mma_xxsetaccz (&acc0);
  285. __builtin_mma_xxsetaccz (&acc1);
  286. BLASLONG l = 0;
  287. for (l = 0; l < temp; l++)
  288. {
  289. vec_t *rowA = (vec_t *) & AO[l << 2];
  290. __vector_pair rowB;
  291. vec_t *rb = (vec_t *) & BO[l << 2];
  292. __builtin_mma_assemble_pair (&rowB, rb[1], rb[0]);
  293. __builtin_mma_xvf64gerpp (&acc0, rowB, rowA[0]);
  294. __builtin_mma_xvf64gerpp (&acc1, rowB, rowA[1]);
  295. }
  296. SAVE_ACC (&acc0, 0);
  297. SAVE_ACC (&acc1, 2);
  298. CO += 4;
  299. AO += temp << 2;
  300. BO += temp << 2;
  301. #if defined(TRMMKERNEL)
  302. REFRESH_AFTER_SAVE (4, 4)
  303. #endif
  304. }
  305. i = (m & 3) >> 1;
  306. for (j = 0; j < i; j++)
  307. {
  308. FLOAT *BO;
  309. #if defined(TRMMKERNEL)
  310. REFRESH_POINTERS (2, 4);
  311. #else
  312. BO = B;
  313. temp = k;
  314. #endif
  315. v4sf_t *rowC;
  316. v4sf_t result[4];
  317. __vector_quad acc0;
  318. __builtin_mma_xxsetaccz (&acc0);
  319. BLASLONG l = 0;
  320. for (l = 0; l < temp; l++)
  321. {
  322. vec_t *rowA = (vec_t *) & AO[l << 1];
  323. __vector_pair rowB;
  324. vec_t *rb = (vec_t *) & BO[l << 2];
  325. __builtin_mma_assemble_pair (&rowB, rb[1], rb[0]);
  326. __builtin_mma_xvf64gerpp (&acc0, rowB, rowA[0]);
  327. }
  328. SAVE_ACC (&acc0, 0);
  329. CO += 2;
  330. AO += temp << 1;
  331. BO += temp << 2;
  332. #if defined(TRMMKERNEL)
  333. REFRESH_AFTER_SAVE (2, 4)
  334. #endif
  335. }
  336. i = (m & 1) >> 0;
  337. for (j = 0; j < i; j++)
  338. {
  339. FLOAT *BO;
  340. #if defined(TRMMKERNEL)
  341. REFRESH_POINTERS (1, 4);
  342. #else
  343. BO = B;
  344. temp = k;
  345. #endif
  346. BLASLONG l = 0;
  347. v4sf_t t = { 0, 0 };
  348. v4sf_t t1 = { 0, 0 };
  349. for (l = 0; l < temp; l++)
  350. {
  351. v4sf_t rowA = { AO[l], AO[l] };
  352. v4sf_t rowB = { BO[l << 2], BO[(l << 2) + 1] };
  353. v4sf_t rowB1 = { BO[(l << 2) + 2], BO[(l << 2) + 3] };
  354. t += rowA * rowB;
  355. t1 += rowA * rowB1;
  356. }
  357. t = t * valpha;
  358. t1 = t1 * valpha;
  359. #if defined(TRMMKERNEL)
  360. CO[0 * ldc] = t[0];
  361. CO[1 * ldc] = t[1];
  362. CO[2 * ldc] = t1[0];
  363. CO[3 * ldc] = t1[1];
  364. #else
  365. CO[0 * ldc] += t[0];
  366. CO[1 * ldc] += t[1];
  367. CO[2 * ldc] += t1[0];
  368. CO[3 * ldc] += t1[1];
  369. #endif
  370. CO += 1;
  371. AO += temp;
  372. BO += temp << 2;
  373. #if defined(TRMMKERNEL)
  374. REFRESH_AFTER_SAVE (1, 4)
  375. #endif
  376. }
  377. #if defined(TRMMKERNEL) && !defined(LEFT)
  378. off += 4; // number of values in A
  379. #endif
  380. B += k << 2;
  381. }
  382. N = (n & 3) >> 1;
  383. for (i1 = 0; i1 < N; i1++)
  384. {
  385. BLASLONG i, j, temp;
  386. #if defined(TRMMKERNEL) && defined(LEFT)
  387. off = offset;
  388. #endif
  389. FLOAT *CO;
  390. FLOAT *AO;
  391. CO = C;
  392. C += ldc << 1;
  393. AO = A;
  394. i = m >> 4;
  395. for (j = 0; j < i; j++)
  396. {
  397. FLOAT *BO;
  398. #if defined(TRMMKERNEL)
  399. REFRESH_POINTERS (16, 2);
  400. #else
  401. BO = B;
  402. temp = k;
  403. #endif
  404. v4sf_t *rowC;
  405. v4sf_t result[4];
  406. __vector_quad acc0, acc1, acc2, acc3, acc4, acc5, acc6, acc7;
  407. SET_ACC_ZERO8 ();
  408. BLASLONG l = 0;
  409. for (l = 0; l < temp; l++)
  410. {
  411. FLOAT t[4] = { 0, 0, 0, 0 };
  412. t[0] = BO[l << 1], t[1] = BO[(l << 1) + 1];
  413. __vector_pair rowB;
  414. vec_t *rb = (vec_t *) & t[0];
  415. __builtin_mma_assemble_pair (&rowB, rb[1], rb[0]);
  416. vec_t *rowA = (vec_t *) & AO[l << 4];
  417. __builtin_mma_xvf64gerpp (&acc0, rowB, rowA[0]);
  418. __builtin_mma_xvf64gerpp (&acc1, rowB, rowA[1]);
  419. __builtin_mma_xvf64gerpp (&acc2, rowB, rowA[2]);
  420. __builtin_mma_xvf64gerpp (&acc3, rowB, rowA[3]);
  421. __builtin_mma_xvf64gerpp (&acc4, rowB, rowA[4]);
  422. __builtin_mma_xvf64gerpp (&acc5, rowB, rowA[5]);
  423. __builtin_mma_xvf64gerpp (&acc6, rowB, rowA[6]);
  424. __builtin_mma_xvf64gerpp (&acc7, rowB, rowA[7]);
  425. }
  426. SAVE2x4_ACC (&acc0, 0);
  427. SAVE2x4_ACC (&acc1, 2);
  428. SAVE2x4_ACC (&acc2, 4);
  429. SAVE2x4_ACC (&acc3, 6);
  430. SAVE2x4_ACC (&acc4, 8);
  431. SAVE2x4_ACC (&acc5, 10);
  432. SAVE2x4_ACC (&acc6, 12);
  433. SAVE2x4_ACC (&acc7, 14);
  434. CO += 16;
  435. AO += temp << 4;
  436. BO += temp << 1;
  437. #if defined(TRMMKERNEL)
  438. REFRESH_AFTER_SAVE (16, 2)
  439. #endif
  440. }
  441. i = (m & 15) >> 3;
  442. for (j = 0; j < i; j++)
  443. {
  444. FLOAT *BO;
  445. #if defined(TRMMKERNEL)
  446. REFRESH_POINTERS (8, 2);
  447. #else
  448. BO = B;
  449. temp = k;
  450. #endif
  451. v4sf_t *rowC;
  452. v4sf_t result[4];
  453. __vector_quad acc0, acc1, acc2, acc3;
  454. SET_ACC_ZERO4 ();
  455. BLASLONG l = 0;
  456. for (l = 0; l < temp; l++)
  457. {
  458. FLOAT t[4] = { 0, 0, 0, 0 };
  459. t[0] = BO[l << 1], t[1] = BO[(l << 1) + 1];
  460. __vector_pair rowB;
  461. vec_t *rb = (vec_t *) & t[0];
  462. __builtin_mma_assemble_pair (&rowB, rb[1], rb[0]);
  463. vec_t *rowA = (vec_t *) & AO[l << 3];
  464. __builtin_mma_xvf64gerpp (&acc0, rowB, rowA[0]);
  465. __builtin_mma_xvf64gerpp (&acc1, rowB, rowA[1]);
  466. __builtin_mma_xvf64gerpp (&acc2, rowB, rowA[2]);
  467. __builtin_mma_xvf64gerpp (&acc3, rowB, rowA[3]);
  468. }
  469. SAVE2x4_ACC (&acc0, 0);
  470. SAVE2x4_ACC (&acc1, 2);
  471. SAVE2x4_ACC (&acc2, 4);
  472. SAVE2x4_ACC (&acc3, 6);
  473. CO += 8;
  474. AO += temp << 3;
  475. BO += temp << 1;
  476. #if defined(TRMMKERNEL)
  477. REFRESH_AFTER_SAVE (8, 2)
  478. #endif
  479. }
  480. i = (m & 7) >> 2;
  481. for (j = 0; j < i; j++)
  482. {
  483. FLOAT *BO;
  484. #if defined(TRMMKERNEL)
  485. REFRESH_POINTERS (4, 2);
  486. #else
  487. BO = B;
  488. temp = k;
  489. #endif
  490. v4sf_t *rowC;
  491. v4sf_t result[4];
  492. __vector_quad acc0, acc1;
  493. __builtin_mma_xxsetaccz (&acc0);
  494. __builtin_mma_xxsetaccz (&acc1);
  495. BLASLONG l = 0;
  496. for (l = 0; l < temp; l++)
  497. {
  498. FLOAT t[4] = { 0, 0, 0, 0 };
  499. t[0] = BO[l << 1], t[1] = BO[(l << 1) + 1];
  500. __vector_pair rowB;
  501. vec_t *rb = (vec_t *) & t[0];
  502. __builtin_mma_assemble_pair (&rowB, rb[1], rb[0]);
  503. vec_t *rowA = (vec_t *) & AO[l << 2];
  504. __builtin_mma_xvf64gerpp (&acc0, rowB, rowA[0]);
  505. __builtin_mma_xvf64gerpp (&acc1, rowB, rowA[1]);
  506. }
  507. SAVE2x4_ACC (&acc0, 0);
  508. SAVE2x4_ACC (&acc1, 2);
  509. CO += 4;
  510. AO += temp << 2;
  511. BO += temp << 1;
  512. #if defined(TRMMKERNEL)
  513. REFRESH_AFTER_SAVE (4, 2)
  514. #endif
  515. }
  516. i = (m & 3) >> 1;
  517. for (j = 0; j < i; j++)
  518. {
  519. FLOAT *BO;
  520. #if defined(TRMMKERNEL)
  521. REFRESH_POINTERS (2, 2);
  522. #else
  523. BO = B;
  524. temp = k;
  525. #endif
  526. v4sf_t *rowC;
  527. v4sf_t result[4];
  528. __vector_quad acc0;
  529. __builtin_mma_xxsetaccz (&acc0);
  530. BLASLONG l = 0;
  531. for (l = 0; l < temp; l++)
  532. {
  533. FLOAT t[4] = { 0, 0, 0, 0 };
  534. t[0] = BO[l << 1], t[1] = BO[(l << 1) + 1];
  535. __vector_pair rowB;
  536. vec_t *rb = (vec_t *) & t[0];
  537. __builtin_mma_assemble_pair (&rowB, rb[1], rb[0]);
  538. vec_t *rowA = (vec_t *) & AO[l << 1];
  539. __builtin_mma_xvf64gerpp (&acc0, rowB, rowA[0]);
  540. }
  541. SAVE2x4_ACC (&acc0, 0);
  542. CO += 2;
  543. AO += temp << 1;
  544. BO += temp << 1;
  545. #if defined(TRMMKERNEL)
  546. REFRESH_AFTER_SAVE (2, 2)
  547. #endif
  548. }
  549. i = (m & 1) >> 0;
  550. for (j = 0; j < i; j++)
  551. {
  552. FLOAT *BO;
  553. #if defined(TRMMKERNEL)
  554. REFRESH_POINTERS (1, 2);
  555. #else
  556. BO = B;
  557. temp = k;
  558. #endif
  559. BLASLONG l = 0;
  560. v4sf_t t = { 0, 0 };
  561. for (l = 0; l < temp; l++)
  562. {
  563. v4sf_t rowA = { AO[l], AO[l] };
  564. v4sf_t rowB = { BO[l << 1], BO[(l << 1) + 1] };
  565. t += rowA * rowB;
  566. }
  567. t = t * valpha;
  568. #if defined(TRMMKERNEL)
  569. CO[0 * ldc] = t[0];
  570. CO[1 * ldc] = t[1];
  571. #else
  572. CO[0 * ldc] += t[0];
  573. CO[1 * ldc] += t[1];
  574. #endif
  575. CO += 1;
  576. AO += temp;
  577. BO += temp << 1;
  578. #if defined(TRMMKERNEL)
  579. REFRESH_AFTER_SAVE (1, 2)
  580. #endif
  581. }
  582. #if defined(TRMMKERNEL) && !defined(LEFT)
  583. off += 2; // number of values in A
  584. #endif
  585. B += k << 1;
  586. }
  587. N = (n & 1) >> 0;
  588. for (i1 = 0; i1 < N; i1++)
  589. {
  590. BLASLONG i, temp;
  591. #if defined(TRMMKERNEL) && defined(LEFT)
  592. off = offset;
  593. #endif
  594. FLOAT *CO;
  595. FLOAT *AO;
  596. CO = C;
  597. C += ldc;
  598. AO = A;
  599. i = m;
  600. while (i >= 16)
  601. {
  602. FLOAT *BO;
  603. #if defined(TRMMKERNEL)
  604. REFRESH_POINTERS (16, 1)
  605. #else
  606. BO = B;
  607. temp = k;
  608. #endif
  609. BLASLONG l = 0;
  610. v4sf_t t = { 0, 0 };
  611. v4sf_t t1 = { 0, 0 };
  612. v4sf_t t2 = { 0, 0 };
  613. v4sf_t t3 = { 0, 0 };
  614. v4sf_t t4 = { 0, 0 };
  615. v4sf_t t5 = { 0, 0 };
  616. v4sf_t t6 = { 0, 0 };
  617. v4sf_t t7 = { 0, 0 };
  618. for (l = 0; l < temp; l++)
  619. {
  620. v4sf_t rowB = { BO[l], BO[l] };
  621. v4sf_t rowA = { AO[l << 4], AO[(l << 4) + 1] };
  622. v4sf_t rowA1 = { AO[(l << 4) + 2], AO[(l << 4) + 3] };
  623. v4sf_t rowA2 = { AO[(l << 4) + 4], AO[(l << 4) + 5] };
  624. v4sf_t rowA3 = { AO[(l << 4) + 6], AO[(l << 4) + 7] };
  625. v4sf_t rowA4 = { AO[(l << 4) + 8], AO[(l << 4) + 9] };
  626. v4sf_t rowA5 = { AO[(l << 4) + 10], AO[(l << 4) + 11] };
  627. v4sf_t rowA6 = { AO[(l << 4) + 12], AO[(l << 4) + 13] };
  628. v4sf_t rowA7 = { AO[(l << 4) + 14], AO[(l << 4) + 15] };
  629. t += rowA * rowB;
  630. t1 += rowA1 * rowB;
  631. t2 += rowA2 * rowB;
  632. t3 += rowA3 * rowB;
  633. t4 += rowA4 * rowB;
  634. t5 += rowA5 * rowB;
  635. t6 += rowA6 * rowB;
  636. t7 += rowA7 * rowB;
  637. }
  638. t = t * valpha;
  639. t1 = t1 * valpha;
  640. t2 = t2 * valpha;
  641. t3 = t3 * valpha;
  642. t4 = t4 * valpha;
  643. t5 = t5 * valpha;
  644. t6 = t6 * valpha;
  645. t7 = t7 * valpha;
  646. #if defined(TRMMKERNEL)
  647. CO[0] = t[0];
  648. CO[1] = t[1];
  649. CO[2] = t1[0];
  650. CO[3] = t1[1];
  651. CO[4] = t2[0];
  652. CO[5] = t2[1];
  653. CO[6] = t3[0];
  654. CO[7] = t3[1];
  655. CO[8] = t4[0];
  656. CO[9] = t4[1];
  657. CO[10] = t5[0];
  658. CO[11] = t5[1];
  659. CO[12] = t6[0];
  660. CO[13] = t6[1];
  661. CO[14] = t7[0];
  662. CO[15] = t7[1];
  663. #else
  664. CO[0] += t[0];
  665. CO[1] += t[1];
  666. CO[2] += t1[0];
  667. CO[3] += t1[1];
  668. CO[4] += t2[0];
  669. CO[5] += t2[1];
  670. CO[6] += t3[0];
  671. CO[7] += t3[1];
  672. CO[8] += t4[0];
  673. CO[9] += t4[1];
  674. CO[10] += t5[0];
  675. CO[11] += t5[1];
  676. CO[12] += t6[0];
  677. CO[13] += t6[1];
  678. CO[14] += t7[0];
  679. CO[15] += t7[1];
  680. #endif
  681. AO += temp << 4;
  682. BO += temp;
  683. CO += 16;
  684. i -= 16;
  685. #if defined(TRMMKERNEL)
  686. REFRESH_AFTER_SAVE (16, 1)
  687. #endif
  688. }
  689. while (i >= 8)
  690. {
  691. FLOAT *BO;
  692. #if defined(TRMMKERNEL)
  693. REFRESH_POINTERS (8, 1)
  694. #else
  695. BO = B;
  696. temp = k;
  697. #endif
  698. BLASLONG l = 0;
  699. v4sf_t t = { 0, 0 };
  700. v4sf_t t1 = { 0, 0 };
  701. v4sf_t t2 = { 0, 0 };
  702. v4sf_t t3 = { 0, 0 };
  703. for (l = 0; l < temp; l++)
  704. {
  705. v4sf_t rowB = { BO[l], BO[l] };
  706. v4sf_t rowA = { AO[l << 3], AO[(l << 3) + 1] };
  707. v4sf_t rowA1 = { AO[(l << 3) + 2], AO[(l << 3) + 3] };
  708. v4sf_t rowA2 = { AO[(l << 3) + 4], AO[(l << 3) + 5] };
  709. v4sf_t rowA3 = { AO[(l << 3) + 6], AO[(l << 3) + 7] };
  710. t += rowA * rowB;
  711. t1 += rowA1 * rowB;
  712. t2 += rowA2 * rowB;
  713. t3 += rowA3 * rowB;
  714. }
  715. t = t * valpha;
  716. t1 = t1 * valpha;
  717. t2 = t2 * valpha;
  718. t3 = t3 * valpha;
  719. #if defined(TRMMKERNEL)
  720. CO[0] = t[0];
  721. CO[1] = t[1];
  722. CO[2] = t1[0];
  723. CO[3] = t1[1];
  724. CO[4] = t2[0];
  725. CO[5] = t2[1];
  726. CO[6] = t3[0];
  727. CO[7] = t3[1];
  728. #else
  729. CO[0] += t[0];
  730. CO[1] += t[1];
  731. CO[2] += t1[0];
  732. CO[3] += t1[1];
  733. CO[4] += t2[0];
  734. CO[5] += t2[1];
  735. CO[6] += t3[0];
  736. CO[7] += t3[1];
  737. #endif
  738. AO += temp << 3;
  739. BO += temp;
  740. CO += 8;
  741. i -= 8;
  742. #if defined(TRMMKERNEL)
  743. REFRESH_AFTER_SAVE (8, 1)
  744. #endif
  745. }
  746. while (i >= 4)
  747. {
  748. FLOAT *BO;
  749. #if defined(TRMMKERNEL)
  750. REFRESH_POINTERS (4, 1)
  751. #else
  752. BO = B;
  753. temp = k;
  754. #endif
  755. BLASLONG l = 0;
  756. v4sf_t t = { 0, 0 };
  757. v4sf_t t1 = { 0, 0 };
  758. for (l = 0; l < temp; l++)
  759. {
  760. v4sf_t rowB = { BO[l], BO[l] };
  761. v4sf_t rowA = { AO[l << 2], AO[(l << 2) + 1] };
  762. v4sf_t rowA1 = { AO[(l << 2) + 2], AO[(l << 2) + 3] };
  763. t += rowA * rowB;
  764. t1 += rowA1 * rowB;
  765. }
  766. t = t * valpha;
  767. t1 = t1 * valpha;
  768. #if defined(TRMMKERNEL)
  769. CO[0] = t[0];
  770. CO[1] = t[1];
  771. CO[2] = t1[0];
  772. CO[3] = t1[1];
  773. #else
  774. CO[0] += t[0];
  775. CO[1] += t[1];
  776. CO[2] += t1[0];
  777. CO[3] += t1[1];
  778. #endif
  779. AO += temp << 2;
  780. BO += temp;
  781. CO += 4;
  782. i -= 4;
  783. #if defined(TRMMKERNEL)
  784. REFRESH_AFTER_SAVE (4, 1)
  785. #endif
  786. }
  787. while (i >= 2)
  788. {
  789. FLOAT *BO;
  790. #if defined(TRMMKERNEL)
  791. REFRESH_POINTERS (2, 1)
  792. #else
  793. BO = B;
  794. temp = k;
  795. #endif
  796. BLASLONG l = 0;
  797. v4sf_t t = { 0, 0 };
  798. for (l = 0; l < temp; l++)
  799. {
  800. v4sf_t rowB = { BO[l], BO[l] };
  801. v4sf_t rowA = { AO[l << 1], AO[(l << 1) + 1] };
  802. t += rowA * rowB;
  803. }
  804. t = t * valpha;
  805. #if defined(TRMMKERNEL)
  806. CO[0] = t[0];
  807. CO[1] = t[1];
  808. #else
  809. CO[0] += t[0];
  810. CO[1] += t[1];
  811. #endif
  812. AO += temp << 1;
  813. BO += temp;
  814. CO += 2;
  815. i -= 2;
  816. #if defined(TRMMKERNEL)
  817. REFRESH_AFTER_SAVE (2, 1)
  818. #endif
  819. }
  820. while (i >= 1)
  821. {
  822. FLOAT *BO;
  823. #if defined(TRMMKERNEL)
  824. REFRESH_POINTERS (1, 1)
  825. #else
  826. BO = B;
  827. temp = k;
  828. #endif
  829. BLASLONG l = 0;
  830. FLOAT t = 0;
  831. for (l = 0; l < temp; l++)
  832. {
  833. t += AO[l] * BO[l];
  834. }
  835. AO += temp;
  836. BO += temp;
  837. #if defined(TRMMKERNEL)
  838. CO[0] = t * alpha;
  839. #else
  840. CO[0] += t * alpha;
  841. #endif
  842. CO += 1;
  843. i -= 1;
  844. #if defined(TRMMKERNEL)
  845. REFRESH_AFTER_SAVE (1, 1)
  846. #endif
  847. }
  848. #if defined(TRMMKERNEL) && !defined(LEFT)
  849. off += 1; // number of values in A
  850. #endif
  851. B += k;
  852. }
  853. return 0;
  854. }