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.

strsm_kernel_LN_bulldozer.c 35 kB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756
  1. /*********************************************************************/
  2. /* Copyright 2009, 2010 The University of Texas at Austin. */
  3. /* All rights reserved. */
  4. /* */
  5. /* Redistribution and use in source and binary forms, with or */
  6. /* without modification, are permitted provided that the following */
  7. /* conditions are met: */
  8. /* */
  9. /* 1. Redistributions of source code must retain the above */
  10. /* copyright notice, this list of conditions and the following */
  11. /* disclaimer. */
  12. /* */
  13. /* 2. Redistributions in binary form must reproduce the above */
  14. /* copyright notice, this list of conditions and the following */
  15. /* disclaimer in the documentation and/or other materials */
  16. /* provided with the distribution. */
  17. /* */
  18. /* THIS SOFTWARE IS PROVIDED BY THE UNIVERSITY OF TEXAS AT */
  19. /* AUSTIN ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, */
  20. /* INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF */
  21. /* MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE */
  22. /* DISCLAIMED. IN NO EVENT SHALL THE UNIVERSITY OF TEXAS AT */
  23. /* AUSTIN OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, */
  24. /* INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES */
  25. /* (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE */
  26. /* GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR */
  27. /* BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF */
  28. /* LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT */
  29. /* (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT */
  30. /* OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE */
  31. /* POSSIBILITY OF SUCH DAMAGE. */
  32. /* */
  33. /* The views and conclusions contained in the software and */
  34. /* documentation are those of the authors and should not be */
  35. /* interpreted as representing official policies, either expressed */
  36. /* or implied, of The University of Texas at Austin. */
  37. /*********************************************************************/
  38. #include "common.h"
  39. static FLOAT dm1 = -1.;
  40. #ifdef CONJ
  41. #define GEMM_KERNEL GEMM_KERNEL_L
  42. #else
  43. #define GEMM_KERNEL GEMM_KERNEL_N
  44. #endif
  45. #if GEMM_DEFAULT_UNROLL_M == 1
  46. #define GEMM_UNROLL_M_SHIFT 0
  47. #endif
  48. #if GEMM_DEFAULT_UNROLL_M == 2
  49. #define GEMM_UNROLL_M_SHIFT 1
  50. #endif
  51. #if GEMM_DEFAULT_UNROLL_M == 4
  52. #define GEMM_UNROLL_M_SHIFT 2
  53. #endif
  54. #if GEMM_DEFAULT_UNROLL_M == 6
  55. #define GEMM_UNROLL_M_SHIFT 2
  56. #endif
  57. #if GEMM_DEFAULT_UNROLL_M == 8
  58. #define GEMM_UNROLL_M_SHIFT 3
  59. #endif
  60. #if GEMM_DEFAULT_UNROLL_M == 16
  61. #define GEMM_UNROLL_M_SHIFT 4
  62. #endif
  63. #if GEMM_DEFAULT_UNROLL_N == 1
  64. #define GEMM_UNROLL_N_SHIFT 0
  65. #endif
  66. #if GEMM_DEFAULT_UNROLL_N == 2
  67. #define GEMM_UNROLL_N_SHIFT 1
  68. #endif
  69. #if GEMM_DEFAULT_UNROLL_N == 4
  70. #define GEMM_UNROLL_N_SHIFT 2
  71. #endif
  72. #if GEMM_DEFAULT_UNROLL_N == 8
  73. #define GEMM_UNROLL_N_SHIFT 3
  74. #endif
  75. #if GEMM_DEFAULT_UNROLL_N == 16
  76. #define GEMM_UNROLL_N_SHIFT 4
  77. #endif
  78. #ifndef COMPLEX
  79. static void strsm_LN_solve_opt(BLASLONG n, FLOAT *a, FLOAT *b, FLOAT *c, BLASLONG ldc, FLOAT *as, FLOAT *bs) __attribute__ ((noinline));
  80. static void strsm_LN_solve_opt(BLASLONG n, FLOAT *a, FLOAT *b, FLOAT *c, BLASLONG ldc, FLOAT *as, FLOAT *bs)
  81. {
  82. FLOAT *c1 = c + ldc ;
  83. BLASLONG n1 = n * 8;
  84. BLASLONG i=0;
  85. as += (16 - 1) * 16;
  86. bs += (16 - 1) * 2;
  87. __asm__ __volatile__
  88. (
  89. " vzeroupper \n\t"
  90. " prefetcht0 (%4) \n\t"
  91. " prefetcht0 (%5) \n\t"
  92. " vxorps %%xmm8 , %%xmm8 , %%xmm8 \n\t"
  93. " vxorps %%xmm9 , %%xmm9 , %%xmm9 \n\t"
  94. " vxorps %%xmm10, %%xmm10, %%xmm10 \n\t"
  95. " vxorps %%xmm11, %%xmm11, %%xmm11 \n\t"
  96. " vxorps %%xmm12, %%xmm12, %%xmm12 \n\t"
  97. " vxorps %%xmm13, %%xmm13, %%xmm13 \n\t"
  98. " vxorps %%xmm14, %%xmm14, %%xmm14 \n\t"
  99. " vxorps %%xmm15, %%xmm15, %%xmm15 \n\t"
  100. " cmpq $0, %0 \n\t"
  101. " je 2f \n\t"
  102. " .align 16 \n\t"
  103. "1: \n\t"
  104. " vbroadcastss (%7,%1,1), %%xmm0 \n\t" // read b
  105. " vmovups (%6,%1,8), %%xmm4 \n\t"
  106. " vbroadcastss 4(%7,%1,1), %%xmm1 \n\t"
  107. " vmovups 16(%6,%1,8), %%xmm5 \n\t"
  108. " vmovups 32(%6,%1,8), %%xmm6 \n\t"
  109. " vmovups 48(%6,%1,8), %%xmm7 \n\t"
  110. " vfmaddps %%xmm8 , %%xmm0 , %%xmm4 , %%xmm8 \n\t"
  111. " vfmaddps %%xmm12, %%xmm1 , %%xmm4 , %%xmm12 \n\t"
  112. " vfmaddps %%xmm9 , %%xmm0 , %%xmm5 , %%xmm9 \n\t"
  113. " vfmaddps %%xmm13, %%xmm1 , %%xmm5 , %%xmm13 \n\t"
  114. " vfmaddps %%xmm10, %%xmm0 , %%xmm6 , %%xmm10 \n\t"
  115. " vfmaddps %%xmm14, %%xmm1 , %%xmm6 , %%xmm14 \n\t"
  116. " addq $8, %1 \n\t"
  117. " vfmaddps %%xmm11, %%xmm0 , %%xmm7 , %%xmm11 \n\t"
  118. " vfmaddps %%xmm15, %%xmm1 , %%xmm7 , %%xmm15 \n\t"
  119. " cmpq %1, %0 \n\t"
  120. " jnz 1b \n\t"
  121. "2: \n\t"
  122. " vmovups (%4) , %%xmm0 \n\t"
  123. " vmovups 16(%4) , %%xmm1 \n\t"
  124. " vmovups 32(%4) , %%xmm2 \n\t"
  125. " vmovups 48(%4) , %%xmm3 \n\t"
  126. " vmovups (%5) , %%xmm4 \n\t"
  127. " vmovups 16(%5) , %%xmm5 \n\t"
  128. " vmovups 32(%5) , %%xmm6 \n\t"
  129. " vmovups 48(%5) , %%xmm7 \n\t"
  130. " vsubps %%xmm8 , %%xmm0 , %%xmm8 \n\t"
  131. " vsubps %%xmm9 , %%xmm1 , %%xmm9 \n\t"
  132. " vsubps %%xmm10, %%xmm2 , %%xmm10 \n\t"
  133. " vsubps %%xmm11, %%xmm3 , %%xmm11 \n\t"
  134. " vsubps %%xmm12, %%xmm4 , %%xmm12 \n\t"
  135. " vsubps %%xmm13, %%xmm5 , %%xmm13 \n\t"
  136. " vsubps %%xmm14, %%xmm6 , %%xmm14 \n\t"
  137. " vsubps %%xmm15, %%xmm7 , %%xmm15 \n\t"
  138. "3: \n\t"
  139. " vbroadcastss 60(%2) , %%xmm0 \n\t" // i=15, read aa[i]
  140. " vshufps $0xff , %%xmm11 , %%xmm11 , %%xmm1 \n\t" // extract bb0
  141. " vshufps $0xff , %%xmm15 , %%xmm15 , %%xmm2 \n\t" // extract bb1
  142. " vmulps %%xmm0 , %%xmm1 , %%xmm1 \n\t" // bb0 * aa
  143. " vmulps %%xmm0 , %%xmm2 , %%xmm2 \n\t" // bb1 * aa
  144. " vmovss %%xmm1 , 60(%4) \n\t" // c[i] = bb0 * aa
  145. " vmovss %%xmm2 , 60(%5) \n\t" // c[i] = bb1 * aa
  146. " vmovss %%xmm1 , (%3) \n\t" // b[0] = bb0 * aa
  147. " vmovss %%xmm2 , 4(%3) \n\t" // b[1] = bb1 * aa
  148. " vmovups 0(%2) , %%xmm4 \n\t" // read a[k]
  149. " vmovups 16(%2) , %%xmm5 \n\t" // read a[k]
  150. " vmovups 32(%2) , %%xmm6 \n\t" // read a[k]
  151. " vmovups 48(%2) , %%xmm7 \n\t" // read a[k]
  152. " vfnmaddps %%xmm8 , %%xmm1 , %%xmm4 , %%xmm8 \n\t"
  153. " vfnmaddps %%xmm12 , %%xmm2 , %%xmm4 , %%xmm12 \n\t"
  154. " vfnmaddps %%xmm9 , %%xmm1 , %%xmm5 , %%xmm9 \n\t"
  155. " vfnmaddps %%xmm13 , %%xmm2 , %%xmm5 , %%xmm13 \n\t"
  156. " vfnmaddps %%xmm10 , %%xmm1 , %%xmm6 , %%xmm10 \n\t"
  157. " vfnmaddps %%xmm14 , %%xmm2 , %%xmm6 , %%xmm14 \n\t"
  158. " vfnmaddps %%xmm11 , %%xmm1 , %%xmm7 , %%xmm11 \n\t"
  159. " vfnmaddps %%xmm15 , %%xmm2 , %%xmm7 , %%xmm15 \n\t"
  160. " subq $64 , %2 \n\t" // a -= m
  161. " subq $8 , %3 \n\t" // b -= n
  162. " vbroadcastss 56(%2) , %%xmm0 \n\t" // i=14, read aa[i]
  163. " vshufps $0xaa , %%xmm11 , %%xmm11 , %%xmm1 \n\t" // extract bb0
  164. " vshufps $0xaa , %%xmm15 , %%xmm15 , %%xmm2 \n\t" // extract bb1
  165. " vmulps %%xmm0 , %%xmm1 , %%xmm1 \n\t" // bb0 * aa
  166. " vmulps %%xmm0 , %%xmm2 , %%xmm2 \n\t" // bb1 * aa
  167. " vmovss %%xmm1 , 56(%4) \n\t" // c[i] = bb0 * aa
  168. " vmovss %%xmm2 , 56(%5) \n\t" // c[i] = bb1 * aa
  169. " vmovss %%xmm1 , (%3) \n\t" // b[0] = bb0 * aa
  170. " vmovss %%xmm2 , 4(%3) \n\t" // b[1] = bb1 * aa
  171. " vmovups 0(%2) , %%xmm4 \n\t" // read a[k]
  172. " vmovups 16(%2) , %%xmm5 \n\t" // read a[k]
  173. " vmovups 32(%2) , %%xmm6 \n\t" // read a[k]
  174. " vmovups 48(%2) , %%xmm7 \n\t" // read a[k]
  175. " vfnmaddps %%xmm8 , %%xmm1 , %%xmm4 , %%xmm8 \n\t"
  176. " vfnmaddps %%xmm12 , %%xmm2 , %%xmm4 , %%xmm12 \n\t"
  177. " vfnmaddps %%xmm9 , %%xmm1 , %%xmm5 , %%xmm9 \n\t"
  178. " vfnmaddps %%xmm13 , %%xmm2 , %%xmm5 , %%xmm13 \n\t"
  179. " vfnmaddps %%xmm10 , %%xmm1 , %%xmm6 , %%xmm10 \n\t"
  180. " vfnmaddps %%xmm14 , %%xmm2 , %%xmm6 , %%xmm14 \n\t"
  181. " vfnmaddps %%xmm11 , %%xmm1 , %%xmm7 , %%xmm11 \n\t"
  182. " vfnmaddps %%xmm15 , %%xmm2 , %%xmm7 , %%xmm15 \n\t"
  183. " subq $64 , %2 \n\t" // a -= m
  184. " subq $8 , %3 \n\t" // b -= n
  185. " vbroadcastss 52(%2) , %%xmm0 \n\t" // i=13, read aa[i]
  186. " vshufps $0x55 , %%xmm11 , %%xmm11 , %%xmm1 \n\t" // extract bb0
  187. " vshufps $0x55 , %%xmm15 , %%xmm15 , %%xmm2 \n\t" // extract bb1
  188. " vmulps %%xmm0 , %%xmm1 , %%xmm1 \n\t" // bb0 * aa
  189. " vmulps %%xmm0 , %%xmm2 , %%xmm2 \n\t" // bb1 * aa
  190. " vmovss %%xmm1 , 52(%4) \n\t" // c[i] = bb0 * aa
  191. " vmovss %%xmm2 , 52(%5) \n\t" // c[i] = bb1 * aa
  192. " vmovss %%xmm1 , (%3) \n\t" // b[0] = bb0 * aa
  193. " vmovss %%xmm2 , 4(%3) \n\t" // b[1] = bb1 * aa
  194. " vmovups 0(%2) , %%xmm4 \n\t" // read a[k]
  195. " vmovups 16(%2) , %%xmm5 \n\t" // read a[k]
  196. " vmovups 32(%2) , %%xmm6 \n\t" // read a[k]
  197. " vmovups 48(%2) , %%xmm7 \n\t" // read a[k]
  198. " vfnmaddps %%xmm8 , %%xmm1 , %%xmm4 , %%xmm8 \n\t"
  199. " vfnmaddps %%xmm12 , %%xmm2 , %%xmm4 , %%xmm12 \n\t"
  200. " vfnmaddps %%xmm9 , %%xmm1 , %%xmm5 , %%xmm9 \n\t"
  201. " vfnmaddps %%xmm13 , %%xmm2 , %%xmm5 , %%xmm13 \n\t"
  202. " vfnmaddps %%xmm10 , %%xmm1 , %%xmm6 , %%xmm10 \n\t"
  203. " vfnmaddps %%xmm14 , %%xmm2 , %%xmm6 , %%xmm14 \n\t"
  204. " vfnmaddps %%xmm11 , %%xmm1 , %%xmm7 , %%xmm11 \n\t"
  205. " vfnmaddps %%xmm15 , %%xmm2 , %%xmm7 , %%xmm15 \n\t"
  206. " subq $64 , %2 \n\t" // a -= m
  207. " subq $8 , %3 \n\t" // b -= n
  208. " vbroadcastss 48(%2) , %%xmm0 \n\t" // i=12, read aa[i]
  209. " vshufps $0x00 , %%xmm11 , %%xmm11 , %%xmm1 \n\t" // extract bb0
  210. " vshufps $0x00 , %%xmm15 , %%xmm15 , %%xmm2 \n\t" // extract bb1
  211. " vmulps %%xmm0 , %%xmm1 , %%xmm1 \n\t" // bb0 * aa
  212. " vmulps %%xmm0 , %%xmm2 , %%xmm2 \n\t" // bb1 * aa
  213. " vmovss %%xmm1 , 48(%4) \n\t" // c[i] = bb0 * aa
  214. " vmovss %%xmm2 , 48(%5) \n\t" // c[i] = bb1 * aa
  215. " vmovss %%xmm1 , (%3) \n\t" // b[0] = bb0 * aa
  216. " vmovss %%xmm2 , 4(%3) \n\t" // b[1] = bb1 * aa
  217. " vmovups 0(%2) , %%xmm4 \n\t" // read a[k]
  218. " vmovups 16(%2) , %%xmm5 \n\t" // read a[k]
  219. " vmovups 32(%2) , %%xmm6 \n\t" // read a[k]
  220. " vfnmaddps %%xmm8 , %%xmm1 , %%xmm4 , %%xmm8 \n\t"
  221. " vfnmaddps %%xmm12 , %%xmm2 , %%xmm4 , %%xmm12 \n\t"
  222. " vfnmaddps %%xmm9 , %%xmm1 , %%xmm5 , %%xmm9 \n\t"
  223. " vfnmaddps %%xmm13 , %%xmm2 , %%xmm5 , %%xmm13 \n\t"
  224. " vfnmaddps %%xmm10 , %%xmm1 , %%xmm6 , %%xmm10 \n\t"
  225. " vfnmaddps %%xmm14 , %%xmm2 , %%xmm6 , %%xmm14 \n\t"
  226. " subq $64 , %2 \n\t" // a -= m
  227. " subq $8 , %3 \n\t" // b -= n
  228. " vbroadcastss 44(%2) , %%xmm0 \n\t" // i=11, read aa[i]
  229. " vshufps $0xff , %%xmm10 , %%xmm10 , %%xmm1 \n\t" // extract bb0
  230. " vshufps $0xff , %%xmm14 , %%xmm14 , %%xmm2 \n\t" // extract bb1
  231. " vmulps %%xmm0 , %%xmm1 , %%xmm1 \n\t" // bb0 * aa
  232. " vmulps %%xmm0 , %%xmm2 , %%xmm2 \n\t" // bb1 * aa
  233. " vmovss %%xmm1 , 44(%4) \n\t" // c[i] = bb0 * aa
  234. " vmovss %%xmm2 , 44(%5) \n\t" // c[i] = bb1 * aa
  235. " vmovss %%xmm1 , (%3) \n\t" // b[0] = bb0 * aa
  236. " vmovss %%xmm2 , 4(%3) \n\t" // b[1] = bb1 * aa
  237. " vmovups 0(%2) , %%xmm4 \n\t" // read a[k]
  238. " vmovups 16(%2) , %%xmm5 \n\t" // read a[k]
  239. " vmovups 32(%2) , %%xmm6 \n\t" // read a[k]
  240. " vfnmaddps %%xmm8 , %%xmm1 , %%xmm4 , %%xmm8 \n\t"
  241. " vfnmaddps %%xmm12 , %%xmm2 , %%xmm4 , %%xmm12 \n\t"
  242. " vfnmaddps %%xmm9 , %%xmm1 , %%xmm5 , %%xmm9 \n\t"
  243. " vfnmaddps %%xmm13 , %%xmm2 , %%xmm5 , %%xmm13 \n\t"
  244. " vfnmaddps %%xmm10 , %%xmm1 , %%xmm6 , %%xmm10 \n\t"
  245. " vfnmaddps %%xmm14 , %%xmm2 , %%xmm6 , %%xmm14 \n\t"
  246. " subq $64 , %2 \n\t" // a -= m
  247. " subq $8 , %3 \n\t" // b -= n
  248. " vbroadcastss 40(%2) , %%xmm0 \n\t" // i=10, read aa[i]
  249. " vshufps $0xaa , %%xmm10 , %%xmm10 , %%xmm1 \n\t" // extract bb0
  250. " vshufps $0xaa , %%xmm14 , %%xmm14 , %%xmm2 \n\t" // extract bb1
  251. " vmulps %%xmm0 , %%xmm1 , %%xmm1 \n\t" // bb0 * aa
  252. " vmulps %%xmm0 , %%xmm2 , %%xmm2 \n\t" // bb1 * aa
  253. " vmovss %%xmm1 , 40(%4) \n\t" // c[i] = bb0 * aa
  254. " vmovss %%xmm2 , 40(%5) \n\t" // c[i] = bb1 * aa
  255. " vmovss %%xmm1 , (%3) \n\t" // b[0] = bb0 * aa
  256. " vmovss %%xmm2 , 4(%3) \n\t" // b[1] = bb1 * aa
  257. " vmovups 0(%2) , %%xmm4 \n\t" // read a[k]
  258. " vmovups 16(%2) , %%xmm5 \n\t" // read a[k]
  259. " vmovups 32(%2) , %%xmm6 \n\t" // read a[k]
  260. " vfnmaddps %%xmm8 , %%xmm1 , %%xmm4 , %%xmm8 \n\t"
  261. " vfnmaddps %%xmm12 , %%xmm2 , %%xmm4 , %%xmm12 \n\t"
  262. " vfnmaddps %%xmm9 , %%xmm1 , %%xmm5 , %%xmm9 \n\t"
  263. " vfnmaddps %%xmm13 , %%xmm2 , %%xmm5 , %%xmm13 \n\t"
  264. " vfnmaddps %%xmm10 , %%xmm1 , %%xmm6 , %%xmm10 \n\t"
  265. " vfnmaddps %%xmm14 , %%xmm2 , %%xmm6 , %%xmm14 \n\t"
  266. " subq $64 , %2 \n\t" // a -= m
  267. " subq $8 , %3 \n\t" // b -= n
  268. " vbroadcastss 36(%2) , %%xmm0 \n\t" // i=9 , read aa[i]
  269. " vshufps $0x55 , %%xmm10 , %%xmm10 , %%xmm1 \n\t" // extract bb0
  270. " vshufps $0x55 , %%xmm14 , %%xmm14 , %%xmm2 \n\t" // extract bb1
  271. " vmulps %%xmm0 , %%xmm1 , %%xmm1 \n\t" // bb0 * aa
  272. " vmulps %%xmm0 , %%xmm2 , %%xmm2 \n\t" // bb1 * aa
  273. " vmovss %%xmm1 , 36(%4) \n\t" // c[i] = bb0 * aa
  274. " vmovss %%xmm2 , 36(%5) \n\t" // c[i] = bb1 * aa
  275. " vmovss %%xmm1 , (%3) \n\t" // b[0] = bb0 * aa
  276. " vmovss %%xmm2 , 4(%3) \n\t" // b[1] = bb1 * aa
  277. " vmovups 0(%2) , %%xmm4 \n\t" // read a[k]
  278. " vmovups 16(%2) , %%xmm5 \n\t" // read a[k]
  279. " vmovups 32(%2) , %%xmm6 \n\t" // read a[k]
  280. " vfnmaddps %%xmm8 , %%xmm1 , %%xmm4 , %%xmm8 \n\t"
  281. " vfnmaddps %%xmm12 , %%xmm2 , %%xmm4 , %%xmm12 \n\t"
  282. " vfnmaddps %%xmm9 , %%xmm1 , %%xmm5 , %%xmm9 \n\t"
  283. " vfnmaddps %%xmm13 , %%xmm2 , %%xmm5 , %%xmm13 \n\t"
  284. " vfnmaddps %%xmm10 , %%xmm1 , %%xmm6 , %%xmm10 \n\t"
  285. " vfnmaddps %%xmm14 , %%xmm2 , %%xmm6 , %%xmm14 \n\t"
  286. " subq $64 , %2 \n\t" // a -= m
  287. " subq $8 , %3 \n\t" // b -= n
  288. " vbroadcastss 32(%2) , %%xmm0 \n\t" // i=8 , read aa[i]
  289. " vshufps $0x00 , %%xmm10 , %%xmm10 , %%xmm1 \n\t" // extract bb0
  290. " vshufps $0x00 , %%xmm14 , %%xmm14 , %%xmm2 \n\t" // extract bb1
  291. " vmulps %%xmm0 , %%xmm1 , %%xmm1 \n\t" // bb0 * aa
  292. " vmulps %%xmm0 , %%xmm2 , %%xmm2 \n\t" // bb1 * aa
  293. " vmovss %%xmm1 , 32(%4) \n\t" // c[i] = bb0 * aa
  294. " vmovss %%xmm2 , 32(%5) \n\t" // c[i] = bb1 * aa
  295. " vmovss %%xmm1 , (%3) \n\t" // b[0] = bb0 * aa
  296. " vmovss %%xmm2 , 4(%3) \n\t" // b[1] = bb1 * aa
  297. " vmovups 0(%2) , %%xmm4 \n\t" // read a[k]
  298. " vmovups 16(%2) , %%xmm5 \n\t" // read a[k]
  299. " vfnmaddps %%xmm8 , %%xmm1 , %%xmm4 , %%xmm8 \n\t"
  300. " vfnmaddps %%xmm12 , %%xmm2 , %%xmm4 , %%xmm12 \n\t"
  301. " vfnmaddps %%xmm9 , %%xmm1 , %%xmm5 , %%xmm9 \n\t"
  302. " vfnmaddps %%xmm13 , %%xmm2 , %%xmm5 , %%xmm13 \n\t"
  303. " subq $64 , %2 \n\t" // a -= m
  304. " subq $8 , %3 \n\t" // b -= n
  305. " vbroadcastss 28(%2) , %%xmm0 \n\t" // i=7 , read aa[i]
  306. " vshufps $0xff , %%xmm9 , %%xmm9 , %%xmm1 \n\t" // extract bb0
  307. " vshufps $0xff , %%xmm13 , %%xmm13 , %%xmm2 \n\t" // extract bb1
  308. " vmulps %%xmm0 , %%xmm1 , %%xmm1 \n\t" // bb0 * aa
  309. " vmulps %%xmm0 , %%xmm2 , %%xmm2 \n\t" // bb1 * aa
  310. " vmovss %%xmm1 , 28(%4) \n\t" // c[i] = bb0 * aa
  311. " vmovss %%xmm2 , 28(%5) \n\t" // c[i] = bb1 * aa
  312. " vmovss %%xmm1 , (%3) \n\t" // b[0] = bb0 * aa
  313. " vmovss %%xmm2 , 4(%3) \n\t" // b[1] = bb1 * aa
  314. " vmovups 0(%2) , %%xmm4 \n\t" // read a[k]
  315. " vmovups 16(%2) , %%xmm5 \n\t" // read a[k]
  316. " vfnmaddps %%xmm8 , %%xmm1 , %%xmm4 , %%xmm8 \n\t"
  317. " vfnmaddps %%xmm12 , %%xmm2 , %%xmm4 , %%xmm12 \n\t"
  318. " vfnmaddps %%xmm9 , %%xmm1 , %%xmm5 , %%xmm9 \n\t"
  319. " vfnmaddps %%xmm13 , %%xmm2 , %%xmm5 , %%xmm13 \n\t"
  320. " subq $64 , %2 \n\t" // a -= m
  321. " subq $8 , %3 \n\t" // b -= n
  322. " vbroadcastss 24(%2) , %%xmm0 \n\t" // i=6 , read aa[i]
  323. " vshufps $0xaa , %%xmm9 , %%xmm9 , %%xmm1 \n\t" // extract bb0
  324. " vshufps $0xaa , %%xmm13 , %%xmm13 , %%xmm2 \n\t" // extract bb1
  325. " vmulps %%xmm0 , %%xmm1 , %%xmm1 \n\t" // bb0 * aa
  326. " vmulps %%xmm0 , %%xmm2 , %%xmm2 \n\t" // bb1 * aa
  327. " vmovss %%xmm1 , 24(%4) \n\t" // c[i] = bb0 * aa
  328. " vmovss %%xmm2 , 24(%5) \n\t" // c[i] = bb1 * aa
  329. " vmovss %%xmm1 , (%3) \n\t" // b[0] = bb0 * aa
  330. " vmovss %%xmm2 , 4(%3) \n\t" // b[1] = bb1 * aa
  331. " vmovups 0(%2) , %%xmm4 \n\t" // read a[k]
  332. " vmovups 16(%2) , %%xmm5 \n\t" // read a[k]
  333. " vfnmaddps %%xmm8 , %%xmm1 , %%xmm4 , %%xmm8 \n\t"
  334. " vfnmaddps %%xmm12 , %%xmm2 , %%xmm4 , %%xmm12 \n\t"
  335. " vfnmaddps %%xmm9 , %%xmm1 , %%xmm5 , %%xmm9 \n\t"
  336. " vfnmaddps %%xmm13 , %%xmm2 , %%xmm5 , %%xmm13 \n\t"
  337. " subq $64 , %2 \n\t" // a -= m
  338. " subq $8 , %3 \n\t" // b -= n
  339. " vbroadcastss 20(%2) , %%xmm0 \n\t" // i=5 , read aa[i]
  340. " vshufps $0x55 , %%xmm9 , %%xmm9 , %%xmm1 \n\t" // extract bb0
  341. " vshufps $0x55 , %%xmm13 , %%xmm13 , %%xmm2 \n\t" // extract bb1
  342. " vmulps %%xmm0 , %%xmm1 , %%xmm1 \n\t" // bb0 * aa
  343. " vmulps %%xmm0 , %%xmm2 , %%xmm2 \n\t" // bb1 * aa
  344. " vmovss %%xmm1 , 20(%4) \n\t" // c[i] = bb0 * aa
  345. " vmovss %%xmm2 , 20(%5) \n\t" // c[i] = bb1 * aa
  346. " vmovss %%xmm1 , (%3) \n\t" // b[0] = bb0 * aa
  347. " vmovss %%xmm2 , 4(%3) \n\t" // b[1] = bb1 * aa
  348. " vmovups 0(%2) , %%xmm4 \n\t" // read a[k]
  349. " vmovups 16(%2) , %%xmm5 \n\t" // read a[k]
  350. " vfnmaddps %%xmm8 , %%xmm1 , %%xmm4 , %%xmm8 \n\t"
  351. " vfnmaddps %%xmm12 , %%xmm2 , %%xmm4 , %%xmm12 \n\t"
  352. " vfnmaddps %%xmm9 , %%xmm1 , %%xmm5 , %%xmm9 \n\t"
  353. " vfnmaddps %%xmm13 , %%xmm2 , %%xmm5 , %%xmm13 \n\t"
  354. " subq $64 , %2 \n\t" // a -= m
  355. " subq $8 , %3 \n\t" // b -= n
  356. " vbroadcastss 16(%2) , %%xmm0 \n\t" // i=4 , read aa[i]
  357. " vshufps $0x00 , %%xmm9 , %%xmm9 , %%xmm1 \n\t" // extract bb0
  358. " vshufps $0x00 , %%xmm13 , %%xmm13 , %%xmm2 \n\t" // extract bb1
  359. " vmulps %%xmm0 , %%xmm1 , %%xmm1 \n\t" // bb0 * aa
  360. " vmulps %%xmm0 , %%xmm2 , %%xmm2 \n\t" // bb1 * aa
  361. " vmovss %%xmm1 , 16(%4) \n\t" // c[i] = bb0 * aa
  362. " vmovss %%xmm2 , 16(%5) \n\t" // c[i] = bb1 * aa
  363. " vmovss %%xmm1 , (%3) \n\t" // b[0] = bb0 * aa
  364. " vmovss %%xmm2 , 4(%3) \n\t" // b[1] = bb1 * aa
  365. " vmovups 0(%2) , %%xmm4 \n\t" // read a[k]
  366. " vfnmaddps %%xmm8 , %%xmm1 , %%xmm4 , %%xmm8 \n\t"
  367. " vfnmaddps %%xmm12 , %%xmm2 , %%xmm4 , %%xmm12 \n\t"
  368. " subq $64 , %2 \n\t" // a -= m
  369. " subq $8 , %3 \n\t" // b -= n
  370. " vbroadcastss 12(%2) , %%xmm0 \n\t" // i=3 , read aa[i]
  371. " vshufps $0xff , %%xmm8 , %%xmm8 , %%xmm1 \n\t" // extract bb0
  372. " vshufps $0xff , %%xmm12 , %%xmm12 , %%xmm2 \n\t" // extract bb1
  373. " vmulps %%xmm0 , %%xmm1 , %%xmm1 \n\t" // bb0 * aa
  374. " vmulps %%xmm0 , %%xmm2 , %%xmm2 \n\t" // bb1 * aa
  375. " vmovss %%xmm1 , 12(%4) \n\t" // c[i] = bb0 * aa
  376. " vmovss %%xmm2 , 12(%5) \n\t" // c[i] = bb1 * aa
  377. " vmovss %%xmm1 , (%3) \n\t" // b[0] = bb0 * aa
  378. " vmovss %%xmm2 , 4(%3) \n\t" // b[1] = bb1 * aa
  379. " vmovups 0(%2) , %%xmm4 \n\t" // read a[k]
  380. " vfnmaddps %%xmm8 , %%xmm1 , %%xmm4 , %%xmm8 \n\t"
  381. " vfnmaddps %%xmm12 , %%xmm2 , %%xmm4 , %%xmm12 \n\t"
  382. " subq $64 , %2 \n\t" // a -= m
  383. " subq $8 , %3 \n\t" // b -= n
  384. " vbroadcastss 8(%2) , %%xmm0 \n\t" // i=2 , read aa[i]
  385. " vshufps $0xaa , %%xmm8 , %%xmm8 , %%xmm1 \n\t" // extract bb0
  386. " vshufps $0xaa , %%xmm12 , %%xmm12 , %%xmm2 \n\t" // extract bb1
  387. " vmulps %%xmm0 , %%xmm1 , %%xmm1 \n\t" // bb0 * aa
  388. " vmulps %%xmm0 , %%xmm2 , %%xmm2 \n\t" // bb1 * aa
  389. " vmovss %%xmm1 , 8(%4) \n\t" // c[i] = bb0 * aa
  390. " vmovss %%xmm2 , 8(%5) \n\t" // c[i] = bb1 * aa
  391. " vmovss %%xmm1 , (%3) \n\t" // b[0] = bb0 * aa
  392. " vmovss %%xmm2 , 4(%3) \n\t" // b[1] = bb1 * aa
  393. " vmovups 0(%2) , %%xmm4 \n\t" // read a[k]
  394. " vfnmaddps %%xmm8 , %%xmm1 , %%xmm4 , %%xmm8 \n\t"
  395. " vfnmaddps %%xmm12 , %%xmm2 , %%xmm4 , %%xmm12 \n\t"
  396. " subq $64 , %2 \n\t" // a -= m
  397. " subq $8 , %3 \n\t" // b -= n
  398. " vbroadcastss 4(%2) , %%xmm0 \n\t" // i=1 , read aa[i]
  399. " vshufps $0x55 , %%xmm8 , %%xmm8 , %%xmm1 \n\t" // extract bb0
  400. " vshufps $0x55 , %%xmm12 , %%xmm12 , %%xmm2 \n\t" // extract bb1
  401. " vmulps %%xmm0 , %%xmm1 , %%xmm1 \n\t" // bb0 * aa
  402. " vmulps %%xmm0 , %%xmm2 , %%xmm2 \n\t" // bb1 * aa
  403. " vmovss %%xmm1 , 4(%4) \n\t" // c[i] = bb0 * aa
  404. " vmovss %%xmm2 , 4(%5) \n\t" // c[i] = bb1 * aa
  405. " vmovss %%xmm1 , (%3) \n\t" // b[0] = bb0 * aa
  406. " vmovss %%xmm2 , 4(%3) \n\t" // b[1] = bb1 * aa
  407. " vmovups 0(%2) , %%xmm4 \n\t" // read a[k]
  408. " vfnmaddps %%xmm8 , %%xmm1 , %%xmm4 , %%xmm8 \n\t"
  409. " vfnmaddps %%xmm12 , %%xmm2 , %%xmm4 , %%xmm12 \n\t"
  410. " subq $64 , %2 \n\t" // a -= m
  411. " subq $8 , %3 \n\t" // b -= n
  412. " vbroadcastss 0(%2) , %%xmm0 \n\t" // i=0 , read aa[i]
  413. " vshufps $0x00 , %%xmm8 , %%xmm8 , %%xmm1 \n\t" // extract bb0
  414. " vshufps $0x00 , %%xmm12 , %%xmm12 , %%xmm2 \n\t" // extract bb1
  415. " vmulps %%xmm0 , %%xmm1 , %%xmm1 \n\t" // bb0 * aa
  416. " vmulps %%xmm0 , %%xmm2 , %%xmm2 \n\t" // bb1 * aa
  417. " vmovss %%xmm1 , 0(%4) \n\t" // c[i] = bb0 * aa
  418. " vmovss %%xmm2 , 0(%5) \n\t" // c[i] = bb1 * aa
  419. " vmovss %%xmm1 , (%3) \n\t" // b[0] = bb0 * aa
  420. " vmovss %%xmm2 , 4(%3) \n\t" // b[1] = bb1 * aa
  421. " vzeroupper \n\t"
  422. :
  423. "+r" (n1), // 0
  424. "+a" (i), // 1
  425. "+r" (as), // 2
  426. "+r" (bs) // 3
  427. :
  428. "r" (c), // 4
  429. "r" (c1), // 5
  430. "r" (a), // 6
  431. "r" (b) // 7
  432. : "cc",
  433. "%xmm0", "%xmm1", "%xmm2", "%xmm3",
  434. "%xmm4", "%xmm5", "%xmm6", "%xmm7",
  435. "%xmm8", "%xmm9", "%xmm10", "%xmm11",
  436. "%xmm12", "%xmm13", "%xmm14", "%xmm15",
  437. "memory"
  438. );
  439. }
  440. static inline void solve(BLASLONG m, BLASLONG n, FLOAT *a, FLOAT *b, FLOAT *c, BLASLONG ldc) {
  441. FLOAT aa, bb;
  442. int i, j, k;
  443. a += (m - 1) * m;
  444. b += (m - 1) * n;
  445. for (i = m - 1; i >= 0; i--) {
  446. aa = *(a + i);
  447. for (j = 0; j < n; j ++) {
  448. bb = *(c + i + j * ldc);
  449. bb *= aa;
  450. *b = bb;
  451. *(c + i + j * ldc) = bb;
  452. b ++;
  453. for (k = 0; k < i; k ++){
  454. *(c + k + j * ldc) -= bb * *(a + k);
  455. }
  456. }
  457. a -= m;
  458. b -= 2 * n;
  459. }
  460. }
  461. #else
  462. static inline void solve(BLASLONG m, BLASLONG n, FLOAT *a, FLOAT *b, FLOAT *c, BLASLONG ldc) {
  463. FLOAT aa1, aa2;
  464. FLOAT bb1, bb2;
  465. FLOAT cc1, cc2;
  466. int i, j, k;
  467. ldc *= 2;
  468. a += (m - 1) * m * 2;
  469. b += (m - 1) * n * 2;
  470. for (i = m - 1; i >= 0; i--) {
  471. aa1 = *(a + i * 2 + 0);
  472. aa2 = *(a + i * 2 + 1);
  473. for (j = 0; j < n; j ++) {
  474. bb1 = *(c + i * 2 + 0 + j * ldc);
  475. bb2 = *(c + i * 2 + 1 + j * ldc);
  476. #ifndef CONJ
  477. cc1 = aa1 * bb1 - aa2 * bb2;
  478. cc2 = aa1 * bb2 + aa2 * bb1;
  479. #else
  480. cc1 = aa1 * bb1 + aa2 * bb2;
  481. cc2 = aa1 * bb2 - aa2 * bb1;
  482. #endif
  483. *(b + 0) = cc1;
  484. *(b + 1) = cc2;
  485. *(c + i * 2 + 0 + j * ldc) = cc1;
  486. *(c + i * 2 + 1 + j * ldc) = cc2;
  487. b += 2;
  488. for (k = 0; k < i; k ++){
  489. #ifndef CONJ
  490. *(c + k * 2 + 0 + j * ldc) -= cc1 * *(a + k * 2 + 0) - cc2 * *(a + k * 2 + 1);
  491. *(c + k * 2 + 1 + j * ldc) -= cc1 * *(a + k * 2 + 1) + cc2 * *(a + k * 2 + 0);
  492. #else
  493. *(c + k * 2 + 0 + j * ldc) -= cc1 * *(a + k * 2 + 0) + cc2 * *(a + k * 2 + 1);
  494. *(c + k * 2 + 1 + j * ldc) -= - cc1 * *(a + k * 2 + 1) + cc2 * *(a + k * 2 + 0);
  495. #endif
  496. }
  497. }
  498. a -= m * 2;
  499. b -= 4 * n;
  500. }
  501. }
  502. #endif
  503. int CNAME(BLASLONG m, BLASLONG n, BLASLONG k, FLOAT dummy1,
  504. #ifdef COMPLEX
  505. FLOAT dummy2,
  506. #endif
  507. FLOAT *a, FLOAT *b, FLOAT *c, BLASLONG ldc, BLASLONG offset){
  508. BLASLONG i, j;
  509. FLOAT *aa, *cc;
  510. BLASLONG kk;
  511. #if 0
  512. fprintf(stderr, "TRSM KERNEL LN : m = %3ld n = %3ld k = %3ld offset = %3ld\n",
  513. m, n, k, offset);
  514. #endif
  515. j = (n >> GEMM_UNROLL_N_SHIFT);
  516. while (j > 0) {
  517. kk = m + offset;
  518. if (m & (GEMM_UNROLL_M - 1)) {
  519. for (i = 1; i < GEMM_UNROLL_M; i *= 2){
  520. if (m & i) {
  521. aa = a + ((m & ~(i - 1)) - i) * k * COMPSIZE;
  522. cc = c + ((m & ~(i - 1)) - i) * COMPSIZE;
  523. if (k - kk > 0) {
  524. GEMM_KERNEL(i, GEMM_UNROLL_N, k - kk, dm1,
  525. #ifdef COMPLEX
  526. ZERO,
  527. #endif
  528. aa + i * kk * COMPSIZE,
  529. b + GEMM_UNROLL_N * kk * COMPSIZE,
  530. cc,
  531. ldc);
  532. }
  533. solve(i, GEMM_UNROLL_N,
  534. aa + (kk - i) * i * COMPSIZE,
  535. b + (kk - i) * GEMM_UNROLL_N * COMPSIZE,
  536. cc, ldc);
  537. kk -= i;
  538. }
  539. }
  540. }
  541. i = (m >> GEMM_UNROLL_M_SHIFT);
  542. if (i > 0) {
  543. aa = a + ((m & ~(GEMM_UNROLL_M - 1)) - GEMM_UNROLL_M) * k * COMPSIZE;
  544. cc = c + ((m & ~(GEMM_UNROLL_M - 1)) - GEMM_UNROLL_M) * COMPSIZE;
  545. do {
  546. strsm_LN_solve_opt(k-kk, aa + GEMM_UNROLL_M * kk * COMPSIZE, b + GEMM_UNROLL_N * kk * COMPSIZE, cc, ldc,
  547. aa + (kk - GEMM_UNROLL_M) * GEMM_UNROLL_M * COMPSIZE,b + (kk - GEMM_UNROLL_M) * GEMM_UNROLL_N * COMPSIZE);
  548. aa -= GEMM_UNROLL_M * k * COMPSIZE;
  549. cc -= GEMM_UNROLL_M * COMPSIZE;
  550. kk -= GEMM_UNROLL_M;
  551. i --;
  552. } while (i > 0);
  553. }
  554. b += GEMM_UNROLL_N * k * COMPSIZE;
  555. c += GEMM_UNROLL_N * ldc * COMPSIZE;
  556. j --;
  557. }
  558. if (n & (GEMM_UNROLL_N - 1)) {
  559. j = (GEMM_UNROLL_N >> 1);
  560. while (j > 0) {
  561. if (n & j) {
  562. kk = m + offset;
  563. if (m & (GEMM_UNROLL_M - 1)) {
  564. for (i = 1; i < GEMM_UNROLL_M; i *= 2){
  565. if (m & i) {
  566. aa = a + ((m & ~(i - 1)) - i) * k * COMPSIZE;
  567. cc = c + ((m & ~(i - 1)) - i) * COMPSIZE;
  568. if (k - kk > 0) {
  569. GEMM_KERNEL(i, j, k - kk, dm1,
  570. #ifdef COMPLEX
  571. ZERO,
  572. #endif
  573. aa + i * kk * COMPSIZE,
  574. b + j * kk * COMPSIZE,
  575. cc, ldc);
  576. }
  577. solve(i, j,
  578. aa + (kk - i) * i * COMPSIZE,
  579. b + (kk - i) * j * COMPSIZE,
  580. cc, ldc);
  581. kk -= i;
  582. }
  583. }
  584. }
  585. i = (m >> GEMM_UNROLL_M_SHIFT);
  586. if (i > 0) {
  587. aa = a + ((m & ~(GEMM_UNROLL_M - 1)) - GEMM_UNROLL_M) * k * COMPSIZE;
  588. cc = c + ((m & ~(GEMM_UNROLL_M - 1)) - GEMM_UNROLL_M) * COMPSIZE;
  589. do {
  590. if (k - kk > 0) {
  591. GEMM_KERNEL(GEMM_UNROLL_M, j, k - kk, dm1,
  592. #ifdef COMPLEX
  593. ZERO,
  594. #endif
  595. aa + GEMM_UNROLL_M * kk * COMPSIZE,
  596. b + j * kk * COMPSIZE,
  597. cc,
  598. ldc);
  599. }
  600. solve(GEMM_UNROLL_M, j,
  601. aa + (kk - GEMM_UNROLL_M) * GEMM_UNROLL_M * COMPSIZE,
  602. b + (kk - GEMM_UNROLL_M) * j * COMPSIZE,
  603. cc, ldc);
  604. aa -= GEMM_UNROLL_M * k * COMPSIZE;
  605. cc -= GEMM_UNROLL_M * COMPSIZE;
  606. kk -= GEMM_UNROLL_M;
  607. i --;
  608. } while (i > 0);
  609. }
  610. b += j * k * COMPSIZE;
  611. c += j * ldc * COMPSIZE;
  612. }
  613. j >>= 1;
  614. }
  615. }
  616. return 0;
  617. }