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.

zgemv_n_msa.c 30 kB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682
  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. #undef OP0
  30. #undef OP1
  31. #undef OP2
  32. #undef OP3
  33. #undef OP4
  34. #if !defined(XCONJ)
  35. #define OP3 -=
  36. #define OP4 +=
  37. #else
  38. #define OP3 +=
  39. #define OP4 -=
  40. #endif
  41. #if !defined(CONJ)
  42. #if !defined(XCONJ)
  43. #define OP0 -=
  44. #define OP1 +=
  45. #define OP2 +=
  46. #else
  47. #define OP0 +=
  48. #define OP1 +=
  49. #define OP2 -=
  50. #endif
  51. #else
  52. #if !defined(XCONJ)
  53. #define OP0 +=
  54. #define OP1 -=
  55. #define OP2 +=
  56. #else
  57. #define OP0 -=
  58. #define OP1 -=
  59. #define OP2 -=
  60. #endif
  61. #endif
  62. #define ZGEMV_N_4x4() \
  63. LD_DP4(pa0 + k, 2, t0, t1, t2, t3); \
  64. LD_DP4(pa1 + k, 2, t4, t5, t6, t7); \
  65. LD_DP4(pa2 + k, 2, t8, t9, t10, t11); \
  66. LD_DP4(pa3 + k, 2, t12, t13, t14, t15); \
  67. \
  68. PCKEVOD_D2_DP(t1, t0, src0r, src0i); \
  69. PCKEVOD_D2_DP(t3, t2, src1r, src1i); \
  70. PCKEVOD_D2_DP(t5, t4, src2r, src2i); \
  71. PCKEVOD_D2_DP(t7, t6, src3r, src3i); \
  72. PCKEVOD_D2_DP(t9, t8, src4r, src4i); \
  73. PCKEVOD_D2_DP(t11, t10, src5r, src5i); \
  74. PCKEVOD_D2_DP(t13, t12, src6r, src6i); \
  75. PCKEVOD_D2_DP(t15, t14, src7r, src7i); \
  76. \
  77. y0r += tp0r * src0r; \
  78. y1r += tp0r * src1r; \
  79. y0r += tp1r * src2r; \
  80. y1r += tp1r * src3r; \
  81. y0r += tp2r * src4r; \
  82. y1r += tp2r * src5r; \
  83. y0r += tp3r * src6r; \
  84. y1r += tp3r * src7r; \
  85. \
  86. y0r OP0 tp0i * src0i; \
  87. y1r OP0 tp0i * src1i; \
  88. y0r OP0 tp1i * src2i; \
  89. y1r OP0 tp1i * src3i; \
  90. y0r OP0 tp2i * src4i; \
  91. y1r OP0 tp2i * src5i; \
  92. y0r OP0 tp3i * src6i; \
  93. y1r OP0 tp3i * src7i; \
  94. \
  95. y0i OP1 tp0r * src0i; \
  96. y1i OP1 tp0r * src1i; \
  97. y0i OP1 tp1r * src2i; \
  98. y1i OP1 tp1r * src3i; \
  99. y0i OP1 tp2r * src4i; \
  100. y1i OP1 tp2r * src5i; \
  101. y0i OP1 tp3r * src6i; \
  102. y1i OP1 tp3r * src7i; \
  103. \
  104. y0i OP2 tp0i * src0r; \
  105. y1i OP2 tp0i * src1r; \
  106. y0i OP2 tp1i * src2r; \
  107. y1i OP2 tp1i * src3r; \
  108. y0i OP2 tp2i * src4r; \
  109. y1i OP2 tp2i * src5r; \
  110. y0i OP2 tp3i * src6r; \
  111. y1i OP2 tp3i * src7r; \
  112. #define ZGEMV_N_2x4() \
  113. LD_DP2(pa0 + k, 2, t0, t1); \
  114. LD_DP2(pa1 + k, 2, t4, t5); \
  115. LD_DP2(pa2 + k, 2, t8, t9); \
  116. LD_DP2(pa3 + k, 2, t12, t13); \
  117. \
  118. PCKEVOD_D2_DP(t1, t0, src0r, src0i); \
  119. PCKEVOD_D2_DP(t5, t4, src2r, src2i); \
  120. PCKEVOD_D2_DP(t9, t8, src4r, src4i); \
  121. PCKEVOD_D2_DP(t13, t12, src6r, src6i); \
  122. \
  123. y0r += tp0r * src0r; \
  124. y0r += tp1r * src2r; \
  125. y0r += tp2r * src4r; \
  126. y0r += tp3r * src6r; \
  127. \
  128. y0r OP0 tp0i * src0i; \
  129. y0r OP0 tp1i * src2i; \
  130. y0r OP0 tp2i * src4i; \
  131. y0r OP0 tp3i * src6i; \
  132. \
  133. y0i OP1 tp0r * src0i; \
  134. y0i OP1 tp1r * src2i; \
  135. y0i OP1 tp2r * src4i; \
  136. y0i OP1 tp3r * src6i; \
  137. \
  138. y0i OP2 tp0i * src0r; \
  139. y0i OP2 tp1i * src2r; \
  140. y0i OP2 tp2i * src4r; \
  141. y0i OP2 tp3i * src6r; \
  142. #define ZGEMV_N_1x4() \
  143. res0 = y[0 * inc_y2]; \
  144. res1 = y[0 * inc_y2 + 1]; \
  145. \
  146. res0 += temp0_r * pa0[k]; \
  147. res0 OP0 temp0_i * pa0[k + 1]; \
  148. res0 += temp1_r * pa1[k]; \
  149. res0 OP0 temp1_i * pa1[k + 1]; \
  150. res0 += temp2_r * pa2[k]; \
  151. res0 OP0 temp2_i * pa2[k + 1]; \
  152. res0 += temp3_r * pa3[k]; \
  153. res0 OP0 temp3_i * pa3[k + 1]; \
  154. \
  155. res1 OP1 temp0_r * pa0[k + 1]; \
  156. res1 OP2 temp0_i * pa0[k]; \
  157. res1 OP1 temp1_r * pa1[k + 1]; \
  158. res1 OP2 temp1_i * pa1[k]; \
  159. res1 OP1 temp2_r * pa2[k + 1]; \
  160. res1 OP2 temp2_i * pa2[k]; \
  161. res1 OP1 temp3_r * pa3[k + 1]; \
  162. res1 OP2 temp3_i * pa3[k]; \
  163. \
  164. y[0 * inc_y2] = res0; \
  165. y[0 * inc_y2 + 1] = res1; \
  166. #define ZGEMV_N_4x2() \
  167. LD_DP4(pa0 + k, 2, t0, t1, t2, t3); \
  168. LD_DP4(pa1 + k, 2, t4, t5, t6, t7); \
  169. \
  170. PCKEVOD_D2_DP(t1, t0, src0r, src0i); \
  171. PCKEVOD_D2_DP(t3, t2, src1r, src1i); \
  172. PCKEVOD_D2_DP(t5, t4, src2r, src2i); \
  173. PCKEVOD_D2_DP(t7, t6, src3r, src3i); \
  174. \
  175. y0r += tp0r * src0r; \
  176. y1r += tp0r * src1r; \
  177. y0r += tp1r * src2r; \
  178. y1r += tp1r * src3r; \
  179. \
  180. y0r OP0 tp0i * src0i; \
  181. y1r OP0 tp0i * src1i; \
  182. y0r OP0 tp1i * src2i; \
  183. y1r OP0 tp1i * src3i; \
  184. \
  185. y0i OP1 tp0r * src0i; \
  186. y1i OP1 tp0r * src1i; \
  187. y0i OP1 tp1r * src2i; \
  188. y1i OP1 tp1r * src3i; \
  189. \
  190. y0i OP2 tp0i * src0r; \
  191. y1i OP2 tp0i * src1r; \
  192. y0i OP2 tp1i * src2r; \
  193. y1i OP2 tp1i * src3r; \
  194. #define ZGEMV_N_2x2() \
  195. LD_DP2(pa0 + k, 2, t0, t1); \
  196. LD_DP2(pa1 + k, 2, t4, t5); \
  197. \
  198. PCKEVOD_D2_DP(t1, t0, src0r, src0i); \
  199. PCKEVOD_D2_DP(t5, t4, src2r, src2i); \
  200. \
  201. y0r += tp0r * src0r; \
  202. y0r += tp1r * src2r; \
  203. \
  204. y0r OP0 tp0i * src0i; \
  205. y0r OP0 tp1i * src2i; \
  206. \
  207. y0i OP1 tp0r * src0i; \
  208. y0i OP1 tp1r * src2i; \
  209. \
  210. y0i OP2 tp0i * src0r; \
  211. y0i OP2 tp1i * src2r; \
  212. #define ZGEMV_N_1x2() \
  213. res0 = y[0 * inc_y2]; \
  214. res1 = y[0 * inc_y2 + 1]; \
  215. \
  216. res0 += temp0_r * pa0[k]; \
  217. res0 OP0 temp0_i * pa0[k + 1]; \
  218. res0 += temp1_r * pa1[k]; \
  219. res0 OP0 temp1_i * pa1[k + 1]; \
  220. \
  221. res1 OP1 temp0_r * pa0[k + 1]; \
  222. res1 OP2 temp0_i * pa0[k]; \
  223. res1 OP1 temp1_r * pa1[k + 1]; \
  224. res1 OP2 temp1_i * pa1[k]; \
  225. \
  226. y[0 * inc_y2] = res0; \
  227. y[0 * inc_y2 + 1] = res1; \
  228. #define ZGEMV_N_4x1() \
  229. LD_DP4(pa0 + k, 2, t0, t1, t2, t3); \
  230. \
  231. PCKEVOD_D2_DP(t1, t0, src0r, src0i); \
  232. PCKEVOD_D2_DP(t3, t2, src1r, src1i); \
  233. \
  234. y0r += tp0r * src0r; \
  235. y1r += tp0r * src1r; \
  236. \
  237. y0r OP0 tp0i * src0i; \
  238. y1r OP0 tp0i * src1i; \
  239. \
  240. y0i OP1 tp0r * src0i; \
  241. y1i OP1 tp0r * src1i; \
  242. \
  243. y0i OP2 tp0i * src0r; \
  244. y1i OP2 tp0i * src1r; \
  245. #define ZGEMV_N_2x1() \
  246. LD_DP2(pa0 + k, 2, t0, t1); \
  247. \
  248. PCKEVOD_D2_DP(t1, t0, src0r, src0i); \
  249. \
  250. y0r += tp0r * src0r; \
  251. y0r OP0 tp0i * src0i; \
  252. y0i OP1 tp0r * src0i; \
  253. y0i OP2 tp0i * src0r; \
  254. #define ZGEMV_N_1x1() \
  255. res0 = y[0 * inc_y2]; \
  256. res1 = y[0 * inc_y2 + 1]; \
  257. \
  258. res0 += temp0_r * pa0[k]; \
  259. res0 OP0 temp0_i * pa0[k + 1]; \
  260. \
  261. res1 OP1 temp0_r * pa0[k + 1]; \
  262. res1 OP2 temp0_i * pa0[k]; \
  263. \
  264. y[0 * inc_y2] = res0; \
  265. y[0 * inc_y2 + 1] = res1; \
  266. #define ZLOAD_X4_SCALE_VECTOR() \
  267. LD_DP4(x, 2, x0, x1, x2, x3); \
  268. \
  269. PCKEVOD_D2_DP(x1, x0, x0r, x0i); \
  270. PCKEVOD_D2_DP(x3, x2, x1r, x1i); \
  271. \
  272. tp4r = alphar * x0r; \
  273. tp4r OP3 alphai * x0i; \
  274. tp4i = alphar * x0i; \
  275. tp4i OP4 alphai * x0r; \
  276. \
  277. tp5r = alphar * x1r; \
  278. tp5r OP3 alphai * x1i; \
  279. tp5i = alphar * x1i; \
  280. tp5i OP4 alphai * x1r; \
  281. \
  282. SPLATI_D2_DP(tp4r, tp0r, tp1r); \
  283. SPLATI_D2_DP(tp5r, tp2r, tp3r); \
  284. SPLATI_D2_DP(tp4i, tp0i, tp1i); \
  285. SPLATI_D2_DP(tp5i, tp2i, tp3i); \
  286. #define ZLOAD_X2_SCALE_VECTOR() \
  287. LD_DP2(x, 2, x0, x1); \
  288. \
  289. PCKEVOD_D2_DP(x1, x0, x0r, x0i); \
  290. \
  291. tp4r = alphar * x0r; \
  292. tp4r OP3 alphai * x0i; \
  293. tp4i = alphar * x0i; \
  294. tp4i OP4 alphai * x0r; \
  295. \
  296. SPLATI_D2_DP(tp4r, tp0r, tp1r); \
  297. SPLATI_D2_DP(tp4i, tp0i, tp1i); \
  298. #define ZLOAD_X4_SCALE_GP() \
  299. x0r = (v2f64) __msa_insert_d((v2i64) tp0r, 0, *((BLASLONG *)(x + 0 * inc_x2))); \
  300. x0r = (v2f64) __msa_insert_d((v2i64) x0r, 1, *((BLASLONG *)(x + 1 * inc_x2))); \
  301. x1r = (v2f64) __msa_insert_d((v2i64) tp0r, 0, *((BLASLONG *)(x + 2 * inc_x2))); \
  302. x1r = (v2f64) __msa_insert_d((v2i64) x1r, 1, *((BLASLONG *)(x + 3 * inc_x2))); \
  303. x0i = (v2f64) __msa_insert_d((v2i64) tp0r, 0, *((BLASLONG *)(x + 0 * inc_x2 + 1))); \
  304. x0i = (v2f64) __msa_insert_d((v2i64) x0i, 1, *((BLASLONG *)(x + 1 * inc_x2 + 1))); \
  305. x1i = (v2f64) __msa_insert_d((v2i64) tp0r, 0, *((BLASLONG *)(x + 2 * inc_x2 + 1))); \
  306. x1i = (v2f64) __msa_insert_d((v2i64) x1i, 1, *((BLASLONG *)(x + 3 * inc_x2 + 1))); \
  307. \
  308. tp4r = alphar * x0r; \
  309. tp4r OP3 alphai * x0i; \
  310. tp4i = alphar * x0i; \
  311. tp4i OP4 alphai * x0r; \
  312. \
  313. tp5r = alphar * x1r; \
  314. tp5r OP3 alphai * x1i; \
  315. tp5i = alphar * x1i; \
  316. tp5i OP4 alphai * x1r; \
  317. \
  318. SPLATI_D2_DP(tp4r, tp0r, tp1r); \
  319. SPLATI_D2_DP(tp5r, tp2r, tp3r); \
  320. SPLATI_D2_DP(tp4i, tp0i, tp1i); \
  321. SPLATI_D2_DP(tp5i, tp2i, tp3i); \
  322. #define ZLOAD_X2_SCALE_GP() \
  323. x0r = (v2f64) __msa_insert_d((v2i64) tp0r, 0, *((BLASLONG *)(x + 0 * inc_x2))); \
  324. x0r = (v2f64) __msa_insert_d((v2i64) x0r, 1, *((BLASLONG *)(x + 1 * inc_x2))); \
  325. x0i = (v2f64) __msa_insert_d((v2i64) tp0r, 0, *((BLASLONG *)(x + 0 * inc_x2 + 1))); \
  326. x0i = (v2f64) __msa_insert_d((v2i64) x0i, 1, *((BLASLONG *)(x + 1 * inc_x2 + 1))); \
  327. \
  328. tp4r = alphar * x0r; \
  329. tp4r OP3 alphai * x0i; \
  330. tp4i = alphar * x0i; \
  331. tp4i OP4 alphai * x0r; \
  332. \
  333. SPLATI_D2_DP(tp4r, tp0r, tp1r); \
  334. SPLATI_D2_DP(tp4i, tp0i, tp1i); \
  335. #define ZLOAD_X1_SCALE_GP() \
  336. temp0_r = alpha_r * x[0 * inc_x2]; \
  337. temp0_r OP3 alpha_i * x[0 * inc_x2 + 1]; \
  338. temp0_i = alpha_r * x[0 * inc_x2 + 1]; \
  339. temp0_i OP4 alpha_i * x[0 * inc_x2]; \
  340. \
  341. tp0r = (v2f64) COPY_DOUBLE_TO_VECTOR(temp0_r); \
  342. tp0i = (v2f64) COPY_DOUBLE_TO_VECTOR(temp0_i); \
  343. #define ZLOAD_Y4_VECTOR() \
  344. LD_DP4(y, 2, y0, y1, y2, y3); \
  345. PCKEVOD_D2_DP(y1, y0, y0r, y0i); \
  346. PCKEVOD_D2_DP(y3, y2, y1r, y1i); \
  347. #define ZLOAD_Y2_VECTOR() \
  348. LD_DP2(y, 2, y0, y1); \
  349. PCKEVOD_D2_DP(y1, y0, y0r, y0i); \
  350. #define ZSTORE_Y4_VECTOR() \
  351. ILVRL_D2_DP(y0i, y0r, y0, y1); \
  352. ILVRL_D2_DP(y1i, y1r, y2, y3); \
  353. ST_DP4(y0, y1, y2, y3, y, 2); \
  354. #define ZSTORE_Y2_VECTOR() \
  355. ILVRL_D2_DP(y0i, y0r, y0, y1); \
  356. ST_DP2(y0, y1, y, 2); \
  357. #define ZLOAD_Y4_GP() \
  358. y0r = (v2f64) __msa_insert_d((v2i64) tp0r, 0, *((BLASLONG *)(y + 0 * inc_y2))); \
  359. y0r = (v2f64) __msa_insert_d((v2i64) y0r, 1, *((BLASLONG *)(y + 1 * inc_y2))); \
  360. y1r = (v2f64) __msa_insert_d((v2i64) tp0r, 0, *((BLASLONG *)(y + 2 * inc_y2))); \
  361. y1r = (v2f64) __msa_insert_d((v2i64) y1r, 1, *((BLASLONG *)(y + 3 * inc_y2))); \
  362. y0i = (v2f64) __msa_insert_d((v2i64) tp0r, 0, *((BLASLONG *)(y + 0 * inc_y2 + 1))); \
  363. y0i = (v2f64) __msa_insert_d((v2i64) y0i, 1, *((BLASLONG *)(y + 1 * inc_y2 + 1))); \
  364. y1i = (v2f64) __msa_insert_d((v2i64) tp0r, 0, *((BLASLONG *)(y + 2 * inc_y2 + 1))); \
  365. y1i = (v2f64) __msa_insert_d((v2i64) y1i, 1, *((BLASLONG *)(y + 3 * inc_y2 + 1))); \
  366. #define ZLOAD_Y2_GP() \
  367. y0r = (v2f64) __msa_insert_d((v2i64) tp0r, 0, *((BLASLONG *)(y + 0 * inc_y2))); \
  368. y0r = (v2f64) __msa_insert_d((v2i64) y0r, 1, *((BLASLONG *)(y + 1 * inc_y2))); \
  369. y0i = (v2f64) __msa_insert_d((v2i64) tp0r, 0, *((BLASLONG *)(y + 0 * inc_y2 + 1))); \
  370. y0i = (v2f64) __msa_insert_d((v2i64) y0i, 1, *((BLASLONG *)(y + 1 * inc_y2 + 1))); \
  371. #define ZSTORE_Y4_GP() \
  372. *((BLASLONG *)(y + 0 * inc_y2)) = __msa_copy_s_d((v2i64) y0r, 0); \
  373. *((BLASLONG *)(y + 1 * inc_y2)) = __msa_copy_s_d((v2i64) y0r, 1); \
  374. *((BLASLONG *)(y + 2 * inc_y2)) = __msa_copy_s_d((v2i64) y1r, 0); \
  375. *((BLASLONG *)(y + 3 * inc_y2)) = __msa_copy_s_d((v2i64) y1r, 1); \
  376. *((BLASLONG *)(y + 0 * inc_y2 + 1)) = __msa_copy_s_d((v2i64) y0i, 0); \
  377. *((BLASLONG *)(y + 1 * inc_y2 + 1)) = __msa_copy_s_d((v2i64) y0i, 1); \
  378. *((BLASLONG *)(y + 2 * inc_y2 + 1)) = __msa_copy_s_d((v2i64) y1i, 0); \
  379. *((BLASLONG *)(y + 3 * inc_y2 + 1)) = __msa_copy_s_d((v2i64) y1i, 1); \
  380. #define ZSTORE_Y2_GP() \
  381. *((BLASLONG *)(y + 0 * inc_y2)) = __msa_copy_s_d((v2i64) y0r, 0); \
  382. *((BLASLONG *)(y + 1 * inc_y2)) = __msa_copy_s_d((v2i64) y0r, 1); \
  383. *((BLASLONG *)(y + 0 * inc_y2 + 1)) = __msa_copy_s_d((v2i64) y0i, 0); \
  384. *((BLASLONG *)(y + 1 * inc_y2 + 1)) = __msa_copy_s_d((v2i64) y0i, 1); \
  385. #define ZGEMV_N_MSA() \
  386. for (j = (n >> 2); j--;) \
  387. { \
  388. ZLOAD_X4_SCALE() \
  389. \
  390. k = 0; \
  391. k_pref = pref_offset; \
  392. y = y_org; \
  393. \
  394. for (i = (m >> 2); i--;) \
  395. { \
  396. PREFETCH(pa0 + k_pref + 8 + 0); \
  397. PREFETCH(pa0 + k_pref + 8 + 4); \
  398. PREFETCH(pa1 + k_pref + 8 + 0); \
  399. PREFETCH(pa1 + k_pref + 8 + 4); \
  400. PREFETCH(pa2 + k_pref + 8 + 0); \
  401. PREFETCH(pa2 + k_pref + 8 + 4); \
  402. PREFETCH(pa3 + k_pref + 8 + 0); \
  403. PREFETCH(pa3 + k_pref + 8 + 4); \
  404. \
  405. ZLOAD_Y4() \
  406. ZGEMV_N_4x4() \
  407. ZSTORE_Y4() \
  408. \
  409. k += 2 * 4; \
  410. k_pref += 2 * 4; \
  411. y += inc_y2 * 4; \
  412. } \
  413. \
  414. if (m & 2) \
  415. { \
  416. ZLOAD_Y2() \
  417. ZGEMV_N_2x4() \
  418. ZSTORE_Y2() \
  419. \
  420. k += 2 * 2; \
  421. y += inc_y2 * 2; \
  422. } \
  423. \
  424. if (m & 1) \
  425. { \
  426. temp0_r = tp4r[0]; \
  427. temp1_r = tp4r[1]; \
  428. temp2_r = tp5r[0]; \
  429. temp3_r = tp5r[1]; \
  430. \
  431. temp0_i = tp4i[0]; \
  432. temp1_i = tp4i[1]; \
  433. temp2_i = tp5i[0]; \
  434. temp3_i = tp5i[1]; \
  435. \
  436. ZGEMV_N_1x4() \
  437. k += 2; \
  438. y += inc_y2; \
  439. } \
  440. \
  441. pa0 += 4 * lda2; \
  442. pa1 += 4 * lda2; \
  443. pa2 += 4 * lda2; \
  444. pa3 += 4 * lda2; \
  445. \
  446. x += 4 * inc_x2; \
  447. } \
  448. \
  449. if (n & 2) \
  450. { \
  451. ZLOAD_X2_SCALE() \
  452. \
  453. k = 0; \
  454. y = y_org; \
  455. \
  456. for (i = (m >> 2); i--;) \
  457. { \
  458. ZLOAD_Y4() \
  459. ZGEMV_N_4x2() \
  460. ZSTORE_Y4() \
  461. \
  462. k += 2 * 4; \
  463. y += inc_y2 * 4; \
  464. } \
  465. \
  466. if (m & 2) \
  467. { \
  468. ZLOAD_Y2() \
  469. ZGEMV_N_2x2() \
  470. ZSTORE_Y2() \
  471. \
  472. k += 2 * 2; \
  473. y += inc_y2 * 2; \
  474. } \
  475. \
  476. if (m & 1) \
  477. { \
  478. temp0_r = tp4r[0]; \
  479. temp1_r = tp4r[1]; \
  480. \
  481. temp0_i = tp4i[0]; \
  482. temp1_i = tp4i[1]; \
  483. \
  484. ZGEMV_N_1x2() \
  485. \
  486. k += 2; \
  487. y += inc_y2; \
  488. } \
  489. \
  490. pa0 += 2 * lda2; \
  491. pa1 += 2 * lda2; \
  492. \
  493. x += 2 * inc_x2; \
  494. } \
  495. \
  496. if (n & 1) \
  497. { \
  498. ZLOAD_X1_SCALE() \
  499. \
  500. k = 0; \
  501. y = y_org; \
  502. \
  503. for (i = (m >> 2); i--;) \
  504. { \
  505. ZLOAD_Y4() \
  506. ZGEMV_N_4x1() \
  507. ZSTORE_Y4() \
  508. \
  509. k += 2 * 4; \
  510. y += inc_y2 * 4; \
  511. } \
  512. \
  513. if (m & 2) \
  514. { \
  515. ZLOAD_Y2() \
  516. ZGEMV_N_2x1() \
  517. ZSTORE_Y2() \
  518. \
  519. k += 2 * 2; \
  520. y += inc_y2 * 2; \
  521. } \
  522. \
  523. if (m & 1) \
  524. { \
  525. ZGEMV_N_1x1() \
  526. \
  527. k += 2; \
  528. y += inc_y2; \
  529. } \
  530. \
  531. pa0 += lda2; \
  532. x += inc_x2; \
  533. } \
  534. int CNAME(BLASLONG m, BLASLONG n, BLASLONG dummy1, FLOAT alpha_r, FLOAT alpha_i,
  535. FLOAT *A, BLASLONG lda2, FLOAT *x, BLASLONG inc_x2, FLOAT *y,
  536. BLASLONG inc_y2, FLOAT *buffer)
  537. {
  538. BLASLONG i, j, k, k_pref, pref_offset;
  539. FLOAT *y_org = y;
  540. FLOAT *pa0, *pa1, *pa2, *pa3;
  541. FLOAT temp0_r, temp1_r, temp2_r, temp3_r, temp0_i, temp1_i, temp2_i;
  542. FLOAT temp3_i, res0, res1;
  543. v2f64 alphar, alphai;
  544. v2f64 x0, x1, x2, x3, y0, y1, y2, y3;
  545. v2f64 x0r, x1r, x0i, x1i, y0r, y1r, y0i, y1i;
  546. v2f64 t0, t1, t2, t3, t4, t5, t6, t7, t8, t9, t10, t11, t12, t13, t14, t15;
  547. v2f64 src0r, src1r, src2r, src3r, src4r, src5r, src6r, src7r;
  548. v2f64 src0i, src1i, src2i, src3i, src4i, src5i, src6i, src7i;
  549. v2f64 tp0r, tp1r, tp2r, tp3r, tp4r, tp5r, tp0i, tp1i, tp2i, tp3i, tp4i, tp5i;
  550. lda2 = 2 * lda2;
  551. inc_x2 = 2 * inc_x2;
  552. inc_y2 = 2 * inc_y2;
  553. pref_offset = (uintptr_t)A & (L1_DATA_LINESIZE - 1);
  554. pref_offset = L1_DATA_LINESIZE - pref_offset;
  555. pref_offset = pref_offset / sizeof(FLOAT);
  556. pa0 = A;
  557. pa1 = A + lda2;
  558. pa2 = A + 2 * lda2;
  559. pa3 = A + 3 * lda2;
  560. alphar = COPY_DOUBLE_TO_VECTOR(alpha_r);
  561. alphai = COPY_DOUBLE_TO_VECTOR(alpha_i);
  562. if ((2 == inc_x2) && (2 == inc_y2))
  563. {
  564. #define ZLOAD_X4_SCALE ZLOAD_X4_SCALE_VECTOR
  565. #define ZLOAD_X2_SCALE ZLOAD_X2_SCALE_VECTOR
  566. #define ZLOAD_X1_SCALE ZLOAD_X1_SCALE_GP
  567. #define ZLOAD_Y4 ZLOAD_Y4_VECTOR
  568. #define ZLOAD_Y2 ZLOAD_Y2_VECTOR
  569. #define ZSTORE_Y4 ZSTORE_Y4_VECTOR
  570. #define ZSTORE_Y2 ZSTORE_Y2_VECTOR
  571. ZGEMV_N_MSA();
  572. #undef ZLOAD_X4_SCALE
  573. #undef ZLOAD_X2_SCALE
  574. #undef ZLOAD_X1_SCALE
  575. #undef ZLOAD_Y4
  576. #undef ZLOAD_Y2
  577. #undef ZSTORE_Y4
  578. #undef ZSTORE_Y2
  579. }
  580. else if (2 == inc_x2)
  581. {
  582. #define ZLOAD_X4_SCALE ZLOAD_X4_SCALE_VECTOR
  583. #define ZLOAD_X2_SCALE ZLOAD_X2_SCALE_VECTOR
  584. #define ZLOAD_X1_SCALE ZLOAD_X1_SCALE_GP
  585. #define ZLOAD_Y4 ZLOAD_Y4_GP
  586. #define ZLOAD_Y2 ZLOAD_Y2_GP
  587. #define ZSTORE_Y4 ZSTORE_Y4_GP
  588. #define ZSTORE_Y2 ZSTORE_Y2_GP
  589. ZGEMV_N_MSA();
  590. #undef ZLOAD_X4_SCALE
  591. #undef ZLOAD_X2_SCALE
  592. #undef ZLOAD_X1_SCALE
  593. #undef ZLOAD_Y4
  594. #undef ZLOAD_Y2
  595. #undef ZSTORE_Y4
  596. #undef ZSTORE_Y2
  597. }
  598. else if (2 == inc_y2)
  599. {
  600. #define ZLOAD_X4_SCALE ZLOAD_X4_SCALE_GP
  601. #define ZLOAD_X2_SCALE ZLOAD_X2_SCALE_GP
  602. #define ZLOAD_X1_SCALE ZLOAD_X1_SCALE_GP
  603. #define ZLOAD_Y4 ZLOAD_Y4_VECTOR
  604. #define ZLOAD_Y2 ZLOAD_Y2_VECTOR
  605. #define ZSTORE_Y4 ZSTORE_Y4_VECTOR
  606. #define ZSTORE_Y2 ZSTORE_Y2_VECTOR
  607. ZGEMV_N_MSA();
  608. #undef ZLOAD_X4_SCALE
  609. #undef ZLOAD_X2_SCALE
  610. #undef ZLOAD_X1_SCALE
  611. #undef ZLOAD_Y4
  612. #undef ZLOAD_Y2
  613. #undef ZSTORE_Y4
  614. #undef ZSTORE_Y2
  615. }
  616. else
  617. {
  618. #define ZLOAD_X4_SCALE ZLOAD_X4_SCALE_GP
  619. #define ZLOAD_X2_SCALE ZLOAD_X2_SCALE_GP
  620. #define ZLOAD_X1_SCALE ZLOAD_X1_SCALE_GP
  621. #define ZLOAD_Y4 ZLOAD_Y4_GP
  622. #define ZLOAD_Y2 ZLOAD_Y2_GP
  623. #define ZSTORE_Y4 ZSTORE_Y4_GP
  624. #define ZSTORE_Y2 ZSTORE_Y2_GP
  625. ZGEMV_N_MSA();
  626. #undef ZLOAD_X4_SCALE
  627. #undef ZLOAD_X2_SCALE
  628. #undef ZLOAD_X1_SCALE
  629. #undef ZLOAD_Y4
  630. #undef ZLOAD_Y2
  631. #undef ZSTORE_Y4
  632. #undef ZSTORE_Y2
  633. }
  634. return(0);
  635. }
  636. #undef OP0
  637. #undef OP1
  638. #undef OP2
  639. #undef OP3
  640. #undef OP4