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_LT_bulldozer.c 34 kB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739
  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. static void strsm_LT_solve_opt(BLASLONG n, FLOAT *a, FLOAT *b, FLOAT *c, BLASLONG ldc, FLOAT *as, FLOAT *bs) __attribute__ ((noinline));
  79. static void strsm_LT_solve_opt(BLASLONG n, FLOAT *a, FLOAT *b, FLOAT *c, BLASLONG ldc, FLOAT *as, FLOAT *bs)
  80. {
  81. FLOAT *c1 = c + ldc ;
  82. BLASLONG n1 = n * 8;
  83. BLASLONG i=0;
  84. __asm__ __volatile__
  85. (
  86. " vzeroupper \n\t"
  87. " prefetcht0 (%4) \n\t"
  88. " prefetcht0 (%5) \n\t"
  89. " vxorps %%xmm8 , %%xmm8 , %%xmm8 \n\t"
  90. " vxorps %%xmm9 , %%xmm9 , %%xmm9 \n\t"
  91. " vxorps %%xmm10, %%xmm10, %%xmm10 \n\t"
  92. " vxorps %%xmm11, %%xmm11, %%xmm11 \n\t"
  93. " vxorps %%xmm12, %%xmm12, %%xmm12 \n\t"
  94. " vxorps %%xmm13, %%xmm13, %%xmm13 \n\t"
  95. " vxorps %%xmm14, %%xmm14, %%xmm14 \n\t"
  96. " vxorps %%xmm15, %%xmm15, %%xmm15 \n\t"
  97. " cmpq $0, %0 \n\t"
  98. " je 2f \n\t"
  99. " .align 16 \n\t"
  100. "1: \n\t"
  101. " vbroadcastss (%3,%1,1), %%xmm0 \n\t" // read b
  102. " vmovups (%2,%1,8), %%xmm4 \n\t"
  103. " vbroadcastss 4(%3,%1,1), %%xmm1 \n\t"
  104. " vmovups 16(%2,%1,8), %%xmm5 \n\t"
  105. " vmovups 32(%2,%1,8), %%xmm6 \n\t"
  106. " vmovups 48(%2,%1,8), %%xmm7 \n\t"
  107. " vfmaddps %%xmm8 , %%xmm0 , %%xmm4 , %%xmm8 \n\t"
  108. " vfmaddps %%xmm12, %%xmm1 , %%xmm4 , %%xmm12 \n\t"
  109. " vfmaddps %%xmm9 , %%xmm0 , %%xmm5 , %%xmm9 \n\t"
  110. " vfmaddps %%xmm13, %%xmm1 , %%xmm5 , %%xmm13 \n\t"
  111. " vfmaddps %%xmm10, %%xmm0 , %%xmm6 , %%xmm10 \n\t"
  112. " vfmaddps %%xmm14, %%xmm1 , %%xmm6 , %%xmm14 \n\t"
  113. " addq $8, %1 \n\t"
  114. " vfmaddps %%xmm11, %%xmm0 , %%xmm7 , %%xmm11 \n\t"
  115. " vfmaddps %%xmm15, %%xmm1 , %%xmm7 , %%xmm15 \n\t"
  116. " cmpq %1, %0 \n\t"
  117. " jnz 1b \n\t"
  118. "2: \n\t"
  119. " vmovups (%4) , %%xmm0 \n\t"
  120. " vmovups 16(%4) , %%xmm1 \n\t"
  121. " vmovups 32(%4) , %%xmm2 \n\t"
  122. " vmovups 48(%4) , %%xmm3 \n\t"
  123. " vmovups (%5) , %%xmm4 \n\t"
  124. " vmovups 16(%5) , %%xmm5 \n\t"
  125. " vmovups 32(%5) , %%xmm6 \n\t"
  126. " vmovups 48(%5) , %%xmm7 \n\t"
  127. " vsubps %%xmm8 , %%xmm0 , %%xmm8 \n\t"
  128. " vsubps %%xmm9 , %%xmm1 , %%xmm9 \n\t"
  129. " vsubps %%xmm10, %%xmm2 , %%xmm10 \n\t"
  130. " vsubps %%xmm11, %%xmm3 , %%xmm11 \n\t"
  131. " vsubps %%xmm12, %%xmm4 , %%xmm12 \n\t"
  132. " vsubps %%xmm13, %%xmm5 , %%xmm13 \n\t"
  133. " vsubps %%xmm14, %%xmm6 , %%xmm14 \n\t"
  134. " vsubps %%xmm15, %%xmm7 , %%xmm15 \n\t"
  135. "3: \n\t"
  136. " vbroadcastss 0(%6) , %%xmm0 \n\t" // i=0, read aa[i]
  137. " vshufps $0x00 , %%xmm8 , %%xmm8 , %%xmm1 \n\t" // extract bb0
  138. " vshufps $0x00 , %%xmm12 , %%xmm12 , %%xmm2 \n\t" // extract bb1
  139. " vmulps %%xmm0 , %%xmm1 , %%xmm1 \n\t" // bb0 * aa
  140. " vmulps %%xmm0 , %%xmm2 , %%xmm2 \n\t" // bb1 * aa
  141. " vmovss %%xmm1 , 0(%4) \n\t" // c[i] = bb0 * aa
  142. " vmovss %%xmm2 , 0(%5) \n\t" // c[i] = bb1 * aa
  143. " vmovss %%xmm1 , (%7) \n\t" // b[0] = bb0 * aa
  144. " vmovss %%xmm2 , 4(%7) \n\t" // b[1] = bb1 * aa
  145. " vmovups 0(%6) , %%xmm4 \n\t" // read a[k]
  146. " vmovups 16(%6) , %%xmm5 \n\t" // read a[k]
  147. " vmovups 32(%6) , %%xmm6 \n\t" // read a[k]
  148. " vmovups 48(%6) , %%xmm7 \n\t" // read a[k]
  149. " vfnmaddps %%xmm8 , %%xmm1 , %%xmm4 , %%xmm8 \n\t"
  150. " vfnmaddps %%xmm12 , %%xmm2 , %%xmm4 , %%xmm12 \n\t"
  151. " vfnmaddps %%xmm9 , %%xmm1 , %%xmm5 , %%xmm9 \n\t"
  152. " vfnmaddps %%xmm13 , %%xmm2 , %%xmm5 , %%xmm13 \n\t"
  153. " vfnmaddps %%xmm10 , %%xmm1 , %%xmm6 , %%xmm10 \n\t"
  154. " vfnmaddps %%xmm14 , %%xmm2 , %%xmm6 , %%xmm14 \n\t"
  155. " vfnmaddps %%xmm11 , %%xmm1 , %%xmm7 , %%xmm11 \n\t"
  156. " vfnmaddps %%xmm15 , %%xmm2 , %%xmm7 , %%xmm15 \n\t"
  157. " addq $64 , %6 \n\t" // a -= m
  158. " addq $8 , %7 \n\t" // b -= n
  159. " vbroadcastss 4(%6) , %%xmm0 \n\t" // i=1, read aa[i]
  160. " vshufps $0x55 , %%xmm8 , %%xmm8 , %%xmm1 \n\t" // extract bb0
  161. " vshufps $0x55 , %%xmm12 , %%xmm12 , %%xmm2 \n\t" // extract bb1
  162. " vmulps %%xmm0 , %%xmm1 , %%xmm1 \n\t" // bb0 * aa
  163. " vmulps %%xmm0 , %%xmm2 , %%xmm2 \n\t" // bb1 * aa
  164. " vmovss %%xmm1 , 4(%4) \n\t" // c[i] = bb0 * aa
  165. " vmovss %%xmm2 , 4(%5) \n\t" // c[i] = bb1 * aa
  166. " vmovss %%xmm1 , (%7) \n\t" // b[0] = bb0 * aa
  167. " vmovss %%xmm2 , 4(%7) \n\t" // b[1] = bb1 * aa
  168. " vmovups 0(%6) , %%xmm4 \n\t" // read a[k]
  169. " vmovups 16(%6) , %%xmm5 \n\t" // read a[k]
  170. " vmovups 32(%6) , %%xmm6 \n\t" // read a[k]
  171. " vmovups 48(%6) , %%xmm7 \n\t" // read a[k]
  172. " vfnmaddps %%xmm8 , %%xmm1 , %%xmm4 , %%xmm8 \n\t"
  173. " vfnmaddps %%xmm12 , %%xmm2 , %%xmm4 , %%xmm12 \n\t"
  174. " vfnmaddps %%xmm9 , %%xmm1 , %%xmm5 , %%xmm9 \n\t"
  175. " vfnmaddps %%xmm13 , %%xmm2 , %%xmm5 , %%xmm13 \n\t"
  176. " vfnmaddps %%xmm10 , %%xmm1 , %%xmm6 , %%xmm10 \n\t"
  177. " vfnmaddps %%xmm14 , %%xmm2 , %%xmm6 , %%xmm14 \n\t"
  178. " vfnmaddps %%xmm11 , %%xmm1 , %%xmm7 , %%xmm11 \n\t"
  179. " vfnmaddps %%xmm15 , %%xmm2 , %%xmm7 , %%xmm15 \n\t"
  180. " addq $64 , %6 \n\t" // a -= m
  181. " addq $8 , %7 \n\t" // b -= n
  182. " vbroadcastss 8(%6) , %%xmm0 \n\t" // i=2, read aa[i]
  183. " vshufps $0xaa , %%xmm8 , %%xmm8 , %%xmm1 \n\t" // extract bb0
  184. " vshufps $0xaa , %%xmm12 , %%xmm12 , %%xmm2 \n\t" // extract bb1
  185. " vmulps %%xmm0 , %%xmm1 , %%xmm1 \n\t" // bb0 * aa
  186. " vmulps %%xmm0 , %%xmm2 , %%xmm2 \n\t" // bb1 * aa
  187. " vmovss %%xmm1 , 8(%4) \n\t" // c[i] = bb0 * aa
  188. " vmovss %%xmm2 , 8(%5) \n\t" // c[i] = bb1 * aa
  189. " vmovss %%xmm1 , (%7) \n\t" // b[0] = bb0 * aa
  190. " vmovss %%xmm2 , 4(%7) \n\t" // b[1] = bb1 * aa
  191. " vmovups 0(%6) , %%xmm4 \n\t" // read a[k]
  192. " vmovups 16(%6) , %%xmm5 \n\t" // read a[k]
  193. " vmovups 32(%6) , %%xmm6 \n\t" // read a[k]
  194. " vmovups 48(%6) , %%xmm7 \n\t" // read a[k]
  195. " vfnmaddps %%xmm8 , %%xmm1 , %%xmm4 , %%xmm8 \n\t"
  196. " vfnmaddps %%xmm12 , %%xmm2 , %%xmm4 , %%xmm12 \n\t"
  197. " vfnmaddps %%xmm9 , %%xmm1 , %%xmm5 , %%xmm9 \n\t"
  198. " vfnmaddps %%xmm13 , %%xmm2 , %%xmm5 , %%xmm13 \n\t"
  199. " vfnmaddps %%xmm10 , %%xmm1 , %%xmm6 , %%xmm10 \n\t"
  200. " vfnmaddps %%xmm14 , %%xmm2 , %%xmm6 , %%xmm14 \n\t"
  201. " vfnmaddps %%xmm11 , %%xmm1 , %%xmm7 , %%xmm11 \n\t"
  202. " vfnmaddps %%xmm15 , %%xmm2 , %%xmm7 , %%xmm15 \n\t"
  203. " addq $64 , %6 \n\t" // a -= m
  204. " addq $8 , %7 \n\t" // b -= n
  205. " vbroadcastss 12(%6) , %%xmm0 \n\t" // i=3, read aa[i]
  206. " vshufps $0xff , %%xmm8 , %%xmm8 , %%xmm1 \n\t" // extract bb0
  207. " vshufps $0xff , %%xmm12 , %%xmm12 , %%xmm2 \n\t" // extract bb1
  208. " vmulps %%xmm0 , %%xmm1 , %%xmm1 \n\t" // bb0 * aa
  209. " vmulps %%xmm0 , %%xmm2 , %%xmm2 \n\t" // bb1 * aa
  210. " vmovss %%xmm1 , 12(%4) \n\t" // c[i] = bb0 * aa
  211. " vmovss %%xmm2 , 12(%5) \n\t" // c[i] = bb1 * aa
  212. " vmovss %%xmm1 , (%7) \n\t" // b[0] = bb0 * aa
  213. " vmovss %%xmm2 , 4(%7) \n\t" // b[1] = bb1 * aa
  214. " vmovups 16(%6) , %%xmm5 \n\t" // read a[k]
  215. " vmovups 32(%6) , %%xmm6 \n\t" // read a[k]
  216. " vmovups 48(%6) , %%xmm7 \n\t" // read a[k]
  217. " vfnmaddps %%xmm9 , %%xmm1 , %%xmm5 , %%xmm9 \n\t"
  218. " vfnmaddps %%xmm13 , %%xmm2 , %%xmm5 , %%xmm13 \n\t"
  219. " vfnmaddps %%xmm10 , %%xmm1 , %%xmm6 , %%xmm10 \n\t"
  220. " vfnmaddps %%xmm14 , %%xmm2 , %%xmm6 , %%xmm14 \n\t"
  221. " vfnmaddps %%xmm11 , %%xmm1 , %%xmm7 , %%xmm11 \n\t"
  222. " vfnmaddps %%xmm15 , %%xmm2 , %%xmm7 , %%xmm15 \n\t"
  223. " addq $64 , %6 \n\t" // a -= m
  224. " addq $8 , %7 \n\t" // b -= n
  225. " vbroadcastss 16(%6) , %%xmm0 \n\t" // i=4, read aa[i]
  226. " vshufps $0x00 , %%xmm9 , %%xmm9 , %%xmm1 \n\t" // extract bb0
  227. " vshufps $0x00 , %%xmm13 , %%xmm13 , %%xmm2 \n\t" // extract bb1
  228. " vmulps %%xmm0 , %%xmm1 , %%xmm1 \n\t" // bb0 * aa
  229. " vmulps %%xmm0 , %%xmm2 , %%xmm2 \n\t" // bb1 * aa
  230. " vmovss %%xmm1 , 16(%4) \n\t" // c[i] = bb0 * aa
  231. " vmovss %%xmm2 , 16(%5) \n\t" // c[i] = bb1 * aa
  232. " vmovss %%xmm1 , (%7) \n\t" // b[0] = bb0 * aa
  233. " vmovss %%xmm2 , 4(%7) \n\t" // b[1] = bb1 * aa
  234. " vmovups 16(%6) , %%xmm5 \n\t" // read a[k]
  235. " vmovups 32(%6) , %%xmm6 \n\t" // read a[k]
  236. " vmovups 48(%6) , %%xmm7 \n\t" // read a[k]
  237. " vfnmaddps %%xmm9 , %%xmm1 , %%xmm5 , %%xmm9 \n\t"
  238. " vfnmaddps %%xmm13 , %%xmm2 , %%xmm5 , %%xmm13 \n\t"
  239. " vfnmaddps %%xmm10 , %%xmm1 , %%xmm6 , %%xmm10 \n\t"
  240. " vfnmaddps %%xmm14 , %%xmm2 , %%xmm6 , %%xmm14 \n\t"
  241. " vfnmaddps %%xmm11 , %%xmm1 , %%xmm7 , %%xmm11 \n\t"
  242. " vfnmaddps %%xmm15 , %%xmm2 , %%xmm7 , %%xmm15 \n\t"
  243. " addq $64 , %6 \n\t" // a -= m
  244. " addq $8 , %7 \n\t" // b -= n
  245. " vbroadcastss 20(%6) , %%xmm0 \n\t" // i=5, read aa[i]
  246. " vshufps $0x55 , %%xmm9 , %%xmm9 , %%xmm1 \n\t" // extract bb0
  247. " vshufps $0x55 , %%xmm13 , %%xmm13 , %%xmm2 \n\t" // extract bb1
  248. " vmulps %%xmm0 , %%xmm1 , %%xmm1 \n\t" // bb0 * aa
  249. " vmulps %%xmm0 , %%xmm2 , %%xmm2 \n\t" // bb1 * aa
  250. " vmovss %%xmm1 , 20(%4) \n\t" // c[i] = bb0 * aa
  251. " vmovss %%xmm2 , 20(%5) \n\t" // c[i] = bb1 * aa
  252. " vmovss %%xmm1 , (%7) \n\t" // b[0] = bb0 * aa
  253. " vmovss %%xmm2 , 4(%7) \n\t" // b[1] = bb1 * aa
  254. " vmovups 16(%6) , %%xmm5 \n\t" // read a[k]
  255. " vmovups 32(%6) , %%xmm6 \n\t" // read a[k]
  256. " vmovups 48(%6) , %%xmm7 \n\t" // read a[k]
  257. " vfnmaddps %%xmm9 , %%xmm1 , %%xmm5 , %%xmm9 \n\t"
  258. " vfnmaddps %%xmm13 , %%xmm2 , %%xmm5 , %%xmm13 \n\t"
  259. " vfnmaddps %%xmm10 , %%xmm1 , %%xmm6 , %%xmm10 \n\t"
  260. " vfnmaddps %%xmm14 , %%xmm2 , %%xmm6 , %%xmm14 \n\t"
  261. " vfnmaddps %%xmm11 , %%xmm1 , %%xmm7 , %%xmm11 \n\t"
  262. " vfnmaddps %%xmm15 , %%xmm2 , %%xmm7 , %%xmm15 \n\t"
  263. " addq $64 , %6 \n\t" // a -= m
  264. " addq $8 , %7 \n\t" // b -= n
  265. " vbroadcastss 24(%6) , %%xmm0 \n\t" // i=6, read aa[i]
  266. " vshufps $0xaa , %%xmm9 , %%xmm9 , %%xmm1 \n\t" // extract bb0
  267. " vshufps $0xaa , %%xmm13 , %%xmm13 , %%xmm2 \n\t" // extract bb1
  268. " vmulps %%xmm0 , %%xmm1 , %%xmm1 \n\t" // bb0 * aa
  269. " vmulps %%xmm0 , %%xmm2 , %%xmm2 \n\t" // bb1 * aa
  270. " vmovss %%xmm1 , 24(%4) \n\t" // c[i] = bb0 * aa
  271. " vmovss %%xmm2 , 24(%5) \n\t" // c[i] = bb1 * aa
  272. " vmovss %%xmm1 , (%7) \n\t" // b[0] = bb0 * aa
  273. " vmovss %%xmm2 , 4(%7) \n\t" // b[1] = bb1 * aa
  274. " vmovups 16(%6) , %%xmm5 \n\t" // read a[k]
  275. " vmovups 32(%6) , %%xmm6 \n\t" // read a[k]
  276. " vmovups 48(%6) , %%xmm7 \n\t" // read a[k]
  277. " vfnmaddps %%xmm9 , %%xmm1 , %%xmm5 , %%xmm9 \n\t"
  278. " vfnmaddps %%xmm13 , %%xmm2 , %%xmm5 , %%xmm13 \n\t"
  279. " vfnmaddps %%xmm10 , %%xmm1 , %%xmm6 , %%xmm10 \n\t"
  280. " vfnmaddps %%xmm14 , %%xmm2 , %%xmm6 , %%xmm14 \n\t"
  281. " vfnmaddps %%xmm11 , %%xmm1 , %%xmm7 , %%xmm11 \n\t"
  282. " vfnmaddps %%xmm15 , %%xmm2 , %%xmm7 , %%xmm15 \n\t"
  283. " addq $64 , %6 \n\t" // a -= m
  284. " addq $8 , %7 \n\t" // b -= n
  285. " vbroadcastss 28(%6) , %%xmm0 \n\t" // i=7, read aa[i]
  286. " vshufps $0xff , %%xmm9 , %%xmm9 , %%xmm1 \n\t" // extract bb0
  287. " vshufps $0xff , %%xmm13 , %%xmm13 , %%xmm2 \n\t" // extract bb1
  288. " vmulps %%xmm0 , %%xmm1 , %%xmm1 \n\t" // bb0 * aa
  289. " vmulps %%xmm0 , %%xmm2 , %%xmm2 \n\t" // bb1 * aa
  290. " vmovss %%xmm1 , 28(%4) \n\t" // c[i] = bb0 * aa
  291. " vmovss %%xmm2 , 28(%5) \n\t" // c[i] = bb1 * aa
  292. " vmovss %%xmm1 , (%7) \n\t" // b[0] = bb0 * aa
  293. " vmovss %%xmm2 , 4(%7) \n\t" // b[1] = bb1 * aa
  294. " vmovups 32(%6) , %%xmm6 \n\t" // read a[k]
  295. " vmovups 48(%6) , %%xmm7 \n\t" // read a[k]
  296. " vfnmaddps %%xmm10 , %%xmm1 , %%xmm6 , %%xmm10 \n\t"
  297. " vfnmaddps %%xmm14 , %%xmm2 , %%xmm6 , %%xmm14 \n\t"
  298. " vfnmaddps %%xmm11 , %%xmm1 , %%xmm7 , %%xmm11 \n\t"
  299. " vfnmaddps %%xmm15 , %%xmm2 , %%xmm7 , %%xmm15 \n\t"
  300. " addq $64 , %6 \n\t" // a -= m
  301. " addq $8 , %7 \n\t" // b -= n
  302. " vbroadcastss 32(%6) , %%xmm0 \n\t" // i=8, read aa[i]
  303. " vshufps $0x00 , %%xmm10 , %%xmm10 , %%xmm1 \n\t" // extract bb0
  304. " vshufps $0x00 , %%xmm14 , %%xmm14 , %%xmm2 \n\t" // extract bb1
  305. " vmulps %%xmm0 , %%xmm1 , %%xmm1 \n\t" // bb0 * aa
  306. " vmulps %%xmm0 , %%xmm2 , %%xmm2 \n\t" // bb1 * aa
  307. " vmovss %%xmm1 , 32(%4) \n\t" // c[i] = bb0 * aa
  308. " vmovss %%xmm2 , 32(%5) \n\t" // c[i] = bb1 * aa
  309. " vmovss %%xmm1 , (%7) \n\t" // b[0] = bb0 * aa
  310. " vmovss %%xmm2 , 4(%7) \n\t" // b[1] = bb1 * aa
  311. " vmovups 32(%6) , %%xmm6 \n\t" // read a[k]
  312. " vmovups 48(%6) , %%xmm7 \n\t" // read a[k]
  313. " vfnmaddps %%xmm10 , %%xmm1 , %%xmm6 , %%xmm10 \n\t"
  314. " vfnmaddps %%xmm14 , %%xmm2 , %%xmm6 , %%xmm14 \n\t"
  315. " vfnmaddps %%xmm11 , %%xmm1 , %%xmm7 , %%xmm11 \n\t"
  316. " vfnmaddps %%xmm15 , %%xmm2 , %%xmm7 , %%xmm15 \n\t"
  317. " addq $64 , %6 \n\t" // a -= m
  318. " addq $8 , %7 \n\t" // b -= n
  319. " vbroadcastss 36(%6) , %%xmm0 \n\t" // i=9, read aa[i]
  320. " vshufps $0x55 , %%xmm10 , %%xmm10 , %%xmm1 \n\t" // extract bb0
  321. " vshufps $0x55 , %%xmm14 , %%xmm14 , %%xmm2 \n\t" // extract bb1
  322. " vmulps %%xmm0 , %%xmm1 , %%xmm1 \n\t" // bb0 * aa
  323. " vmulps %%xmm0 , %%xmm2 , %%xmm2 \n\t" // bb1 * aa
  324. " vmovss %%xmm1 , 36(%4) \n\t" // c[i] = bb0 * aa
  325. " vmovss %%xmm2 , 36(%5) \n\t" // c[i] = bb1 * aa
  326. " vmovss %%xmm1 , (%7) \n\t" // b[0] = bb0 * aa
  327. " vmovss %%xmm2 , 4(%7) \n\t" // b[1] = bb1 * aa
  328. " vmovups 32(%6) , %%xmm6 \n\t" // read a[k]
  329. " vmovups 48(%6) , %%xmm7 \n\t" // read a[k]
  330. " vfnmaddps %%xmm10 , %%xmm1 , %%xmm6 , %%xmm10 \n\t"
  331. " vfnmaddps %%xmm14 , %%xmm2 , %%xmm6 , %%xmm14 \n\t"
  332. " vfnmaddps %%xmm11 , %%xmm1 , %%xmm7 , %%xmm11 \n\t"
  333. " vfnmaddps %%xmm15 , %%xmm2 , %%xmm7 , %%xmm15 \n\t"
  334. " addq $64 , %6 \n\t" // a -= m
  335. " addq $8 , %7 \n\t" // b -= n
  336. " vbroadcastss 40(%6) , %%xmm0 \n\t" // i=10, read aa[i]
  337. " vshufps $0xaa , %%xmm10 , %%xmm10 , %%xmm1 \n\t" // extract bb0
  338. " vshufps $0xaa , %%xmm14 , %%xmm14 , %%xmm2 \n\t" // extract bb1
  339. " vmulps %%xmm0 , %%xmm1 , %%xmm1 \n\t" // bb0 * aa
  340. " vmulps %%xmm0 , %%xmm2 , %%xmm2 \n\t" // bb1 * aa
  341. " vmovss %%xmm1 , 40(%4) \n\t" // c[i] = bb0 * aa
  342. " vmovss %%xmm2 , 40(%5) \n\t" // c[i] = bb1 * aa
  343. " vmovss %%xmm1 , (%7) \n\t" // b[0] = bb0 * aa
  344. " vmovss %%xmm2 , 4(%7) \n\t" // b[1] = bb1 * aa
  345. " vmovups 32(%6) , %%xmm6 \n\t" // read a[k]
  346. " vmovups 48(%6) , %%xmm7 \n\t" // read a[k]
  347. " vfnmaddps %%xmm10 , %%xmm1 , %%xmm6 , %%xmm10 \n\t"
  348. " vfnmaddps %%xmm14 , %%xmm2 , %%xmm6 , %%xmm14 \n\t"
  349. " vfnmaddps %%xmm11 , %%xmm1 , %%xmm7 , %%xmm11 \n\t"
  350. " vfnmaddps %%xmm15 , %%xmm2 , %%xmm7 , %%xmm15 \n\t"
  351. " addq $64 , %6 \n\t" // a -= m
  352. " addq $8 , %7 \n\t" // b -= n
  353. " vbroadcastss 44(%6) , %%xmm0 \n\t" // i=11, read aa[i]
  354. " vshufps $0xff , %%xmm10 , %%xmm10 , %%xmm1 \n\t" // extract bb0
  355. " vshufps $0xff , %%xmm14 , %%xmm14 , %%xmm2 \n\t" // extract bb1
  356. " vmulps %%xmm0 , %%xmm1 , %%xmm1 \n\t" // bb0 * aa
  357. " vmulps %%xmm0 , %%xmm2 , %%xmm2 \n\t" // bb1 * aa
  358. " vmovss %%xmm1 , 44(%4) \n\t" // c[i] = bb0 * aa
  359. " vmovss %%xmm2 , 44(%5) \n\t" // c[i] = bb1 * aa
  360. " vmovss %%xmm1 , (%7) \n\t" // b[0] = bb0 * aa
  361. " vmovss %%xmm2 , 4(%7) \n\t" // b[1] = bb1 * aa
  362. " vmovups 48(%6) , %%xmm7 \n\t" // read a[k]
  363. " vfnmaddps %%xmm11 , %%xmm1 , %%xmm7 , %%xmm11 \n\t"
  364. " vfnmaddps %%xmm15 , %%xmm2 , %%xmm7 , %%xmm15 \n\t"
  365. " addq $64 , %6 \n\t" // a -= m
  366. " addq $8 , %7 \n\t" // b -= n
  367. " vbroadcastss 48(%6) , %%xmm0 \n\t" // i=12, read aa[i]
  368. " vshufps $0x00 , %%xmm11 , %%xmm11 , %%xmm1 \n\t" // extract bb0
  369. " vshufps $0x00 , %%xmm15 , %%xmm15 , %%xmm2 \n\t" // extract bb1
  370. " vmulps %%xmm0 , %%xmm1 , %%xmm1 \n\t" // bb0 * aa
  371. " vmulps %%xmm0 , %%xmm2 , %%xmm2 \n\t" // bb1 * aa
  372. " vmovss %%xmm1 , 48(%4) \n\t" // c[i] = bb0 * aa
  373. " vmovss %%xmm2 , 48(%5) \n\t" // c[i] = bb1 * aa
  374. " vmovss %%xmm1 , (%7) \n\t" // b[0] = bb0 * aa
  375. " vmovss %%xmm2 , 4(%7) \n\t" // b[1] = bb1 * aa
  376. " vmovups 48(%6) , %%xmm7 \n\t" // read a[k]
  377. " vfnmaddps %%xmm11 , %%xmm1 , %%xmm7 , %%xmm11 \n\t"
  378. " vfnmaddps %%xmm15 , %%xmm2 , %%xmm7 , %%xmm15 \n\t"
  379. " addq $64 , %6 \n\t" // a -= m
  380. " addq $8 , %7 \n\t" // b -= n
  381. " vbroadcastss 52(%6) , %%xmm0 \n\t" // i=13, read aa[i]
  382. " vshufps $0x55 , %%xmm11 , %%xmm11 , %%xmm1 \n\t" // extract bb0
  383. " vshufps $0x55 , %%xmm15 , %%xmm15 , %%xmm2 \n\t" // extract bb1
  384. " vmulps %%xmm0 , %%xmm1 , %%xmm1 \n\t" // bb0 * aa
  385. " vmulps %%xmm0 , %%xmm2 , %%xmm2 \n\t" // bb1 * aa
  386. " vmovss %%xmm1 , 52(%4) \n\t" // c[i] = bb0 * aa
  387. " vmovss %%xmm2 , 52(%5) \n\t" // c[i] = bb1 * aa
  388. " vmovss %%xmm1 , (%7) \n\t" // b[0] = bb0 * aa
  389. " vmovss %%xmm2 , 4(%7) \n\t" // b[1] = bb1 * aa
  390. " vmovups 48(%6) , %%xmm7 \n\t" // read a[k]
  391. " vfnmaddps %%xmm11 , %%xmm1 , %%xmm7 , %%xmm11 \n\t"
  392. " vfnmaddps %%xmm15 , %%xmm2 , %%xmm7 , %%xmm15 \n\t"
  393. " addq $64 , %6 \n\t" // a -= m
  394. " addq $8 , %7 \n\t" // b -= n
  395. " vbroadcastss 56(%6) , %%xmm0 \n\t" // i=14, read aa[i]
  396. " vshufps $0xaa , %%xmm11 , %%xmm11 , %%xmm1 \n\t" // extract bb0
  397. " vshufps $0xaa , %%xmm15 , %%xmm15 , %%xmm2 \n\t" // extract bb1
  398. " vmulps %%xmm0 , %%xmm1 , %%xmm1 \n\t" // bb0 * aa
  399. " vmulps %%xmm0 , %%xmm2 , %%xmm2 \n\t" // bb1 * aa
  400. " vmovss %%xmm1 , 56(%4) \n\t" // c[i] = bb0 * aa
  401. " vmovss %%xmm2 , 56(%5) \n\t" // c[i] = bb1 * aa
  402. " vmovss %%xmm1 , (%7) \n\t" // b[0] = bb0 * aa
  403. " vmovss %%xmm2 , 4(%7) \n\t" // b[1] = bb1 * aa
  404. " vmovups 48(%6) , %%xmm7 \n\t" // read a[k]
  405. " vfnmaddps %%xmm11 , %%xmm1 , %%xmm7 , %%xmm11 \n\t"
  406. " vfnmaddps %%xmm15 , %%xmm2 , %%xmm7 , %%xmm15 \n\t"
  407. " addq $64 , %6 \n\t" // a -= m
  408. " addq $8 , %7 \n\t" // b -= n
  409. " vbroadcastss 60(%6) , %%xmm0 \n\t" // i=15, read aa[i]
  410. " vshufps $0xff , %%xmm11 , %%xmm11 , %%xmm1 \n\t" // extract bb0
  411. " vshufps $0xff , %%xmm15 , %%xmm15 , %%xmm2 \n\t" // extract bb1
  412. " vmulps %%xmm0 , %%xmm1 , %%xmm1 \n\t" // bb0 * aa
  413. " vmulps %%xmm0 , %%xmm2 , %%xmm2 \n\t" // bb1 * aa
  414. " vmovss %%xmm1 , 60(%4) \n\t" // c[i] = bb0 * aa
  415. " vmovss %%xmm2 , 60(%5) \n\t" // c[i] = bb1 * aa
  416. " vmovss %%xmm1 , (%7) \n\t" // b[0] = bb0 * aa
  417. " vmovss %%xmm2 , 4(%7) \n\t" // b[1] = bb1 * aa
  418. " vzeroupper \n\t"
  419. :
  420. :
  421. "r" (n1), // 0
  422. "a" (i), // 1
  423. "r" (a), // 2
  424. "r" (b), // 3
  425. "r" (c), // 4
  426. "r" (c1), // 5
  427. "r" (as), // 6
  428. "r" (bs) // 7
  429. : "cc",
  430. "%xmm0", "%xmm1", "%xmm2", "%xmm3",
  431. "%xmm4", "%xmm5", "%xmm6", "%xmm7",
  432. "%xmm8", "%xmm9", "%xmm10", "%xmm11",
  433. "%xmm12", "%xmm13", "%xmm14", "%xmm15",
  434. "memory"
  435. );
  436. }
  437. #ifndef COMPLEX
  438. static inline void solve(BLASLONG m, BLASLONG n, FLOAT *a, FLOAT *b, FLOAT *c, BLASLONG ldc) {
  439. FLOAT aa, bb;
  440. int i, j, k;
  441. for (i = 0; i < m; i++) {
  442. aa = *(a + i);
  443. for (j = 0; j < n; j ++) {
  444. bb = *(c + i + j * ldc);
  445. bb *= aa;
  446. *b = bb;
  447. *(c + i + j * ldc) = bb;
  448. b ++;
  449. for (k = i + 1; k < m; k ++){
  450. *(c + k + j * ldc) -= bb * *(a + k);
  451. }
  452. }
  453. a += m;
  454. }
  455. }
  456. #else
  457. static inline void solve(BLASLONG m, BLASLONG n, FLOAT *a, FLOAT *b, FLOAT *c, BLASLONG ldc) {
  458. FLOAT aa1, aa2;
  459. FLOAT bb1, bb2;
  460. FLOAT cc1, cc2;
  461. int i, j, k;
  462. ldc *= 2;
  463. for (i = 0; i < m; i++) {
  464. aa1 = *(a + i * 2 + 0);
  465. aa2 = *(a + i * 2 + 1);
  466. for (j = 0; j < n; j ++) {
  467. bb1 = *(c + i * 2 + 0 + j * ldc);
  468. bb2 = *(c + i * 2 + 1 + j * ldc);
  469. #ifndef CONJ
  470. cc1 = aa1 * bb1 - aa2 * bb2;
  471. cc2 = aa1 * bb2 + aa2 * bb1;
  472. #else
  473. cc1 = aa1 * bb1 + aa2 * bb2;
  474. cc2 = aa1 * bb2 - aa2 * bb1;
  475. #endif
  476. *(b + 0) = cc1;
  477. *(b + 1) = cc2;
  478. *(c + i * 2 + 0 + j * ldc) = cc1;
  479. *(c + i * 2 + 1 + j * ldc) = cc2;
  480. b += 2;
  481. for (k = i + 1; k < m; k ++){
  482. #ifndef CONJ
  483. *(c + k * 2 + 0 + j * ldc) -= cc1 * *(a + k * 2 + 0) - cc2 * *(a + k * 2 + 1);
  484. *(c + k * 2 + 1 + j * ldc) -= cc1 * *(a + k * 2 + 1) + cc2 * *(a + k * 2 + 0);
  485. #else
  486. *(c + k * 2 + 0 + j * ldc) -= cc1 * *(a + k * 2 + 0) + cc2 * *(a + k * 2 + 1);
  487. *(c + k * 2 + 1 + j * ldc) -= -cc1 * *(a + k * 2 + 1) + cc2 * *(a + k * 2 + 0);
  488. #endif
  489. }
  490. }
  491. a += m * 2;
  492. }
  493. }
  494. #endif
  495. int CNAME(BLASLONG m, BLASLONG n, BLASLONG k, FLOAT dummy1,
  496. #ifdef COMPLEX
  497. FLOAT dummy2,
  498. #endif
  499. FLOAT *a, FLOAT *b, FLOAT *c, BLASLONG ldc, BLASLONG offset){
  500. FLOAT *aa, *cc;
  501. BLASLONG kk;
  502. BLASLONG i, j, jj;
  503. #if 0
  504. fprintf(stderr, "TRSM KERNEL LT : m = %3ld n = %3ld k = %3ld offset = %3ld\n",
  505. m, n, k, offset);
  506. #endif
  507. jj = 0;
  508. j = (n >> GEMM_UNROLL_N_SHIFT);
  509. while (j > 0) {
  510. kk = offset;
  511. aa = a;
  512. cc = c;
  513. i = (m >> GEMM_UNROLL_M_SHIFT);
  514. while (i > 0) {
  515. strsm_LT_solve_opt(kk , aa , b , cc, ldc, aa + kk * GEMM_UNROLL_M * COMPSIZE, b + kk * GEMM_UNROLL_N * COMPSIZE);
  516. aa += GEMM_UNROLL_M * k * COMPSIZE;
  517. cc += GEMM_UNROLL_M * COMPSIZE;
  518. kk += GEMM_UNROLL_M;
  519. i --;
  520. }
  521. if (m & (GEMM_UNROLL_M - 1)) {
  522. i = (GEMM_UNROLL_M >> 1);
  523. while (i > 0) {
  524. if (m & i) {
  525. if (kk > 0) {
  526. GEMM_KERNEL(i, GEMM_UNROLL_N, kk, dm1,
  527. #ifdef COMPLEX
  528. ZERO,
  529. #endif
  530. aa, b, cc, ldc);
  531. }
  532. solve(i, GEMM_UNROLL_N,
  533. aa + kk * i * COMPSIZE,
  534. b + kk * GEMM_UNROLL_N * COMPSIZE,
  535. cc, ldc);
  536. aa += i * k * COMPSIZE;
  537. cc += i * COMPSIZE;
  538. kk += i;
  539. }
  540. i >>= 1;
  541. }
  542. }
  543. b += GEMM_UNROLL_N * k * COMPSIZE;
  544. c += GEMM_UNROLL_N * ldc * COMPSIZE;
  545. j --;
  546. jj += GEMM_UNROLL_M;
  547. }
  548. if (n & (GEMM_UNROLL_N - 1)) {
  549. j = (GEMM_UNROLL_N >> 1);
  550. while (j > 0) {
  551. if (n & j) {
  552. kk = offset;
  553. aa = a;
  554. cc = c;
  555. i = (m >> GEMM_UNROLL_M_SHIFT);
  556. while (i > 0) {
  557. if (kk > 0) {
  558. GEMM_KERNEL(GEMM_UNROLL_M, j, kk, dm1,
  559. #ifdef COMPLEX
  560. ZERO,
  561. #endif
  562. aa,
  563. b,
  564. cc,
  565. ldc);
  566. }
  567. solve(GEMM_UNROLL_M, j,
  568. aa + kk * GEMM_UNROLL_M * COMPSIZE,
  569. b + kk * j * COMPSIZE, cc, ldc);
  570. aa += GEMM_UNROLL_M * k * COMPSIZE;
  571. cc += GEMM_UNROLL_M * COMPSIZE;
  572. kk += GEMM_UNROLL_M;
  573. i --;
  574. }
  575. if (m & (GEMM_UNROLL_M - 1)) {
  576. i = (GEMM_UNROLL_M >> 1);
  577. while (i > 0) {
  578. if (m & i) {
  579. if (kk > 0) {
  580. GEMM_KERNEL(i, j, kk, dm1,
  581. #ifdef COMPLEX
  582. ZERO,
  583. #endif
  584. aa,
  585. b,
  586. cc,
  587. ldc);
  588. }
  589. solve(i, j,
  590. aa + kk * i * COMPSIZE,
  591. b + kk * j * COMPSIZE, cc, ldc);
  592. aa += i * k * COMPSIZE;
  593. cc += i * COMPSIZE;
  594. kk += i;
  595. }
  596. i >>= 1;
  597. }
  598. }
  599. b += j * k * COMPSIZE;
  600. c += j * ldc * COMPSIZE;
  601. }
  602. j >>= 1;
  603. }
  604. }
  605. return 0;
  606. }