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.

dtrsm_kernel_RT_8x4_msa.c 27 kB

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