InstEmitSimdHelper.cs 61 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476147714781479148014811482148314841485148614871488148914901491149214931494149514961497149814991500150115021503150415051506150715081509151015111512151315141515151615171518151915201521152215231524152515261527152815291530153115321533153415351536153715381539154015411542154315441545154615471548154915501551155215531554155515561557155815591560156115621563156415651566156715681569157015711572157315741575157615771578157915801581158215831584158515861587158815891590159115921593159415951596159715981599160016011602160316041605160616071608160916101611161216131614161516161617161816191620162116221623162416251626162716281629163016311632163316341635163616371638163916401641164216431644164516461647164816491650165116521653165416551656165716581659166016611662166316641665166616671668166916701671167216731674167516761677167816791680168116821683168416851686168716881689169016911692169316941695169616971698169917001701170217031704
  1. using ARMeilleure.Decoders;
  2. using ARMeilleure.IntermediateRepresentation;
  3. using ARMeilleure.State;
  4. using ARMeilleure.Translation;
  5. using System;
  6. using System.Diagnostics;
  7. using System.Reflection;
  8. using static ARMeilleure.Instructions.InstEmitHelper;
  9. using static ARMeilleure.IntermediateRepresentation.OperandHelper;
  10. namespace ARMeilleure.Instructions
  11. {
  12. using Func1I = Func<Operand, Operand>;
  13. using Func2I = Func<Operand, Operand, Operand>;
  14. using Func3I = Func<Operand, Operand, Operand, Operand>;
  15. static class InstEmitSimdHelper
  16. {
  17. #region "Masks"
  18. public static readonly long[] EvenMasks = new long[]
  19. {
  20. 14L << 56 | 12L << 48 | 10L << 40 | 08L << 32 | 06L << 24 | 04L << 16 | 02L << 8 | 00L << 0, // B
  21. 13L << 56 | 12L << 48 | 09L << 40 | 08L << 32 | 05L << 24 | 04L << 16 | 01L << 8 | 00L << 0, // H
  22. 11L << 56 | 10L << 48 | 09L << 40 | 08L << 32 | 03L << 24 | 02L << 16 | 01L << 8 | 00L << 0 // S
  23. };
  24. public static readonly long[] OddMasks = new long[]
  25. {
  26. 15L << 56 | 13L << 48 | 11L << 40 | 09L << 32 | 07L << 24 | 05L << 16 | 03L << 8 | 01L << 0, // B
  27. 15L << 56 | 14L << 48 | 11L << 40 | 10L << 32 | 07L << 24 | 06L << 16 | 03L << 8 | 02L << 0, // H
  28. 15L << 56 | 14L << 48 | 13L << 40 | 12L << 32 | 07L << 24 | 06L << 16 | 05L << 8 | 04L << 0 // S
  29. };
  30. public static readonly long ZeroMask = 128L << 56 | 128L << 48 | 128L << 40 | 128L << 32 | 128L << 24 | 128L << 16 | 128L << 8 | 128L << 0;
  31. #endregion
  32. #region "X86 SSE Intrinsics"
  33. public static readonly Intrinsic[] X86PaddInstruction = new Intrinsic[]
  34. {
  35. Intrinsic.X86Paddb,
  36. Intrinsic.X86Paddw,
  37. Intrinsic.X86Paddd,
  38. Intrinsic.X86Paddq
  39. };
  40. public static readonly Intrinsic[] X86PcmpeqInstruction = new Intrinsic[]
  41. {
  42. Intrinsic.X86Pcmpeqb,
  43. Intrinsic.X86Pcmpeqw,
  44. Intrinsic.X86Pcmpeqd,
  45. Intrinsic.X86Pcmpeqq
  46. };
  47. public static readonly Intrinsic[] X86PcmpgtInstruction = new Intrinsic[]
  48. {
  49. Intrinsic.X86Pcmpgtb,
  50. Intrinsic.X86Pcmpgtw,
  51. Intrinsic.X86Pcmpgtd,
  52. Intrinsic.X86Pcmpgtq
  53. };
  54. public static readonly Intrinsic[] X86PmaxsInstruction = new Intrinsic[]
  55. {
  56. Intrinsic.X86Pmaxsb,
  57. Intrinsic.X86Pmaxsw,
  58. Intrinsic.X86Pmaxsd
  59. };
  60. public static readonly Intrinsic[] X86PmaxuInstruction = new Intrinsic[]
  61. {
  62. Intrinsic.X86Pmaxub,
  63. Intrinsic.X86Pmaxuw,
  64. Intrinsic.X86Pmaxud
  65. };
  66. public static readonly Intrinsic[] X86PminsInstruction = new Intrinsic[]
  67. {
  68. Intrinsic.X86Pminsb,
  69. Intrinsic.X86Pminsw,
  70. Intrinsic.X86Pminsd
  71. };
  72. public static readonly Intrinsic[] X86PminuInstruction = new Intrinsic[]
  73. {
  74. Intrinsic.X86Pminub,
  75. Intrinsic.X86Pminuw,
  76. Intrinsic.X86Pminud
  77. };
  78. public static readonly Intrinsic[] X86PmovsxInstruction = new Intrinsic[]
  79. {
  80. Intrinsic.X86Pmovsxbw,
  81. Intrinsic.X86Pmovsxwd,
  82. Intrinsic.X86Pmovsxdq
  83. };
  84. public static readonly Intrinsic[] X86PmovzxInstruction = new Intrinsic[]
  85. {
  86. Intrinsic.X86Pmovzxbw,
  87. Intrinsic.X86Pmovzxwd,
  88. Intrinsic.X86Pmovzxdq
  89. };
  90. public static readonly Intrinsic[] X86PsllInstruction = new Intrinsic[]
  91. {
  92. 0,
  93. Intrinsic.X86Psllw,
  94. Intrinsic.X86Pslld,
  95. Intrinsic.X86Psllq
  96. };
  97. public static readonly Intrinsic[] X86PsraInstruction = new Intrinsic[]
  98. {
  99. 0,
  100. Intrinsic.X86Psraw,
  101. Intrinsic.X86Psrad
  102. };
  103. public static readonly Intrinsic[] X86PsrlInstruction = new Intrinsic[]
  104. {
  105. 0,
  106. Intrinsic.X86Psrlw,
  107. Intrinsic.X86Psrld,
  108. Intrinsic.X86Psrlq
  109. };
  110. public static readonly Intrinsic[] X86PsubInstruction = new Intrinsic[]
  111. {
  112. Intrinsic.X86Psubb,
  113. Intrinsic.X86Psubw,
  114. Intrinsic.X86Psubd,
  115. Intrinsic.X86Psubq
  116. };
  117. public static readonly Intrinsic[] X86PunpckhInstruction = new Intrinsic[]
  118. {
  119. Intrinsic.X86Punpckhbw,
  120. Intrinsic.X86Punpckhwd,
  121. Intrinsic.X86Punpckhdq,
  122. Intrinsic.X86Punpckhqdq
  123. };
  124. public static readonly Intrinsic[] X86PunpcklInstruction = new Intrinsic[]
  125. {
  126. Intrinsic.X86Punpcklbw,
  127. Intrinsic.X86Punpcklwd,
  128. Intrinsic.X86Punpckldq,
  129. Intrinsic.X86Punpcklqdq
  130. };
  131. #endregion
  132. public static int GetImmShl(OpCodeSimdShImm op)
  133. {
  134. return op.Imm - (8 << op.Size);
  135. }
  136. public static int GetImmShr(OpCodeSimdShImm op)
  137. {
  138. return (8 << (op.Size + 1)) - op.Imm;
  139. }
  140. public static Operand X86GetScalar(ArmEmitterContext context, float value)
  141. {
  142. return X86GetScalar(context, BitConverter.SingleToInt32Bits(value));
  143. }
  144. public static Operand X86GetScalar(ArmEmitterContext context, double value)
  145. {
  146. return X86GetScalar(context, BitConverter.DoubleToInt64Bits(value));
  147. }
  148. public static Operand X86GetScalar(ArmEmitterContext context, int value)
  149. {
  150. return context.VectorCreateScalar(Const(value));
  151. }
  152. public static Operand X86GetScalar(ArmEmitterContext context, long value)
  153. {
  154. return context.VectorCreateScalar(Const(value));
  155. }
  156. public static Operand X86GetAllElements(ArmEmitterContext context, float value)
  157. {
  158. return X86GetAllElements(context, BitConverter.SingleToInt32Bits(value));
  159. }
  160. public static Operand X86GetAllElements(ArmEmitterContext context, double value)
  161. {
  162. return X86GetAllElements(context, BitConverter.DoubleToInt64Bits(value));
  163. }
  164. public static Operand X86GetAllElements(ArmEmitterContext context, short value)
  165. {
  166. ulong value1 = (ushort)value;
  167. ulong value2 = value1 << 16 | value1;
  168. ulong value4 = value2 << 32 | value2;
  169. return X86GetAllElements(context, (long)value4);
  170. }
  171. public static Operand X86GetAllElements(ArmEmitterContext context, int value)
  172. {
  173. Operand vector = context.VectorCreateScalar(Const(value));
  174. vector = context.AddIntrinsic(Intrinsic.X86Shufps, vector, vector, Const(0));
  175. return vector;
  176. }
  177. public static Operand X86GetAllElements(ArmEmitterContext context, long value)
  178. {
  179. Operand vector = context.VectorCreateScalar(Const(value));
  180. vector = context.AddIntrinsic(Intrinsic.X86Movlhps, vector, vector);
  181. return vector;
  182. }
  183. public static Operand X86GetElements(ArmEmitterContext context, long e1, long e0)
  184. {
  185. return X86GetElements(context, (ulong)e1, (ulong)e0);
  186. }
  187. public static Operand X86GetElements(ArmEmitterContext context, ulong e1, ulong e0)
  188. {
  189. Operand vector0 = context.VectorCreateScalar(Const(e0));
  190. Operand vector1 = context.VectorCreateScalar(Const(e1));
  191. return context.AddIntrinsic(Intrinsic.X86Punpcklqdq, vector0, vector1);
  192. }
  193. public static int X86GetRoundControl(FPRoundingMode roundMode)
  194. {
  195. switch (roundMode)
  196. {
  197. case FPRoundingMode.ToNearest: return 8 | 0; // even
  198. case FPRoundingMode.TowardsPlusInfinity: return 8 | 2;
  199. case FPRoundingMode.TowardsMinusInfinity: return 8 | 1;
  200. case FPRoundingMode.TowardsZero: return 8 | 3;
  201. }
  202. throw new ArgumentException($"Invalid rounding mode \"{roundMode}\".");
  203. }
  204. public static Operand EmitCountSetBits8(ArmEmitterContext context, Operand op) // "size" is 8 (SIMD&FP Inst.).
  205. {
  206. Debug.Assert(op.Type == OperandType.I32 || op.Type == OperandType.I64);
  207. Operand op0 = context.Subtract(op, context.BitwiseAnd(context.ShiftRightUI(op, Const(1)), Const(op.Type, 0x55L)));
  208. Operand c1 = Const(op.Type, 0x33L);
  209. Operand op1 = context.Add(context.BitwiseAnd(context.ShiftRightUI(op0, Const(2)), c1), context.BitwiseAnd(op0, c1));
  210. return context.BitwiseAnd(context.Add(op1, context.ShiftRightUI(op1, Const(4))), Const(op.Type, 0x0fL));
  211. }
  212. public static void EmitScalarUnaryOpF(ArmEmitterContext context, Intrinsic inst32, Intrinsic inst64)
  213. {
  214. OpCodeSimd op = (OpCodeSimd)context.CurrOp;
  215. Operand n = GetVec(op.Rn);
  216. Intrinsic inst = (op.Size & 1) != 0 ? inst64 : inst32;
  217. Operand res = context.AddIntrinsic(inst, n);
  218. if ((op.Size & 1) != 0)
  219. {
  220. res = context.VectorZeroUpper64(res);
  221. }
  222. else
  223. {
  224. res = context.VectorZeroUpper96(res);
  225. }
  226. context.Copy(GetVec(op.Rd), res);
  227. }
  228. public static void EmitScalarBinaryOpF(ArmEmitterContext context, Intrinsic inst32, Intrinsic inst64)
  229. {
  230. OpCodeSimdReg op = (OpCodeSimdReg)context.CurrOp;
  231. Operand n = GetVec(op.Rn);
  232. Operand m = GetVec(op.Rm);
  233. Intrinsic inst = (op.Size & 1) != 0 ? inst64 : inst32;
  234. Operand res = context.AddIntrinsic(inst, n, m);
  235. if ((op.Size & 1) != 0)
  236. {
  237. res = context.VectorZeroUpper64(res);
  238. }
  239. else
  240. {
  241. res = context.VectorZeroUpper96(res);
  242. }
  243. context.Copy(GetVec(op.Rd), res);
  244. }
  245. public static void EmitVectorUnaryOpF(ArmEmitterContext context, Intrinsic inst32, Intrinsic inst64)
  246. {
  247. OpCodeSimd op = (OpCodeSimd)context.CurrOp;
  248. Operand n = GetVec(op.Rn);
  249. Intrinsic inst = (op.Size & 1) != 0 ? inst64 : inst32;
  250. Operand res = context.AddIntrinsic(inst, n);
  251. if (op.RegisterSize == RegisterSize.Simd64)
  252. {
  253. res = context.VectorZeroUpper64(res);
  254. }
  255. context.Copy(GetVec(op.Rd), res);
  256. }
  257. public static void EmitVectorBinaryOpF(ArmEmitterContext context, Intrinsic inst32, Intrinsic inst64)
  258. {
  259. OpCodeSimdReg op = (OpCodeSimdReg)context.CurrOp;
  260. Operand n = GetVec(op.Rn);
  261. Operand m = GetVec(op.Rm);
  262. Intrinsic inst = (op.Size & 1) != 0 ? inst64 : inst32;
  263. Operand res = context.AddIntrinsic(inst, n, m);
  264. if (op.RegisterSize == RegisterSize.Simd64)
  265. {
  266. res = context.VectorZeroUpper64(res);
  267. }
  268. context.Copy(GetVec(op.Rd), res);
  269. }
  270. public static Operand EmitUnaryMathCall(ArmEmitterContext context, string name, Operand n)
  271. {
  272. IOpCodeSimd op = (IOpCodeSimd)context.CurrOp;
  273. MethodInfo info = (op.Size & 1) == 0
  274. ? typeof(MathF).GetMethod(name, new Type[] { typeof(float) })
  275. : typeof(Math). GetMethod(name, new Type[] { typeof(double) });
  276. return context.Call(info, n);
  277. }
  278. public static Operand EmitRoundMathCall(ArmEmitterContext context, MidpointRounding roundMode, Operand n)
  279. {
  280. IOpCodeSimd op = (IOpCodeSimd)context.CurrOp;
  281. string name = nameof(Math.Round);
  282. MethodInfo info = (op.Size & 1) == 0
  283. ? typeof(MathF).GetMethod(name, new Type[] { typeof(float), typeof(MidpointRounding) })
  284. : typeof(Math). GetMethod(name, new Type[] { typeof(double), typeof(MidpointRounding) });
  285. return context.Call(info, n, Const((int)roundMode));
  286. }
  287. public static Operand EmitSoftFloatCall(ArmEmitterContext context, string name, params Operand[] callArgs)
  288. {
  289. IOpCodeSimd op = (IOpCodeSimd)context.CurrOp;
  290. MethodInfo info = (op.Size & 1) == 0
  291. ? typeof(SoftFloat32).GetMethod(name)
  292. : typeof(SoftFloat64).GetMethod(name);
  293. return context.Call(info, callArgs);
  294. }
  295. public static void EmitScalarBinaryOpByElemF(ArmEmitterContext context, Func2I emit)
  296. {
  297. OpCodeSimdRegElemF op = (OpCodeSimdRegElemF)context.CurrOp;
  298. OperandType type = (op.Size & 1) != 0 ? OperandType.FP64 : OperandType.FP32;
  299. Operand n = context.VectorExtract(type, GetVec(op.Rn), 0);
  300. Operand m = context.VectorExtract(type, GetVec(op.Rm), op.Index);
  301. context.Copy(GetVec(op.Rd), context.VectorInsert(context.VectorZero(), emit(n, m), 0));
  302. }
  303. public static void EmitScalarTernaryOpByElemF(ArmEmitterContext context, Func3I emit)
  304. {
  305. OpCodeSimdRegElemF op = (OpCodeSimdRegElemF)context.CurrOp;
  306. OperandType type = (op.Size & 1) != 0 ? OperandType.FP64 : OperandType.FP32;
  307. Operand d = context.VectorExtract(type, GetVec(op.Rd), 0);
  308. Operand n = context.VectorExtract(type, GetVec(op.Rn), 0);
  309. Operand m = context.VectorExtract(type, GetVec(op.Rm), op.Index);
  310. context.Copy(GetVec(op.Rd), context.VectorInsert(context.VectorZero(), emit(d, n, m), 0));
  311. }
  312. public static void EmitScalarUnaryOpSx(ArmEmitterContext context, Func1I emit)
  313. {
  314. OpCodeSimd op = (OpCodeSimd)context.CurrOp;
  315. Operand n = EmitVectorExtractSx(context, op.Rn, 0, op.Size);
  316. Operand d = EmitVectorInsert(context, context.VectorZero(), emit(n), 0, op.Size);
  317. context.Copy(GetVec(op.Rd), d);
  318. }
  319. public static void EmitScalarBinaryOpSx(ArmEmitterContext context, Func2I emit)
  320. {
  321. OpCodeSimdReg op = (OpCodeSimdReg)context.CurrOp;
  322. Operand n = EmitVectorExtractSx(context, op.Rn, 0, op.Size);
  323. Operand m = EmitVectorExtractSx(context, op.Rm, 0, op.Size);
  324. Operand d = EmitVectorInsert(context, context.VectorZero(), emit(n, m), 0, op.Size);
  325. context.Copy(GetVec(op.Rd), d);
  326. }
  327. public static void EmitScalarUnaryOpZx(ArmEmitterContext context, Func1I emit)
  328. {
  329. OpCodeSimd op = (OpCodeSimd)context.CurrOp;
  330. Operand n = EmitVectorExtractZx(context, op.Rn, 0, op.Size);
  331. Operand d = EmitVectorInsert(context, context.VectorZero(), emit(n), 0, op.Size);
  332. context.Copy(GetVec(op.Rd), d);
  333. }
  334. public static void EmitScalarBinaryOpZx(ArmEmitterContext context, Func2I emit)
  335. {
  336. OpCodeSimdReg op = (OpCodeSimdReg)context.CurrOp;
  337. Operand n = EmitVectorExtractZx(context, op.Rn, 0, op.Size);
  338. Operand m = EmitVectorExtractZx(context, op.Rm, 0, op.Size);
  339. Operand d = EmitVectorInsert(context, context.VectorZero(), emit(n, m), 0, op.Size);
  340. context.Copy(GetVec(op.Rd), d);
  341. }
  342. public static void EmitScalarTernaryOpZx(ArmEmitterContext context, Func3I emit)
  343. {
  344. OpCodeSimdReg op = (OpCodeSimdReg)context.CurrOp;
  345. Operand d = EmitVectorExtractZx(context, op.Rd, 0, op.Size);
  346. Operand n = EmitVectorExtractZx(context, op.Rn, 0, op.Size);
  347. Operand m = EmitVectorExtractZx(context, op.Rm, 0, op.Size);
  348. d = EmitVectorInsert(context, context.VectorZero(), emit(d, n, m), 0, op.Size);
  349. context.Copy(GetVec(op.Rd), d);
  350. }
  351. public static void EmitScalarUnaryOpF(ArmEmitterContext context, Func1I emit)
  352. {
  353. OpCodeSimd op = (OpCodeSimd)context.CurrOp;
  354. OperandType type = (op.Size & 1) != 0 ? OperandType.FP64 : OperandType.FP32;
  355. Operand n = context.VectorExtract(type, GetVec(op.Rn), 0);
  356. context.Copy(GetVec(op.Rd), context.VectorInsert(context.VectorZero(), emit(n), 0));
  357. }
  358. public static void EmitScalarBinaryOpF(ArmEmitterContext context, Func2I emit)
  359. {
  360. OpCodeSimdReg op = (OpCodeSimdReg)context.CurrOp;
  361. OperandType type = (op.Size & 1) != 0 ? OperandType.FP64 : OperandType.FP32;
  362. Operand n = context.VectorExtract(type, GetVec(op.Rn), 0);
  363. Operand m = context.VectorExtract(type, GetVec(op.Rm), 0);
  364. context.Copy(GetVec(op.Rd), context.VectorInsert(context.VectorZero(), emit(n, m), 0));
  365. }
  366. public static void EmitScalarTernaryRaOpF(ArmEmitterContext context, Func3I emit)
  367. {
  368. OpCodeSimdReg op = (OpCodeSimdReg)context.CurrOp;
  369. OperandType type = (op.Size & 1) != 0 ? OperandType.FP64 : OperandType.FP32;
  370. Operand a = context.VectorExtract(type, GetVec(op.Ra), 0);
  371. Operand n = context.VectorExtract(type, GetVec(op.Rn), 0);
  372. Operand m = context.VectorExtract(type, GetVec(op.Rm), 0);
  373. context.Copy(GetVec(op.Rd), context.VectorInsert(context.VectorZero(), emit(a, n, m), 0));
  374. }
  375. public static void EmitVectorUnaryOpF(ArmEmitterContext context, Func1I emit)
  376. {
  377. OpCodeSimd op = (OpCodeSimd)context.CurrOp;
  378. Operand res = context.VectorZero();
  379. int sizeF = op.Size & 1;
  380. OperandType type = sizeF != 0 ? OperandType.FP64 : OperandType.FP32;
  381. int elems = op.GetBytesCount() >> sizeF + 2;
  382. for (int index = 0; index < elems; index++)
  383. {
  384. Operand ne = context.VectorExtract(type, GetVec(op.Rn), index);
  385. res = context.VectorInsert(res, emit(ne), index);
  386. }
  387. context.Copy(GetVec(op.Rd), res);
  388. }
  389. public static void EmitVectorBinaryOpF(ArmEmitterContext context, Func2I emit)
  390. {
  391. OpCodeSimdReg op = (OpCodeSimdReg)context.CurrOp;
  392. Operand res = context.VectorZero();
  393. int sizeF = op.Size & 1;
  394. OperandType type = sizeF != 0 ? OperandType.FP64 : OperandType.FP32;
  395. int elems = op.GetBytesCount() >> sizeF + 2;
  396. for (int index = 0; index < elems; index++)
  397. {
  398. Operand ne = context.VectorExtract(type, GetVec(op.Rn), index);
  399. Operand me = context.VectorExtract(type, GetVec(op.Rm), index);
  400. res = context.VectorInsert(res, emit(ne, me), index);
  401. }
  402. context.Copy(GetVec(op.Rd), res);
  403. }
  404. public static void EmitVectorTernaryOpF(ArmEmitterContext context, Func3I emit)
  405. {
  406. OpCodeSimdReg op = (OpCodeSimdReg)context.CurrOp;
  407. Operand res = context.VectorZero();
  408. int sizeF = op.Size & 1;
  409. OperandType type = sizeF != 0 ? OperandType.FP64 : OperandType.FP32;
  410. int elems = op.GetBytesCount() >> sizeF + 2;
  411. for (int index = 0; index < elems; index++)
  412. {
  413. Operand de = context.VectorExtract(type, GetVec(op.Rd), index);
  414. Operand ne = context.VectorExtract(type, GetVec(op.Rn), index);
  415. Operand me = context.VectorExtract(type, GetVec(op.Rm), index);
  416. res = context.VectorInsert(res, emit(de, ne, me), index);
  417. }
  418. context.Copy(GetVec(op.Rd), res);
  419. }
  420. public static void EmitVectorBinaryOpByElemF(ArmEmitterContext context, Func2I emit)
  421. {
  422. OpCodeSimdRegElemF op = (OpCodeSimdRegElemF)context.CurrOp;
  423. Operand res = context.VectorZero();
  424. int sizeF = op.Size & 1;
  425. OperandType type = sizeF != 0 ? OperandType.FP64 : OperandType.FP32;
  426. int elems = op.GetBytesCount() >> sizeF + 2;
  427. for (int index = 0; index < elems; index++)
  428. {
  429. Operand ne = context.VectorExtract(type, GetVec(op.Rn), index);
  430. Operand me = context.VectorExtract(type, GetVec(op.Rm), op.Index);
  431. res = context.VectorInsert(res, emit(ne, me), index);
  432. }
  433. context.Copy(GetVec(op.Rd), res);
  434. }
  435. public static void EmitVectorTernaryOpByElemF(ArmEmitterContext context, Func3I emit)
  436. {
  437. OpCodeSimdRegElemF op = (OpCodeSimdRegElemF)context.CurrOp;
  438. Operand res = context.VectorZero();
  439. int sizeF = op.Size & 1;
  440. OperandType type = sizeF != 0 ? OperandType.FP64 : OperandType.FP32;
  441. int elems = op.GetBytesCount() >> sizeF + 2;
  442. for (int index = 0; index < elems; index++)
  443. {
  444. Operand de = context.VectorExtract(type, GetVec(op.Rd), index);
  445. Operand ne = context.VectorExtract(type, GetVec(op.Rn), index);
  446. Operand me = context.VectorExtract(type, GetVec(op.Rm), op.Index);
  447. res = context.VectorInsert(res, emit(de, ne, me), index);
  448. }
  449. context.Copy(GetVec(op.Rd), res);
  450. }
  451. public static void EmitVectorUnaryOpSx(ArmEmitterContext context, Func1I emit)
  452. {
  453. OpCodeSimd op = (OpCodeSimd)context.CurrOp;
  454. Operand res = context.VectorZero();
  455. int elems = op.GetBytesCount() >> op.Size;
  456. for (int index = 0; index < elems; index++)
  457. {
  458. Operand ne = EmitVectorExtractSx(context, op.Rn, index, op.Size);
  459. res = EmitVectorInsert(context, res, emit(ne), index, op.Size);
  460. }
  461. context.Copy(GetVec(op.Rd), res);
  462. }
  463. public static void EmitVectorBinaryOpSx(ArmEmitterContext context, Func2I emit)
  464. {
  465. OpCodeSimdReg op = (OpCodeSimdReg)context.CurrOp;
  466. Operand res = context.VectorZero();
  467. int elems = op.GetBytesCount() >> op.Size;
  468. for (int index = 0; index < elems; index++)
  469. {
  470. Operand ne = EmitVectorExtractSx(context, op.Rn, index, op.Size);
  471. Operand me = EmitVectorExtractSx(context, op.Rm, index, op.Size);
  472. res = EmitVectorInsert(context, res, emit(ne, me), index, op.Size);
  473. }
  474. context.Copy(GetVec(op.Rd), res);
  475. }
  476. public static void EmitVectorTernaryOpSx(ArmEmitterContext context, Func3I emit)
  477. {
  478. OpCodeSimdReg op = (OpCodeSimdReg)context.CurrOp;
  479. Operand res = context.VectorZero();
  480. int elems = op.GetBytesCount() >> op.Size;
  481. for (int index = 0; index < elems; index++)
  482. {
  483. Operand de = EmitVectorExtractSx(context, op.Rd, index, op.Size);
  484. Operand ne = EmitVectorExtractSx(context, op.Rn, index, op.Size);
  485. Operand me = EmitVectorExtractSx(context, op.Rm, index, op.Size);
  486. res = EmitVectorInsert(context, res, emit(de, ne, me), index, op.Size);
  487. }
  488. context.Copy(GetVec(op.Rd), res);
  489. }
  490. public static void EmitVectorUnaryOpZx(ArmEmitterContext context, Func1I emit)
  491. {
  492. OpCodeSimd op = (OpCodeSimd)context.CurrOp;
  493. Operand res = context.VectorZero();
  494. int elems = op.GetBytesCount() >> op.Size;
  495. for (int index = 0; index < elems; index++)
  496. {
  497. Operand ne = EmitVectorExtractZx(context, op.Rn, index, op.Size);
  498. res = EmitVectorInsert(context, res, emit(ne), index, op.Size);
  499. }
  500. context.Copy(GetVec(op.Rd), res);
  501. }
  502. public static void EmitVectorBinaryOpZx(ArmEmitterContext context, Func2I emit)
  503. {
  504. OpCodeSimdReg op = (OpCodeSimdReg)context.CurrOp;
  505. Operand res = context.VectorZero();
  506. int elems = op.GetBytesCount() >> op.Size;
  507. for (int index = 0; index < elems; index++)
  508. {
  509. Operand ne = EmitVectorExtractZx(context, op.Rn, index, op.Size);
  510. Operand me = EmitVectorExtractZx(context, op.Rm, index, op.Size);
  511. res = EmitVectorInsert(context, res, emit(ne, me), index, op.Size);
  512. }
  513. context.Copy(GetVec(op.Rd), res);
  514. }
  515. public static void EmitVectorTernaryOpZx(ArmEmitterContext context, Func3I emit)
  516. {
  517. OpCodeSimdReg op = (OpCodeSimdReg)context.CurrOp;
  518. Operand res = context.VectorZero();
  519. int elems = op.GetBytesCount() >> op.Size;
  520. for (int index = 0; index < elems; index++)
  521. {
  522. Operand de = EmitVectorExtractZx(context, op.Rd, index, op.Size);
  523. Operand ne = EmitVectorExtractZx(context, op.Rn, index, op.Size);
  524. Operand me = EmitVectorExtractZx(context, op.Rm, index, op.Size);
  525. res = EmitVectorInsert(context, res, emit(de, ne, me), index, op.Size);
  526. }
  527. context.Copy(GetVec(op.Rd), res);
  528. }
  529. public static void EmitVectorBinaryOpByElemSx(ArmEmitterContext context, Func2I emit)
  530. {
  531. OpCodeSimdRegElem op = (OpCodeSimdRegElem)context.CurrOp;
  532. Operand res = context.VectorZero();
  533. Operand me = EmitVectorExtractSx(context, op.Rm, op.Index, op.Size);
  534. int elems = op.GetBytesCount() >> op.Size;
  535. for (int index = 0; index < elems; index++)
  536. {
  537. Operand ne = EmitVectorExtractSx(context, op.Rn, index, op.Size);
  538. res = EmitVectorInsert(context, res, emit(ne, me), index, op.Size);
  539. }
  540. context.Copy(GetVec(op.Rd), res);
  541. }
  542. public static void EmitVectorBinaryOpByElemZx(ArmEmitterContext context, Func2I emit)
  543. {
  544. OpCodeSimdRegElem op = (OpCodeSimdRegElem)context.CurrOp;
  545. Operand res = context.VectorZero();
  546. Operand me = EmitVectorExtractZx(context, op.Rm, op.Index, op.Size);
  547. int elems = op.GetBytesCount() >> op.Size;
  548. for (int index = 0; index < elems; index++)
  549. {
  550. Operand ne = EmitVectorExtractZx(context, op.Rn, index, op.Size);
  551. res = EmitVectorInsert(context, res, emit(ne, me), index, op.Size);
  552. }
  553. context.Copy(GetVec(op.Rd), res);
  554. }
  555. public static void EmitVectorTernaryOpByElemZx(ArmEmitterContext context, Func3I emit)
  556. {
  557. OpCodeSimdRegElem op = (OpCodeSimdRegElem)context.CurrOp;
  558. Operand res = context.VectorZero();
  559. Operand me = EmitVectorExtractZx(context, op.Rm, op.Index, op.Size);
  560. int elems = op.GetBytesCount() >> op.Size;
  561. for (int index = 0; index < elems; index++)
  562. {
  563. Operand de = EmitVectorExtractZx(context, op.Rd, index, op.Size);
  564. Operand ne = EmitVectorExtractZx(context, op.Rn, index, op.Size);
  565. res = EmitVectorInsert(context, res, emit(de, ne, me), index, op.Size);
  566. }
  567. context.Copy(GetVec(op.Rd), res);
  568. }
  569. public static void EmitVectorImmUnaryOp(ArmEmitterContext context, Func1I emit)
  570. {
  571. OpCodeSimdImm op = (OpCodeSimdImm)context.CurrOp;
  572. Operand imm = Const(op.Immediate);
  573. Operand res = context.VectorZero();
  574. int elems = op.GetBytesCount() >> op.Size;
  575. for (int index = 0; index < elems; index++)
  576. {
  577. res = EmitVectorInsert(context, res, emit(imm), index, op.Size);
  578. }
  579. context.Copy(GetVec(op.Rd), res);
  580. }
  581. public static void EmitVectorImmBinaryOp(ArmEmitterContext context, Func2I emit)
  582. {
  583. OpCodeSimdImm op = (OpCodeSimdImm)context.CurrOp;
  584. Operand imm = Const(op.Immediate);
  585. Operand res = context.VectorZero();
  586. int elems = op.GetBytesCount() >> op.Size;
  587. for (int index = 0; index < elems; index++)
  588. {
  589. Operand de = EmitVectorExtractZx(context, op.Rd, index, op.Size);
  590. res = EmitVectorInsert(context, res, emit(de, imm), index, op.Size);
  591. }
  592. context.Copy(GetVec(op.Rd), res);
  593. }
  594. public static void EmitVectorWidenRmBinaryOpSx(ArmEmitterContext context, Func2I emit)
  595. {
  596. EmitVectorWidenRmBinaryOp(context, emit, signed: true);
  597. }
  598. public static void EmitVectorWidenRmBinaryOpZx(ArmEmitterContext context, Func2I emit)
  599. {
  600. EmitVectorWidenRmBinaryOp(context, emit, signed: false);
  601. }
  602. private static void EmitVectorWidenRmBinaryOp(ArmEmitterContext context, Func2I emit, bool signed)
  603. {
  604. OpCodeSimdReg op = (OpCodeSimdReg)context.CurrOp;
  605. Operand res = context.VectorZero();
  606. int elems = 8 >> op.Size;
  607. int part = op.RegisterSize == RegisterSize.Simd128 ? elems : 0;
  608. for (int index = 0; index < elems; index++)
  609. {
  610. Operand ne = EmitVectorExtract(context, op.Rn, index, op.Size + 1, signed);
  611. Operand me = EmitVectorExtract(context, op.Rm, part + index, op.Size, signed);
  612. res = EmitVectorInsert(context, res, emit(ne, me), index, op.Size + 1);
  613. }
  614. context.Copy(GetVec(op.Rd), res);
  615. }
  616. public static void EmitVectorWidenRnRmBinaryOpSx(ArmEmitterContext context, Func2I emit)
  617. {
  618. EmitVectorWidenRnRmBinaryOp(context, emit, signed: true);
  619. }
  620. public static void EmitVectorWidenRnRmBinaryOpZx(ArmEmitterContext context, Func2I emit)
  621. {
  622. EmitVectorWidenRnRmBinaryOp(context, emit, signed: false);
  623. }
  624. private static void EmitVectorWidenRnRmBinaryOp(ArmEmitterContext context, Func2I emit, bool signed)
  625. {
  626. OpCodeSimdReg op = (OpCodeSimdReg)context.CurrOp;
  627. Operand res = context.VectorZero();
  628. int elems = 8 >> op.Size;
  629. int part = op.RegisterSize == RegisterSize.Simd128 ? elems : 0;
  630. for (int index = 0; index < elems; index++)
  631. {
  632. Operand ne = EmitVectorExtract(context, op.Rn, part + index, op.Size, signed);
  633. Operand me = EmitVectorExtract(context, op.Rm, part + index, op.Size, signed);
  634. res = EmitVectorInsert(context, res, emit(ne, me), index, op.Size + 1);
  635. }
  636. context.Copy(GetVec(op.Rd), res);
  637. }
  638. public static void EmitVectorWidenRnRmTernaryOpSx(ArmEmitterContext context, Func3I emit)
  639. {
  640. EmitVectorWidenRnRmTernaryOp(context, emit, signed: true);
  641. }
  642. public static void EmitVectorWidenRnRmTernaryOpZx(ArmEmitterContext context, Func3I emit)
  643. {
  644. EmitVectorWidenRnRmTernaryOp(context, emit, signed: false);
  645. }
  646. private static void EmitVectorWidenRnRmTernaryOp(ArmEmitterContext context, Func3I emit, bool signed)
  647. {
  648. OpCodeSimdReg op = (OpCodeSimdReg)context.CurrOp;
  649. Operand res = context.VectorZero();
  650. int elems = 8 >> op.Size;
  651. int part = op.RegisterSize == RegisterSize.Simd128 ? elems : 0;
  652. for (int index = 0; index < elems; index++)
  653. {
  654. Operand de = EmitVectorExtract(context, op.Rd, index, op.Size + 1, signed);
  655. Operand ne = EmitVectorExtract(context, op.Rn, part + index, op.Size, signed);
  656. Operand me = EmitVectorExtract(context, op.Rm, part + index, op.Size, signed);
  657. res = EmitVectorInsert(context, res, emit(de, ne, me), index, op.Size + 1);
  658. }
  659. context.Copy(GetVec(op.Rd), res);
  660. }
  661. public static void EmitVectorWidenBinaryOpByElemSx(ArmEmitterContext context, Func2I emit)
  662. {
  663. EmitVectorWidenBinaryOpByElem(context, emit, signed: true);
  664. }
  665. public static void EmitVectorWidenBinaryOpByElemZx(ArmEmitterContext context, Func2I emit)
  666. {
  667. EmitVectorWidenBinaryOpByElem(context, emit, signed: false);
  668. }
  669. private static void EmitVectorWidenBinaryOpByElem(ArmEmitterContext context, Func2I emit, bool signed)
  670. {
  671. OpCodeSimdRegElem op = (OpCodeSimdRegElem)context.CurrOp;
  672. Operand res = context.VectorZero();
  673. Operand me = EmitVectorExtract(context, op.Rm, op.Index, op.Size, signed);
  674. int elems = 8 >> op.Size;
  675. int part = op.RegisterSize == RegisterSize.Simd128 ? elems : 0;
  676. for (int index = 0; index < elems; index++)
  677. {
  678. Operand ne = EmitVectorExtract(context, op.Rn, part + index, op.Size, signed);
  679. res = EmitVectorInsert(context, res, emit(ne, me), index, op.Size + 1);
  680. }
  681. context.Copy(GetVec(op.Rd), res);
  682. }
  683. public static void EmitVectorWidenTernaryOpByElemSx(ArmEmitterContext context, Func3I emit)
  684. {
  685. EmitVectorWidenTernaryOpByElem(context, emit, signed: true);
  686. }
  687. public static void EmitVectorWidenTernaryOpByElemZx(ArmEmitterContext context, Func3I emit)
  688. {
  689. EmitVectorWidenTernaryOpByElem(context, emit, signed: false);
  690. }
  691. private static void EmitVectorWidenTernaryOpByElem(ArmEmitterContext context, Func3I emit, bool signed)
  692. {
  693. OpCodeSimdRegElem op = (OpCodeSimdRegElem)context.CurrOp;
  694. Operand res = context.VectorZero();
  695. Operand me = EmitVectorExtract(context, op.Rm, op.Index, op.Size, signed);
  696. int elems = 8 >> op.Size;
  697. int part = op.RegisterSize == RegisterSize.Simd128 ? elems : 0;
  698. for (int index = 0; index < elems; index++)
  699. {
  700. Operand de = EmitVectorExtract(context, op.Rd, index, op.Size + 1, signed);
  701. Operand ne = EmitVectorExtract(context, op.Rn, part + index, op.Size, signed);
  702. res = EmitVectorInsert(context, res, emit(de, ne, me), index, op.Size + 1);
  703. }
  704. context.Copy(GetVec(op.Rd), res);
  705. }
  706. public static void EmitVectorPairwiseOpSx(ArmEmitterContext context, Func2I emit)
  707. {
  708. EmitVectorPairwiseOp(context, emit, signed: true);
  709. }
  710. public static void EmitVectorPairwiseOpZx(ArmEmitterContext context, Func2I emit)
  711. {
  712. EmitVectorPairwiseOp(context, emit, signed: false);
  713. }
  714. private static void EmitVectorPairwiseOp(ArmEmitterContext context, Func2I emit, bool signed)
  715. {
  716. OpCodeSimdReg op = (OpCodeSimdReg)context.CurrOp;
  717. Operand res = context.VectorZero();
  718. int pairs = op.GetPairsCount() >> op.Size;
  719. for (int index = 0; index < pairs; index++)
  720. {
  721. int pairIndex = index << 1;
  722. Operand n0 = EmitVectorExtract(context, op.Rn, pairIndex, op.Size, signed);
  723. Operand n1 = EmitVectorExtract(context, op.Rn, pairIndex + 1, op.Size, signed);
  724. Operand m0 = EmitVectorExtract(context, op.Rm, pairIndex, op.Size, signed);
  725. Operand m1 = EmitVectorExtract(context, op.Rm, pairIndex + 1, op.Size, signed);
  726. res = EmitVectorInsert(context, res, emit(n0, n1), index, op.Size);
  727. res = EmitVectorInsert(context, res, emit(m0, m1), pairs + index, op.Size);
  728. }
  729. context.Copy(GetVec(op.Rd), res);
  730. }
  731. public static void EmitSsse3VectorPairwiseOp(ArmEmitterContext context, Intrinsic[] inst)
  732. {
  733. OpCodeSimdReg op = (OpCodeSimdReg)context.CurrOp;
  734. Operand n = GetVec(op.Rn);
  735. Operand m = GetVec(op.Rm);
  736. if (op.RegisterSize == RegisterSize.Simd64)
  737. {
  738. Operand zeroEvenMask = X86GetElements(context, ZeroMask, EvenMasks[op.Size]);
  739. Operand zeroOddMask = X86GetElements(context, ZeroMask, OddMasks [op.Size]);
  740. Operand mN = context.AddIntrinsic(Intrinsic.X86Punpcklqdq, n, m); // m:n
  741. Operand left = context.AddIntrinsic(Intrinsic.X86Pshufb, mN, zeroEvenMask); // 0:even from m:n
  742. Operand right = context.AddIntrinsic(Intrinsic.X86Pshufb, mN, zeroOddMask); // 0:odd from m:n
  743. context.Copy(GetVec(op.Rd), context.AddIntrinsic(inst[op.Size], left, right));
  744. }
  745. else if (op.Size < 3)
  746. {
  747. Operand oddEvenMask = X86GetElements(context, OddMasks[op.Size], EvenMasks[op.Size]);
  748. Operand oddEvenN = context.AddIntrinsic(Intrinsic.X86Pshufb, n, oddEvenMask); // odd:even from n
  749. Operand oddEvenM = context.AddIntrinsic(Intrinsic.X86Pshufb, m, oddEvenMask); // odd:even from m
  750. Operand left = context.AddIntrinsic(Intrinsic.X86Punpcklqdq, oddEvenN, oddEvenM);
  751. Operand right = context.AddIntrinsic(Intrinsic.X86Punpckhqdq, oddEvenN, oddEvenM);
  752. context.Copy(GetVec(op.Rd), context.AddIntrinsic(inst[op.Size], left, right));
  753. }
  754. else
  755. {
  756. Operand left = context.AddIntrinsic(Intrinsic.X86Punpcklqdq, n, m);
  757. Operand right = context.AddIntrinsic(Intrinsic.X86Punpckhqdq, n, m);
  758. context.Copy(GetVec(op.Rd), context.AddIntrinsic(inst[3], left, right));
  759. }
  760. }
  761. public static void EmitVectorAcrossVectorOpSx(ArmEmitterContext context, Func2I emit)
  762. {
  763. EmitVectorAcrossVectorOp(context, emit, signed: true, isLong: false);
  764. }
  765. public static void EmitVectorAcrossVectorOpZx(ArmEmitterContext context, Func2I emit)
  766. {
  767. EmitVectorAcrossVectorOp(context, emit, signed: false, isLong: false);
  768. }
  769. public static void EmitVectorLongAcrossVectorOpSx(ArmEmitterContext context, Func2I emit)
  770. {
  771. EmitVectorAcrossVectorOp(context, emit, signed: true, isLong: true);
  772. }
  773. public static void EmitVectorLongAcrossVectorOpZx(ArmEmitterContext context, Func2I emit)
  774. {
  775. EmitVectorAcrossVectorOp(context, emit, signed: false, isLong: true);
  776. }
  777. private static void EmitVectorAcrossVectorOp(
  778. ArmEmitterContext context,
  779. Func2I emit,
  780. bool signed,
  781. bool isLong)
  782. {
  783. OpCodeSimd op = (OpCodeSimd)context.CurrOp;
  784. int elems = op.GetBytesCount() >> op.Size;
  785. Operand res = EmitVectorExtract(context, op.Rn, 0, op.Size, signed);
  786. for (int index = 1; index < elems; index++)
  787. {
  788. Operand n = EmitVectorExtract(context, op.Rn, index, op.Size, signed);
  789. res = emit(res, n);
  790. }
  791. int size = isLong ? op.Size + 1 : op.Size;
  792. Operand d = EmitVectorInsert(context, context.VectorZero(), res, 0, size);
  793. context.Copy(GetVec(op.Rd), d);
  794. }
  795. public static void EmitVectorAcrossVectorOpF(ArmEmitterContext context, Func2I emit)
  796. {
  797. OpCodeSimd op = (OpCodeSimd)context.CurrOp;
  798. Debug.Assert((op.Size & 1) == 0 && op.RegisterSize == RegisterSize.Simd128);
  799. Operand res = context.VectorExtract(OperandType.FP32, GetVec(op.Rn), 0);
  800. for (int index = 1; index < 4; index++)
  801. {
  802. Operand n = context.VectorExtract(OperandType.FP32, GetVec(op.Rn), index);
  803. res = emit(res, n);
  804. }
  805. Operand d = context.VectorInsert(context.VectorZero(), res, 0);
  806. context.Copy(GetVec(op.Rd), d);
  807. }
  808. public static void EmitSse2VectorAcrossVectorOpF(ArmEmitterContext context, Func2I emit)
  809. {
  810. OpCodeSimd op = (OpCodeSimd)context.CurrOp;
  811. Debug.Assert((op.Size & 1) == 0 && op.RegisterSize == RegisterSize.Simd128);
  812. const int sm0 = 0 << 6 | 0 << 4 | 0 << 2 | 0 << 0;
  813. const int sm1 = 1 << 6 | 1 << 4 | 1 << 2 | 1 << 0;
  814. const int sm2 = 2 << 6 | 2 << 4 | 2 << 2 | 2 << 0;
  815. const int sm3 = 3 << 6 | 3 << 4 | 3 << 2 | 3 << 0;
  816. Operand nCopy = context.Copy(GetVec(op.Rn));
  817. Operand part0 = context.AddIntrinsic(Intrinsic.X86Shufps, nCopy, nCopy, Const(sm0));
  818. Operand part1 = context.AddIntrinsic(Intrinsic.X86Shufps, nCopy, nCopy, Const(sm1));
  819. Operand part2 = context.AddIntrinsic(Intrinsic.X86Shufps, nCopy, nCopy, Const(sm2));
  820. Operand part3 = context.AddIntrinsic(Intrinsic.X86Shufps, nCopy, nCopy, Const(sm3));
  821. Operand res = emit(emit(part0, part1), emit(part2, part3));
  822. context.Copy(GetVec(op.Rd), context.VectorZeroUpper96(res));
  823. }
  824. public static void EmitScalarPairwiseOpF(ArmEmitterContext context, Func2I emit)
  825. {
  826. OpCodeSimd op = (OpCodeSimd)context.CurrOp;
  827. OperandType type = (op.Size & 1) != 0 ? OperandType.FP64 : OperandType.FP32;
  828. Operand ne0 = context.VectorExtract(type, GetVec(op.Rn), 0);
  829. Operand ne1 = context.VectorExtract(type, GetVec(op.Rn), 1);
  830. Operand res = context.VectorInsert(context.VectorZero(), emit(ne0, ne1), 0);
  831. context.Copy(GetVec(op.Rd), res);
  832. }
  833. public static void EmitSse2ScalarPairwiseOpF(ArmEmitterContext context, Func2I emit)
  834. {
  835. OpCodeSimd op = (OpCodeSimd)context.CurrOp;
  836. Operand n = GetVec(op.Rn);
  837. Operand op0, op1;
  838. if ((op.Size & 1) == 0)
  839. {
  840. const int sm0 = 2 << 6 | 2 << 4 | 2 << 2 | 0 << 0;
  841. const int sm1 = 2 << 6 | 2 << 4 | 2 << 2 | 1 << 0;
  842. Operand zeroN = context.VectorZeroUpper64(n);
  843. op0 = context.AddIntrinsic(Intrinsic.X86Pshufd, zeroN, Const(sm0));
  844. op1 = context.AddIntrinsic(Intrinsic.X86Pshufd, zeroN, Const(sm1));
  845. }
  846. else /* if ((op.Size & 1) == 1) */
  847. {
  848. Operand zero = context.VectorZero();
  849. op0 = context.AddIntrinsic(Intrinsic.X86Movlhps, n, zero);
  850. op1 = context.AddIntrinsic(Intrinsic.X86Movhlps, zero, n);
  851. }
  852. context.Copy(GetVec(op.Rd), emit(op0, op1));
  853. }
  854. public static void EmitVectorPairwiseOpF(ArmEmitterContext context, Func2I emit)
  855. {
  856. OpCodeSimdReg op = (OpCodeSimdReg)context.CurrOp;
  857. Operand res = context.VectorZero();
  858. int sizeF = op.Size & 1;
  859. OperandType type = sizeF != 0 ? OperandType.FP64 : OperandType.FP32;
  860. int pairs = op.GetPairsCount() >> sizeF + 2;
  861. for (int index = 0; index < pairs; index++)
  862. {
  863. int pairIndex = index << 1;
  864. Operand n0 = context.VectorExtract(type, GetVec(op.Rn), pairIndex);
  865. Operand n1 = context.VectorExtract(type, GetVec(op.Rn), pairIndex + 1);
  866. Operand m0 = context.VectorExtract(type, GetVec(op.Rm), pairIndex);
  867. Operand m1 = context.VectorExtract(type, GetVec(op.Rm), pairIndex + 1);
  868. res = context.VectorInsert(res, emit(n0, n1), index);
  869. res = context.VectorInsert(res, emit(m0, m1), pairs + index);
  870. }
  871. context.Copy(GetVec(op.Rd), res);
  872. }
  873. public static void EmitSse2VectorPairwiseOpF(ArmEmitterContext context, Func2I emit)
  874. {
  875. OpCodeSimdReg op = (OpCodeSimdReg)context.CurrOp;
  876. Operand nCopy = context.Copy(GetVec(op.Rn));
  877. Operand mCopy = context.Copy(GetVec(op.Rm));
  878. int sizeF = op.Size & 1;
  879. if (sizeF == 0)
  880. {
  881. if (op.RegisterSize == RegisterSize.Simd64)
  882. {
  883. Operand unpck = context.AddIntrinsic(Intrinsic.X86Unpcklps, nCopy, mCopy);
  884. Operand zero = context.VectorZero();
  885. Operand part0 = context.AddIntrinsic(Intrinsic.X86Movlhps, unpck, zero);
  886. Operand part1 = context.AddIntrinsic(Intrinsic.X86Movhlps, zero, unpck);
  887. context.Copy(GetVec(op.Rd), emit(part0, part1));
  888. }
  889. else /* if (op.RegisterSize == RegisterSize.Simd128) */
  890. {
  891. const int sm0 = 2 << 6 | 0 << 4 | 2 << 2 | 0 << 0;
  892. const int sm1 = 3 << 6 | 1 << 4 | 3 << 2 | 1 << 0;
  893. Operand part0 = context.AddIntrinsic(Intrinsic.X86Shufps, nCopy, mCopy, Const(sm0));
  894. Operand part1 = context.AddIntrinsic(Intrinsic.X86Shufps, nCopy, mCopy, Const(sm1));
  895. context.Copy(GetVec(op.Rd), emit(part0, part1));
  896. }
  897. }
  898. else /* if (sizeF == 1) */
  899. {
  900. Operand part0 = context.AddIntrinsic(Intrinsic.X86Unpcklpd, nCopy, mCopy);
  901. Operand part1 = context.AddIntrinsic(Intrinsic.X86Unpckhpd, nCopy, mCopy);
  902. context.Copy(GetVec(op.Rd), emit(part0, part1));
  903. }
  904. }
  905. [Flags]
  906. public enum Mxcsr
  907. {
  908. Ftz = 1 << 15, // Flush To Zero.
  909. Um = 1 << 11, // Underflow Mask.
  910. Dm = 1 << 8, // Denormal Mask.
  911. Daz = 1 << 6 // Denormals Are Zero.
  912. }
  913. public static void EmitSseOrAvxEnterFtzAndDazModesOpF(ArmEmitterContext context, out Operand isTrue)
  914. {
  915. isTrue = context.Call(typeof(NativeInterface).GetMethod(nameof(NativeInterface.GetFpcrFz)));
  916. Operand lblTrue = Label();
  917. context.BranchIfFalse(lblTrue, isTrue);
  918. context.AddIntrinsicNoRet(Intrinsic.X86Mxcsrmb, Const((int)(Mxcsr.Ftz | Mxcsr.Um | Mxcsr.Dm | Mxcsr.Daz)));
  919. context.MarkLabel(lblTrue);
  920. }
  921. public static void EmitSseOrAvxExitFtzAndDazModesOpF(ArmEmitterContext context, Operand isTrue = null)
  922. {
  923. isTrue ??= context.Call(typeof(NativeInterface).GetMethod(nameof(NativeInterface.GetFpcrFz)));
  924. Operand lblTrue = Label();
  925. context.BranchIfFalse(lblTrue, isTrue);
  926. context.AddIntrinsicNoRet(Intrinsic.X86Mxcsrub, Const((int)(Mxcsr.Ftz | Mxcsr.Daz)));
  927. context.MarkLabel(lblTrue);
  928. }
  929. public enum CmpCondition
  930. {
  931. // Legacy Sse.
  932. Equal = 0, // Ordered, non-signaling.
  933. LessThan = 1, // Ordered, signaling.
  934. LessThanOrEqual = 2, // Ordered, signaling.
  935. UnorderedQ = 3, // Non-signaling.
  936. NotLessThan = 5, // Unordered, signaling.
  937. NotLessThanOrEqual = 6, // Unordered, signaling.
  938. OrderedQ = 7, // Non-signaling.
  939. // Vex.
  940. GreaterThanOrEqual = 13, // Ordered, signaling.
  941. GreaterThan = 14, // Ordered, signaling.
  942. OrderedS = 23 // Signaling.
  943. }
  944. [Flags]
  945. public enum SaturatingFlags
  946. {
  947. None = 0,
  948. ByElem = 1 << 0,
  949. Scalar = 1 << 1,
  950. Signed = 1 << 2,
  951. Add = 1 << 3,
  952. Sub = 1 << 4,
  953. Accumulate = 1 << 5
  954. }
  955. public static void EmitScalarSaturatingUnaryOpSx(ArmEmitterContext context, Func1I emit)
  956. {
  957. EmitSaturatingUnaryOpSx(context, emit, SaturatingFlags.Scalar | SaturatingFlags.Signed);
  958. }
  959. public static void EmitVectorSaturatingUnaryOpSx(ArmEmitterContext context, Func1I emit)
  960. {
  961. EmitSaturatingUnaryOpSx(context, emit, SaturatingFlags.Signed);
  962. }
  963. public static void EmitSaturatingUnaryOpSx(ArmEmitterContext context, Func1I emit, SaturatingFlags flags)
  964. {
  965. OpCodeSimd op = (OpCodeSimd)context.CurrOp;
  966. Operand res = context.VectorZero();
  967. bool scalar = (flags & SaturatingFlags.Scalar) != 0;
  968. int elems = !scalar ? op.GetBytesCount() >> op.Size : 1;
  969. for (int index = 0; index < elems; index++)
  970. {
  971. Operand ne = EmitVectorExtractSx(context, op.Rn, index, op.Size);
  972. Operand de;
  973. if (op.Size <= 2)
  974. {
  975. de = EmitSatQ(context, emit(ne), op.Size, signedSrc: true, signedDst: true);
  976. }
  977. else /* if (op.Size == 3) */
  978. {
  979. de = EmitUnarySignedSatQAbsOrNeg(context, emit(ne));
  980. }
  981. res = EmitVectorInsert(context, res, de, index, op.Size);
  982. }
  983. context.Copy(GetVec(op.Rd), res);
  984. }
  985. public static void EmitScalarSaturatingBinaryOpSx(ArmEmitterContext context, Func2I emit = null, SaturatingFlags flags = SaturatingFlags.None)
  986. {
  987. EmitSaturatingBinaryOp(context, emit, SaturatingFlags.Scalar | SaturatingFlags.Signed | flags);
  988. }
  989. public static void EmitScalarSaturatingBinaryOpZx(ArmEmitterContext context, SaturatingFlags flags)
  990. {
  991. EmitSaturatingBinaryOp(context, null, SaturatingFlags.Scalar | flags);
  992. }
  993. public static void EmitVectorSaturatingBinaryOpSx(ArmEmitterContext context, Func2I emit = null, SaturatingFlags flags = SaturatingFlags.None)
  994. {
  995. EmitSaturatingBinaryOp(context, emit, SaturatingFlags.Signed | flags);
  996. }
  997. public static void EmitVectorSaturatingBinaryOpZx(ArmEmitterContext context, SaturatingFlags flags)
  998. {
  999. EmitSaturatingBinaryOp(context, null, flags);
  1000. }
  1001. public static void EmitVectorSaturatingBinaryOpByElemSx(ArmEmitterContext context, Func2I emit)
  1002. {
  1003. EmitSaturatingBinaryOp(context, emit, SaturatingFlags.ByElem | SaturatingFlags.Signed);
  1004. }
  1005. public static void EmitSaturatingBinaryOp(ArmEmitterContext context, Func2I emit, SaturatingFlags flags)
  1006. {
  1007. OpCodeSimd op = (OpCodeSimd)context.CurrOp;
  1008. Operand res = context.VectorZero();
  1009. bool byElem = (flags & SaturatingFlags.ByElem) != 0;
  1010. bool scalar = (flags & SaturatingFlags.Scalar) != 0;
  1011. bool signed = (flags & SaturatingFlags.Signed) != 0;
  1012. bool add = (flags & SaturatingFlags.Add) != 0;
  1013. bool sub = (flags & SaturatingFlags.Sub) != 0;
  1014. bool accumulate = (flags & SaturatingFlags.Accumulate) != 0;
  1015. int elems = !scalar ? op.GetBytesCount() >> op.Size : 1;
  1016. if (add || sub)
  1017. {
  1018. for (int index = 0; index < elems; index++)
  1019. {
  1020. Operand de;
  1021. Operand ne = EmitVectorExtract(context, op.Rn, index, op.Size, signed);
  1022. Operand me = EmitVectorExtract(context, ((OpCodeSimdReg)op).Rm, index, op.Size, signed);
  1023. if (op.Size <= 2)
  1024. {
  1025. Operand temp = add ? context.Add(ne, me) : context.Subtract(ne, me);
  1026. de = EmitSatQ(context, temp, op.Size, signedSrc: true, signedDst: signed);
  1027. }
  1028. else if (add) /* if (op.Size == 3) */
  1029. {
  1030. de = EmitBinarySatQAdd(context, ne, me, signed);
  1031. }
  1032. else /* if (sub) */
  1033. {
  1034. de = EmitBinarySatQSub(context, ne, me, signed);
  1035. }
  1036. res = EmitVectorInsert(context, res, de, index, op.Size);
  1037. }
  1038. }
  1039. else if (accumulate)
  1040. {
  1041. for (int index = 0; index < elems; index++)
  1042. {
  1043. Operand de;
  1044. Operand ne = EmitVectorExtract(context, op.Rn, index, op.Size, !signed);
  1045. Operand me = EmitVectorExtract(context, op.Rd, index, op.Size, signed);
  1046. if (op.Size <= 2)
  1047. {
  1048. Operand temp = context.Add(ne, me);
  1049. de = EmitSatQ(context, temp, op.Size, signedSrc: true, signedDst: signed);
  1050. }
  1051. else /* if (op.Size == 3) */
  1052. {
  1053. de = EmitBinarySatQAccumulate(context, ne, me, signed);
  1054. }
  1055. res = EmitVectorInsert(context, res, de, index, op.Size);
  1056. }
  1057. }
  1058. else
  1059. {
  1060. Operand me = null;
  1061. if (byElem)
  1062. {
  1063. OpCodeSimdRegElem opRegElem = (OpCodeSimdRegElem)op;
  1064. me = EmitVectorExtract(context, opRegElem.Rm, opRegElem.Index, op.Size, signed);
  1065. }
  1066. for (int index = 0; index < elems; index++)
  1067. {
  1068. Operand ne = EmitVectorExtract(context, op.Rn, index, op.Size, signed);
  1069. if (!byElem)
  1070. {
  1071. me = EmitVectorExtract(context, ((OpCodeSimdReg)op).Rm, index, op.Size, signed);
  1072. }
  1073. Operand de = EmitSatQ(context, emit(ne, me), op.Size, true, signed);
  1074. res = EmitVectorInsert(context, res, de, index, op.Size);
  1075. }
  1076. }
  1077. context.Copy(GetVec(op.Rd), res);
  1078. }
  1079. [Flags]
  1080. public enum SaturatingNarrowFlags
  1081. {
  1082. Scalar = 1 << 0,
  1083. SignedSrc = 1 << 1,
  1084. SignedDst = 1 << 2,
  1085. ScalarSxSx = Scalar | SignedSrc | SignedDst,
  1086. ScalarSxZx = Scalar | SignedSrc,
  1087. ScalarZxZx = Scalar,
  1088. VectorSxSx = SignedSrc | SignedDst,
  1089. VectorSxZx = SignedSrc,
  1090. VectorZxZx = 0
  1091. }
  1092. public static void EmitSaturatingNarrowOp(ArmEmitterContext context, SaturatingNarrowFlags flags)
  1093. {
  1094. OpCodeSimd op = (OpCodeSimd)context.CurrOp;
  1095. bool scalar = (flags & SaturatingNarrowFlags.Scalar) != 0;
  1096. bool signedSrc = (flags & SaturatingNarrowFlags.SignedSrc) != 0;
  1097. bool signedDst = (flags & SaturatingNarrowFlags.SignedDst) != 0;
  1098. int elems = !scalar ? 8 >> op.Size : 1;
  1099. int part = !scalar && (op.RegisterSize == RegisterSize.Simd128) ? elems : 0;
  1100. Operand d = GetVec(op.Rd);
  1101. Operand res = part == 0 ? context.VectorZero() : context.Copy(d);
  1102. for (int index = 0; index < elems; index++)
  1103. {
  1104. Operand ne = EmitVectorExtract(context, op.Rn, index, op.Size + 1, signedSrc);
  1105. Operand temp = EmitSatQ(context, ne, op.Size, signedSrc, signedDst);
  1106. res = EmitVectorInsert(context, res, temp, part + index, op.Size);
  1107. }
  1108. context.Copy(d, res);
  1109. }
  1110. // TSrc (16bit, 32bit, 64bit; signed, unsigned) > TDst (8bit, 16bit, 32bit; signed, unsigned).
  1111. public static Operand EmitSatQ(ArmEmitterContext context, Operand op, int sizeDst, bool signedSrc, bool signedDst)
  1112. {
  1113. if ((uint)sizeDst > 2u)
  1114. {
  1115. throw new ArgumentOutOfRangeException(nameof(sizeDst));
  1116. }
  1117. MethodInfo info;
  1118. if (signedSrc)
  1119. {
  1120. info = signedDst
  1121. ? typeof(SoftFallback).GetMethod(nameof(SoftFallback.SignedSrcSignedDstSatQ))
  1122. : typeof(SoftFallback).GetMethod(nameof(SoftFallback.SignedSrcUnsignedDstSatQ));
  1123. }
  1124. else
  1125. {
  1126. info = signedDst
  1127. ? typeof(SoftFallback).GetMethod(nameof(SoftFallback.UnsignedSrcSignedDstSatQ))
  1128. : typeof(SoftFallback).GetMethod(nameof(SoftFallback.UnsignedSrcUnsignedDstSatQ));
  1129. }
  1130. return context.Call(info, op, Const(sizeDst));
  1131. }
  1132. // TSrc (64bit) == TDst (64bit); signed.
  1133. public static Operand EmitUnarySignedSatQAbsOrNeg(ArmEmitterContext context, Operand op)
  1134. {
  1135. Debug.Assert(((OpCodeSimd)context.CurrOp).Size == 3, "Invalid element size.");
  1136. return context.Call(typeof(SoftFallback).GetMethod(nameof(SoftFallback.UnarySignedSatQAbsOrNeg)), op);
  1137. }
  1138. // TSrcs (64bit) == TDst (64bit); signed, unsigned.
  1139. public static Operand EmitBinarySatQAdd(ArmEmitterContext context, Operand op1, Operand op2, bool signed)
  1140. {
  1141. Debug.Assert(((OpCodeSimd)context.CurrOp).Size == 3, "Invalid element size.");
  1142. MethodInfo info = signed
  1143. ? typeof(SoftFallback).GetMethod(nameof(SoftFallback.BinarySignedSatQAdd))
  1144. : typeof(SoftFallback).GetMethod(nameof(SoftFallback.BinaryUnsignedSatQAdd));
  1145. return context.Call(info, op1, op2);
  1146. }
  1147. // TSrcs (64bit) == TDst (64bit); signed, unsigned.
  1148. public static Operand EmitBinarySatQSub(ArmEmitterContext context, Operand op1, Operand op2, bool signed)
  1149. {
  1150. Debug.Assert(((OpCodeSimd)context.CurrOp).Size == 3, "Invalid element size.");
  1151. MethodInfo info = signed
  1152. ? typeof(SoftFallback).GetMethod(nameof(SoftFallback.BinarySignedSatQSub))
  1153. : typeof(SoftFallback).GetMethod(nameof(SoftFallback.BinaryUnsignedSatQSub));
  1154. return context.Call(info, op1, op2);
  1155. }
  1156. // TSrcs (64bit) == TDst (64bit); signed, unsigned.
  1157. public static Operand EmitBinarySatQAccumulate(ArmEmitterContext context, Operand op1, Operand op2, bool signed)
  1158. {
  1159. Debug.Assert(((OpCodeSimd)context.CurrOp).Size == 3, "Invalid element size.");
  1160. MethodInfo info = signed
  1161. ? typeof(SoftFallback).GetMethod(nameof(SoftFallback.BinarySignedSatQAcc))
  1162. : typeof(SoftFallback).GetMethod(nameof(SoftFallback.BinaryUnsignedSatQAcc));
  1163. return context.Call(info, op1, op2);
  1164. }
  1165. public static Operand EmitFloatAbs(ArmEmitterContext context, Operand value, bool single, bool vector)
  1166. {
  1167. Operand mask;
  1168. if (single)
  1169. {
  1170. mask = vector ? X86GetAllElements(context, -0f) : X86GetScalar(context, -0f);
  1171. }
  1172. else
  1173. {
  1174. mask = vector ? X86GetAllElements(context, -0d) : X86GetScalar(context, -0d);
  1175. }
  1176. return context.AddIntrinsic(single ? Intrinsic.X86Andnps : Intrinsic.X86Andnpd, mask, value);
  1177. }
  1178. public static Operand EmitVectorExtractSx(ArmEmitterContext context, int reg, int index, int size)
  1179. {
  1180. return EmitVectorExtract(context, reg, index, size, true);
  1181. }
  1182. public static Operand EmitVectorExtractZx(ArmEmitterContext context, int reg, int index, int size)
  1183. {
  1184. return EmitVectorExtract(context, reg, index, size, false);
  1185. }
  1186. public static Operand EmitVectorExtract(ArmEmitterContext context, int reg, int index, int size, bool signed)
  1187. {
  1188. ThrowIfInvalid(index, size);
  1189. Operand res = null;
  1190. switch (size)
  1191. {
  1192. case 0:
  1193. res = context.VectorExtract8(GetVec(reg), index);
  1194. break;
  1195. case 1:
  1196. res = context.VectorExtract16(GetVec(reg), index);
  1197. break;
  1198. case 2:
  1199. res = context.VectorExtract(OperandType.I32, GetVec(reg), index);
  1200. break;
  1201. case 3:
  1202. res = context.VectorExtract(OperandType.I64, GetVec(reg), index);
  1203. break;
  1204. }
  1205. if (signed)
  1206. {
  1207. switch (size)
  1208. {
  1209. case 0: res = context.SignExtend8 (OperandType.I64, res); break;
  1210. case 1: res = context.SignExtend16(OperandType.I64, res); break;
  1211. case 2: res = context.SignExtend32(OperandType.I64, res); break;
  1212. }
  1213. }
  1214. else
  1215. {
  1216. switch (size)
  1217. {
  1218. case 0: res = context.ZeroExtend8 (OperandType.I64, res); break;
  1219. case 1: res = context.ZeroExtend16(OperandType.I64, res); break;
  1220. case 2: res = context.ZeroExtend32(OperandType.I64, res); break;
  1221. }
  1222. }
  1223. return res;
  1224. }
  1225. public static Operand EmitVectorInsert(ArmEmitterContext context, Operand vector, Operand value, int index, int size)
  1226. {
  1227. ThrowIfInvalid(index, size);
  1228. if (size < 3 && value.Type == OperandType.I64)
  1229. {
  1230. value = context.ConvertI64ToI32(value);
  1231. }
  1232. switch (size)
  1233. {
  1234. case 0: vector = context.VectorInsert8 (vector, value, index); break;
  1235. case 1: vector = context.VectorInsert16(vector, value, index); break;
  1236. case 2: vector = context.VectorInsert (vector, value, index); break;
  1237. case 3: vector = context.VectorInsert (vector, value, index); break;
  1238. }
  1239. return vector;
  1240. }
  1241. public static void ThrowIfInvalid(int index, int size)
  1242. {
  1243. if ((uint)size > 3u)
  1244. {
  1245. throw new ArgumentOutOfRangeException(nameof(size));
  1246. }
  1247. if ((uint)index >= 16u >> size)
  1248. {
  1249. throw new ArgumentOutOfRangeException(nameof(index));
  1250. }
  1251. }
  1252. }
  1253. }