| 12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484148514861487148814891490149114921493149414951496149714981499150015011502150315041505150615071508150915101511151215131514151515161517151815191520152115221523152415251526152715281529153015311532153315341535153615371538153915401541154215431544154515461547154815491550155115521553155415551556155715581559156015611562156315641565156615671568156915701571157215731574157515761577157815791580 |
- using ARMeilleure.CodeGen.Linking;
- using ARMeilleure.CodeGen.Optimizations;
- using ARMeilleure.CodeGen.RegisterAllocators;
- using ARMeilleure.CodeGen.Unwinding;
- using ARMeilleure.Common;
- using ARMeilleure.Diagnostics;
- using ARMeilleure.IntermediateRepresentation;
- using ARMeilleure.Translation;
- using System;
- using System.Collections.Generic;
- using System.Diagnostics;
- using System.Numerics;
- using static ARMeilleure.IntermediateRepresentation.Operand;
- using static ARMeilleure.IntermediateRepresentation.Operand.Factory;
- namespace ARMeilleure.CodeGen.Arm64
- {
- static class CodeGenerator
- {
- private const int DWordScale = 3;
- private const int RegistersCount = 32;
- private const int FpRegister = 29;
- private const int LrRegister = 30;
- private const int SpRegister = 31;
- private const int ZrRegister = 31;
- private enum AccessSize
- {
- Byte,
- Hword,
- Auto
- }
- private static Action<CodeGenContext, Operation>[] _instTable;
- static CodeGenerator()
- {
- _instTable = new Action<CodeGenContext, Operation>[EnumUtils.GetCount(typeof(Instruction))];
- Add(Instruction.Add, GenerateAdd);
- Add(Instruction.BitwiseAnd, GenerateBitwiseAnd);
- Add(Instruction.BitwiseExclusiveOr, GenerateBitwiseExclusiveOr);
- Add(Instruction.BitwiseNot, GenerateBitwiseNot);
- Add(Instruction.BitwiseOr, GenerateBitwiseOr);
- Add(Instruction.BranchIf, GenerateBranchIf);
- Add(Instruction.ByteSwap, GenerateByteSwap);
- Add(Instruction.Call, GenerateCall);
- //Add(Instruction.Clobber, GenerateClobber);
- Add(Instruction.Compare, GenerateCompare);
- Add(Instruction.CompareAndSwap, GenerateCompareAndSwap);
- Add(Instruction.CompareAndSwap16, GenerateCompareAndSwap16);
- Add(Instruction.CompareAndSwap8, GenerateCompareAndSwap8);
- Add(Instruction.ConditionalSelect, GenerateConditionalSelect);
- Add(Instruction.ConvertI64ToI32, GenerateConvertI64ToI32);
- Add(Instruction.ConvertToFP, GenerateConvertToFP);
- Add(Instruction.ConvertToFPUI, GenerateConvertToFPUI);
- Add(Instruction.Copy, GenerateCopy);
- Add(Instruction.CountLeadingZeros, GenerateCountLeadingZeros);
- Add(Instruction.Divide, GenerateDivide);
- Add(Instruction.DivideUI, GenerateDivideUI);
- Add(Instruction.Fill, GenerateFill);
- Add(Instruction.Load, GenerateLoad);
- Add(Instruction.Load16, GenerateLoad16);
- Add(Instruction.Load8, GenerateLoad8);
- Add(Instruction.MemoryBarrier, GenerateMemoryBarrier);
- Add(Instruction.Multiply, GenerateMultiply);
- Add(Instruction.Multiply64HighSI, GenerateMultiply64HighSI);
- Add(Instruction.Multiply64HighUI, GenerateMultiply64HighUI);
- Add(Instruction.Negate, GenerateNegate);
- Add(Instruction.Return, GenerateReturn);
- Add(Instruction.RotateRight, GenerateRotateRight);
- Add(Instruction.ShiftLeft, GenerateShiftLeft);
- Add(Instruction.ShiftRightSI, GenerateShiftRightSI);
- Add(Instruction.ShiftRightUI, GenerateShiftRightUI);
- Add(Instruction.SignExtend16, GenerateSignExtend16);
- Add(Instruction.SignExtend32, GenerateSignExtend32);
- Add(Instruction.SignExtend8, GenerateSignExtend8);
- Add(Instruction.Spill, GenerateSpill);
- Add(Instruction.SpillArg, GenerateSpillArg);
- Add(Instruction.StackAlloc, GenerateStackAlloc);
- Add(Instruction.Store, GenerateStore);
- Add(Instruction.Store16, GenerateStore16);
- Add(Instruction.Store8, GenerateStore8);
- Add(Instruction.Subtract, GenerateSubtract);
- Add(Instruction.Tailcall, GenerateTailcall);
- Add(Instruction.VectorCreateScalar, GenerateVectorCreateScalar);
- Add(Instruction.VectorExtract, GenerateVectorExtract);
- Add(Instruction.VectorExtract16, GenerateVectorExtract16);
- Add(Instruction.VectorExtract8, GenerateVectorExtract8);
- Add(Instruction.VectorInsert, GenerateVectorInsert);
- Add(Instruction.VectorInsert16, GenerateVectorInsert16);
- Add(Instruction.VectorInsert8, GenerateVectorInsert8);
- Add(Instruction.VectorOne, GenerateVectorOne);
- Add(Instruction.VectorZero, GenerateVectorZero);
- Add(Instruction.VectorZeroUpper64, GenerateVectorZeroUpper64);
- Add(Instruction.VectorZeroUpper96, GenerateVectorZeroUpper96);
- Add(Instruction.ZeroExtend16, GenerateZeroExtend16);
- Add(Instruction.ZeroExtend32, GenerateZeroExtend32);
- Add(Instruction.ZeroExtend8, GenerateZeroExtend8);
- static void Add(Instruction inst, Action<CodeGenContext, Operation> func)
- {
- _instTable[(int)inst] = func;
- }
- }
- public static CompiledFunction Generate(CompilerContext cctx)
- {
- ControlFlowGraph cfg = cctx.Cfg;
- Logger.StartPass(PassName.Optimization);
- if (cctx.Options.HasFlag(CompilerOptions.Optimize))
- {
- if (cctx.Options.HasFlag(CompilerOptions.SsaForm))
- {
- Optimizer.RunPass(cfg);
- }
- BlockPlacement.RunPass(cfg);
- }
- Arm64Optimizer.RunPass(cfg);
- Logger.EndPass(PassName.Optimization, cfg);
- Logger.StartPass(PassName.PreAllocation);
- StackAllocator stackAlloc = new();
- PreAllocator.RunPass(cctx, stackAlloc, out int maxCallArgs);
- Logger.EndPass(PassName.PreAllocation, cfg);
- Logger.StartPass(PassName.RegisterAllocation);
- if (cctx.Options.HasFlag(CompilerOptions.SsaForm))
- {
- Ssa.Deconstruct(cfg);
- }
- IRegisterAllocator regAlloc;
- if (cctx.Options.HasFlag(CompilerOptions.Lsra))
- {
- regAlloc = new LinearScanAllocator();
- }
- else
- {
- regAlloc = new HybridAllocator();
- }
- RegisterMasks regMasks = new(
- CallingConvention.GetIntAvailableRegisters(),
- CallingConvention.GetVecAvailableRegisters(),
- CallingConvention.GetIntCallerSavedRegisters(),
- CallingConvention.GetVecCallerSavedRegisters(),
- CallingConvention.GetIntCalleeSavedRegisters(),
- CallingConvention.GetVecCalleeSavedRegisters(),
- RegistersCount);
- AllocationResult allocResult = regAlloc.RunPass(cfg, stackAlloc, regMasks);
- Logger.EndPass(PassName.RegisterAllocation, cfg);
- Logger.StartPass(PassName.CodeGeneration);
- //Console.Error.WriteLine(IRDumper.GetDump(cfg));
- bool relocatable = (cctx.Options & CompilerOptions.Relocatable) != 0;
- CodeGenContext context = new(allocResult, maxCallArgs, cfg.Blocks.Count, relocatable);
- UnwindInfo unwindInfo = WritePrologue(context);
- for (BasicBlock block = cfg.Blocks.First; block != null; block = block.ListNext)
- {
- context.EnterBlock(block);
- for (Operation node = block.Operations.First; node != default;)
- {
- node = GenerateOperation(context, node);
- }
- if (block.SuccessorsCount == 0)
- {
- // The only blocks which can have 0 successors are exit blocks.
- Operation last = block.Operations.Last;
- Debug.Assert(last.Instruction == Instruction.Tailcall ||
- last.Instruction == Instruction.Return);
- }
- else
- {
- BasicBlock succ = block.GetSuccessor(0);
- if (succ != block.ListNext)
- {
- context.JumpTo(succ);
- }
- }
- }
- (byte[] code, RelocInfo relocInfo) = context.GetCode();
- Logger.EndPass(PassName.CodeGeneration);
- return new CompiledFunction(code, unwindInfo, relocInfo);
- }
- private static Operation GenerateOperation(CodeGenContext context, Operation operation)
- {
- if (operation.Instruction == Instruction.Extended)
- {
- CodeGeneratorIntrinsic.GenerateOperation(context, operation);
- }
- else
- {
- if (IsLoadOrStore(operation) &&
- operation.ListNext != default &&
- operation.ListNext.Instruction == operation.Instruction &&
- TryPairMemoryOp(context, operation, operation.ListNext))
- {
- // Skip next operation if we managed to pair them.
- return operation.ListNext.ListNext;
- }
- Action<CodeGenContext, Operation> func = _instTable[(int)operation.Instruction];
- if (func != null)
- {
- func(context, operation);
- }
- else
- {
- throw new ArgumentException($"Invalid instruction \"{operation.Instruction}\".");
- }
- }
- return operation.ListNext;
- }
- private static void GenerateAdd(CodeGenContext context, Operation operation)
- {
- Operand dest = operation.Destination;
- Operand src1 = operation.GetSource(0);
- Operand src2 = operation.GetSource(1);
- // ValidateBinOp(dest, src1, src2);
- if (dest.Type.IsInteger())
- {
- context.Assembler.Add(dest, src1, src2);
- }
- else
- {
- context.Assembler.FaddScalar(dest, src1, src2);
- }
- }
- private static void GenerateBitwiseAnd(CodeGenContext context, Operation operation)
- {
- Operand dest = operation.Destination;
- Operand src1 = operation.GetSource(0);
- Operand src2 = operation.GetSource(1);
- ValidateBinOp(dest, src1, src2);
- Debug.Assert(dest.Type.IsInteger());
- context.Assembler.And(dest, src1, src2);
- }
- private static void GenerateBitwiseExclusiveOr(CodeGenContext context, Operation operation)
- {
- Operand dest = operation.Destination;
- Operand src1 = operation.GetSource(0);
- Operand src2 = operation.GetSource(1);
- ValidateBinOp(dest, src1, src2);
- if (dest.Type.IsInteger())
- {
- context.Assembler.Eor(dest, src1, src2);
- }
- else
- {
- context.Assembler.EorVector(dest, src1, src2);
- }
- }
- private static void GenerateBitwiseNot(CodeGenContext context, Operation operation)
- {
- Operand dest = operation.Destination;
- Operand source = operation.GetSource(0);
- ValidateUnOp(dest, source);
- Debug.Assert(dest.Type.IsInteger());
- context.Assembler.Mvn(dest, source);
- }
- private static void GenerateBitwiseOr(CodeGenContext context, Operation operation)
- {
- Operand dest = operation.Destination;
- Operand src1 = operation.GetSource(0);
- Operand src2 = operation.GetSource(1);
- ValidateBinOp(dest, src1, src2);
- Debug.Assert(dest.Type.IsInteger());
- context.Assembler.Orr(dest, src1, src2);
- }
- private static void GenerateBranchIf(CodeGenContext context, Operation operation)
- {
- Operand comp = operation.GetSource(2);
- Debug.Assert(comp.Kind == OperandKind.Constant);
- var cond = ((Comparison)comp.AsInt32()).ToArmCondition();
- GenerateCompareCommon(context, operation);
- context.JumpTo(cond, context.CurrBlock.GetSuccessor(1));
- }
- private static void GenerateByteSwap(CodeGenContext context, Operation operation)
- {
- Operand dest = operation.Destination;
- Operand source = operation.GetSource(0);
- ValidateUnOp(dest, source);
- Debug.Assert(dest.Type.IsInteger());
- context.Assembler.Rev(dest, source);
- }
- private static void GenerateCall(CodeGenContext context, Operation operation)
- {
- context.Assembler.Blr(operation.GetSource(0));
- }
- private static void GenerateCompare(CodeGenContext context, Operation operation)
- {
- Operand dest = operation.Destination;
- Operand comp = operation.GetSource(2);
- Debug.Assert(dest.Type == OperandType.I32);
- Debug.Assert(comp.Kind == OperandKind.Constant);
- var cond = ((Comparison)comp.AsInt32()).ToArmCondition();
- GenerateCompareCommon(context, operation);
- context.Assembler.Cset(dest, cond);
- }
- private static void GenerateCompareAndSwap(CodeGenContext context, Operation operation)
- {
- if (operation.SourcesCount == 5) // CompareAndSwap128 has 5 sources, compared to CompareAndSwap64/32's 3.
- {
- Operand actualLow = operation.GetDestination(0);
- Operand actualHigh = operation.GetDestination(1);
- Operand temp0 = operation.GetDestination(2);
- Operand temp1 = operation.GetDestination(3);
- Operand address = operation.GetSource(0);
- Operand expectedLow = operation.GetSource(1);
- Operand expectedHigh = operation.GetSource(2);
- Operand desiredLow = operation.GetSource(3);
- Operand desiredHigh = operation.GetSource(4);
- GenerateAtomicDcas(
- context,
- address,
- expectedLow,
- expectedHigh,
- desiredLow,
- desiredHigh,
- actualLow,
- actualHigh,
- temp0,
- temp1);
- }
- else
- {
- Operand actual = operation.GetDestination(0);
- Operand result = operation.GetDestination(1);
- Operand address = operation.GetSource(0);
- Operand expected = operation.GetSource(1);
- Operand desired = operation.GetSource(2);
- GenerateAtomicCas(context, address, expected, desired, actual, result, AccessSize.Auto);
- }
- }
- private static void GenerateCompareAndSwap16(CodeGenContext context, Operation operation)
- {
- Operand actual = operation.GetDestination(0);
- Operand result = operation.GetDestination(1);
- Operand address = operation.GetSource(0);
- Operand expected = operation.GetSource(1);
- Operand desired = operation.GetSource(2);
- GenerateAtomicCas(context, address, expected, desired, actual, result, AccessSize.Hword);
- }
- private static void GenerateCompareAndSwap8(CodeGenContext context, Operation operation)
- {
- Operand actual = operation.GetDestination(0);
- Operand result = operation.GetDestination(1);
- Operand address = operation.GetSource(0);
- Operand expected = operation.GetSource(1);
- Operand desired = operation.GetSource(2);
- GenerateAtomicCas(context, address, expected, desired, actual, result, AccessSize.Byte);
- }
- private static void GenerateCompareCommon(CodeGenContext context, Operation operation)
- {
- Operand src1 = operation.GetSource(0);
- Operand src2 = operation.GetSource(1);
- EnsureSameType(src1, src2);
- Debug.Assert(src1.Type.IsInteger());
- context.Assembler.Cmp(src1, src2);
- }
- private static void GenerateConditionalSelect(CodeGenContext context, Operation operation)
- {
- Operand dest = operation.Destination;
- Operand src1 = operation.GetSource(0);
- Operand src2 = operation.GetSource(1);
- Operand src3 = operation.GetSource(2);
- EnsureSameType(dest, src2, src3);
- Debug.Assert(dest.Type.IsInteger());
- Debug.Assert(src1.Type == OperandType.I32);
- context.Assembler.Cmp (src1, Const(src1.Type, 0));
- context.Assembler.Csel(dest, src2, src3, ArmCondition.Ne);
- }
- private static void GenerateConvertI64ToI32(CodeGenContext context, Operation operation)
- {
- Operand dest = operation.Destination;
- Operand source = operation.GetSource(0);
- Debug.Assert(dest.Type == OperandType.I32 && source.Type == OperandType.I64);
- context.Assembler.Mov(dest, Register(source, OperandType.I32));
- }
- private static void GenerateConvertToFP(CodeGenContext context, Operation operation)
- {
- Operand dest = operation.Destination;
- Operand source = operation.GetSource(0);
- Debug.Assert(dest.Type == OperandType.FP32 || dest.Type == OperandType.FP64);
- Debug.Assert(dest.Type != source.Type);
- Debug.Assert(source.Type != OperandType.V128);
- if (source.Type.IsInteger())
- {
- context.Assembler.ScvtfScalar(dest, source);
- }
- else
- {
- context.Assembler.FcvtScalar(dest, source);
- }
- }
- private static void GenerateConvertToFPUI(CodeGenContext context, Operation operation)
- {
- Operand dest = operation.Destination;
- Operand source = operation.GetSource(0);
- Debug.Assert(dest.Type == OperandType.FP32 || dest.Type == OperandType.FP64);
- Debug.Assert(dest.Type != source.Type);
- Debug.Assert(source.Type.IsInteger());
- context.Assembler.UcvtfScalar(dest, source);
- }
- private static void GenerateCopy(CodeGenContext context, Operation operation)
- {
- Operand dest = operation.Destination;
- Operand source = operation.GetSource(0);
- EnsureSameType(dest, source);
- Debug.Assert(dest.Type.IsInteger() || source.Kind != OperandKind.Constant);
- // Moves to the same register are useless.
- if (dest.Kind == source.Kind && dest.Value == source.Value)
- {
- return;
- }
- if (dest.Kind == OperandKind.Register && source.Kind == OperandKind.Constant)
- {
- if (source.Relocatable)
- {
- context.ReserveRelocatableConstant(dest, source.Symbol, source.Value);
- }
- else
- {
- GenerateConstantCopy(context, dest, source.Value);
- }
- }
- else
- {
- context.Assembler.Mov(dest, source);
- }
- }
- private static void GenerateCountLeadingZeros(CodeGenContext context, Operation operation)
- {
- Operand dest = operation.Destination;
- Operand source = operation.GetSource(0);
- EnsureSameType(dest, source);
- Debug.Assert(dest.Type.IsInteger());
- context.Assembler.Clz(dest, source);
- }
- private static void GenerateDivide(CodeGenContext context, Operation operation)
- {
- Operand dest = operation.Destination;
- Operand dividend = operation.GetSource(0);
- Operand divisor = operation.GetSource(1);
- ValidateBinOp(dest, dividend, divisor);
- if (dest.Type.IsInteger())
- {
- context.Assembler.Sdiv(dest, dividend, divisor);
- }
- else
- {
- context.Assembler.FdivScalar(dest, dividend, divisor);
- }
- }
- private static void GenerateDivideUI(CodeGenContext context, Operation operation)
- {
- Operand dest = operation.Destination;
- Operand dividend = operation.GetSource(0);
- Operand divisor = operation.GetSource(1);
- ValidateBinOp(dest, dividend, divisor);
- context.Assembler.Udiv(dest, dividend, divisor);
- }
- private static void GenerateLoad(CodeGenContext context, Operation operation)
- {
- Operand value = operation.Destination;
- Operand address = operation.GetSource(0);
- context.Assembler.Ldr(value, address);
- }
- private static void GenerateLoad16(CodeGenContext context, Operation operation)
- {
- Operand value = operation.Destination;
- Operand address = operation.GetSource(0);
- Debug.Assert(value.Type.IsInteger());
- context.Assembler.LdrhRiUn(value, address, 0);
- }
- private static void GenerateLoad8(CodeGenContext context, Operation operation)
- {
- Operand value = operation.Destination;
- Operand address = operation.GetSource(0);
- Debug.Assert(value.Type.IsInteger());
- context.Assembler.LdrbRiUn(value, address, 0);
- }
- private static void GenerateMemoryBarrier(CodeGenContext context, Operation operation)
- {
- context.Assembler.Dmb(0xf);
- }
- private static void GenerateMultiply(CodeGenContext context, Operation operation)
- {
- Operand dest = operation.Destination;
- Operand src1 = operation.GetSource(0);
- Operand src2 = operation.GetSource(1);
- EnsureSameType(dest, src1, src2);
- if (dest.Type.IsInteger())
- {
- context.Assembler.Mul(dest, src1, src2);
- }
- else
- {
- context.Assembler.FmulScalar(dest, src1, src2);
- }
- }
- private static void GenerateMultiply64HighSI(CodeGenContext context, Operation operation)
- {
- Operand dest = operation.Destination;
- Operand src1 = operation.GetSource(0);
- Operand src2 = operation.GetSource(1);
- EnsureSameType(dest, src1, src2);
- Debug.Assert(dest.Type == OperandType.I64);
- context.Assembler.Smulh(dest, src1, src2);
- }
- private static void GenerateMultiply64HighUI(CodeGenContext context, Operation operation)
- {
- Operand dest = operation.Destination;
- Operand src1 = operation.GetSource(0);
- Operand src2 = operation.GetSource(1);
- EnsureSameType(dest, src1, src2);
- Debug.Assert(dest.Type == OperandType.I64);
- context.Assembler.Umulh(dest, src1, src2);
- }
- private static void GenerateNegate(CodeGenContext context, Operation operation)
- {
- Operand dest = operation.Destination;
- Operand source = operation.GetSource(0);
- ValidateUnOp(dest, source);
- if (dest.Type.IsInteger())
- {
- context.Assembler.Neg(dest, source);
- }
- else
- {
- context.Assembler.FnegScalar(dest, source);
- }
- }
- private static void GenerateLoad(CodeGenContext context, Operand value, Operand address, int offset)
- {
- if (CodeGenCommon.ConstFitsOnUImm12(offset, value.Type))
- {
- context.Assembler.LdrRiUn(value, address, offset);
- }
- else if (CodeGenCommon.ConstFitsOnSImm9(offset))
- {
- context.Assembler.Ldur(value, address, offset);
- }
- else
- {
- Operand tempAddress = Register(CodeGenCommon.ReservedRegister);
- GenerateConstantCopy(context, tempAddress, (ulong)offset);
- context.Assembler.Add(tempAddress, address, tempAddress, ArmExtensionType.Uxtx); // Address might be SP and must be the first input.
- context.Assembler.LdrRiUn(value, tempAddress, 0);
- }
- }
- private static void GenerateReturn(CodeGenContext context, Operation operation)
- {
- WriteEpilogue(context);
- context.Assembler.Ret(Register(LrRegister));
- }
- private static void GenerateRotateRight(CodeGenContext context, Operation operation)
- {
- Operand dest = operation.Destination;
- Operand src1 = operation.GetSource(0);
- Operand src2 = operation.GetSource(1);
- ValidateShift(dest, src1, src2);
- context.Assembler.Ror(dest, src1, src2);
- }
- private static void GenerateShiftLeft(CodeGenContext context, Operation operation)
- {
- Operand dest = operation.Destination;
- Operand src1 = operation.GetSource(0);
- Operand src2 = operation.GetSource(1);
- ValidateShift(dest, src1, src2);
- context.Assembler.Lsl(dest, src1, src2);
- }
- private static void GenerateShiftRightSI(CodeGenContext context, Operation operation)
- {
- Operand dest = operation.Destination;
- Operand src1 = operation.GetSource(0);
- Operand src2 = operation.GetSource(1);
- ValidateShift(dest, src1, src2);
- context.Assembler.Asr(dest, src1, src2);
- }
- private static void GenerateShiftRightUI(CodeGenContext context, Operation operation)
- {
- Operand dest = operation.Destination;
- Operand src1 = operation.GetSource(0);
- Operand src2 = operation.GetSource(1);
- ValidateShift(dest, src1, src2);
- context.Assembler.Lsr(dest, src1, src2);
- }
- private static void GenerateSignExtend16(CodeGenContext context, Operation operation)
- {
- Operand dest = operation.Destination;
- Operand source = operation.GetSource(0);
- Debug.Assert(dest.Type.IsInteger() && source.Type.IsInteger());
- context.Assembler.Sxth(dest, source);
- }
- private static void GenerateSignExtend32(CodeGenContext context, Operation operation)
- {
- Operand dest = operation.Destination;
- Operand source = operation.GetSource(0);
- Debug.Assert(dest.Type.IsInteger() && source.Type.IsInteger());
- context.Assembler.Sxtw(dest, source);
- }
- private static void GenerateSignExtend8(CodeGenContext context, Operation operation)
- {
- Operand dest = operation.Destination;
- Operand source = operation.GetSource(0);
- Debug.Assert(dest.Type.IsInteger() && source.Type.IsInteger());
- context.Assembler.Sxtb(dest, source);
- }
- private static void GenerateFill(CodeGenContext context, Operation operation)
- {
- Operand dest = operation.Destination;
- Operand offset = operation.GetSource(0);
- Debug.Assert(offset.Kind == OperandKind.Constant);
- int offs = offset.AsInt32() + context.CallArgsRegionSize + context.FpLrSaveRegionSize;
- GenerateLoad(context, dest, Register(SpRegister), offs);
- }
- private static void GenerateStore(CodeGenContext context, Operand value, Operand address, int offset)
- {
- if (CodeGenCommon.ConstFitsOnUImm12(offset, value.Type))
- {
- context.Assembler.StrRiUn(value, address, offset);
- }
- else if (CodeGenCommon.ConstFitsOnSImm9(offset))
- {
- context.Assembler.Stur(value, address, offset);
- }
- else
- {
- Operand tempAddress = Register(CodeGenCommon.ReservedRegister);
- GenerateConstantCopy(context, tempAddress, (ulong)offset);
- context.Assembler.Add(tempAddress, address, tempAddress, ArmExtensionType.Uxtx); // Address might be SP and must be the first input.
- context.Assembler.StrRiUn(value, tempAddress, 0);
- }
- }
- private static void GenerateSpill(CodeGenContext context, Operation operation)
- {
- GenerateSpill(context, operation, context.CallArgsRegionSize + context.FpLrSaveRegionSize);
- }
- private static void GenerateSpillArg(CodeGenContext context, Operation operation)
- {
- GenerateSpill(context, operation, 0);
- }
- private static void GenerateStackAlloc(CodeGenContext context, Operation operation)
- {
- Operand dest = operation.Destination;
- Operand offset = operation.GetSource(0);
- Debug.Assert(offset.Kind == OperandKind.Constant);
- int offs = offset.AsInt32() + context.CallArgsRegionSize + context.FpLrSaveRegionSize;
- context.Assembler.Add(dest, Register(SpRegister), Const(dest.Type, offs));
- }
- private static void GenerateStore(CodeGenContext context, Operation operation)
- {
- Operand value = operation.GetSource(1);
- Operand address = operation.GetSource(0);
- context.Assembler.Str(value, address);
- }
- private static void GenerateStore16(CodeGenContext context, Operation operation)
- {
- Operand value = operation.GetSource(1);
- Operand address = operation.GetSource(0);
- Debug.Assert(value.Type.IsInteger());
- context.Assembler.StrhRiUn(value, address, 0);
- }
- private static void GenerateStore8(CodeGenContext context, Operation operation)
- {
- Operand value = operation.GetSource(1);
- Operand address = operation.GetSource(0);
- Debug.Assert(value.Type.IsInteger());
- context.Assembler.StrbRiUn(value, address, 0);
- }
- private static void GenerateSpill(CodeGenContext context, Operation operation, int baseOffset)
- {
- Operand offset = operation.GetSource(0);
- Operand source = operation.GetSource(1);
- Debug.Assert(offset.Kind == OperandKind.Constant);
- int offs = offset.AsInt32() + baseOffset;
- GenerateStore(context, source, Register(SpRegister), offs);
- }
- private static void GenerateSubtract(CodeGenContext context, Operation operation)
- {
- Operand dest = operation.Destination;
- Operand src1 = operation.GetSource(0);
- Operand src2 = operation.GetSource(1);
- // ValidateBinOp(dest, src1, src2);
- if (dest.Type.IsInteger())
- {
- context.Assembler.Sub(dest, src1, src2);
- }
- else
- {
- context.Assembler.FsubScalar(dest, src1, src2);
- }
- }
- private static void GenerateTailcall(CodeGenContext context, Operation operation)
- {
- WriteEpilogue(context);
- context.Assembler.Br(operation.GetSource(0));
- }
- private static void GenerateVectorCreateScalar(CodeGenContext context, Operation operation)
- {
- Operand dest = operation.Destination;
- Operand source = operation.GetSource(0);
- if (dest != default)
- {
- Debug.Assert(!dest.Type.IsInteger() && source.Type.IsInteger());
- OperandType destType = source.Type == OperandType.I64 ? OperandType.FP64 : OperandType.FP32;
- context.Assembler.Fmov(Register(dest, destType), source, topHalf: false);
- }
- }
- private static void GenerateVectorExtract(CodeGenContext context, Operation operation)
- {
- Operand dest = operation.Destination; // Value
- Operand src1 = operation.GetSource(0); // Vector
- Operand src2 = operation.GetSource(1); // Index
- Debug.Assert(src1.Type == OperandType.V128);
- Debug.Assert(src2.Kind == OperandKind.Constant);
- byte index = src2.AsByte();
- Debug.Assert(index < OperandType.V128.GetSizeInBytes() / dest.Type.GetSizeInBytes());
- if (dest.Type.IsInteger())
- {
- context.Assembler.Umov(dest, src1, index, dest.Type == OperandType.I64 ? 3 : 2);
- }
- else
- {
- context.Assembler.DupScalar(dest, src1, index, dest.Type == OperandType.FP64 ? 3 : 2);
- }
- }
- private static void GenerateVectorExtract16(CodeGenContext context, Operation operation)
- {
- Operand dest = operation.Destination; // Value
- Operand src1 = operation.GetSource(0); // Vector
- Operand src2 = operation.GetSource(1); // Index
- Debug.Assert(src1.Type == OperandType.V128);
- Debug.Assert(src2.Kind == OperandKind.Constant);
- byte index = src2.AsByte();
- Debug.Assert(index < 8);
- context.Assembler.Umov(dest, src1, index, 1);
- }
- private static void GenerateVectorExtract8(CodeGenContext context, Operation operation)
- {
- Operand dest = operation.Destination; // Value
- Operand src1 = operation.GetSource(0); // Vector
- Operand src2 = operation.GetSource(1); // Index
- Debug.Assert(src1.Type == OperandType.V128);
- Debug.Assert(src2.Kind == OperandKind.Constant);
- byte index = src2.AsByte();
- Debug.Assert(index < 16);
- context.Assembler.Umov(dest, src1, index, 0);
- }
- private static void GenerateVectorInsert(CodeGenContext context, Operation operation)
- {
- Operand dest = operation.Destination;
- Operand src1 = operation.GetSource(0); // Vector
- Operand src2 = operation.GetSource(1); // Value
- Operand src3 = operation.GetSource(2); // Index
- EnsureSameReg(dest, src1);
- Debug.Assert(src1.Type == OperandType.V128);
- Debug.Assert(src3.Kind == OperandKind.Constant);
- byte index = src3.AsByte();
- if (src2.Type.IsInteger())
- {
- context.Assembler.Ins(dest, src2, index, src2.Type == OperandType.I64 ? 3 : 2);
- }
- else
- {
- context.Assembler.Ins(dest, src2, 0, index, src2.Type == OperandType.FP64 ? 3 : 2);
- }
- }
- private static void GenerateVectorInsert16(CodeGenContext context, Operation operation)
- {
- Operand dest = operation.Destination;
- Operand src1 = operation.GetSource(0); // Vector
- Operand src2 = operation.GetSource(1); // Value
- Operand src3 = operation.GetSource(2); // Index
- EnsureSameReg(dest, src1);
- Debug.Assert(src1.Type == OperandType.V128);
- Debug.Assert(src3.Kind == OperandKind.Constant);
- byte index = src3.AsByte();
- context.Assembler.Ins(dest, src2, index, 1);
- }
- private static void GenerateVectorInsert8(CodeGenContext context, Operation operation)
- {
- Operand dest = operation.Destination;
- Operand src1 = operation.GetSource(0); // Vector
- Operand src2 = operation.GetSource(1); // Value
- Operand src3 = operation.GetSource(2); // Index
- EnsureSameReg(dest, src1);
- Debug.Assert(src1.Type == OperandType.V128);
- Debug.Assert(src3.Kind == OperandKind.Constant);
- byte index = src3.AsByte();
- context.Assembler.Ins(dest, src2, index, 0);
- }
- private static void GenerateVectorOne(CodeGenContext context, Operation operation)
- {
- Operand dest = operation.Destination;
- Debug.Assert(!dest.Type.IsInteger());
- context.Assembler.CmeqVector(dest, dest, dest, 2);
- }
- private static void GenerateVectorZero(CodeGenContext context, Operation operation)
- {
- Operand dest = operation.Destination;
- Debug.Assert(!dest.Type.IsInteger());
- context.Assembler.EorVector(dest, dest, dest);
- }
- private static void GenerateVectorZeroUpper64(CodeGenContext context, Operation operation)
- {
- Operand dest = operation.Destination;
- Operand source = operation.GetSource(0);
- Debug.Assert(dest.Type == OperandType.V128 && source.Type == OperandType.V128);
- context.Assembler.Fmov(Register(dest, OperandType.FP64), Register(source, OperandType.FP64));
- }
- private static void GenerateVectorZeroUpper96(CodeGenContext context, Operation operation)
- {
- Operand dest = operation.Destination;
- Operand source = operation.GetSource(0);
- Debug.Assert(dest.Type == OperandType.V128 && source.Type == OperandType.V128);
- context.Assembler.Fmov(Register(dest, OperandType.FP32), Register(source, OperandType.FP32));
- }
- private static void GenerateZeroExtend16(CodeGenContext context, Operation operation)
- {
- Operand dest = operation.Destination;
- Operand source = operation.GetSource(0);
- Debug.Assert(dest.Type.IsInteger() && source.Type.IsInteger());
- context.Assembler.Uxth(dest, source);
- }
- private static void GenerateZeroExtend32(CodeGenContext context, Operation operation)
- {
- Operand dest = operation.Destination;
- Operand source = operation.GetSource(0);
- Debug.Assert(dest.Type.IsInteger() && source.Type.IsInteger());
- // We can eliminate the move if source is already 32-bit and the registers are the same.
- if (dest.Value == source.Value && source.Type == OperandType.I32)
- {
- return;
- }
- context.Assembler.Mov(Register(dest.GetRegister().Index, OperandType.I32), source);
- }
- private static void GenerateZeroExtend8(CodeGenContext context, Operation operation)
- {
- Operand dest = operation.Destination;
- Operand source = operation.GetSource(0);
- Debug.Assert(dest.Type.IsInteger() && source.Type.IsInteger());
- context.Assembler.Uxtb(dest, source);
- }
- private static UnwindInfo WritePrologue(CodeGenContext context)
- {
- List<UnwindPushEntry> pushEntries = new List<UnwindPushEntry>();
- Operand rsp = Register(SpRegister);
- int intMask = CallingConvention.GetIntCalleeSavedRegisters() & context.AllocResult.IntUsedRegisters;
- int vecMask = CallingConvention.GetFpCalleeSavedRegisters() & context.AllocResult.VecUsedRegisters;
- int intCalleeSavedRegsCount = BitOperations.PopCount((uint)intMask);
- int vecCalleeSavedRegsCount = BitOperations.PopCount((uint)vecMask);
- int calleeSaveRegionSize = Align16(intCalleeSavedRegsCount * 8 + vecCalleeSavedRegsCount * 8);
- int offset = 0;
- WritePrologueCalleeSavesPreIndexed(context, pushEntries, ref intMask, ref offset, calleeSaveRegionSize, OperandType.I64);
- WritePrologueCalleeSavesPreIndexed(context, pushEntries, ref vecMask, ref offset, calleeSaveRegionSize, OperandType.FP64);
- int localSize = Align16(context.AllocResult.SpillRegionSize + context.FpLrSaveRegionSize);
- int outArgsSize = context.CallArgsRegionSize;
- if (CodeGenCommon.ConstFitsOnSImm7(localSize, DWordScale))
- {
- if (context.HasCall)
- {
- context.Assembler.StpRiPre(Register(FpRegister), Register(LrRegister), rsp, -localSize);
- context.Assembler.MovSp(Register(FpRegister), rsp);
- }
- if (outArgsSize != 0)
- {
- context.Assembler.Sub(rsp, rsp, Const(OperandType.I64, outArgsSize));
- }
- }
- else
- {
- int frameSize = localSize + outArgsSize;
- if (frameSize != 0)
- {
- if (CodeGenCommon.ConstFitsOnUImm12(frameSize))
- {
- context.Assembler.Sub(rsp, rsp, Const(OperandType.I64, frameSize));
- }
- else
- {
- Operand tempSize = Register(CodeGenCommon.ReservedRegister);
- GenerateConstantCopy(context, tempSize, (ulong)frameSize);
- context.Assembler.Sub(rsp, rsp, tempSize, ArmExtensionType.Uxtx);
- }
- }
- context.Assembler.StpRiUn(Register(FpRegister), Register(LrRegister), rsp, outArgsSize);
- if (outArgsSize != 0)
- {
- context.Assembler.Add(Register(FpRegister), Register(SpRegister), Const(OperandType.I64, outArgsSize));
- }
- else
- {
- context.Assembler.MovSp(Register(FpRegister), Register(SpRegister));
- }
- }
- return new UnwindInfo(pushEntries.ToArray(), context.StreamOffset);
- }
- private static void WritePrologueCalleeSavesPreIndexed(
- CodeGenContext context,
- List<UnwindPushEntry> pushEntries,
- ref int mask,
- ref int offset,
- int calleeSaveRegionSize,
- OperandType type)
- {
- if ((BitOperations.PopCount((uint)mask) & 1) != 0)
- {
- int reg = BitOperations.TrailingZeroCount(mask);
- pushEntries.Add(new UnwindPushEntry(UnwindPseudoOp.PushReg, context.StreamOffset, regIndex: reg));
- mask &= ~(1 << reg);
- if (offset != 0)
- {
- context.Assembler.StrRiUn(Register(reg, type), Register(SpRegister), offset);
- }
- else
- {
- context.Assembler.StrRiPre(Register(reg, type), Register(SpRegister), -calleeSaveRegionSize);
- }
- offset += type.GetSizeInBytes();
- }
- while (mask != 0)
- {
- int reg = BitOperations.TrailingZeroCount(mask);
- pushEntries.Add(new UnwindPushEntry(UnwindPseudoOp.PushReg, context.StreamOffset, regIndex: reg));
- mask &= ~(1 << reg);
- int reg2 = BitOperations.TrailingZeroCount(mask);
- pushEntries.Add(new UnwindPushEntry(UnwindPseudoOp.PushReg, context.StreamOffset, regIndex: reg2));
- mask &= ~(1 << reg2);
- if (offset != 0)
- {
- context.Assembler.StpRiUn(Register(reg, type), Register(reg2, type), Register(SpRegister), offset);
- }
- else
- {
- context.Assembler.StpRiPre(Register(reg, type), Register(reg2, type), Register(SpRegister), -calleeSaveRegionSize);
- }
- offset += type.GetSizeInBytes() * 2;
- }
- }
- private static void WriteEpilogue(CodeGenContext context)
- {
- Operand rsp = Register(SpRegister);
- int localSize = Align16(context.AllocResult.SpillRegionSize + context.FpLrSaveRegionSize);
- int outArgsSize = context.CallArgsRegionSize;
- if (CodeGenCommon.ConstFitsOnSImm7(localSize, DWordScale))
- {
- if (outArgsSize != 0)
- {
- context.Assembler.Add(rsp, rsp, Const(OperandType.I64, outArgsSize));
- }
- if (context.HasCall)
- {
- context.Assembler.LdpRiPost(Register(FpRegister), Register(LrRegister), rsp, localSize);
- }
- }
- else
- {
- if (context.HasCall)
- {
- context.Assembler.LdpRiUn(Register(FpRegister), Register(LrRegister), rsp, outArgsSize);
- }
- int frameSize = localSize + outArgsSize;
- if (frameSize != 0)
- {
- if (CodeGenCommon.ConstFitsOnUImm12(frameSize))
- {
- context.Assembler.Add(rsp, rsp, Const(OperandType.I64, frameSize));
- }
- else
- {
- Operand tempSize = Register(CodeGenCommon.ReservedRegister);
- GenerateConstantCopy(context, tempSize, (ulong)frameSize);
- context.Assembler.Add(rsp, rsp, tempSize, ArmExtensionType.Uxtx);
- }
- }
- }
- int intMask = CallingConvention.GetIntCalleeSavedRegisters() & context.AllocResult.IntUsedRegisters;
- int vecMask = CallingConvention.GetFpCalleeSavedRegisters() & context.AllocResult.VecUsedRegisters;
- int intCalleeSavedRegsCount = BitOperations.PopCount((uint)intMask);
- int vecCalleeSavedRegsCount = BitOperations.PopCount((uint)vecMask);
- int offset = intCalleeSavedRegsCount * 8 + vecCalleeSavedRegsCount * 8;
- int calleeSaveRegionSize = Align16(offset);
- WriteEpilogueCalleeSavesPostIndexed(context, ref vecMask, ref offset, calleeSaveRegionSize, OperandType.FP64);
- WriteEpilogueCalleeSavesPostIndexed(context, ref intMask, ref offset, calleeSaveRegionSize, OperandType.I64);
- }
- private static void WriteEpilogueCalleeSavesPostIndexed(
- CodeGenContext context,
- ref int mask,
- ref int offset,
- int calleeSaveRegionSize,
- OperandType type)
- {
- while (mask != 0)
- {
- int reg = BitUtils.HighestBitSet(mask);
- mask &= ~(1 << reg);
- if (mask != 0)
- {
- int reg2 = BitUtils.HighestBitSet(mask);
- mask &= ~(1 << reg2);
- offset -= type.GetSizeInBytes() * 2;
- if (offset != 0)
- {
- context.Assembler.LdpRiUn(Register(reg2, type), Register(reg, type), Register(SpRegister), offset);
- }
- else
- {
- context.Assembler.LdpRiPost(Register(reg2, type), Register(reg, type), Register(SpRegister), calleeSaveRegionSize);
- }
- }
- else
- {
- offset -= type.GetSizeInBytes();
- if (offset != 0)
- {
- context.Assembler.LdrRiUn(Register(reg, type), Register(SpRegister), offset);
- }
- else
- {
- context.Assembler.LdrRiPost(Register(reg, type), Register(SpRegister), calleeSaveRegionSize);
- }
- }
- }
- }
- private static void GenerateConstantCopy(CodeGenContext context, Operand dest, ulong value)
- {
- if (value == 0)
- {
- context.Assembler.Mov(dest, Register(ZrRegister, dest.Type));
- }
- else if (CodeGenCommon.TryEncodeBitMask(dest.Type, value, out _, out _, out _))
- {
- context.Assembler.Orr(dest, Register(ZrRegister, dest.Type), Const(dest.Type, (long)value));
- }
- else
- {
- int hw = 0;
- bool first = true;
- while (value != 0)
- {
- int valueLow = (ushort)value;
- if (valueLow != 0)
- {
- if (first)
- {
- context.Assembler.Movz(dest, valueLow, hw);
- first = false;
- }
- else
- {
- context.Assembler.Movk(dest, valueLow, hw);
- }
- }
- hw++;
- value >>= 16;
- }
- }
- }
- private static void GenerateAtomicCas(
- CodeGenContext context,
- Operand address,
- Operand expected,
- Operand desired,
- Operand actual,
- Operand result,
- AccessSize accessSize)
- {
- int startOffset = context.StreamOffset;
- switch (accessSize)
- {
- case AccessSize.Byte:
- context.Assembler.Ldaxrb(actual, address);
- break;
- case AccessSize.Hword:
- context.Assembler.Ldaxrh(actual, address);
- break;
- default:
- context.Assembler.Ldaxr(actual, address);
- break;
- }
- context.Assembler.Cmp(actual, expected);
- context.JumpToNear(ArmCondition.Ne);
- switch (accessSize)
- {
- case AccessSize.Byte:
- context.Assembler.Stlxrb(desired, address, result);
- break;
- case AccessSize.Hword:
- context.Assembler.Stlxrh(desired, address, result);
- break;
- default:
- context.Assembler.Stlxr(desired, address, result);
- break;
- }
- context.Assembler.Cbnz(result, startOffset - context.StreamOffset); // Retry if store failed.
- context.JumpHere();
- context.Assembler.Clrex();
- }
- private static void GenerateAtomicDcas(
- CodeGenContext context,
- Operand address,
- Operand expectedLow,
- Operand expectedHigh,
- Operand desiredLow,
- Operand desiredHigh,
- Operand actualLow,
- Operand actualHigh,
- Operand temp0,
- Operand temp1)
- {
- int startOffset = context.StreamOffset;
- context.Assembler.Ldaxp(actualLow, actualHigh, address);
- context.Assembler.Eor(temp0, actualHigh, expectedHigh);
- context.Assembler.Eor(temp1, actualLow, expectedLow);
- context.Assembler.Orr(temp0, temp1, temp0);
- context.JumpToNearIfNotZero(temp0);
- Operand result = Register(temp0, OperandType.I32);
- context.Assembler.Stlxp(desiredLow, desiredHigh, address, result);
- context.Assembler.Cbnz(result, startOffset - context.StreamOffset); // Retry if store failed.
- context.JumpHere();
- context.Assembler.Clrex();
- }
- private static bool TryPairMemoryOp(CodeGenContext context, Operation currentOp, Operation nextOp)
- {
- if (!TryGetMemOpBaseAndOffset(currentOp, out Operand op1Base, out int op1Offset))
- {
- return false;
- }
- if (!TryGetMemOpBaseAndOffset(nextOp, out Operand op2Base, out int op2Offset))
- {
- return false;
- }
- if (op1Base != op2Base)
- {
- return false;
- }
- OperandType valueType = GetMemOpValueType(currentOp);
- if (valueType != GetMemOpValueType(nextOp) || op1Offset + valueType.GetSizeInBytes() != op2Offset)
- {
- return false;
- }
- if (!CodeGenCommon.ConstFitsOnSImm7(op1Offset, valueType.GetSizeInBytesLog2()))
- {
- return false;
- }
- if (currentOp.Instruction == Instruction.Load)
- {
- context.Assembler.LdpRiUn(currentOp.Destination, nextOp.Destination, op1Base, op1Offset);
- }
- else if (currentOp.Instruction == Instruction.Store)
- {
- context.Assembler.StpRiUn(currentOp.GetSource(1), nextOp.GetSource(1), op1Base, op1Offset);
- }
- else
- {
- return false;
- }
- return true;
- }
- private static bool IsLoadOrStore(Operation operation)
- {
- return operation.Instruction == Instruction.Load || operation.Instruction == Instruction.Store;
- }
- private static OperandType GetMemOpValueType(Operation operation)
- {
- if (operation.Destination != default)
- {
- return operation.Destination.Type;
- }
- return operation.GetSource(1).Type;
- }
- private static bool TryGetMemOpBaseAndOffset(Operation operation, out Operand baseAddress, out int offset)
- {
- baseAddress = default;
- offset = 0;
- Operand address = operation.GetSource(0);
- if (address.Kind != OperandKind.Memory)
- {
- return false;
- }
- MemoryOperand memOp = address.GetMemory();
- Operand baseOp = memOp.BaseAddress;
- if (baseOp == default)
- {
- baseOp = memOp.Index;
- if (baseOp == default || memOp.Scale != Multiplier.x1)
- {
- return false;
- }
- }
- if (memOp.Index != default)
- {
- return false;
- }
- baseAddress = memOp.BaseAddress;
- offset = memOp.Displacement;
- return true;
- }
- private static Operand Register(Operand operand, OperandType type = OperandType.I64)
- {
- return Register(operand.GetRegister().Index, type);
- }
- private static Operand Register(int register, OperandType type = OperandType.I64)
- {
- return Factory.Register(register, RegisterType.Integer, type);
- }
- private static int Align16(int value)
- {
- return (value + 0xf) & ~0xf;
- }
- [Conditional("DEBUG")]
- private static void ValidateUnOp(Operand dest, Operand source)
- {
- // Destination and source aren't forced to be equals
- // EnsureSameReg (dest, source);
- EnsureSameType(dest, source);
- }
- [Conditional("DEBUG")]
- private static void ValidateBinOp(Operand dest, Operand src1, Operand src2)
- {
- // Destination and source aren't forced to be equals
- // EnsureSameReg (dest, src1);
- EnsureSameType(dest, src1, src2);
- }
- [Conditional("DEBUG")]
- private static void ValidateShift(Operand dest, Operand src1, Operand src2)
- {
- // Destination and source aren't forced to be equals
- // EnsureSameReg (dest, src1);
- EnsureSameType(dest, src1);
- Debug.Assert(dest.Type.IsInteger() && src2.Type == OperandType.I32);
- }
- private static void EnsureSameReg(Operand op1, Operand op2)
- {
- Debug.Assert(op1.Kind == OperandKind.Register || op1.Kind == OperandKind.Memory);
- Debug.Assert(op1.Kind == op2.Kind);
- Debug.Assert(op1.Value == op2.Value);
- }
- private static void EnsureSameType(Operand op1, Operand op2)
- {
- Debug.Assert(op1.Type == op2.Type);
- }
- private static void EnsureSameType(Operand op1, Operand op2, Operand op3)
- {
- Debug.Assert(op1.Type == op2.Type);
- Debug.Assert(op1.Type == op3.Type);
- }
- private static void EnsureSameType(Operand op1, Operand op2, Operand op3, Operand op4)
- {
- Debug.Assert(op1.Type == op2.Type);
- Debug.Assert(op1.Type == op3.Type);
- Debug.Assert(op1.Type == op4.Type);
- }
- }
- }
|