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

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