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 24 kB

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