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.

cscal_msa.c 37 kB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008
  1. /*******************************************************************************
  2. Copyright (c) 2017, 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. /* This will shuffle the elements in 'in' vector as (mask needed :: 10 11 00 01)
  30. 0 1 2 3 => 1 0 3 2 */
  31. #define SHF_177 177
  32. int CNAME(BLASLONG n, BLASLONG dummy0, BLASLONG dummy1, FLOAT da_r, FLOAT da_i,
  33. FLOAT *x, BLASLONG inc_x, FLOAT *y, BLASLONG inc_y, FLOAT *dummy,
  34. BLASLONG dummy2)
  35. {
  36. BLASLONG i, inc_x2;
  37. FLOAT *px;
  38. FLOAT tp0, tp1, tp2, tp3, f0, f1, f2, f3;
  39. v4f32 x0, x1, x2, x3, x4, x5, x6, x7, x8, x9, x10, x11, x12, x13, x14, x15;
  40. v4f32 d0, d1, d2, d3, d4, d5, d6, d7, d8, d9, d10, d11, d12, d13, d14, d15;
  41. v4f32 da_i_vec, da_i_vec_neg, da_r_vec;
  42. px = x;
  43. if (1 == inc_x)
  44. {
  45. if ((0.0 == da_r) && (0.0 == da_i))
  46. {
  47. v4f32 zero_v = {0.0, 0.0, 0.0, 0.0};
  48. for (i = (n >> 5); i--;)
  49. {
  50. ST_SP8_INC(zero_v, zero_v, zero_v, zero_v, zero_v, zero_v,
  51. zero_v, zero_v, x, 4);
  52. ST_SP8_INC(zero_v, zero_v, zero_v, zero_v, zero_v, zero_v,
  53. zero_v, zero_v, x, 4);
  54. }
  55. if (n & 31)
  56. {
  57. if (n & 16)
  58. {
  59. ST_SP8_INC(zero_v, zero_v, zero_v, zero_v, zero_v, zero_v,
  60. zero_v, zero_v, x, 4);
  61. }
  62. if (n & 8)
  63. {
  64. ST_SP4_INC(zero_v, zero_v, zero_v, zero_v, x, 4);
  65. }
  66. if (n & 4)
  67. {
  68. ST_SP2_INC(zero_v, zero_v, x, 4);
  69. }
  70. if (n & 2)
  71. {
  72. ST_SP(zero_v, x); x += 4;
  73. }
  74. if (n & 1)
  75. {
  76. *x = 0; x += 1;
  77. *x = 0;
  78. }
  79. }
  80. }
  81. else if (0.0 == da_r)
  82. {
  83. da_i_vec = COPY_FLOAT_TO_VECTOR(da_i);
  84. da_i_vec_neg = -da_i_vec;
  85. da_i_vec = (v4f32) __msa_ilvev_w((v4i32) da_i_vec_neg, (v4i32) da_i_vec);
  86. if (n > 31)
  87. {
  88. FLOAT *x_pref;
  89. BLASLONG pref_offset;
  90. pref_offset = (BLASLONG)x & (L1_DATA_LINESIZE - 1);
  91. if (pref_offset > 0)
  92. {
  93. pref_offset = L1_DATA_LINESIZE - pref_offset;
  94. pref_offset = pref_offset / sizeof(FLOAT);
  95. }
  96. x_pref = x + pref_offset + 64 + 32;
  97. LD_SP8_INC(px, 4, x0, x1, x2, x3, x4, x5, x6, x7);
  98. for (i = (n >> 5)- 1; i--;)
  99. {
  100. PREF_OFFSET(x_pref, 0);
  101. PREF_OFFSET(x_pref, 32);
  102. PREF_OFFSET(x_pref, 64);
  103. PREF_OFFSET(x_pref, 96);
  104. PREF_OFFSET(x_pref, 128);
  105. PREF_OFFSET(x_pref, 160);
  106. PREF_OFFSET(x_pref, 192);
  107. PREF_OFFSET(x_pref, 224);
  108. x_pref += 64;
  109. x8 = LD_SP(px); px += 4;
  110. x0 *= da_i_vec;
  111. x9 = LD_SP(px); px += 4;
  112. x1 *= da_i_vec;
  113. x10 = LD_SP(px); px += 4;
  114. x2 *= da_i_vec;
  115. x11 = LD_SP(px); px += 4;
  116. x3 *= da_i_vec;
  117. x12 = LD_SP(px); px += 4;
  118. x4 *= da_i_vec;
  119. x13 = LD_SP(px); px += 4;
  120. x5 *= da_i_vec;
  121. x0 = (v4f32) __msa_shf_w((v4i32) x0, SHF_177);
  122. x14 = LD_SP(px); px += 4;
  123. x6 *= da_i_vec;
  124. x1 = (v4f32) __msa_shf_w((v4i32) x1, SHF_177);
  125. x15 = LD_SP(px); px += 4;
  126. x7 *= da_i_vec;
  127. x2 = (v4f32) __msa_shf_w((v4i32) x2, SHF_177);
  128. x8 *= da_i_vec;
  129. x3 = (v4f32) __msa_shf_w((v4i32) x3, SHF_177);
  130. ST_SP(x0, x); x += 4;
  131. x9 *= da_i_vec;
  132. x4 = (v4f32) __msa_shf_w((v4i32) x4, SHF_177);
  133. ST_SP(x1, x); x += 4;
  134. x10 *= da_i_vec;
  135. x5 = (v4f32) __msa_shf_w((v4i32) x5, SHF_177);
  136. ST_SP(x2, x); x += 4;
  137. x11 *= da_i_vec;
  138. x6 = (v4f32) __msa_shf_w((v4i32) x6, SHF_177);
  139. ST_SP(x3, x); x += 4;
  140. x12 *= da_i_vec;
  141. x7 = (v4f32) __msa_shf_w((v4i32) x7, SHF_177);
  142. ST_SP(x4, x); x += 4;
  143. x13 *= da_i_vec;
  144. x8 = (v4f32) __msa_shf_w((v4i32) x8, SHF_177);
  145. ST_SP(x5, x); x += 4;
  146. x14 *= da_i_vec;
  147. x9 = (v4f32) __msa_shf_w((v4i32) x9, SHF_177);
  148. ST_SP(x6, x); x += 4;
  149. x15 *= da_i_vec;
  150. x10 = (v4f32) __msa_shf_w((v4i32) x10, SHF_177);
  151. ST_SP(x7, x); x += 4;
  152. x11 = (v4f32) __msa_shf_w((v4i32) x11, SHF_177);
  153. ST_SP(x8, x); x += 4;
  154. x0 = LD_SP(px); px += 4;
  155. x12 = (v4f32) __msa_shf_w((v4i32) x12, SHF_177);
  156. ST_SP(x9, x); x += 4;
  157. x1 = LD_SP(px); px += 4;
  158. x13 = (v4f32) __msa_shf_w((v4i32) x13, SHF_177);
  159. ST_SP(x10, x); x += 4;
  160. x2 = LD_SP(px); px += 4;
  161. x14 = (v4f32) __msa_shf_w((v4i32) x14, SHF_177);
  162. ST_SP(x11, x); x += 4;
  163. x3 = LD_SP(px); px += 4;
  164. x15 = (v4f32) __msa_shf_w((v4i32) x15, SHF_177);
  165. ST_SP(x12, x); x += 4;
  166. x4 = LD_SP(px); px += 4;
  167. ST_SP(x13, x); x += 4;
  168. x5 = LD_SP(px); px += 4;
  169. ST_SP(x14, x); x += 4;
  170. x6 = LD_SP(px); px += 4;
  171. ST_SP(x15, x); x += 4;
  172. x7 = LD_SP(px); px += 4;
  173. }
  174. LD_SP8_INC(px, 4, x8, x9, x10, x11, x12, x13, x14, x15);
  175. MUL4(x0, da_i_vec, x1, da_i_vec, x2, da_i_vec, x3, da_i_vec,
  176. x0, x1, x2, x3);
  177. MUL4(x4, da_i_vec, x5, da_i_vec, x6, da_i_vec, x7, da_i_vec,
  178. x4, x5, x6, x7);
  179. MUL4(x8, da_i_vec, x9, da_i_vec, x10, da_i_vec, x11, da_i_vec,
  180. x8, x9, x10, x11);
  181. MUL4(x12, da_i_vec, x13, da_i_vec, x14, da_i_vec, x15, da_i_vec,
  182. x12, x13, x14, x15);
  183. SHF_W4_SP(x0, x1, x2, x3, x0, x1, x2, x3, SHF_177);
  184. SHF_W4_SP(x4, x5, x6, x7, x4, x5, x6, x7, SHF_177);
  185. SHF_W4_SP(x8, x9, x10, x11, x8, x9, x10, x11, SHF_177);
  186. SHF_W4_SP(x12, x13, x14, x15, x12, x13, x14, x15, SHF_177);
  187. ST_SP16_INC(x0, x1, x2, x3, x4, x5, x6, x7, x8, x9, x10, x11,
  188. x12, x13, x14, x15, x, 4);
  189. }
  190. if (n & 31)
  191. {
  192. if (n & 16)
  193. {
  194. LD_SP8_INC(px, 4, x0, x1, x2, x3, x4, x5, x6, x7);
  195. MUL4(x0, da_i_vec, x1, da_i_vec, x2, da_i_vec, x3, da_i_vec,
  196. x0, x1, x2, x3);
  197. MUL4(x4, da_i_vec, x5, da_i_vec, x6, da_i_vec, x7, da_i_vec,
  198. x4, x5, x6, x7);
  199. SHF_W4_SP(x0, x1, x2, x3, x0, x1, x2, x3, SHF_177);
  200. SHF_W4_SP(x4, x5, x6, x7, x4, x5, x6, x7, SHF_177);
  201. ST_SP8_INC(x0, x1, x2, x3, x4, x5, x6, x7, x, 4);
  202. }
  203. if (n & 8)
  204. {
  205. LD_SP4_INC(px, 4, x0, x1, x2, x3);
  206. MUL4(x0, da_i_vec, x1, da_i_vec, x2, da_i_vec, x3, da_i_vec,
  207. x0, x1, x2, x3);
  208. SHF_W4_SP(x0, x1, x2, x3, x0, x1, x2, x3, SHF_177);
  209. ST_SP4_INC(x0, x1, x2, x3, x, 4);
  210. }
  211. if (n & 4)
  212. {
  213. LD_SP2_INC(px, 4, x0, x1);
  214. MUL2(x0, da_i_vec, x1, da_i_vec, x0, x1);
  215. SHF_W2_SP(x0, x1, x0, x1, SHF_177);
  216. ST_SP2_INC(x0, x1, x, 4);
  217. }
  218. if (n & 2)
  219. {
  220. LD_GP4_INC(px, 1, f0, f1, f2, f3);
  221. MUL4(f0, da_i, f1, -da_i, f2, da_i, f3, -da_i,
  222. f0, f1, f2, f3);
  223. ST_GP4_INC(f1, f0, f3, f2, x, 1);
  224. }
  225. if (n & 1)
  226. {
  227. LD_GP2_INC(px, 1, f0, f1);
  228. MUL2(f0, da_i, f1, -da_i, f0, f1);
  229. ST_GP2_INC(f1, f0, x, 1);
  230. }
  231. }
  232. }
  233. else if (0.0 == da_i)
  234. {
  235. da_r_vec = COPY_FLOAT_TO_VECTOR(da_r);
  236. if (n > 31)
  237. {
  238. FLOAT *x_pref;
  239. BLASLONG pref_offset;
  240. pref_offset = (BLASLONG)x & (L1_DATA_LINESIZE - 1);
  241. if (pref_offset > 0)
  242. {
  243. pref_offset = L1_DATA_LINESIZE - pref_offset;
  244. pref_offset = pref_offset / sizeof(FLOAT);
  245. }
  246. x_pref = x + pref_offset + 64 + 32;
  247. LD_SP8_INC(px, 4, x0, x1, x2, x3, x4, x5, x6, x7);
  248. for (i = (n >> 5)- 1; i--;)
  249. {
  250. PREF_OFFSET(x_pref, 0);
  251. PREF_OFFSET(x_pref, 32);
  252. PREF_OFFSET(x_pref, 64);
  253. PREF_OFFSET(x_pref, 96);
  254. PREF_OFFSET(x_pref, 128);
  255. PREF_OFFSET(x_pref, 160);
  256. PREF_OFFSET(x_pref, 192);
  257. PREF_OFFSET(x_pref, 224);
  258. x_pref += 64;
  259. x8 = LD_SP(px); px += 4;
  260. x0 *= da_r_vec;
  261. x9 = LD_SP(px); px += 4;
  262. x1 *= da_r_vec;
  263. x10 = LD_SP(px); px += 4;
  264. x2 *= da_r_vec;
  265. x11 = LD_SP(px); px += 4;
  266. x3 *= da_r_vec;
  267. x12 = LD_SP(px); px += 4;
  268. x4 *= da_r_vec;
  269. x13 = LD_SP(px); px += 4;
  270. x5 *= da_r_vec;
  271. ST_SP(x0, x); x += 4;
  272. x14 = LD_SP(px); px += 4;
  273. x6 *= da_r_vec;
  274. ST_SP(x1, x); x += 4;
  275. x15 = LD_SP(px); px += 4;
  276. x7 *= da_r_vec;
  277. ST_SP(x2, x); x += 4;
  278. x8 *= da_r_vec;
  279. ST_SP(x3, x); x += 4;
  280. x9 *= da_r_vec;
  281. ST_SP(x4, x); x += 4;
  282. x10 *= da_r_vec;
  283. ST_SP(x5, x); x += 4;
  284. x11 *= da_r_vec;
  285. ST_SP(x6, x); x += 4;
  286. x12 *= da_r_vec;
  287. ST_SP(x7, x); x += 4;
  288. x13 *= da_r_vec;
  289. ST_SP(x8, x); x += 4;
  290. x0 = LD_SP(px); px += 4;
  291. x14 *= da_r_vec;
  292. ST_SP(x9, x); x += 4;
  293. x1 = LD_SP(px); px += 4;
  294. x15 *= da_r_vec;
  295. ST_SP(x10, x); x += 4;
  296. x2 = LD_SP(px); px += 4;
  297. ST_SP(x11, x); x += 4;
  298. x3 = LD_SP(px); px += 4;
  299. ST_SP(x12, x); x += 4;
  300. x4 = LD_SP(px); px += 4;
  301. ST_SP(x13, x); x += 4;
  302. x5 = LD_SP(px); px += 4;
  303. ST_SP(x14, x); x += 4;
  304. x6 = LD_SP(px); px += 4;
  305. ST_SP(x15, x); x += 4;
  306. x7 = LD_SP(px); px += 4;
  307. }
  308. LD_SP8_INC(px, 4, x8, x9, x10, x11, x12, x13, x14, x15);
  309. MUL4(x0, da_r_vec, x1, da_r_vec, x2, da_r_vec, x3, da_r_vec,
  310. x0, x1, x2, x3);
  311. MUL4(x4, da_r_vec, x5, da_r_vec, x6, da_r_vec, x7, da_r_vec,
  312. x4, x5, x6, x7);
  313. MUL4(x8, da_r_vec, x9, da_r_vec, x10, da_r_vec, x11, da_r_vec,
  314. x8, x9, x10, x11);
  315. MUL4(x12, da_r_vec, x13, da_r_vec, x14, da_r_vec, x15, da_r_vec,
  316. x12, x13, x14, x15);
  317. ST_SP16_INC(x0, x1, x2, x3, x4, x5, x6, x7, x8, x9, x10, x11,
  318. x12, x13, x14, x15, x, 4);
  319. }
  320. if (n & 31)
  321. {
  322. if (n & 16)
  323. {
  324. LD_SP8_INC(px, 4, x0, x1, x2, x3, x4, x5, x6, x7);
  325. MUL4(x0, da_r_vec, x1, da_r_vec, x2, da_r_vec, x3, da_r_vec,
  326. x0, x1, x2, x3);
  327. MUL4(x4, da_r_vec, x5, da_r_vec, x6, da_r_vec, x7, da_r_vec,
  328. x4, x5, x6, x7);
  329. ST_SP8_INC(x0, x1, x2, x3, x4, x5, x6, x7, x, 4);
  330. }
  331. if (n & 8)
  332. {
  333. LD_SP4_INC(px, 4, x0, x1, x2, x3);
  334. MUL4(x0, da_r_vec, x1, da_r_vec, x2, da_r_vec, x3, da_r_vec,
  335. x0, x1, x2, x3);
  336. ST_SP4_INC(x0, x1, x2, x3, x, 4);
  337. }
  338. if (n & 4)
  339. {
  340. LD_SP2_INC(px, 4, x0, x1);
  341. MUL2(x0, da_r_vec, x1, da_r_vec, x0, x1);
  342. ST_SP2_INC(x0, x1, x, 4);
  343. }
  344. if (n & 2)
  345. {
  346. LD_GP4_INC(px, 1, f0, f1, f2, f3);
  347. MUL4(f0, da_r, f1, da_r, f2, da_r, f3, da_r, f0, f1, f2, f3);
  348. ST_GP4_INC(f0, f1, f2, f3, x, 1);
  349. }
  350. if (n & 1)
  351. {
  352. LD_GP2_INC(px, 1, f0, f1);
  353. MUL2(f0, da_r, f1, da_r, f0, f1);
  354. ST_GP2_INC(f0, f1, x, 1);
  355. }
  356. }
  357. }
  358. else
  359. {
  360. FLOAT *x_pref;
  361. BLASLONG pref_offset;
  362. pref_offset = (BLASLONG)x & (L1_DATA_LINESIZE - 1);
  363. if (pref_offset > 0)
  364. {
  365. pref_offset = L1_DATA_LINESIZE - pref_offset;
  366. pref_offset = pref_offset / sizeof(FLOAT);
  367. }
  368. x_pref = x + pref_offset + 64;
  369. da_i_vec = COPY_FLOAT_TO_VECTOR(da_i);
  370. da_i_vec_neg = -da_i_vec;
  371. da_i_vec = (v4f32) __msa_ilvev_w((v4i32) da_i_vec_neg, (v4i32) da_i_vec);
  372. da_r_vec = COPY_FLOAT_TO_VECTOR(da_r);
  373. for (i = (n >> 5); i--;)
  374. {
  375. PREF_OFFSET(x_pref, 0);
  376. PREF_OFFSET(x_pref, 32);
  377. PREF_OFFSET(x_pref, 64);
  378. PREF_OFFSET(x_pref, 96);
  379. PREF_OFFSET(x_pref, 128);
  380. PREF_OFFSET(x_pref, 160);
  381. PREF_OFFSET(x_pref, 192);
  382. PREF_OFFSET(x_pref, 224);
  383. x_pref += 64;
  384. LD_SP16_INC(px, 4, x0, x1, x2, x3, x4, x5, x6, x7, x8, x9, x10,
  385. x11, x12, x13, x14, x15);
  386. MUL4(x0, da_i_vec, x1, da_i_vec, x2, da_i_vec, x3, da_i_vec,
  387. d0, d1, d2, d3);
  388. MUL4(x4, da_i_vec, x5, da_i_vec, x6, da_i_vec, x7, da_i_vec,
  389. d4, d5, d6, d7);
  390. MUL4(x8, da_i_vec, x9, da_i_vec, x10, da_i_vec, x11, da_i_vec,
  391. d8, d9, d10, d11);
  392. MUL4(x12, da_i_vec, x13, da_i_vec, x14, da_i_vec, x15, da_i_vec,
  393. d12, d13, d14, d15);
  394. SHF_W4_SP(d0, d1, d2, d3, d0, d1, d2, d3, SHF_177);
  395. SHF_W4_SP(d4, d5, d6, d7, d4, d5, d6, d7, SHF_177);
  396. SHF_W4_SP(d8, d9, d10, d11, d8, d9, d10, d11, SHF_177);
  397. SHF_W4_SP(d12, d13, d14, d15, d12, d13, d14, d15, SHF_177);
  398. FMADD4(x0, x1, x2, x3, da_r_vec, d0, d1, d2, d3);
  399. FMADD4(x4, x5, x6, x7, da_r_vec, d4, d5, d6, d7);
  400. FMADD4(x8, x9, x10, x11, da_r_vec, d8, d9, d10, d11);
  401. FMADD4(x12, x13, x14, x15, da_r_vec, d12, d13, d14, d15);
  402. ST_SP16_INC(d0, d1, d2, d3, d4, d5, d6, d7, d8, d9, d10, d11,
  403. d12, d13, d14, d15, x, 4);
  404. }
  405. if (n & 31)
  406. {
  407. if (n & 16)
  408. {
  409. LD_SP8_INC(px, 4, x0, x1, x2, x3, x4, x5, x6, x7);
  410. MUL4(x0, da_i_vec, x1, da_i_vec, x2, da_i_vec, x3, da_i_vec,
  411. d0, d1, d2, d3);
  412. MUL4(x4, da_i_vec, x5, da_i_vec, x6, da_i_vec, x7, da_i_vec,
  413. d4, d5, d6, d7);
  414. SHF_W4_SP(d0, d1, d2, d3, d0, d1, d2, d3, SHF_177);
  415. SHF_W4_SP(d4, d5, d6, d7, d4, d5, d6, d7, SHF_177);
  416. FMADD4(x0, x1, x2, x3, da_r_vec, d0, d1, d2, d3);
  417. FMADD4(x4, x5, x6, x7, da_r_vec, d4, d5, d6, d7);
  418. ST_SP8_INC(d0, d1, d2, d3, d4, d5, d6, d7, x, 4);
  419. }
  420. if (n & 8)
  421. {
  422. LD_SP4_INC(px, 4, x0, x1, x2, x3);
  423. MUL4(x0, da_i_vec, x1, da_i_vec, x2, da_i_vec, x3, da_i_vec,
  424. d0, d1, d2, d3);
  425. SHF_W4_SP(d0, d1, d2, d3, d0, d1, d2, d3, SHF_177);
  426. FMADD4(x0, x1, x2, x3, da_r_vec, d0, d1, d2, d3);
  427. ST_SP4_INC(d0, d1, d2, d3, x, 4);
  428. }
  429. if (n & 4)
  430. {
  431. LD_SP2_INC(px, 4, x0, x1);
  432. MUL2(x0, da_i_vec, x1, da_i_vec, d0, d1);
  433. SHF_W2_SP(d0, d1, d0, d1, SHF_177);
  434. FMADD2(x0, x1, da_r_vec, d0, d1);
  435. ST_SP2_INC(d0, d1, x, 4);
  436. }
  437. if (n & 2)
  438. {
  439. LD_GP4_INC(px, 1, f0, f1, f2, f3);
  440. tp0 = da_r * f0;
  441. tp0 -= da_i * f1;
  442. tp1 = da_r * f1;
  443. tp1 += da_i * f0;
  444. tp2 = da_r * f2;
  445. tp2 -= da_i * f3;
  446. tp3 = da_r * f3;
  447. tp3 += da_i * f2;
  448. ST_GP4_INC(tp0, tp1, tp2, tp3, x, 1);
  449. }
  450. if (n & 1)
  451. {
  452. LD_GP2_INC(px, 1, f0, f1);
  453. tp0 = da_r * f0;
  454. tp0 -= da_i * f1;
  455. tp1 = da_r * f1;
  456. tp1 += da_i * f0;
  457. ST_GP2_INC(tp0, tp1, x, 1);
  458. }
  459. }
  460. }
  461. }
  462. else
  463. {
  464. inc_x2 = 2 * inc_x;
  465. if ((0.0 == da_r) && (0.0 == da_i))
  466. {
  467. for (i = n; i--;)
  468. {
  469. *x = 0;
  470. *(x + 1) = 0;
  471. x += inc_x2;
  472. }
  473. }
  474. else if (0.0 == da_r)
  475. {
  476. da_i_vec = COPY_FLOAT_TO_VECTOR(da_i);
  477. da_i_vec_neg = -da_i_vec;
  478. da_i_vec = (v4f32) __msa_ilvev_w((v4i32) da_i_vec_neg, (v4i32) da_i_vec);
  479. for (i = (n >> 4); i--;)
  480. {
  481. LD_SP16_INC(px, inc_x2, x0, x1, x2, x3, x4, x5, x6, x7, x8, x9,
  482. x10, x11, x12, x13, x14, x15);
  483. PCKEV_D4_SP(x1, x0, x3, x2, x5, x4, x7, x6, d0, d1, d2, d3);
  484. PCKEV_D4_SP(x9, x8, x11, x10, x13, x12, x15, x14, d4, d5, d6, d7);
  485. MUL4(d0, da_i_vec, d1, da_i_vec, d2, da_i_vec, d3, da_i_vec,
  486. d0, d1, d2, d3);
  487. MUL4(d4, da_i_vec, d5, da_i_vec, d6, da_i_vec, d7, da_i_vec,
  488. d4, d5, d6, d7);
  489. *x = d0[1];
  490. *(x + 1) = d0[0];
  491. x += inc_x2;
  492. *x = d0[3];
  493. *(x + 1) = d0[2];
  494. x += inc_x2;
  495. *x = d1[1];
  496. *(x + 1) = d1[0];
  497. x += inc_x2;
  498. *x = d1[3];
  499. *(x + 1) = d1[2];
  500. x += inc_x2;
  501. *x = d2[1];
  502. *(x + 1) = d2[0];
  503. x += inc_x2;
  504. *x = d2[3];
  505. *(x + 1) = d2[2];
  506. x += inc_x2;
  507. *x = d3[1];
  508. *(x + 1) = d3[0];
  509. x += inc_x2;
  510. *x = d3[3];
  511. *(x + 1) = d3[2];
  512. x += inc_x2;
  513. *x = d4[1];
  514. *(x + 1) = d4[0];
  515. x += inc_x2;
  516. *x = d4[3];
  517. *(x + 1) = d4[2];
  518. x += inc_x2;
  519. *x = d5[1];
  520. *(x + 1) = d5[0];
  521. x += inc_x2;
  522. *x = d5[3];
  523. *(x + 1) = d5[2];
  524. x += inc_x2;
  525. *x = d6[1];
  526. *(x + 1) = d6[0];
  527. x += inc_x2;
  528. *x = d6[3];
  529. *(x + 1) = d6[2];
  530. x += inc_x2;
  531. *x = d7[1];
  532. *(x + 1) = d7[0];
  533. x += inc_x2;
  534. *x = d7[3];
  535. *(x + 1) = d7[2];
  536. x += inc_x2;
  537. }
  538. if (n & 15)
  539. {
  540. if (n & 8)
  541. {
  542. LD_SP8_INC(px, inc_x2, x0, x1, x2, x3, x4, x5, x6, x7);
  543. PCKEV_D4_SP(x1, x0, x3, x2, x5, x4, x7, x6, d0, d1, d2, d3);
  544. MUL4(d0, da_i_vec, d1, da_i_vec, d2, da_i_vec, d3, da_i_vec,
  545. d0, d1, d2, d3);
  546. *x = d0[1];
  547. *(x + 1) = d0[0];
  548. x += inc_x2;
  549. *x = d0[3];
  550. *(x + 1) = d0[2];
  551. x += inc_x2;
  552. *x = d1[1];
  553. *(x + 1) = d1[0];
  554. x += inc_x2;
  555. *x = d1[3];
  556. *(x + 1) = d1[2];
  557. x += inc_x2;
  558. *x = d2[1];
  559. *(x + 1) = d2[0];
  560. x += inc_x2;
  561. *x = d2[3];
  562. *(x + 1) = d2[2];
  563. x += inc_x2;
  564. *x = d3[1];
  565. *(x + 1) = d3[0];
  566. x += inc_x2;
  567. *x = d3[3];
  568. *(x + 1) = d3[2];
  569. x += inc_x2;
  570. }
  571. if (n & 4)
  572. {
  573. LD_SP4_INC(px, inc_x2, x0, x1, x2, x3);
  574. PCKEV_D2_SP(x1, x0, x3, x2, d0, d1);
  575. MUL2(d0, da_i_vec, d1, da_i_vec, d0, d1);
  576. *x = d0[1];
  577. *(x + 1) = d0[0];
  578. x += inc_x2;
  579. *x = d0[3];
  580. *(x + 1) = d0[2];
  581. x += inc_x2;
  582. *x = d1[1];
  583. *(x + 1) = d1[0];
  584. x += inc_x2;
  585. *x = d1[3];
  586. *(x + 1) = d1[2];
  587. x += inc_x2;
  588. }
  589. if (n & 2)
  590. {
  591. f0 = *px;
  592. f1 = *(px + 1);
  593. px += inc_x2;
  594. f2 = *px;
  595. f3 = *(px + 1);
  596. px += inc_x2;
  597. MUL4(f0, da_i, f1, -da_i, f2, da_i, f3, -da_i, f0, f1, f2, f3);
  598. *x = f1;
  599. *(x + 1) = f0;
  600. x += inc_x2;
  601. *x = f3;
  602. *(x + 1) = f2;
  603. x += inc_x2;
  604. }
  605. if (n & 1)
  606. {
  607. f0 = *x;
  608. f1 = *(x + 1);
  609. MUL2(f0, da_i, f1, -da_i, f0, f1);
  610. *x = f1;
  611. *(x + 1) = f0;
  612. }
  613. }
  614. }
  615. else if (0.0 == da_i)
  616. {
  617. da_r_vec = COPY_FLOAT_TO_VECTOR(da_r);
  618. for (i = (n >> 4); i--;)
  619. {
  620. LD_SP16_INC(px, inc_x2, x0, x1, x2, x3, x4, x5, x6, x7, x8, x9,
  621. x10, x11, x12, x13, x14, x15);
  622. PCKEV_D4_SP(x1, x0, x3, x2, x5, x4, x7, x6, d0, d1, d2, d3);
  623. PCKEV_D4_SP(x9, x8, x11, x10, x13, x12, x15, x14, d4, d5, d6, d7);
  624. MUL4(d0, da_r_vec, d1, da_r_vec, d2, da_r_vec, d3, da_r_vec,
  625. d0, d1, d2, d3);
  626. MUL4(d4, da_r_vec, d5, da_r_vec, d6, da_r_vec, d7, da_r_vec,
  627. d4, d5, d6, d7);
  628. *x = d0[0];
  629. *(x + 1) = d0[1];
  630. x += inc_x2;
  631. *x = d0[2];
  632. *(x + 1) = d0[3];
  633. x += inc_x2;
  634. *x = d1[0];
  635. *(x + 1) = d1[1];
  636. x += inc_x2;
  637. *x = d1[2];
  638. *(x + 1) = d1[3];
  639. x += inc_x2;
  640. *x = d2[0];
  641. *(x + 1) = d2[1];
  642. x += inc_x2;
  643. *x = d2[2];
  644. *(x + 1) = d2[3];
  645. x += inc_x2;
  646. *x = d3[0];
  647. *(x + 1) = d3[1];
  648. x += inc_x2;
  649. *x = d3[2];
  650. *(x + 1) = d3[3];
  651. x += inc_x2;
  652. *x = d4[0];
  653. *(x + 1) = d4[1];
  654. x += inc_x2;
  655. *x = d4[2];
  656. *(x + 1) = d4[3];
  657. x += inc_x2;
  658. *x = d5[0];
  659. *(x + 1) = d5[1];
  660. x += inc_x2;
  661. *x = d5[2];
  662. *(x + 1) = d5[3];
  663. x += inc_x2;
  664. *x = d6[0];
  665. *(x + 1) = d6[1];
  666. x += inc_x2;
  667. *x = d6[2];
  668. *(x + 1) = d6[3];
  669. x += inc_x2;
  670. *x = d7[0];
  671. *(x + 1) = d7[1];
  672. x += inc_x2;
  673. *x = d7[2];
  674. *(x + 1) = d7[3];
  675. x += inc_x2;
  676. }
  677. if (n & 15)
  678. {
  679. if (n & 8)
  680. {
  681. LD_SP8_INC(px, inc_x2, x0, x1, x2, x3, x4, x5, x6, x7);
  682. PCKEV_D4_SP(x1, x0, x3, x2, x5, x4, x7, x6, d0, d1, d2, d3);
  683. MUL4(d0, da_r_vec, d1, da_r_vec, d2, da_r_vec, d3, da_r_vec,
  684. d0, d1, d2, d3);
  685. *x = d0[0];
  686. *(x + 1) = d0[1];
  687. x += inc_x2;
  688. *x = d0[2];
  689. *(x + 1) = d0[3];
  690. x += inc_x2;
  691. *x = d1[0];
  692. *(x + 1) = d1[1];
  693. x += inc_x2;
  694. *x = d1[2];
  695. *(x + 1) = d1[3];
  696. x += inc_x2;
  697. *x = d2[0];
  698. *(x + 1) = d2[1];
  699. x += inc_x2;
  700. *x = d2[2];
  701. *(x + 1) = d2[3];
  702. x += inc_x2;
  703. *x = d3[0];
  704. *(x + 1) = d3[1];
  705. x += inc_x2;
  706. *x = d3[2];
  707. *(x + 1) = d3[3];
  708. x += inc_x2;
  709. }
  710. if (n & 4)
  711. {
  712. LD_SP4_INC(px, inc_x2, x0, x1, x2, x3);
  713. PCKEV_D2_SP(x1, x0, x3, x2, d0, d1);
  714. MUL2(d0, da_r_vec, d1, da_r_vec, d0, d1);
  715. *x = d0[0];
  716. *(x + 1) = d0[1];
  717. x += inc_x2;
  718. *x = d0[2];
  719. *(x + 1) = d0[3];
  720. x += inc_x2;
  721. *x = d1[0];
  722. *(x + 1) = d1[1];
  723. x += inc_x2;
  724. *x = d1[2];
  725. *(x + 1) = d1[3];
  726. x += inc_x2;
  727. }
  728. if (n & 2)
  729. {
  730. f0 = *px;
  731. f1 = *(px + 1);
  732. px += inc_x2;
  733. f2 = *px;
  734. f3 = *(px + 1);
  735. px += inc_x2;
  736. MUL4(f0, da_r, f1, da_r, f2, da_r, f3, da_r, f0, f1, f2, f3);
  737. *x = f0;
  738. *(x + 1) = f1;
  739. x += inc_x2;
  740. *x = f2;
  741. *(x + 1) = f3;
  742. x += inc_x2;
  743. }
  744. if (n & 1)
  745. {
  746. f0 = *x;
  747. f1 = *(x + 1);
  748. MUL2(f0, da_r, f1, da_r, f0, f1);
  749. *x = f0;
  750. *(x + 1) = f1;
  751. }
  752. }
  753. }
  754. else
  755. {
  756. da_i_vec = COPY_FLOAT_TO_VECTOR(da_i);
  757. da_i_vec_neg = -da_i_vec;
  758. da_i_vec = (v4f32) __msa_ilvev_w((v4i32) da_i_vec_neg, (v4i32) da_i_vec);
  759. da_r_vec = COPY_FLOAT_TO_VECTOR(da_r);
  760. for (i = (n >> 4); i--;)
  761. {
  762. LD_SP16_INC(px, inc_x2, x0, x1, x2, x3, x4, x5, x6, x7, x8, x9,
  763. x10, x11, x12, x13, x14, x15);
  764. PCKEV_D4_SP(x1, x0, x3, x2, x5, x4, x7, x6, d0, d1, d2, d3);
  765. PCKEV_D4_SP(x9, x8, x11, x10, x13, x12, x15, x14, d4, d5, d6, d7);
  766. MUL4(d0, da_i_vec, d1, da_i_vec, d2, da_i_vec, d3, da_i_vec,
  767. x0, x1, x2, x3);
  768. MUL4(d4, da_i_vec, d5, da_i_vec, d6, da_i_vec, d7, da_i_vec,
  769. x4, x5, x6, x7);
  770. MUL4(d0, da_r_vec, d1, da_r_vec, d2, da_r_vec, d3, da_r_vec,
  771. d0, d1, d2, d3);
  772. MUL4(d4, da_r_vec, d5, da_r_vec, d6, da_r_vec, d7, da_r_vec,
  773. d4, d5, d6, d7);
  774. SHF_W4_SP(x0, x1, x2, x3, x0, x1, x2, x3, SHF_177);
  775. SHF_W4_SP(x4, x5, x6, x7, x4, x5, x6, x7, SHF_177);
  776. ADD4(d0, x0, d1, x1, d2, x2, d3, x3, d0, d1, d2, d3);
  777. ADD4(d4, x4, d5, x5, d6, x6, d7, x7, d4, d5, d6, d7);
  778. *x = d0[0];
  779. *(x + 1) = d0[1];
  780. x += inc_x2;
  781. *x = d0[2];
  782. *(x + 1) = d0[3];
  783. x += inc_x2;
  784. *x = d1[0];
  785. *(x + 1) = d1[1];
  786. x += inc_x2;
  787. *x = d1[2];
  788. *(x + 1) = d1[3];
  789. x += inc_x2;
  790. *x = d2[0];
  791. *(x + 1) = d2[1];
  792. x += inc_x2;
  793. *x = d2[2];
  794. *(x + 1) = d2[3];
  795. x += inc_x2;
  796. *x = d3[0];
  797. *(x + 1) = d3[1];
  798. x += inc_x2;
  799. *x = d3[2];
  800. *(x + 1) = d3[3];
  801. x += inc_x2;
  802. *x = d4[0];
  803. *(x + 1) = d4[1];
  804. x += inc_x2;
  805. *x = d4[2];
  806. *(x + 1) = d4[3];
  807. x += inc_x2;
  808. *x = d5[0];
  809. *(x + 1) = d5[1];
  810. x += inc_x2;
  811. *x = d5[2];
  812. *(x + 1) = d5[3];
  813. x += inc_x2;
  814. *x = d6[0];
  815. *(x + 1) = d6[1];
  816. x += inc_x2;
  817. *x = d6[2];
  818. *(x + 1) = d6[3];
  819. x += inc_x2;
  820. *x = d7[0];
  821. *(x + 1) = d7[1];
  822. x += inc_x2;
  823. *x = d7[2];
  824. *(x + 1) = d7[3];
  825. x += inc_x2;
  826. }
  827. if (n & 15)
  828. {
  829. if (n & 8)
  830. {
  831. LD_SP8_INC(px, inc_x2, x0, x1, x2, x3, x4, x5, x6, x7);
  832. PCKEV_D4_SP(x1, x0, x3, x2, x5, x4, x7, x6, d0, d1, d2, d3);
  833. MUL4(d0, da_i_vec, d1, da_i_vec, d2, da_i_vec, d3, da_i_vec,
  834. x0, x1, x2, x3);
  835. MUL4(d0, da_r_vec, d1, da_r_vec, d2, da_r_vec, d3, da_r_vec,
  836. d0, d1, d2, d3);
  837. SHF_W4_SP(x0, x1, x2, x3, x0, x1, x2, x3, SHF_177);
  838. ADD4(d0, x0, d1, x1, d2, x2, d3, x3, d0, d1, d2, d3);
  839. *x = d0[0];
  840. *(x + 1) = d0[1];
  841. x += inc_x2;
  842. *x = d0[2];
  843. *(x + 1) = d0[3];
  844. x += inc_x2;
  845. *x = d1[0];
  846. *(x + 1) = d1[1];
  847. x += inc_x2;
  848. *x = d1[2];
  849. *(x + 1) = d1[3];
  850. x += inc_x2;
  851. *x = d2[0];
  852. *(x + 1) = d2[1];
  853. x += inc_x2;
  854. *x = d2[2];
  855. *(x + 1) = d2[3];
  856. x += inc_x2;
  857. *x = d3[0];
  858. *(x + 1) = d3[1];
  859. x += inc_x2;
  860. *x = d3[2];
  861. *(x + 1) = d3[3];
  862. x += inc_x2;
  863. }
  864. if (n & 4)
  865. {
  866. LD_SP4_INC(px, inc_x2, x0, x1, x2, x3);
  867. PCKEV_D2_SP(x1, x0, x3, x2, d0, d1);
  868. MUL2(d0, da_i_vec, d1, da_i_vec, x0, x1);
  869. MUL2(d0, da_r_vec, d1, da_r_vec, d0, d1);
  870. SHF_W2_SP(x0, x1, x0, x1, SHF_177);
  871. ADD2(d0, x0, d1, x1, d0, d1);
  872. *x = d0[0];
  873. *(x + 1) = d0[1];
  874. x += inc_x2;
  875. *x = d0[2];
  876. *(x + 1) = d0[3];
  877. x += inc_x2;
  878. *x = d1[0];
  879. *(x + 1) = d1[1];
  880. x += inc_x2;
  881. *x = d1[2];
  882. *(x + 1) = d1[3];
  883. x += inc_x2;
  884. }
  885. if (n & 2)
  886. {
  887. f0 = *px;;
  888. f1 = *(px + 1);
  889. px += inc_x2;
  890. f2 = *px;
  891. f3 = *(px + 1);
  892. px += inc_x2;
  893. tp0 = da_r * f0;
  894. tp0 -= da_i * f1;
  895. tp1 = da_r * f1;
  896. tp1 += da_i * f0;
  897. tp2 = da_r * f2;
  898. tp2 -= da_i * f3;
  899. tp3 = da_r * f3;
  900. tp3 += da_i * f2;
  901. *x = tp0;
  902. *(x + 1) = tp1;
  903. x += inc_x2;
  904. *x = tp2;
  905. *(x + 1) = tp3;
  906. x += inc_x2;
  907. }
  908. if (n & 1)
  909. {
  910. f0 = *px; px += 1;
  911. f1 = *px;
  912. tp0 = da_r * f0;
  913. tp0 -= da_i * f1;
  914. tp1 = da_r * f1;
  915. tp1 += da_i * f0;
  916. *x = tp0; x += 1;
  917. *x = tp1;
  918. }
  919. }
  920. }
  921. }
  922. return (0);
  923. }