InstEmitSimdHelper.cs 61 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484148514861487148814891490149114921493149414951496149714981499150015011502150315041505150615071508150915101511151215131514151515161517151815191520152115221523152415251526152715281529153015311532153315341535153615371538153915401541154215431544154515461547154815491550155115521553155415551556155715581559156015611562156315641565156615671568156915701571157215731574157515761577157815791580158115821583158415851586158715881589159015911592159315941595159615971598159916001601160216031604160516061607160816091610161116121613161416151616161716181619162016211622162316241625162616271628162916301631163216331634163516361637163816391640164116421643164416451646164716481649165016511652165316541655165616571658165916601661166216631664166516661667166816691670167116721673167416751676167716781679168016811682168316841685168616871688168916901691169216931694169516961697169816991700170117021703170417051706
  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.Operand.Factory;
  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 = default)
  922. {
  923. isTrue = isTrue == default
  924. ? context.Call(typeof(NativeInterface).GetMethod(nameof(NativeInterface.GetFpcrFz)))
  925. : isTrue;
  926. Operand lblTrue = Label();
  927. context.BranchIfFalse(lblTrue, isTrue);
  928. context.AddIntrinsicNoRet(Intrinsic.X86Mxcsrub, Const((int)(Mxcsr.Ftz | Mxcsr.Daz)));
  929. context.MarkLabel(lblTrue);
  930. }
  931. public enum CmpCondition
  932. {
  933. // Legacy Sse.
  934. Equal = 0, // Ordered, non-signaling.
  935. LessThan = 1, // Ordered, signaling.
  936. LessThanOrEqual = 2, // Ordered, signaling.
  937. UnorderedQ = 3, // Non-signaling.
  938. NotLessThan = 5, // Unordered, signaling.
  939. NotLessThanOrEqual = 6, // Unordered, signaling.
  940. OrderedQ = 7, // Non-signaling.
  941. // Vex.
  942. GreaterThanOrEqual = 13, // Ordered, signaling.
  943. GreaterThan = 14, // Ordered, signaling.
  944. OrderedS = 23 // Signaling.
  945. }
  946. [Flags]
  947. public enum SaturatingFlags
  948. {
  949. None = 0,
  950. ByElem = 1 << 0,
  951. Scalar = 1 << 1,
  952. Signed = 1 << 2,
  953. Add = 1 << 3,
  954. Sub = 1 << 4,
  955. Accumulate = 1 << 5
  956. }
  957. public static void EmitScalarSaturatingUnaryOpSx(ArmEmitterContext context, Func1I emit)
  958. {
  959. EmitSaturatingUnaryOpSx(context, emit, SaturatingFlags.Scalar | SaturatingFlags.Signed);
  960. }
  961. public static void EmitVectorSaturatingUnaryOpSx(ArmEmitterContext context, Func1I emit)
  962. {
  963. EmitSaturatingUnaryOpSx(context, emit, SaturatingFlags.Signed);
  964. }
  965. public static void EmitSaturatingUnaryOpSx(ArmEmitterContext context, Func1I emit, SaturatingFlags flags)
  966. {
  967. OpCodeSimd op = (OpCodeSimd)context.CurrOp;
  968. Operand res = context.VectorZero();
  969. bool scalar = (flags & SaturatingFlags.Scalar) != 0;
  970. int elems = !scalar ? op.GetBytesCount() >> op.Size : 1;
  971. for (int index = 0; index < elems; index++)
  972. {
  973. Operand ne = EmitVectorExtractSx(context, op.Rn, index, op.Size);
  974. Operand de;
  975. if (op.Size <= 2)
  976. {
  977. de = EmitSatQ(context, emit(ne), op.Size, signedSrc: true, signedDst: true);
  978. }
  979. else /* if (op.Size == 3) */
  980. {
  981. de = EmitUnarySignedSatQAbsOrNeg(context, emit(ne));
  982. }
  983. res = EmitVectorInsert(context, res, de, index, op.Size);
  984. }
  985. context.Copy(GetVec(op.Rd), res);
  986. }
  987. public static void EmitScalarSaturatingBinaryOpSx(ArmEmitterContext context, Func2I emit = null, SaturatingFlags flags = SaturatingFlags.None)
  988. {
  989. EmitSaturatingBinaryOp(context, emit, SaturatingFlags.Scalar | SaturatingFlags.Signed | flags);
  990. }
  991. public static void EmitScalarSaturatingBinaryOpZx(ArmEmitterContext context, SaturatingFlags flags)
  992. {
  993. EmitSaturatingBinaryOp(context, null, SaturatingFlags.Scalar | flags);
  994. }
  995. public static void EmitVectorSaturatingBinaryOpSx(ArmEmitterContext context, Func2I emit = null, SaturatingFlags flags = SaturatingFlags.None)
  996. {
  997. EmitSaturatingBinaryOp(context, emit, SaturatingFlags.Signed | flags);
  998. }
  999. public static void EmitVectorSaturatingBinaryOpZx(ArmEmitterContext context, SaturatingFlags flags)
  1000. {
  1001. EmitSaturatingBinaryOp(context, null, flags);
  1002. }
  1003. public static void EmitVectorSaturatingBinaryOpByElemSx(ArmEmitterContext context, Func2I emit)
  1004. {
  1005. EmitSaturatingBinaryOp(context, emit, SaturatingFlags.ByElem | SaturatingFlags.Signed);
  1006. }
  1007. public static void EmitSaturatingBinaryOp(ArmEmitterContext context, Func2I emit, SaturatingFlags flags)
  1008. {
  1009. OpCodeSimd op = (OpCodeSimd)context.CurrOp;
  1010. Operand res = context.VectorZero();
  1011. bool byElem = (flags & SaturatingFlags.ByElem) != 0;
  1012. bool scalar = (flags & SaturatingFlags.Scalar) != 0;
  1013. bool signed = (flags & SaturatingFlags.Signed) != 0;
  1014. bool add = (flags & SaturatingFlags.Add) != 0;
  1015. bool sub = (flags & SaturatingFlags.Sub) != 0;
  1016. bool accumulate = (flags & SaturatingFlags.Accumulate) != 0;
  1017. int elems = !scalar ? op.GetBytesCount() >> op.Size : 1;
  1018. if (add || sub)
  1019. {
  1020. for (int index = 0; index < elems; index++)
  1021. {
  1022. Operand de;
  1023. Operand ne = EmitVectorExtract(context, op.Rn, index, op.Size, signed);
  1024. Operand me = EmitVectorExtract(context, ((OpCodeSimdReg)op).Rm, index, op.Size, signed);
  1025. if (op.Size <= 2)
  1026. {
  1027. Operand temp = add ? context.Add(ne, me) : context.Subtract(ne, me);
  1028. de = EmitSatQ(context, temp, op.Size, signedSrc: true, signedDst: signed);
  1029. }
  1030. else if (add) /* if (op.Size == 3) */
  1031. {
  1032. de = EmitBinarySatQAdd(context, ne, me, signed);
  1033. }
  1034. else /* if (sub) */
  1035. {
  1036. de = EmitBinarySatQSub(context, ne, me, signed);
  1037. }
  1038. res = EmitVectorInsert(context, res, de, index, op.Size);
  1039. }
  1040. }
  1041. else if (accumulate)
  1042. {
  1043. for (int index = 0; index < elems; index++)
  1044. {
  1045. Operand de;
  1046. Operand ne = EmitVectorExtract(context, op.Rn, index, op.Size, !signed);
  1047. Operand me = EmitVectorExtract(context, op.Rd, index, op.Size, signed);
  1048. if (op.Size <= 2)
  1049. {
  1050. Operand temp = context.Add(ne, me);
  1051. de = EmitSatQ(context, temp, op.Size, signedSrc: true, signedDst: signed);
  1052. }
  1053. else /* if (op.Size == 3) */
  1054. {
  1055. de = EmitBinarySatQAccumulate(context, ne, me, signed);
  1056. }
  1057. res = EmitVectorInsert(context, res, de, index, op.Size);
  1058. }
  1059. }
  1060. else
  1061. {
  1062. Operand me = default;
  1063. if (byElem)
  1064. {
  1065. OpCodeSimdRegElem opRegElem = (OpCodeSimdRegElem)op;
  1066. me = EmitVectorExtract(context, opRegElem.Rm, opRegElem.Index, op.Size, signed);
  1067. }
  1068. for (int index = 0; index < elems; index++)
  1069. {
  1070. Operand ne = EmitVectorExtract(context, op.Rn, index, op.Size, signed);
  1071. if (!byElem)
  1072. {
  1073. me = EmitVectorExtract(context, ((OpCodeSimdReg)op).Rm, index, op.Size, signed);
  1074. }
  1075. Operand de = EmitSatQ(context, emit(ne, me), op.Size, true, signed);
  1076. res = EmitVectorInsert(context, res, de, index, op.Size);
  1077. }
  1078. }
  1079. context.Copy(GetVec(op.Rd), res);
  1080. }
  1081. [Flags]
  1082. public enum SaturatingNarrowFlags
  1083. {
  1084. Scalar = 1 << 0,
  1085. SignedSrc = 1 << 1,
  1086. SignedDst = 1 << 2,
  1087. ScalarSxSx = Scalar | SignedSrc | SignedDst,
  1088. ScalarSxZx = Scalar | SignedSrc,
  1089. ScalarZxZx = Scalar,
  1090. VectorSxSx = SignedSrc | SignedDst,
  1091. VectorSxZx = SignedSrc,
  1092. VectorZxZx = 0
  1093. }
  1094. public static void EmitSaturatingNarrowOp(ArmEmitterContext context, SaturatingNarrowFlags flags)
  1095. {
  1096. OpCodeSimd op = (OpCodeSimd)context.CurrOp;
  1097. bool scalar = (flags & SaturatingNarrowFlags.Scalar) != 0;
  1098. bool signedSrc = (flags & SaturatingNarrowFlags.SignedSrc) != 0;
  1099. bool signedDst = (flags & SaturatingNarrowFlags.SignedDst) != 0;
  1100. int elems = !scalar ? 8 >> op.Size : 1;
  1101. int part = !scalar && (op.RegisterSize == RegisterSize.Simd128) ? elems : 0;
  1102. Operand d = GetVec(op.Rd);
  1103. Operand res = part == 0 ? context.VectorZero() : context.Copy(d);
  1104. for (int index = 0; index < elems; index++)
  1105. {
  1106. Operand ne = EmitVectorExtract(context, op.Rn, index, op.Size + 1, signedSrc);
  1107. Operand temp = EmitSatQ(context, ne, op.Size, signedSrc, signedDst);
  1108. res = EmitVectorInsert(context, res, temp, part + index, op.Size);
  1109. }
  1110. context.Copy(d, res);
  1111. }
  1112. // TSrc (16bit, 32bit, 64bit; signed, unsigned) > TDst (8bit, 16bit, 32bit; signed, unsigned).
  1113. public static Operand EmitSatQ(ArmEmitterContext context, Operand op, int sizeDst, bool signedSrc, bool signedDst)
  1114. {
  1115. if ((uint)sizeDst > 2u)
  1116. {
  1117. throw new ArgumentOutOfRangeException(nameof(sizeDst));
  1118. }
  1119. MethodInfo info;
  1120. if (signedSrc)
  1121. {
  1122. info = signedDst
  1123. ? typeof(SoftFallback).GetMethod(nameof(SoftFallback.SignedSrcSignedDstSatQ))
  1124. : typeof(SoftFallback).GetMethod(nameof(SoftFallback.SignedSrcUnsignedDstSatQ));
  1125. }
  1126. else
  1127. {
  1128. info = signedDst
  1129. ? typeof(SoftFallback).GetMethod(nameof(SoftFallback.UnsignedSrcSignedDstSatQ))
  1130. : typeof(SoftFallback).GetMethod(nameof(SoftFallback.UnsignedSrcUnsignedDstSatQ));
  1131. }
  1132. return context.Call(info, op, Const(sizeDst));
  1133. }
  1134. // TSrc (64bit) == TDst (64bit); signed.
  1135. public static Operand EmitUnarySignedSatQAbsOrNeg(ArmEmitterContext context, Operand op)
  1136. {
  1137. Debug.Assert(((OpCodeSimd)context.CurrOp).Size == 3, "Invalid element size.");
  1138. return context.Call(typeof(SoftFallback).GetMethod(nameof(SoftFallback.UnarySignedSatQAbsOrNeg)), op);
  1139. }
  1140. // TSrcs (64bit) == TDst (64bit); signed, unsigned.
  1141. public static Operand EmitBinarySatQAdd(ArmEmitterContext context, Operand op1, Operand op2, bool signed)
  1142. {
  1143. Debug.Assert(((OpCodeSimd)context.CurrOp).Size == 3, "Invalid element size.");
  1144. MethodInfo info = signed
  1145. ? typeof(SoftFallback).GetMethod(nameof(SoftFallback.BinarySignedSatQAdd))
  1146. : typeof(SoftFallback).GetMethod(nameof(SoftFallback.BinaryUnsignedSatQAdd));
  1147. return context.Call(info, op1, op2);
  1148. }
  1149. // TSrcs (64bit) == TDst (64bit); signed, unsigned.
  1150. public static Operand EmitBinarySatQSub(ArmEmitterContext context, Operand op1, Operand op2, bool signed)
  1151. {
  1152. Debug.Assert(((OpCodeSimd)context.CurrOp).Size == 3, "Invalid element size.");
  1153. MethodInfo info = signed
  1154. ? typeof(SoftFallback).GetMethod(nameof(SoftFallback.BinarySignedSatQSub))
  1155. : typeof(SoftFallback).GetMethod(nameof(SoftFallback.BinaryUnsignedSatQSub));
  1156. return context.Call(info, op1, op2);
  1157. }
  1158. // TSrcs (64bit) == TDst (64bit); signed, unsigned.
  1159. public static Operand EmitBinarySatQAccumulate(ArmEmitterContext context, Operand op1, Operand op2, bool signed)
  1160. {
  1161. Debug.Assert(((OpCodeSimd)context.CurrOp).Size == 3, "Invalid element size.");
  1162. MethodInfo info = signed
  1163. ? typeof(SoftFallback).GetMethod(nameof(SoftFallback.BinarySignedSatQAcc))
  1164. : typeof(SoftFallback).GetMethod(nameof(SoftFallback.BinaryUnsignedSatQAcc));
  1165. return context.Call(info, op1, op2);
  1166. }
  1167. public static Operand EmitFloatAbs(ArmEmitterContext context, Operand value, bool single, bool vector)
  1168. {
  1169. Operand mask;
  1170. if (single)
  1171. {
  1172. mask = vector ? X86GetAllElements(context, -0f) : X86GetScalar(context, -0f);
  1173. }
  1174. else
  1175. {
  1176. mask = vector ? X86GetAllElements(context, -0d) : X86GetScalar(context, -0d);
  1177. }
  1178. return context.AddIntrinsic(single ? Intrinsic.X86Andnps : Intrinsic.X86Andnpd, mask, value);
  1179. }
  1180. public static Operand EmitVectorExtractSx(ArmEmitterContext context, int reg, int index, int size)
  1181. {
  1182. return EmitVectorExtract(context, reg, index, size, true);
  1183. }
  1184. public static Operand EmitVectorExtractZx(ArmEmitterContext context, int reg, int index, int size)
  1185. {
  1186. return EmitVectorExtract(context, reg, index, size, false);
  1187. }
  1188. public static Operand EmitVectorExtract(ArmEmitterContext context, int reg, int index, int size, bool signed)
  1189. {
  1190. ThrowIfInvalid(index, size);
  1191. Operand res = default;
  1192. switch (size)
  1193. {
  1194. case 0:
  1195. res = context.VectorExtract8(GetVec(reg), index);
  1196. break;
  1197. case 1:
  1198. res = context.VectorExtract16(GetVec(reg), index);
  1199. break;
  1200. case 2:
  1201. res = context.VectorExtract(OperandType.I32, GetVec(reg), index);
  1202. break;
  1203. case 3:
  1204. res = context.VectorExtract(OperandType.I64, GetVec(reg), index);
  1205. break;
  1206. }
  1207. if (signed)
  1208. {
  1209. switch (size)
  1210. {
  1211. case 0: res = context.SignExtend8 (OperandType.I64, res); break;
  1212. case 1: res = context.SignExtend16(OperandType.I64, res); break;
  1213. case 2: res = context.SignExtend32(OperandType.I64, res); break;
  1214. }
  1215. }
  1216. else
  1217. {
  1218. switch (size)
  1219. {
  1220. case 0: res = context.ZeroExtend8 (OperandType.I64, res); break;
  1221. case 1: res = context.ZeroExtend16(OperandType.I64, res); break;
  1222. case 2: res = context.ZeroExtend32(OperandType.I64, res); break;
  1223. }
  1224. }
  1225. return res;
  1226. }
  1227. public static Operand EmitVectorInsert(ArmEmitterContext context, Operand vector, Operand value, int index, int size)
  1228. {
  1229. ThrowIfInvalid(index, size);
  1230. if (size < 3 && value.Type == OperandType.I64)
  1231. {
  1232. value = context.ConvertI64ToI32(value);
  1233. }
  1234. switch (size)
  1235. {
  1236. case 0: vector = context.VectorInsert8 (vector, value, index); break;
  1237. case 1: vector = context.VectorInsert16(vector, value, index); break;
  1238. case 2: vector = context.VectorInsert (vector, value, index); break;
  1239. case 3: vector = context.VectorInsert (vector, value, index); break;
  1240. }
  1241. return vector;
  1242. }
  1243. public static void ThrowIfInvalid(int index, int size)
  1244. {
  1245. if ((uint)size > 3u)
  1246. {
  1247. throw new ArgumentOutOfRangeException(nameof(size));
  1248. }
  1249. if ((uint)index >= 16u >> size)
  1250. {
  1251. throw new ArgumentOutOfRangeException(nameof(index));
  1252. }
  1253. }
  1254. }
  1255. }