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_8x4_msa.c 31 kB

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