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.

cgemm_kernel_power10.c 53 kB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154
  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. #define SET_ACC_ZERO() \
  33. __builtin_mma_xxsetaccz (&acc0); \
  34. __builtin_mma_xxsetaccz (&acc1); \
  35. __builtin_mma_xxsetaccz (&acc2); \
  36. __builtin_mma_xxsetaccz (&acc3); \
  37. __builtin_mma_xxsetaccz (&acc4); \
  38. __builtin_mma_xxsetaccz (&acc5); \
  39. __builtin_mma_xxsetaccz (&acc6); \
  40. __builtin_mma_xxsetaccz (&acc7);
  41. #if (defined(NN) || defined(NT) || defined(TN) || defined(TT))
  42. #define COMP_MUL(_real, _arbr, _aibi, _imag, _arbi, _aibr) { _real = _arbr - _aibi; _imag = _arbi + _aibr; }
  43. #define COMP_MAC(_real, _arbr, _aibi, _imag, _arbi, _aibr) { _real += _arbr - _aibi; _imag += _arbi + _aibr; }
  44. #endif
  45. #if (defined(NR) || defined(NC) || defined(TR) || defined(TC))
  46. #define COMP_MUL(_real, _arbr, _aibi, _imag, _arbi, _aibr) { _real = _arbr + _aibi; _imag = -_arbi + _aibr; }
  47. #define COMP_MAC(_real, _arbr, _aibi, _imag, _arbi, _aibr) { _real += _arbr + _aibi; _imag += -_arbi + _aibr; }
  48. #endif
  49. #if (defined(RN) || defined(RT) || defined(CN) || defined(CT))
  50. #define COMP_MUL(_real, _arbr, _aibi, _imag, _arbi, _aibr) { _real = _arbr + _aibi; _imag = _arbi - _aibr; }
  51. #define COMP_MAC(_real, _arbr, _aibi, _imag, _arbi, _aibr) { _real += _arbr + _aibi; _imag += _arbi - _aibr; }
  52. #endif
  53. #if (defined(RR) || defined(RC) || defined(CR) || defined(CC))
  54. #define COMP_MUL(_real, _arbr, _aibi, _imag, _arbi, _aibr) { _real = _arbr - _aibi; _imag = -_arbi - _aibr; }
  55. #define COMP_MAC(_real, _arbr, _aibi, _imag, _arbi, _aibr) { _real += _arbr - _aibi; _imag += -_arbi - _aibr; }
  56. #endif
  57. #if defined (TRMMKERNEL)
  58. #define A_OP =
  59. #else
  60. #define A_OP +=
  61. #endif
  62. #define BUILTIN_MMA_DISASSEMBLE_ACC_8 \
  63. __builtin_mma_disassemble_acc ((void *)result, &acc0); \
  64. __builtin_mma_disassemble_acc ((void *)&result[ 4], &acc1); \
  65. __builtin_mma_disassemble_acc ((void *)&result[ 8], &acc2); \
  66. __builtin_mma_disassemble_acc ((void *)&result[12], &acc3); \
  67. __builtin_mma_disassemble_acc ((void *)&result[16], &acc4); \
  68. __builtin_mma_disassemble_acc ((void *)&result[20], &acc5); \
  69. __builtin_mma_disassemble_acc ((void *)&result[24], &acc6); \
  70. __builtin_mma_disassemble_acc ((void *)&result[28], &acc7);
  71. #define COMP_MUL_1 \
  72. COMP_MUL(tr[0], res[ 0], res[ 5], ti[0], res[ 1], res[ 4])
  73. #define COMP_MAC_1(_offset) { \
  74. FLOAT *_ro = &res[_offset]; \
  75. COMP_MAC(tr[0], _ro[ 0], _ro[ 5], ti[0], _ro[ 1], _ro[ 4]) \
  76. }
  77. #define COMP_MUL_2A \
  78. COMP_MUL(tr[0], res[ 0], res[ 5], ti[0], res[ 1], res[ 4]) \
  79. COMP_MUL(tr[1], res[ 2], res[ 7], ti[1], res[ 3], res[ 6])
  80. #define COMP_MAC_2A(_offset) { \
  81. FLOAT *_ro = &res[_offset]; \
  82. COMP_MAC(tr[0], _ro[ 0], _ro[ 5], ti[0], _ro[ 1], _ro[ 4]) \
  83. COMP_MAC(tr[1], _ro[ 2], _ro[ 7], ti[1], _ro[ 3], _ro[ 6]) \
  84. }
  85. #define COMP_MUL_2B \
  86. COMP_MUL(tr[0], res[ 0], res[ 5], ti[0], res[ 1], res[ 4]) \
  87. COMP_MUL(tr[1], res[ 8], res[13], ti[1], res[ 9], res[12])
  88. #define COMP_MAC_2B(_offset) { \
  89. FLOAT *_ro = &res[_offset]; \
  90. COMP_MAC(tr[0], _ro[ 0], _ro[ 5], ti[0], _ro[ 1], _ro[ 4]) \
  91. COMP_MAC(tr[1], _ro[ 8], _ro[13], ti[1], _ro[ 9], _ro[12]) \
  92. }
  93. #define COMP_MUL_4A(_offset) { \
  94. FLOAT *_ro = &res[_offset]; \
  95. COMP_MUL(tr[0], _ro[ 0], _ro[ 5], ti[0], _ro[ 1], _ro[ 4]) \
  96. COMP_MUL(tr[1], _ro[ 8], _ro[13], ti[1], _ro[ 9], _ro[12]) \
  97. COMP_MUL(tr[2], _ro[16], _ro[21], ti[2], _ro[17], _ro[20]) \
  98. COMP_MUL(tr[3], _ro[24], _ro[29], ti[3], _ro[25], _ro[28]) \
  99. }
  100. #define COMP_MAC_4A(_offset) { \
  101. FLOAT *_ro = &res[_offset]; \
  102. COMP_MAC(tr[0], _ro[ 0], _ro[ 5], ti[0], _ro[ 1], _ro[ 4]) \
  103. COMP_MAC(tr[1], _ro[ 8], _ro[13], ti[1], _ro[ 9], _ro[12]) \
  104. COMP_MAC(tr[2], _ro[16], _ro[21], ti[2], _ro[17], _ro[20]) \
  105. COMP_MAC(tr[3], _ro[24], _ro[29], ti[3], _ro[25], _ro[28]) \
  106. }
  107. #define COMP_MUL_4B(_offset) { \
  108. FLOAT *_ro = &res[_offset]; \
  109. COMP_MUL(tr[0], _ro[ 0], _ro[ 5], ti[0], _ro[ 1], _ro[ 4]) \
  110. COMP_MUL(tr[1], _ro[ 8], _ro[13], ti[1], _ro[ 9], _ro[12]) \
  111. COMP_MUL(tr[2], _ro[ 2], _ro[ 7], ti[2], _ro[ 3], _ro[ 6]) \
  112. COMP_MUL(tr[3], _ro[10], _ro[15], ti[3], _ro[11], _ro[14]) \
  113. }
  114. #define COMP_MAC_4B(_offset) { \
  115. FLOAT *_ro = &res[_offset]; \
  116. COMP_MAC(tr[0], _ro[ 0], _ro[ 5], ti[0], _ro[ 1], _ro[ 4]) \
  117. COMP_MAC(tr[1], _ro[ 8], _ro[13], ti[1], _ro[ 9], _ro[12]) \
  118. COMP_MAC(tr[2], _ro[ 2], _ro[ 7], ti[2], _ro[ 3], _ro[ 6]) \
  119. COMP_MAC(tr[3], _ro[10], _ro[15], ti[3], _ro[11], _ro[14]) \
  120. }
  121. #define SAVE_ACC_COMPLEX_11 \
  122. BUILTIN_MMA_DISASSEMBLE_ACC_8 \
  123. COMP_MUL_1 \
  124. COMP_MAC_1(16) \
  125. COMP_MAC_1(32) \
  126. COMP_MAC_1(48) \
  127. COMP_MAC_1(64) \
  128. COMP_MAC_1(80) \
  129. COMP_MAC_1(96) \
  130. COMP_MAC_1(112) \
  131. CO[0] A_OP tr[0] * alpha_r - ti[0] * alpha_i; \
  132. CO[1] A_OP ti[0] * alpha_r + tr[0] * alpha_i;
  133. #define SAVE_ACC_COMPLEX_12 \
  134. BUILTIN_MMA_DISASSEMBLE_ACC_8 \
  135. COMP_MUL_2A \
  136. COMP_MAC_2A(16) \
  137. COMP_MAC_2A(32) \
  138. COMP_MAC_2A(48) \
  139. COMP_MAC_2A(64) \
  140. COMP_MAC_2A(80) \
  141. COMP_MAC_2A(96) \
  142. COMP_MAC_2A(112) \
  143. CO[0] A_OP tr[0] * alpha_r - ti[0] * alpha_i; \
  144. CO[1] A_OP ti[0] * alpha_r + tr[0] * alpha_i; \
  145. CO[2*ldc+0] A_OP tr[1] * alpha_r - ti[1] * alpha_i; \
  146. CO[2*ldc+1] A_OP ti[1] * alpha_r + tr[1] * alpha_i;
  147. #define SAVE_ACC_COMPLEX_21_1 \
  148. BUILTIN_MMA_DISASSEMBLE_ACC_8 \
  149. COMP_MUL_2B \
  150. COMP_MAC_2B(16) \
  151. COMP_MAC_2B(32) \
  152. COMP_MAC_2B(48) \
  153. COMP_MAC_2B(64) \
  154. COMP_MAC_2B(80) \
  155. COMP_MAC_2B(96) \
  156. COMP_MAC_2B(112) \
  157. CO[0] A_OP tr[0] * alpha_r - ti[0] * alpha_i; \
  158. CO[1] A_OP ti[0] * alpha_r + tr[0] * alpha_i; \
  159. CO[2] A_OP tr[1] * alpha_r - ti[1] * alpha_i; \
  160. CO[3] A_OP ti[1] * alpha_r + tr[1] * alpha_i;
  161. #define SAVE_ACC_COMPLEX_21_2 \
  162. BUILTIN_MMA_DISASSEMBLE_ACC_8 \
  163. COMP_MUL_4A(0) \
  164. COMP_MAC_4A(32) \
  165. COMP_MAC_4A(64) \
  166. COMP_MAC_4A(96) \
  167. CO[0] A_OP tr[0] * alpha_r - ti[0] * alpha_i; \
  168. CO[1] A_OP ti[0] * alpha_r + tr[0] * alpha_i; \
  169. CO[2] A_OP tr[1] * alpha_r - ti[1] * alpha_i; \
  170. CO[3] A_OP ti[1] * alpha_r + tr[1] * alpha_i; \
  171. CO[4] A_OP tr[2] * alpha_r - ti[2] * alpha_i; \
  172. CO[5] A_OP ti[2] * alpha_r + tr[2] * alpha_i; \
  173. CO[6] A_OP tr[3] * alpha_r - ti[3] * alpha_i; \
  174. CO[7] A_OP ti[3] * alpha_r + tr[3] * alpha_i;
  175. #define SAVE_ACC_COMPLEX_21_4 \
  176. BUILTIN_MMA_DISASSEMBLE_ACC_8 \
  177. COMP_MUL_4A(0) \
  178. COMP_MAC_4A(64) \
  179. CO[ 0] A_OP tr[0] * alpha_r - ti[0] * alpha_i; \
  180. CO[ 1] A_OP ti[0] * alpha_r + tr[0] * alpha_i; \
  181. CO[ 2] A_OP tr[1] * alpha_r - ti[1] * alpha_i; \
  182. CO[ 3] A_OP ti[1] * alpha_r + tr[1] * alpha_i; \
  183. CO[ 4] A_OP tr[2] * alpha_r - ti[2] * alpha_i; \
  184. CO[ 5] A_OP ti[2] * alpha_r + tr[2] * alpha_i; \
  185. CO[ 6] A_OP tr[3] * alpha_r - ti[3] * alpha_i; \
  186. CO[ 7] A_OP ti[3] * alpha_r + tr[3] * alpha_i; \
  187. COMP_MUL_4A(32) \
  188. COMP_MAC_4A(96) \
  189. CO[ 8] A_OP tr[0] * alpha_r - ti[0] * alpha_i; \
  190. CO[ 9] A_OP ti[0] * alpha_r + tr[0] * alpha_i; \
  191. CO[10] A_OP tr[1] * alpha_r - ti[1] * alpha_i; \
  192. CO[11] A_OP ti[1] * alpha_r + tr[1] * alpha_i; \
  193. CO[12] A_OP tr[2] * alpha_r - ti[2] * alpha_i; \
  194. CO[13] A_OP ti[2] * alpha_r + tr[2] * alpha_i; \
  195. CO[14] A_OP tr[3] * alpha_r - ti[3] * alpha_i; \
  196. CO[15] A_OP ti[3] * alpha_r + tr[3] * alpha_i;
  197. #define SAVE_ACC_COMPLEX_22_4 \
  198. BUILTIN_MMA_DISASSEMBLE_ACC_8 \
  199. COMP_MUL_4B(0) \
  200. CO[ 0] A_OP tr[0] * alpha_r - ti[0] * alpha_i; \
  201. CO[ 1] A_OP ti[0] * alpha_r + tr[0] * alpha_i; \
  202. CO[ 2] A_OP tr[1] * alpha_r - ti[1] * alpha_i; \
  203. CO[ 3] A_OP ti[1] * alpha_r + tr[1] * alpha_i; \
  204. CO[2*ldc+ 0] A_OP tr[2] * alpha_r - ti[2] * alpha_i; \
  205. CO[2*ldc+ 1] A_OP ti[2] * alpha_r + tr[2] * alpha_i; \
  206. CO[2*ldc+ 2] A_OP tr[3] * alpha_r - ti[3] * alpha_i; \
  207. CO[2*ldc+ 3] A_OP ti[3] * alpha_r + tr[3] * alpha_i; \
  208. COMP_MUL_4B(16) \
  209. CO[ 4] A_OP tr[0] * alpha_r - ti[0] * alpha_i; \
  210. CO[ 5] A_OP ti[0] * alpha_r + tr[0] * alpha_i; \
  211. CO[ 6] A_OP tr[1] * alpha_r - ti[1] * alpha_i; \
  212. CO[ 7] A_OP ti[1] * alpha_r + tr[1] * alpha_i; \
  213. CO[2*ldc+ 4] A_OP tr[2] * alpha_r - ti[2] * alpha_i; \
  214. CO[2*ldc+ 5] A_OP ti[2] * alpha_r + tr[2] * alpha_i; \
  215. CO[2*ldc+ 6] A_OP tr[3] * alpha_r - ti[3] * alpha_i; \
  216. CO[2*ldc+ 7] A_OP ti[3] * alpha_r + tr[3] * alpha_i; \
  217. COMP_MUL_4B(32) \
  218. CO[ 8] A_OP tr[0] * alpha_r - ti[0] * alpha_i; \
  219. CO[ 9] A_OP ti[0] * alpha_r + tr[0] * alpha_i; \
  220. CO[10] A_OP tr[1] * alpha_r - ti[1] * alpha_i; \
  221. CO[11] A_OP ti[1] * alpha_r + tr[1] * alpha_i; \
  222. CO[2*ldc+ 8] A_OP tr[2] * alpha_r - ti[2] * alpha_i; \
  223. CO[2*ldc+ 9] A_OP ti[2] * alpha_r + tr[2] * alpha_i; \
  224. CO[2*ldc+10] A_OP tr[3] * alpha_r - ti[3] * alpha_i; \
  225. CO[2*ldc+11] A_OP ti[3] * alpha_r + tr[3] * alpha_i; \
  226. COMP_MUL_4B(48) \
  227. CO[12] A_OP tr[0] * alpha_r - ti[0] * alpha_i; \
  228. CO[13] A_OP ti[0] * alpha_r + tr[0] * alpha_i; \
  229. CO[14] A_OP tr[1] * alpha_r - ti[1] * alpha_i; \
  230. CO[15] A_OP ti[1] * alpha_r + tr[1] * alpha_i; \
  231. CO[2*ldc+12] A_OP tr[2] * alpha_r - ti[2] * alpha_i; \
  232. CO[2*ldc+13] A_OP ti[2] * alpha_r + tr[2] * alpha_i; \
  233. CO[2*ldc+14] A_OP tr[3] * alpha_r - ti[3] * alpha_i; \
  234. CO[2*ldc+15] A_OP ti[3] * alpha_r + tr[3] * alpha_i;
  235. #define SAVE_ACC_COMPLEX_22_2 \
  236. BUILTIN_MMA_DISASSEMBLE_ACC_8 \
  237. COMP_MUL_4B(0) \
  238. CO[0] A_OP tr[0] * alpha_r - ti[0] * alpha_i; \
  239. CO[1] A_OP ti[0] * alpha_r + tr[0] * alpha_i; \
  240. CO[2] A_OP tr[1] * alpha_r - ti[1] * alpha_i; \
  241. CO[3] A_OP ti[1] * alpha_r + tr[1] * alpha_i; \
  242. CO[2*ldc+0] A_OP tr[2] * alpha_r - ti[2] * alpha_i; \
  243. CO[2*ldc+1] A_OP ti[2] * alpha_r + tr[2] * alpha_i; \
  244. CO[2*ldc+2] A_OP tr[3] * alpha_r - ti[3] * alpha_i; \
  245. CO[2*ldc+3] A_OP ti[3] * alpha_r + tr[3] * alpha_i; \
  246. COMP_MUL_4B(16) \
  247. CO[4] A_OP tr[0] * alpha_r - ti[0] * alpha_i; \
  248. CO[5] A_OP ti[0] * alpha_r + tr[0] * alpha_i; \
  249. CO[6] A_OP tr[1] * alpha_r - ti[1] * alpha_i; \
  250. CO[7] A_OP ti[1] * alpha_r + tr[1] * alpha_i; \
  251. CO[2*ldc+4] A_OP tr[2] * alpha_r - ti[2] * alpha_i; \
  252. CO[2*ldc+5] A_OP ti[2] * alpha_r + tr[2] * alpha_i; \
  253. CO[2*ldc+6] A_OP tr[3] * alpha_r - ti[3] * alpha_i; \
  254. CO[2*ldc+7] A_OP ti[3] * alpha_r + tr[3] * alpha_i;
  255. #define SAVE_ACC_COMPLEX_22_1 \
  256. BUILTIN_MMA_DISASSEMBLE_ACC_8 \
  257. COMP_MUL_4B(0) \
  258. CO[0] A_OP tr[0] * alpha_r - ti[0] * alpha_i; \
  259. CO[1] A_OP ti[0] * alpha_r + tr[0] * alpha_i; \
  260. CO[2] A_OP tr[1] * alpha_r - ti[1] * alpha_i; \
  261. CO[3] A_OP ti[1] * alpha_r + tr[1] * alpha_i; \
  262. CO[2*ldc+0] A_OP tr[2] * alpha_r - ti[2] * alpha_i; \
  263. CO[2*ldc+1] A_OP ti[2] * alpha_r + tr[2] * alpha_i; \
  264. CO[2*ldc+2] A_OP tr[3] * alpha_r - ti[3] * alpha_i; \
  265. CO[2*ldc+3] A_OP ti[3] * alpha_r + tr[3] * alpha_i;
  266. #define SAVE_ACC_COMPLEX_24_ALL \
  267. __builtin_mma_disassemble_acc ((void *)result, &acc0); \
  268. __builtin_mma_disassemble_acc ((void *)(&result[4]), &acc4); \
  269. __builtin_mma_disassemble_acc ((void *)(&result[8]), &acc1); \
  270. __builtin_mma_disassemble_acc ((void *)(&result[12]), &acc5); \
  271. __builtin_mma_disassemble_acc ((void *)(&result[16]), &acc2); \
  272. __builtin_mma_disassemble_acc ((void *)(&result[20]), &acc6); \
  273. __builtin_mma_disassemble_acc ((void *)(&result[24]), &acc3); \
  274. __builtin_mma_disassemble_acc ((void *)(&result[28]), &acc7); \
  275. COMP_MUL(tr[ 0], res[ 0], res[ 5], ti[ 0], res[ 1], res[ 4]) \
  276. COMP_MUL(tr[ 1], res[ 8], res[ 13], ti[ 1], res[ 9], res[ 12]) \
  277. COMP_MUL(tr[ 2], res[ 2], res[ 7], ti[ 2], res[ 3], res[ 6]) \
  278. COMP_MUL(tr[ 3], res[ 10], res[ 15], ti[ 3], res[ 11], res[ 14]) \
  279. COMP_MUL(tr[ 4], res[ 16], res[ 21], ti[ 4], res[ 17], res[ 20]) \
  280. COMP_MUL(tr[ 5], res[ 24], res[ 29], ti[ 5], res[ 25], res[ 28]) \
  281. COMP_MUL(tr[ 6], res[ 18], res[ 23], ti[ 6], res[ 19], res[ 22]) \
  282. COMP_MUL(tr[ 7], res[ 26], res[ 31], ti[ 7], res[ 27], res[ 30]) \
  283. COMP_MUL(tr[ 8], res[ 32], res[ 37], ti[ 8], res[ 33], res[ 36]) \
  284. COMP_MUL(tr[ 9], res[ 40], res[ 45], ti[ 9], res[ 41], res[ 44]) \
  285. COMP_MUL(tr[10], res[ 34], res[ 39], ti[10], res[ 35], res[ 38]) \
  286. COMP_MUL(tr[11], res[ 42], res[ 47], ti[11], res[ 43], res[ 46]) \
  287. COMP_MUL(tr[12], res[ 48], res[ 53], ti[12], res[ 49], res[ 52]) \
  288. COMP_MUL(tr[13], res[ 56], res[ 61], ti[13], res[ 57], res[ 60]) \
  289. COMP_MUL(tr[14], res[ 50], res[ 55], ti[14], res[ 51], res[ 54]) \
  290. COMP_MUL(tr[15], res[ 58], res[ 63], ti[15], res[ 59], res[ 62]) \
  291. COMP_MUL(tr[16], res[ 64], res[ 69], ti[16], res[ 65], res[ 68]) \
  292. COMP_MUL(tr[17], res[ 72], res[ 77], ti[17], res[ 73], res[ 76]) \
  293. COMP_MUL(tr[18], res[ 66], res[ 71], ti[18], res[ 67], res[ 70]) \
  294. COMP_MUL(tr[19], res[ 74], res[ 79], ti[19], res[ 75], res[ 78]) \
  295. COMP_MUL(tr[20], res[ 80], res[ 85], ti[20], res[ 81], res[ 84]) \
  296. COMP_MUL(tr[21], res[ 88], res[ 93], ti[21], res[ 89], res[ 92]) \
  297. COMP_MUL(tr[22], res[ 82], res[ 87], ti[22], res[ 83], res[ 86]) \
  298. COMP_MUL(tr[23], res[ 90], res[ 95], ti[23], res[ 91], res[ 94]) \
  299. COMP_MUL(tr[24], res[ 96], res[101], ti[24], res[ 97], res[100]) \
  300. COMP_MUL(tr[25], res[104], res[109], ti[25], res[105], res[108]) \
  301. COMP_MUL(tr[26], res[ 98], res[103], ti[26], res[ 99], res[102]) \
  302. COMP_MUL(tr[27], res[106], res[111], ti[27], res[107], res[110]) \
  303. COMP_MUL(tr[28], res[112], res[117], ti[28], res[113], res[116]) \
  304. COMP_MUL(tr[29], res[120], res[125], ti[29], res[121], res[124]) \
  305. COMP_MUL(tr[30], res[114], res[119], ti[30], res[115], res[118]) \
  306. COMP_MUL(tr[31], res[122], res[127], ti[31], res[123], res[126]) \
  307. CO[ 0] A_OP tr[ 0] * alpha_r - ti[ 0] * alpha_i; \
  308. CO[ 1] A_OP ti[ 0] * alpha_r + tr[ 0] * alpha_i; \
  309. CO[ 2] A_OP tr[ 1] * alpha_r - ti[ 1] * alpha_i; \
  310. CO[ 3] A_OP ti[ 1] * alpha_r + tr[ 1] * alpha_i; \
  311. CO[2*ldc+ 0] A_OP tr[ 2] * alpha_r - ti[ 2] * alpha_i; \
  312. CO[2*ldc+ 1] A_OP ti[ 2] * alpha_r + tr[ 2] * alpha_i; \
  313. CO[2*ldc+ 2] A_OP tr[ 3] * alpha_r - ti[ 3] * alpha_i; \
  314. CO[2*ldc+ 3] A_OP ti[ 3] * alpha_r + tr[ 3] * alpha_i; \
  315. CO[4*ldc+ 0] A_OP tr[ 4] * alpha_r - ti[ 4] * alpha_i; \
  316. CO[4*ldc+ 1] A_OP ti[ 4] * alpha_r + tr[ 4] * alpha_i; \
  317. CO[4*ldc+ 2] A_OP tr[ 5] * alpha_r - ti[ 5] * alpha_i; \
  318. CO[4*ldc+ 3] A_OP ti[ 5] * alpha_r + tr[ 5] * alpha_i; \
  319. CO[6*ldc+ 0] A_OP tr[ 6] * alpha_r - ti[ 6] * alpha_i; \
  320. CO[6*ldc+ 1] A_OP ti[ 6] * alpha_r + tr[ 6] * alpha_i; \
  321. CO[6*ldc+ 2] A_OP tr[ 7] * alpha_r - ti[ 7] * alpha_i; \
  322. CO[6*ldc+ 3] A_OP ti[ 7] * alpha_r + tr[ 7] * alpha_i; \
  323. CO[ 4] A_OP tr[ 8] * alpha_r - ti[ 8] * alpha_i; \
  324. CO[ 5] A_OP ti[ 8] * alpha_r + tr[ 8] * alpha_i; \
  325. CO[ 6] A_OP tr[ 9] * alpha_r - ti[ 9] * alpha_i; \
  326. CO[ 7] A_OP ti[ 9] * alpha_r + tr[ 9] * alpha_i; \
  327. CO[2*ldc+ 4] A_OP tr[10] * alpha_r - ti[10] * alpha_i; \
  328. CO[2*ldc+ 5] A_OP ti[10] * alpha_r + tr[10] * alpha_i; \
  329. CO[2*ldc+ 6] A_OP tr[11] * alpha_r - ti[11] * alpha_i; \
  330. CO[2*ldc+ 7] A_OP ti[11] * alpha_r + tr[11] * alpha_i; \
  331. CO[4*ldc+ 4] A_OP tr[12] * alpha_r - ti[12] * alpha_i; \
  332. CO[4*ldc+ 5] A_OP ti[12] * alpha_r + tr[12] * alpha_i; \
  333. CO[4*ldc+ 6] A_OP tr[13] * alpha_r - ti[13] * alpha_i; \
  334. CO[4*ldc+ 7] A_OP ti[13] * alpha_r + tr[13] * alpha_i; \
  335. CO[6*ldc+ 4] A_OP tr[14] * alpha_r - ti[14] * alpha_i; \
  336. CO[6*ldc+ 5] A_OP ti[14] * alpha_r + tr[14] * alpha_i; \
  337. CO[6*ldc+ 6] A_OP tr[15] * alpha_r - ti[15] * alpha_i; \
  338. CO[6*ldc+ 7] A_OP ti[15] * alpha_r + tr[15] * alpha_i; \
  339. CO[ 8] A_OP tr[16] * alpha_r - ti[16] * alpha_i; \
  340. CO[ 9] A_OP ti[16] * alpha_r + tr[16] * alpha_i; \
  341. CO[ 10] A_OP tr[17] * alpha_r - ti[17] * alpha_i; \
  342. CO[ 11] A_OP ti[17] * alpha_r + tr[17] * alpha_i; \
  343. CO[2*ldc+ 8] A_OP tr[18] * alpha_r - ti[18] * alpha_i; \
  344. CO[2*ldc+ 9] A_OP ti[18] * alpha_r + tr[18] * alpha_i; \
  345. CO[2*ldc+10] A_OP tr[19] * alpha_r - ti[19] * alpha_i; \
  346. CO[2*ldc+11] A_OP ti[19] * alpha_r + tr[19] * alpha_i; \
  347. CO[4*ldc+ 8] A_OP tr[20] * alpha_r - ti[20] * alpha_i; \
  348. CO[4*ldc+ 9] A_OP ti[20] * alpha_r + tr[20] * alpha_i; \
  349. CO[4*ldc+10] A_OP tr[21] * alpha_r - ti[21] * alpha_i; \
  350. CO[4*ldc+11] A_OP ti[21] * alpha_r + tr[21] * alpha_i; \
  351. CO[6*ldc+ 8] A_OP tr[22] * alpha_r - ti[22] * alpha_i; \
  352. CO[6*ldc+ 9] A_OP ti[22] * alpha_r + tr[22] * alpha_i; \
  353. CO[6*ldc+10] A_OP tr[23] * alpha_r - ti[23] * alpha_i; \
  354. CO[6*ldc+11] A_OP ti[23] * alpha_r + tr[23] * alpha_i; \
  355. CO[ 12] A_OP tr[24] * alpha_r - ti[24] * alpha_i; \
  356. CO[ 13] A_OP ti[24] * alpha_r + tr[24] * alpha_i; \
  357. CO[ 14] A_OP tr[25] * alpha_r - ti[25] * alpha_i; \
  358. CO[ 15] A_OP ti[25] * alpha_r + tr[25] * alpha_i; \
  359. CO[2*ldc+12] A_OP tr[26] * alpha_r - ti[26] * alpha_i; \
  360. CO[2*ldc+13] A_OP ti[26] * alpha_r + tr[26] * alpha_i; \
  361. CO[2*ldc+14] A_OP tr[27] * alpha_r - ti[27] * alpha_i; \
  362. CO[2*ldc+15] A_OP ti[27] * alpha_r + tr[27] * alpha_i; \
  363. CO[4*ldc+12] A_OP tr[28] * alpha_r - ti[28] * alpha_i; \
  364. CO[4*ldc+13] A_OP ti[28] * alpha_r + tr[28] * alpha_i; \
  365. CO[4*ldc+14] A_OP tr[29] * alpha_r - ti[29] * alpha_i; \
  366. CO[4*ldc+15] A_OP ti[29] * alpha_r + tr[29] * alpha_i; \
  367. CO[6*ldc+12] A_OP tr[30] * alpha_r - ti[30] * alpha_i; \
  368. CO[6*ldc+13] A_OP ti[30] * alpha_r + tr[30] * alpha_i; \
  369. CO[6*ldc+14] A_OP tr[31] * alpha_r - ti[31] * alpha_i; \
  370. CO[6*ldc+15] A_OP ti[31] * alpha_r + tr[31] * alpha_i;
  371. #define SAVE_ACC_COMPLEX_24(ACC1, ACC2, CI) \
  372. __builtin_mma_disassemble_acc ((void *)result, ACC1); \
  373. __builtin_mma_disassemble_acc ((void *)(&result[4]), ACC2); \
  374. COMP_MUL(tr[0], res[0], res[5], ti[0], res[1], res[4]) \
  375. COMP_MUL(tr[1], res[8], res[13], ti[1], res[9], res[12]) \
  376. COMP_MUL(tr[2], res[2], res[7], ti[2], res[3], res[6]) \
  377. COMP_MUL(tr[3], res[10], res[15], ti[3], res[11], res[14]) \
  378. COMP_MUL(tr[4], res[16], res[21], ti[4], res[17], res[20]) \
  379. COMP_MUL(tr[5], res[24], res[29], ti[5], res[25], res[28]) \
  380. COMP_MUL(tr[6], res[18], res[23], ti[6], res[19], res[22]) \
  381. COMP_MUL(tr[7], res[26], res[31], ti[7], res[27], res[30]) \
  382. CO[CI+0] A_OP tr[0] * alpha_r - ti[0] * alpha_i; \
  383. CO[CI+1] A_OP ti[0] * alpha_r + tr[0] * alpha_i; \
  384. CO[CI+2] A_OP tr[1] * alpha_r - ti[1] * alpha_i; \
  385. CO[CI+3] A_OP ti[1] * alpha_r + tr[1] * alpha_i; \
  386. CO[CI+2*ldc+0] A_OP tr[2] * alpha_r - ti[2] * alpha_i; \
  387. CO[CI+2*ldc+1] A_OP ti[2] * alpha_r + tr[2] * alpha_i; \
  388. CO[CI+2*ldc+2] A_OP tr[3] * alpha_r - ti[3] * alpha_i; \
  389. CO[CI+2*ldc+3] A_OP ti[3] * alpha_r + tr[3] * alpha_i; \
  390. CO[CI+4*ldc+0] A_OP tr[4] * alpha_r - ti[4] * alpha_i; \
  391. CO[CI+4*ldc+1] A_OP ti[4] * alpha_r + tr[4] * alpha_i; \
  392. CO[CI+4*ldc+2] A_OP tr[5] * alpha_r - ti[5] * alpha_i; \
  393. CO[CI+4*ldc+3] A_OP ti[5] * alpha_r + tr[5] * alpha_i; \
  394. CO[CI+6*ldc+0] A_OP tr[6] * alpha_r - ti[6] * alpha_i; \
  395. CO[CI+6*ldc+1] A_OP ti[6] * alpha_r + tr[6] * alpha_i; \
  396. CO[CI+6*ldc+2] A_OP tr[7] * alpha_r - ti[7] * alpha_i; \
  397. CO[CI+6*ldc+3] A_OP ti[7] * alpha_r + tr[7] * alpha_i;
  398. #define SAVE_ACC_COMPLEX_14 \
  399. BUILTIN_MMA_DISASSEMBLE_ACC_8 \
  400. COMP_MUL(tr[0], res[ 0], res[ 5], ti[0], res[ 1], res[ 4]) \
  401. COMP_MUL(tr[1], res[ 2], res[ 7], ti[1], res[ 3], res[ 6]) \
  402. COMP_MUL(tr[2], res[ 16], res[ 21], ti[2], res[ 17], res[ 20]) \
  403. COMP_MUL(tr[3], res[ 18], res[ 23], ti[3], res[ 19], res[ 22]) \
  404. COMP_MAC(tr[0], res[ 32], res[ 37], ti[0], res[ 33], res[ 36]) \
  405. COMP_MAC(tr[1], res[ 34], res[ 39], ti[1], res[ 35], res[ 38]) \
  406. COMP_MAC(tr[2], res[ 48], res[ 53], ti[2], res[ 49], res[ 52]) \
  407. COMP_MAC(tr[3], res[ 50], res[ 55], ti[3], res[ 51], res[ 54]) \
  408. COMP_MAC(tr[0], res[ 64], res[ 69], ti[0], res[ 65], res[ 68]) \
  409. COMP_MAC(tr[1], res[ 66], res[ 71], ti[1], res[ 67], res[ 70]) \
  410. COMP_MAC(tr[2], res[ 80], res[ 85], ti[2], res[ 81], res[ 84]) \
  411. COMP_MAC(tr[3], res[ 82], res[ 87], ti[3], res[ 83], res[ 86]) \
  412. COMP_MAC(tr[0], res[ 96], res[101], ti[0], res[ 97], res[100]) \
  413. COMP_MAC(tr[1], res[ 98], res[103], ti[1], res[ 99], res[102]) \
  414. COMP_MAC(tr[2], res[112], res[117], ti[2], res[113], res[116]) \
  415. COMP_MAC(tr[3], res[114], res[119], ti[3], res[115], res[118]) \
  416. CO[0] A_OP tr[0] * alpha_r - ti[0] * alpha_i; \
  417. CO[1] A_OP ti[0] * alpha_r + tr[0] * alpha_i; \
  418. CO[2*ldc+0] A_OP tr[1] * alpha_r - ti[1] * alpha_i; \
  419. CO[2*ldc+1] A_OP ti[1] * alpha_r + tr[1] * alpha_i; \
  420. CO[4*ldc+0] A_OP tr[2] * alpha_r - ti[2] * alpha_i; \
  421. CO[4*ldc+1] A_OP ti[2] * alpha_r + tr[2] * alpha_i; \
  422. CO[6*ldc+0] A_OP tr[3] * alpha_r - ti[3] * alpha_i; \
  423. CO[6*ldc+1] A_OP ti[3] * alpha_r + tr[3] * alpha_i;
  424. #define PREFETCH1(x, y) asm volatile ("dcbt %0, %1" : : "r" (x), "b" (y) : "memory");
  425. #if (defined(LEFT) && !defined(TRANSA)) || (!defined(LEFT) && defined(TRANSA))
  426. #define REFRESH_TEMP_BK(x, y) \
  427. temp = k - off;
  428. #elif defined(LEFT)
  429. #define REFRESH_TEMP_BK(x, y) \
  430. temp = off + x;
  431. #else
  432. #define REFRESH_TEMP_BK(x, y) \
  433. temp = off + y;
  434. #endif
  435. #if (defined(LEFT) && defined(TRANSA)) || (!defined(LEFT) && !defined(TRANSA))
  436. #define REFRESH_POINTERS(x, y) \
  437. BO = B; \
  438. REFRESH_TEMP_BK(x, y)
  439. #else
  440. #define REFRESH_POINTERS(x, y) \
  441. AO += off * (2*x); \
  442. BO = B + off * (2*y); \
  443. REFRESH_TEMP_BK(x, y)
  444. #endif
  445. #ifdef LEFT
  446. #define REFRESH_OFF(x) \
  447. off += x;
  448. #else
  449. #define REFRESH_OFF(x)
  450. #endif
  451. #ifdef LEFT
  452. #define UPDATE_TEMP(x, y) \
  453. temp -= x;
  454. #else
  455. #define UPDATE_TEMP(x, y) \
  456. temp -= y;
  457. #endif
  458. #if (defined(LEFT) && defined(TRANSA)) || (!defined(LEFT) && !defined(TRANSA))
  459. #define REFRESH_TMP_AFTER_SAVE(x, y) \
  460. temp = k - off; \
  461. UPDATE_TEMP(x, y) \
  462. AO += temp * (2*x); \
  463. BO += temp * (2*y);
  464. #else
  465. #define REFRESH_TMP_AFTER_SAVE(x, y)
  466. #endif
  467. #define REFRESH_AFTER_SAVE(x,y) \
  468. REFRESH_TMP_AFTER_SAVE(x, y) \
  469. REFRESH_OFF(x)
  470. /*************************************************************************************
  471. * GEMM Kernel
  472. *************************************************************************************/
  473. int
  474. CNAME (BLASLONG m, BLASLONG n, BLASLONG k, FLOAT alpha_r, FLOAT alpha_i, FLOAT * A, FLOAT * B,
  475. FLOAT * C, BLASLONG ldc
  476. #ifdef TRMMKERNEL
  477. , BLASLONG offset
  478. #endif
  479. )
  480. {
  481. BLASLONG i1, i, l, temp;
  482. FLOAT *AO, *BO, *CO;
  483. #if defined(TRMMKERNEL)
  484. BLASLONG off;
  485. #endif
  486. #if defined(TRMMKERNEL) && !defined(LEFT)
  487. off = -offset;
  488. #endif
  489. __vector_quad acc0, acc1, acc2, acc3, acc4, acc5, acc6, acc7;
  490. v4sf_t result[32];
  491. FLOAT *res, tr[64], ti[64];
  492. res = (FLOAT *) result;
  493. for (i1 = 0; i1 < (n >> 2); i1++)
  494. {
  495. #if defined(TRMMKERNEL) && defined(LEFT)
  496. off = offset;
  497. #endif
  498. AO = A;
  499. CO = C;
  500. C += ldc << 3;
  501. for (i = 0; i < (m >> 3); i++)
  502. {
  503. #if defined(TRMMKERNEL)
  504. REFRESH_POINTERS (8, 4);
  505. #else
  506. BO = B;
  507. temp = k;
  508. #endif
  509. SET_ACC_ZERO()
  510. for (l = 0; l < temp; ++l)
  511. {
  512. vec_t rowA1 = *(vec_t *) & AO[l<<4];
  513. vec_t rowB1 = *(vec_t *) & BO[l<<3];
  514. vec_t rowA2 = *(vec_t *) & AO[(l<<4)+4];
  515. vec_t rowB2 = *(vec_t *) & BO[(l<<3)+4];
  516. vec_t rowA3 = *(vec_t *) & AO[(l<<4)+8];
  517. vec_t rowA4 = *(vec_t *) & AO[(l<<4)+12];
  518. __builtin_mma_xvf32gerpp(&acc0, rowA1, rowB1);
  519. __builtin_mma_xvf32gerpp(&acc1, rowA2, rowB1);
  520. __builtin_mma_xvf32gerpp(&acc2, rowA3, rowB1);
  521. __builtin_mma_xvf32gerpp(&acc3, rowA4, rowB1);
  522. __builtin_mma_xvf32gerpp(&acc4, rowA1, rowB2);
  523. __builtin_mma_xvf32gerpp(&acc5, rowA2, rowB2);
  524. __builtin_mma_xvf32gerpp(&acc6, rowA3, rowB2);
  525. __builtin_mma_xvf32gerpp(&acc7, rowA4, rowB2);
  526. }
  527. SAVE_ACC_COMPLEX_24_ALL
  528. CO += 16;
  529. AO += temp << 4;
  530. BO += temp << 3;
  531. #if defined(TRMMKERNEL)
  532. REFRESH_AFTER_SAVE (8, 4)
  533. #endif
  534. }
  535. if (m & 4)
  536. {
  537. #if defined(TRMMKERNEL)
  538. REFRESH_POINTERS (4, 4);
  539. #else
  540. BO = B;
  541. temp = k;
  542. #endif
  543. SET_ACC_ZERO()
  544. for (l = 0; l < (temp & (~1)); l+=2)
  545. {
  546. vec_t rowA1 = *(vec_t *) & AO[l<<3];
  547. vec_t rowA2 = *(vec_t *) & AO[(l<<3)+4];
  548. vec_t rowA3 = *(vec_t *) & AO[(l<<3)+8];
  549. vec_t rowA4 = *(vec_t *) & AO[(l<<3)+12];
  550. vec_t rowB1 = *(vec_t *) & BO[l<<3];
  551. vec_t rowB2 = *(vec_t *) & BO[(l<<3)+4];
  552. vec_t rowB3 = *(vec_t *) & BO[(l<<3)+8];
  553. vec_t rowB4 = *(vec_t *) & BO[(l<<3)+12];
  554. __builtin_mma_xvf32gerpp(&acc0, rowA1, rowB1);
  555. __builtin_mma_xvf32gerpp(&acc1, rowA2, rowB1);
  556. __builtin_mma_xvf32gerpp(&acc2, rowA1, rowB2);
  557. __builtin_mma_xvf32gerpp(&acc3, rowA2, rowB2);
  558. __builtin_mma_xvf32gerpp(&acc0, rowA3, rowB3);
  559. __builtin_mma_xvf32gerpp(&acc1, rowA4, rowB3);
  560. __builtin_mma_xvf32gerpp(&acc2, rowA3, rowB4);
  561. __builtin_mma_xvf32gerpp(&acc3, rowA4, rowB4);
  562. }
  563. for (l = (temp & (~1)); l < temp; ++l)
  564. {
  565. vec_t rowA1 = *(vec_t *) & AO[l<<3];
  566. vec_t rowA2 = *(vec_t *) & AO[(l<<3)+4];
  567. vec_t rowB1 = *(vec_t *) & BO[l<<3];
  568. vec_t rowB2 = *(vec_t *) & BO[(l<<3)+4];
  569. __builtin_mma_xvf32gerpp(&acc0, rowA1, rowB1);
  570. __builtin_mma_xvf32gerpp(&acc1, rowA2, rowB1);
  571. __builtin_mma_xvf32gerpp(&acc2, rowA1, rowB2);
  572. __builtin_mma_xvf32gerpp(&acc3, rowA2, rowB2);
  573. }
  574. SAVE_ACC_COMPLEX_24(&acc0, &acc2, 0)
  575. SAVE_ACC_COMPLEX_24(&acc1, &acc3, 4)
  576. CO += 8;
  577. AO += temp << 3;
  578. BO += temp << 3;
  579. #if defined(TRMMKERNEL)
  580. REFRESH_AFTER_SAVE (4, 4)
  581. #endif
  582. }
  583. if (m & 2)
  584. {
  585. #if defined(TRMMKERNEL)
  586. REFRESH_POINTERS (2, 4);
  587. #else
  588. BO = B;
  589. temp = k;
  590. #endif
  591. SET_ACC_ZERO()
  592. for (l = 0; l < (temp & (~3)); l+=4)
  593. {
  594. vec_t rowA1 = *(vec_t *) & AO[l<<2];
  595. vec_t rowA2 = *(vec_t *) & AO[(l<<2)+4];
  596. vec_t rowA3 = *(vec_t *) & AO[(l<<2)+8];
  597. vec_t rowA4 = *(vec_t *) & AO[(l<<2)+12];
  598. vec_t rowB1 = *(vec_t *) & BO[l<<3];
  599. vec_t rowB2 = *(vec_t *) & BO[(l<<3)+4];
  600. vec_t rowB3 = *(vec_t *) & BO[(l<<3)+8];
  601. vec_t rowB4 = *(vec_t *) & BO[(l<<3)+12];
  602. vec_t rowB5 = *(vec_t *) & BO[(l<<3)+16];
  603. vec_t rowB6 = *(vec_t *) & BO[(l<<3)+20];
  604. vec_t rowB7 = *(vec_t *) & BO[(l<<3)+24];
  605. vec_t rowB8 = *(vec_t *) & BO[(l<<3)+28];
  606. __builtin_mma_xvf32gerpp(&acc0, rowA1, rowB1);
  607. __builtin_mma_xvf32gerpp(&acc1, rowA1, rowB2);
  608. __builtin_mma_xvf32gerpp(&acc0, rowA2, rowB3);
  609. __builtin_mma_xvf32gerpp(&acc1, rowA2, rowB4);
  610. __builtin_mma_xvf32gerpp(&acc0, rowA3, rowB5);
  611. __builtin_mma_xvf32gerpp(&acc1, rowA3, rowB6);
  612. __builtin_mma_xvf32gerpp(&acc0, rowA4, rowB7);
  613. __builtin_mma_xvf32gerpp(&acc1, rowA4, rowB8);
  614. }
  615. for (l = (temp & (~3)); l < temp; ++l)
  616. {
  617. vec_t rowA1 = *(vec_t *) & AO[l<<2];
  618. vec_t rowB1 = *(vec_t *) & BO[l<<3];
  619. vec_t rowB2 = *(vec_t *) & BO[(l<<3)+4];
  620. __builtin_mma_xvf32gerpp(&acc0, rowA1, rowB1);
  621. __builtin_mma_xvf32gerpp(&acc1, rowA1, rowB2);
  622. }
  623. SAVE_ACC_COMPLEX_24(&acc0, &acc1, 0)
  624. CO += 4;
  625. AO += temp << 2;
  626. BO += temp << 3;
  627. #if defined(TRMMKERNEL)
  628. REFRESH_AFTER_SAVE (2, 4)
  629. #endif
  630. }
  631. if (m & 1)
  632. {
  633. #if defined(TRMMKERNEL)
  634. REFRESH_POINTERS (1, 4)
  635. #else
  636. BO = B;
  637. temp = k;
  638. #endif
  639. SET_ACC_ZERO()
  640. for (l = 0; l < (temp & (~3)); l+=4)
  641. {
  642. vec_t rowA1 = *(vec_t *) & AO[l<<1];
  643. vec_t rowA2 = *(vec_t *) & AO[(l<<1)+2];
  644. vec_t rowA3 = *(vec_t *) & AO[(l<<1)+4];
  645. vec_t rowA4 = *(vec_t *) & AO[(l<<1)+6];
  646. vec_t rowB1 = *(vec_t *) & BO[l<<3];
  647. vec_t rowB2 = *(vec_t *) & BO[(l<<3)+4];
  648. vec_t rowB3 = *(vec_t *) & BO[(l<<3)+8];
  649. vec_t rowB4 = *(vec_t *) & BO[(l<<3)+12];
  650. vec_t rowB5 = *(vec_t *) & BO[(l<<3)+16];
  651. vec_t rowB6 = *(vec_t *) & BO[(l<<3)+20];
  652. vec_t rowB7 = *(vec_t *) & BO[(l<<3)+24];
  653. vec_t rowB8 = *(vec_t *) & BO[(l<<3)+28];
  654. __builtin_mma_xvf32gerpp(&acc0, rowA1, rowB1);
  655. __builtin_mma_xvf32gerpp(&acc1, rowA1, rowB2);
  656. __builtin_mma_xvf32gerpp(&acc2, rowA2, rowB3);
  657. __builtin_mma_xvf32gerpp(&acc3, rowA2, rowB4);
  658. __builtin_mma_xvf32gerpp(&acc4, rowA3, rowB5);
  659. __builtin_mma_xvf32gerpp(&acc5, rowA3, rowB6);
  660. __builtin_mma_xvf32gerpp(&acc6, rowA4, rowB7);
  661. __builtin_mma_xvf32gerpp(&acc7, rowA4, rowB8);
  662. }
  663. for (l = (temp & (~3)); l < temp; ++l)
  664. {
  665. vec_t rowA1 = *(vec_t *) & AO[l<<1];
  666. vec_t rowB1 = *(vec_t *) & BO[l<<3];
  667. vec_t rowB2 = *(vec_t *) & BO[(l<<3)+4];
  668. __builtin_mma_xvf32gerpp(&acc0, rowA1, rowB1);
  669. __builtin_mma_xvf32gerpp(&acc1, rowA1, rowB2);
  670. }
  671. SAVE_ACC_COMPLEX_14
  672. CO += 2;
  673. AO += temp << 1;
  674. BO += temp << 3;
  675. #if defined(TRMMKERNEL)
  676. REFRESH_AFTER_SAVE (1, 4)
  677. #endif
  678. }
  679. #if defined(TRMMKERNEL) && !defined(LEFT)
  680. off += 4; // number of values in A
  681. #endif
  682. B += k << 3;
  683. }
  684. if (n & 2)
  685. {
  686. #if defined(TRMMKERNEL) && defined(LEFT)
  687. off = offset;
  688. #endif
  689. AO = A;
  690. CO = C;
  691. C += ldc << 2;
  692. for (i = 0; i < (m >> 3); i++)
  693. {
  694. #if defined(TRMMKERNEL)
  695. REFRESH_POINTERS (8, 2)
  696. #else
  697. BO = B;
  698. temp = k;
  699. #endif
  700. SET_ACC_ZERO()
  701. for (l = 0; l < (temp & (~1)); l+=2)
  702. {
  703. vec_t rowA1 = *(vec_t *) & AO[l<<4];
  704. vec_t rowA2 = *(vec_t *) & AO[(l<<4)+4];
  705. vec_t rowA3 = *(vec_t *) & AO[(l<<4)+8];
  706. vec_t rowA4 = *(vec_t *) & AO[(l<<4)+12];
  707. vec_t rowA5 = *(vec_t *) & AO[(l<<4)+16];
  708. vec_t rowA6 = *(vec_t *) & AO[(l<<4)+20];
  709. vec_t rowA7 = *(vec_t *) & AO[(l<<4)+24];
  710. vec_t rowA8 = *(vec_t *) & AO[(l<<4)+28];
  711. vec_t rowB1 = *(vec_t *) & BO[l<<2];
  712. vec_t rowB2 = *(vec_t *) & BO[(l<<2)+4];
  713. __builtin_mma_xvf32gerpp(&acc0, rowA1, rowB1);
  714. __builtin_mma_xvf32gerpp(&acc1, rowA2, rowB1);
  715. __builtin_mma_xvf32gerpp(&acc2, rowA3, rowB1);
  716. __builtin_mma_xvf32gerpp(&acc3, rowA4, rowB1);
  717. __builtin_mma_xvf32gerpp(&acc0, rowA5, rowB2);
  718. __builtin_mma_xvf32gerpp(&acc1, rowA6, rowB2);
  719. __builtin_mma_xvf32gerpp(&acc2, rowA7, rowB2);
  720. __builtin_mma_xvf32gerpp(&acc3, rowA8, rowB2);
  721. }
  722. for (l = (temp & (~1)); l < temp; ++l)
  723. {
  724. vec_t rowA1 = *(vec_t *) & AO[l<<4];
  725. vec_t rowA2 = *(vec_t *) & AO[(l<<4)+4];
  726. vec_t rowA3 = *(vec_t *) & AO[(l<<4)+8];
  727. vec_t rowA4 = *(vec_t *) & AO[(l<<4)+12];
  728. vec_t rowB1 = *(vec_t *) & BO[l<<2];
  729. __builtin_mma_xvf32gerpp(&acc0, rowA1, rowB1);
  730. __builtin_mma_xvf32gerpp(&acc1, rowA2, rowB1);
  731. __builtin_mma_xvf32gerpp(&acc2, rowA3, rowB1);
  732. __builtin_mma_xvf32gerpp(&acc3, rowA4, rowB1);
  733. }
  734. SAVE_ACC_COMPLEX_22_4
  735. AO += temp << 4;
  736. BO += temp << 2;
  737. CO += 16;
  738. #if defined(TRMMKERNEL)
  739. REFRESH_AFTER_SAVE (8, 2)
  740. #endif
  741. }
  742. if (m & 4)
  743. {
  744. #if defined(TRMMKERNEL)
  745. REFRESH_POINTERS (4, 2)
  746. #else
  747. BO = B;
  748. temp = k;
  749. #endif
  750. SET_ACC_ZERO()
  751. for (l = 0; l < (temp & (~3)); l+=4)
  752. {
  753. vec_t rowA1 = *(vec_t *) & AO[l<<3];
  754. vec_t rowA2 = *(vec_t *) & AO[(l<<3)+4];
  755. vec_t rowA3 = *(vec_t *) & AO[(l<<3)+8];
  756. vec_t rowA4 = *(vec_t *) & AO[(l<<3)+12];
  757. vec_t rowA5 = *(vec_t *) & AO[(l<<3)+16];
  758. vec_t rowA6 = *(vec_t *) & AO[(l<<3)+20];
  759. vec_t rowA7 = *(vec_t *) & AO[(l<<3)+24];
  760. vec_t rowA8 = *(vec_t *) & AO[(l<<3)+28];
  761. vec_t rowB1 = *(vec_t *) & BO[l<<2];
  762. vec_t rowB2 = *(vec_t *) & BO[(l<<2)+4];
  763. vec_t rowB3 = *(vec_t *) & BO[(l<<2)+8];
  764. vec_t rowB4 = *(vec_t *) & BO[(l<<2)+12];
  765. __builtin_mma_xvf32gerpp(&acc0, rowA1, rowB1);
  766. __builtin_mma_xvf32gerpp(&acc1, rowA2, rowB1);
  767. __builtin_mma_xvf32gerpp(&acc0, rowA3, rowB2);
  768. __builtin_mma_xvf32gerpp(&acc1, rowA4, rowB2);
  769. __builtin_mma_xvf32gerpp(&acc0, rowA5, rowB3);
  770. __builtin_mma_xvf32gerpp(&acc1, rowA6, rowB3);
  771. __builtin_mma_xvf32gerpp(&acc0, rowA7, rowB4);
  772. __builtin_mma_xvf32gerpp(&acc1, rowA8, rowB4);
  773. }
  774. for (l = (temp & (~3)); l < temp; ++l)
  775. {
  776. vec_t rowA1 = *(vec_t *) & AO[l<<3];
  777. vec_t rowA2 = *(vec_t *) & AO[(l<<3)+4];
  778. vec_t rowB1 = *(vec_t *) & BO[l<<2];
  779. __builtin_mma_xvf32gerpp(&acc0, rowA1, rowB1);
  780. __builtin_mma_xvf32gerpp(&acc1, rowA2, rowB1);
  781. }
  782. SAVE_ACC_COMPLEX_22_2
  783. AO += temp << 3;
  784. BO += temp << 2;
  785. CO += 8;
  786. #if defined(TRMMKERNEL)
  787. REFRESH_AFTER_SAVE (4, 2)
  788. #endif
  789. } if (m & 2)
  790. {
  791. #if defined(TRMMKERNEL)
  792. REFRESH_POINTERS (2, 2)
  793. #else
  794. BO = B;
  795. temp = k;
  796. #endif
  797. SET_ACC_ZERO()
  798. for (l = 0; l < (temp & (~7)); l+=8)
  799. {
  800. vec_t rowA1 = *(vec_t *) & AO[l<<2];
  801. vec_t rowA2 = *(vec_t *) & AO[(l<<2)+4];
  802. vec_t rowA3 = *(vec_t *) & AO[(l<<2)+8];
  803. vec_t rowA4 = *(vec_t *) & AO[(l<<2)+12];
  804. vec_t rowA5 = *(vec_t *) & AO[(l<<2)+16];
  805. vec_t rowA6 = *(vec_t *) & AO[(l<<2)+20];
  806. vec_t rowA7 = *(vec_t *) & AO[(l<<2)+24];
  807. vec_t rowA8 = *(vec_t *) & AO[(l<<2)+28];
  808. vec_t rowB1 = *(vec_t *) & BO[l<<2];
  809. vec_t rowB2 = *(vec_t *) & BO[(l<<2)+4];
  810. vec_t rowB3 = *(vec_t *) & BO[(l<<2)+8];
  811. vec_t rowB4 = *(vec_t *) & BO[(l<<2)+12];
  812. vec_t rowB5 = *(vec_t *) & BO[(l<<2)+16];
  813. vec_t rowB6 = *(vec_t *) & BO[(l<<2)+20];
  814. vec_t rowB7 = *(vec_t *) & BO[(l<<2)+24];
  815. vec_t rowB8 = *(vec_t *) & BO[(l<<2)+28];
  816. __builtin_mma_xvf32gerpp(&acc0, rowA1, rowB1);
  817. __builtin_mma_xvf32gerpp(&acc0, rowA2, rowB2);
  818. __builtin_mma_xvf32gerpp(&acc0, rowA3, rowB3);
  819. __builtin_mma_xvf32gerpp(&acc0, rowA4, rowB4);
  820. __builtin_mma_xvf32gerpp(&acc0, rowA5, rowB5);
  821. __builtin_mma_xvf32gerpp(&acc0, rowA6, rowB6);
  822. __builtin_mma_xvf32gerpp(&acc0, rowA7, rowB7);
  823. __builtin_mma_xvf32gerpp(&acc0, rowA8, rowB8);
  824. }
  825. for (l = (temp & (~7)); l < temp; ++l)
  826. {
  827. vec_t rowA1 = *(vec_t *) & AO[l<<2];
  828. vec_t rowB1 = *(vec_t *) & BO[l<<2];
  829. __builtin_mma_xvf32gerpp(&acc0, rowA1, rowB1);
  830. }
  831. SAVE_ACC_COMPLEX_22_1
  832. AO += temp << 2;
  833. BO += temp << 2;
  834. CO += 4;
  835. #if defined(TRMMKERNEL)
  836. REFRESH_AFTER_SAVE (2, 2)
  837. #endif
  838. }
  839. if (m & 1)
  840. {
  841. #if defined(TRMMKERNEL)
  842. REFRESH_POINTERS (1, 2)
  843. #else
  844. BO = B;
  845. temp = k;
  846. #endif
  847. // RIP OUT MMA STUFF!
  848. SET_ACC_ZERO()
  849. for (l = 0; l < (temp & (~7)); l+=8)
  850. {
  851. vec_t rowA1 = *(vec_t *) & AO[l<<1];
  852. vec_t rowA2 = *(vec_t *) & AO[(l<<1)+2];
  853. vec_t rowA3 = *(vec_t *) & AO[(l<<1)+4];
  854. vec_t rowA4 = *(vec_t *) & AO[(l<<1)+6];
  855. vec_t rowA5 = *(vec_t *) & AO[(l<<1)+8];
  856. vec_t rowA6 = *(vec_t *) & AO[(l<<1)+10];
  857. vec_t rowA7 = *(vec_t *) & AO[(l<<1)+12];
  858. vec_t rowA8 = *(vec_t *) & AO[(l<<1)+14];
  859. vec_t rowB1 = *(vec_t *) & BO[l<<2];
  860. vec_t rowB2 = *(vec_t *) & BO[(l<<2)+4];
  861. vec_t rowB3 = *(vec_t *) & BO[(l<<2)+8];
  862. vec_t rowB4 = *(vec_t *) & BO[(l<<2)+12];
  863. vec_t rowB5 = *(vec_t *) & BO[(l<<2)+16];
  864. vec_t rowB6 = *(vec_t *) & BO[(l<<2)+20];
  865. vec_t rowB7 = *(vec_t *) & BO[(l<<2)+24];
  866. vec_t rowB8 = *(vec_t *) & BO[(l<<2)+28];
  867. __builtin_mma_xvf32gerpp(&acc0, rowA1, rowB1);
  868. __builtin_mma_xvf32gerpp(&acc1, rowA2, rowB2);
  869. __builtin_mma_xvf32gerpp(&acc2, rowA3, rowB3);
  870. __builtin_mma_xvf32gerpp(&acc3, rowA4, rowB4);
  871. __builtin_mma_xvf32gerpp(&acc4, rowA5, rowB5);
  872. __builtin_mma_xvf32gerpp(&acc5, rowA6, rowB6);
  873. __builtin_mma_xvf32gerpp(&acc6, rowA7, rowB7);
  874. __builtin_mma_xvf32gerpp(&acc7, rowA8, rowB8);
  875. }
  876. for (l = (temp & (~7)); l < temp; ++l)
  877. {
  878. vec_t rowA1 = *(vec_t *) & AO[l<<1];
  879. vec_t rowB1 = *(vec_t *) & BO[l<<2];
  880. __builtin_mma_xvf32gerpp(&acc0, rowA1, rowB1);
  881. }
  882. SAVE_ACC_COMPLEX_12
  883. AO += temp<<1;
  884. BO += temp<<2;
  885. CO += 2;
  886. #if defined(TRMMKERNEL)
  887. REFRESH_AFTER_SAVE (1, 2)
  888. #endif
  889. }
  890. #if defined(TRMMKERNEL) && !defined(LEFT)
  891. off += 2; // number of values in A
  892. #endif
  893. B += k << 2;
  894. }
  895. if (n & 1)
  896. {
  897. #if defined(TRMMKERNEL) && defined(LEFT)
  898. off = offset;
  899. #endif
  900. AO = A;
  901. CO = C;
  902. C += ldc << 1;
  903. for (i = 0; i < (m >> 3); i++)
  904. {
  905. #if defined(TRMMKERNEL)
  906. REFRESH_POINTERS (8, 1)
  907. #else
  908. BO = B;
  909. temp = k;
  910. #endif
  911. SET_ACC_ZERO()
  912. for (l = 0; l < (temp & (~1)); l+=2)
  913. {
  914. vec_t rowA1 = *(vec_t *) & AO[l<<4];
  915. vec_t rowA2 = *(vec_t *) & AO[(l<<4)+4];
  916. vec_t rowA3 = *(vec_t *) & AO[(l<<4)+8];
  917. vec_t rowA4 = *(vec_t *) & AO[(l<<4)+12];
  918. vec_t rowA5 = *(vec_t *) & AO[(l<<4)+16];
  919. vec_t rowA6 = *(vec_t *) & AO[(l<<4)+20];
  920. vec_t rowA7 = *(vec_t *) & AO[(l<<4)+24];
  921. vec_t rowA8 = *(vec_t *) & AO[(l<<4)+28];
  922. vec_t rowB1 = *(vec_t *) & BO[l<<1];
  923. vec_t rowB2 = *(vec_t *) & BO[(l<<1)+2];
  924. __builtin_mma_xvf32gerpp(&acc0, rowA1, rowB1);
  925. __builtin_mma_xvf32gerpp(&acc1, rowA2, rowB1);
  926. __builtin_mma_xvf32gerpp(&acc2, rowA3, rowB1);
  927. __builtin_mma_xvf32gerpp(&acc3, rowA4, rowB1);
  928. __builtin_mma_xvf32gerpp(&acc4, rowA5, rowB2);
  929. __builtin_mma_xvf32gerpp(&acc5, rowA6, rowB2);
  930. __builtin_mma_xvf32gerpp(&acc6, rowA7, rowB2);
  931. __builtin_mma_xvf32gerpp(&acc7, rowA8, rowB2);
  932. }
  933. for (l = (temp & (~1)); l < temp; ++l)
  934. {
  935. vec_t rowA1 = *(vec_t *) & AO[l<<4];
  936. vec_t rowA2 = *(vec_t *) & AO[(l<<4)+4];
  937. vec_t rowA3 = *(vec_t *) & AO[(l<<4)+8];
  938. vec_t rowA4 = *(vec_t *) & AO[(l<<4)+12];
  939. vec_t rowB1 = *(vec_t *) & BO[l<<1];
  940. __builtin_mma_xvf32gerpp(&acc0, rowA1, rowB1);
  941. __builtin_mma_xvf32gerpp(&acc1, rowA2, rowB1);
  942. __builtin_mma_xvf32gerpp(&acc2, rowA3, rowB1);
  943. __builtin_mma_xvf32gerpp(&acc3, rowA4, rowB1);
  944. }
  945. SAVE_ACC_COMPLEX_21_4
  946. AO += temp << 4;
  947. BO += temp << 1;
  948. CO += 16;
  949. #if defined(TRMMKERNEL)
  950. REFRESH_AFTER_SAVE (8, 1)
  951. #endif
  952. }
  953. if (m & 4)
  954. {
  955. #if defined(TRMMKERNEL)
  956. REFRESH_POINTERS (4, 1)
  957. #else
  958. BO = B;
  959. temp = k;
  960. #endif
  961. SET_ACC_ZERO()
  962. for (l = 0; l < (temp & (~3)); l+=4)
  963. {
  964. vec_t rowA1 = *(vec_t *) & AO[l<<3];
  965. vec_t rowA2 = *(vec_t *) & AO[(l<<3)+4];
  966. vec_t rowA3 = *(vec_t *) & AO[(l<<3)+8];
  967. vec_t rowA4 = *(vec_t *) & AO[(l<<3)+12];
  968. vec_t rowA5 = *(vec_t *) & AO[(l<<3)+16];
  969. vec_t rowA6 = *(vec_t *) & AO[(l<<3)+20];
  970. vec_t rowA7 = *(vec_t *) & AO[(l<<3)+24];
  971. vec_t rowA8 = *(vec_t *) & AO[(l<<3)+28];
  972. vec_t rowB1 = *(vec_t *) & BO[l<<1];
  973. vec_t rowB2 = *(vec_t *) & BO[(l<<1)+2];
  974. vec_t rowB3 = *(vec_t *) & BO[(l<<1)+4];
  975. vec_t rowB4 = *(vec_t *) & BO[(l<<1)+6];
  976. __builtin_mma_xvf32gerpp(&acc0, rowA1, rowB1);
  977. __builtin_mma_xvf32gerpp(&acc1, rowA2, rowB1);
  978. __builtin_mma_xvf32gerpp(&acc2, rowA3, rowB2);
  979. __builtin_mma_xvf32gerpp(&acc3, rowA4, rowB2);
  980. __builtin_mma_xvf32gerpp(&acc4, rowA5, rowB3);
  981. __builtin_mma_xvf32gerpp(&acc5, rowA6, rowB3);
  982. __builtin_mma_xvf32gerpp(&acc6, rowA7, rowB4);
  983. __builtin_mma_xvf32gerpp(&acc7, rowA8, rowB4);
  984. }
  985. for (l = (temp & (~3)); l < temp; ++l)
  986. {
  987. vec_t rowA1 = *(vec_t *) & AO[l<<3];
  988. vec_t rowA2 = *(vec_t *) & AO[(l<<3)+4];
  989. vec_t rowB1 = *(vec_t *) & BO[l<<1];
  990. __builtin_mma_xvf32gerpp(&acc0, rowA1, rowB1);
  991. __builtin_mma_xvf32gerpp(&acc1, rowA2, rowB1);
  992. }
  993. SAVE_ACC_COMPLEX_21_2
  994. AO += temp << 3;
  995. BO += temp << 1;
  996. CO += 8;
  997. #if defined(TRMMKERNEL)
  998. REFRESH_AFTER_SAVE (4, 1)
  999. #endif
  1000. }
  1001. if (m & 2)
  1002. {
  1003. #if defined(TRMMKERNEL)
  1004. REFRESH_POINTERS (2, 1)
  1005. #else
  1006. BO = B;
  1007. temp = k;
  1008. #endif
  1009. // RIP OUT MMA STUFF!
  1010. SET_ACC_ZERO()
  1011. for (l = 0; l < (temp & (~7)); l+=8)
  1012. {
  1013. vec_t rowA1 = *(vec_t *) & AO[l<<2];
  1014. vec_t rowA2 = *(vec_t *) & AO[(l<<2)+4];
  1015. vec_t rowA3 = *(vec_t *) & AO[(l<<2)+8];
  1016. vec_t rowA4 = *(vec_t *) & AO[(l<<2)+12];
  1017. vec_t rowA5 = *(vec_t *) & AO[(l<<2)+16];
  1018. vec_t rowA6 = *(vec_t *) & AO[(l<<2)+20];
  1019. vec_t rowA7 = *(vec_t *) & AO[(l<<2)+24];
  1020. vec_t rowA8 = *(vec_t *) & AO[(l<<2)+28];
  1021. vec_t rowB1 = *(vec_t *) & BO[l<<1];
  1022. vec_t rowB2 = *(vec_t *) & BO[(l<<1)+2];
  1023. vec_t rowB3 = *(vec_t *) & BO[(l<<1)+4];
  1024. vec_t rowB4 = *(vec_t *) & BO[(l<<1)+6];
  1025. vec_t rowB5 = *(vec_t *) & BO[(l<<1)+8];
  1026. vec_t rowB6 = *(vec_t *) & BO[(l<<1)+10];
  1027. vec_t rowB7 = *(vec_t *) & BO[(l<<1)+12];
  1028. vec_t rowB8 = *(vec_t *) & BO[(l<<1)+14];
  1029. __builtin_mma_xvf32gerpp(&acc0, rowA1, rowB1);
  1030. __builtin_mma_xvf32gerpp(&acc1, rowA2, rowB2);
  1031. __builtin_mma_xvf32gerpp(&acc2, rowA3, rowB3);
  1032. __builtin_mma_xvf32gerpp(&acc3, rowA4, rowB4);
  1033. __builtin_mma_xvf32gerpp(&acc4, rowA5, rowB5);
  1034. __builtin_mma_xvf32gerpp(&acc5, rowA6, rowB6);
  1035. __builtin_mma_xvf32gerpp(&acc6, rowA7, rowB7);
  1036. __builtin_mma_xvf32gerpp(&acc7, rowA8, rowB8);
  1037. }
  1038. for (l = (temp & (~7)); l < temp; ++l)
  1039. {
  1040. vec_t rowA1 = *(vec_t *) & AO[l<<2];
  1041. vec_t rowB1 = *(vec_t *) & BO[l<<1];
  1042. __builtin_mma_xvf32gerpp(&acc0, rowA1, rowB1);
  1043. }
  1044. SAVE_ACC_COMPLEX_21_1
  1045. AO += temp << 2;
  1046. BO += temp << 1;
  1047. CO += 4;
  1048. #if defined(TRMMKERNEL)
  1049. REFRESH_AFTER_SAVE (2, 1)
  1050. #endif
  1051. }
  1052. if (m & 1)
  1053. {
  1054. #if defined(TRMMKERNEL)
  1055. REFRESH_POINTERS (1, 1)
  1056. #else
  1057. BO = B;
  1058. temp = k;
  1059. #endif
  1060. // RIP OUT MMA STUFF!
  1061. SET_ACC_ZERO()
  1062. for (l = 0; l < (temp & (~7)); l+=8)
  1063. {
  1064. vec_t rowA1 = *(vec_t *) & AO[l<<1];
  1065. vec_t rowA2 = *(vec_t *) & AO[(l<<1)+2];
  1066. vec_t rowA3 = *(vec_t *) & AO[(l<<1)+4];
  1067. vec_t rowA4 = *(vec_t *) & AO[(l<<1)+6];
  1068. vec_t rowA5 = *(vec_t *) & AO[(l<<1)+8];
  1069. vec_t rowA6 = *(vec_t *) & AO[(l<<1)+10];
  1070. vec_t rowA7 = *(vec_t *) & AO[(l<<1)+12];
  1071. vec_t rowA8 = *(vec_t *) & AO[(l<<1)+14];
  1072. vec_t rowB1 = *(vec_t *) & BO[l<<1];
  1073. vec_t rowB2 = *(vec_t *) & BO[(l<<1)+2];
  1074. vec_t rowB3 = *(vec_t *) & BO[(l<<1)+4];
  1075. vec_t rowB4 = *(vec_t *) & BO[(l<<1)+6];
  1076. vec_t rowB5 = *(vec_t *) & BO[(l<<1)+8];
  1077. vec_t rowB6 = *(vec_t *) & BO[(l<<1)+10];
  1078. vec_t rowB7 = *(vec_t *) & BO[(l<<1)+12];
  1079. vec_t rowB8 = *(vec_t *) & BO[(l<<1)+14];
  1080. __builtin_mma_xvf32gerpp(&acc0, rowA1, rowB1);
  1081. __builtin_mma_xvf32gerpp(&acc1, rowA2, rowB2);
  1082. __builtin_mma_xvf32gerpp(&acc2, rowA3, rowB3);
  1083. __builtin_mma_xvf32gerpp(&acc3, rowA4, rowB4);
  1084. __builtin_mma_xvf32gerpp(&acc4, rowA5, rowB5);
  1085. __builtin_mma_xvf32gerpp(&acc5, rowA6, rowB6);
  1086. __builtin_mma_xvf32gerpp(&acc6, rowA7, rowB7);
  1087. __builtin_mma_xvf32gerpp(&acc7, rowA8, rowB8);
  1088. }
  1089. for (l = (temp & (~7)); l < temp; ++l)
  1090. {
  1091. vec_t rowA1 = *(vec_t *) & AO[l<<1];
  1092. vec_t rowB1 = *(vec_t *) & BO[l<<1];
  1093. __builtin_mma_xvf32gerpp(&acc0, rowA1, rowB1);
  1094. }
  1095. SAVE_ACC_COMPLEX_11
  1096. AO += temp<<1;
  1097. BO += temp<<1;
  1098. CO += 2;
  1099. #if defined(TRMMKERNEL)
  1100. REFRESH_AFTER_SAVE (1, 1)
  1101. #endif
  1102. }
  1103. #if defined(TRMMKERNEL) && !defined(LEFT)
  1104. off += 1; // number of values in A
  1105. #endif
  1106. B += k << 1;
  1107. }
  1108. return 0;
  1109. }