Index: include/llvm/MC/MCAsmBackend.h =================================================================== --- include/llvm/MC/MCAsmBackend.h +++ include/llvm/MC/MCAsmBackend.h @@ -132,6 +132,10 @@ /// \return - True on success. virtual bool writeNopData(uint64_t Count, MCObjectWriter *OW) const = 0; + /// Give backend an opportunity to finish layout after relaxation + virtual void finishLayout(MCAssembler const &Asm, + MCAsmLayout &Layout) const {} + /// Handle any target-specific assembler flags. By default, do nothing. virtual void handleAssemblerFlag(MCAssemblerFlag Flag) {} Index: lib/MC/MCAssembler.cpp =================================================================== --- lib/MC/MCAssembler.cpp +++ lib/MC/MCAssembler.cpp @@ -914,4 +914,5 @@ for (unsigned int i = 0, n = Layout.getSectionOrder().size(); i != n; ++i) { Layout.getFragmentOffset(&*Layout.getSectionOrder()[i]->rbegin()); } + getBackend().finishLayout(*this, Layout); } Index: lib/Target/Hexagon/MCTargetDesc/HexagonAsmBackend.cpp =================================================================== --- lib/Target/Hexagon/MCTargetDesc/HexagonAsmBackend.cpp +++ lib/Target/Hexagon/MCTargetDesc/HexagonAsmBackend.cpp @@ -11,6 +11,8 @@ #include "HexagonFixupKinds.h" #include "HexagonMCTargetDesc.h" #include "MCTargetDesc/HexagonBaseInfo.h" +#include "MCTargetDesc/HexagonMCChecker.h" +#include "MCTargetDesc/HexagonMCCodeEmitter.h" #include "MCTargetDesc/HexagonMCInstrInfo.h" #include "llvm/MC/MCAsmBackend.h" #include "llvm/MC/MCAsmLayout.h" @@ -41,6 +43,19 @@ std::unique_ptr MCII; std::unique_ptr RelaxTarget; MCInst * Extender; + + void ReplaceInstruction(MCCodeEmitter &E, MCRelaxableFragment &RF, + MCInst &HMB) const { + SmallVector Fixups; + SmallString<256> Code; + raw_svector_ostream VecOS(Code); + E.encodeInstruction(HMB, VecOS, Fixups, RF.getSubtargetInfo()); + + // Update the fragment. + RF.setInst(HMB); + RF.getContents() = Code; + RF.getFixups() = Fixups; + } public: HexagonAsmBackend(const Target &T, uint8_t OSABI, StringRef CPU) : OSABI(OSABI), MCII (T.createMCInstrInfo()), RelaxTarget(new MCInst *), @@ -525,28 +540,8 @@ return Relaxable; } - /// MayNeedRelaxation - Check whether the given instruction may need - /// relaxation. - /// - /// \param Inst - The instruction to test. bool mayNeedRelaxation(MCInst const &Inst) const override { - assert(HexagonMCInstrInfo::isBundle(Inst)); - bool PreviousIsExtender = false; - for (auto const &I : HexagonMCInstrInfo::bundleInstructions(Inst)) { - auto const &Inst = *I.getInst(); - if (!PreviousIsExtender) { - if (HexagonMCInstrInfo::isDuplex(*MCII, Inst)) { - if (isInstRelaxable(*Inst.getOperand(0).getInst()) || - isInstRelaxable(*Inst.getOperand(1).getInst())) - return true; - } else { - if (isInstRelaxable(Inst)) - return true; - } - } - PreviousIsExtender = HexagonMCInstrInfo::isImmext(Inst); - } - return false; + return true; } /// fixupNeedsRelaxation - Target specific predicate for whether a given @@ -686,6 +681,56 @@ } return true; } + + void finishLayout(MCAssembler const &Asm, + MCAsmLayout &Layout) const override { + for (auto I : Layout.getSectionOrder()) { + auto &Fragments = I->getFragmentList(); + for (auto &J : Fragments) { + switch (J.getKind()) { + default: + break; + case MCFragment::FT_Align: { + auto Size = Asm.computeFragmentSize(Layout, J); + for (auto K = J.getIterator(); + K != Fragments.begin() && Size >= HEXAGON_PACKET_SIZE;) { + --K; + switch (K->getKind()) { + default: + break; + case MCFragment::FT_Align: { + // Don't pad before other alignments + Size = 0; + break; + } + case MCFragment::FT_Relaxable: { + auto &RF = cast(*K); + auto &Inst = const_cast(RF.getInst()); + while (Size > 0) { + MCInst *Nop = new (Asm.getContext()) MCInst; + Nop->setOpcode(Hexagon::A2_nop); + Inst.addOperand(MCOperand::createInst(Nop)); + Size -= 4; + if (!HexagonMCChecker( + *MCII, RF.getSubtargetInfo(), Inst, Inst, + *Asm.getContext().getRegisterInfo()).check()) { + Inst.erase(Inst.end() - 1); + Size = 0; + } + } + HexagonMCShuffle(*MCII, RF.getSubtargetInfo(), Inst); + ReplaceInstruction(Asm.getEmitter(), RF, Inst); + Layout.invalidateFragmentsFrom(&RF); + Size = 0; // Only look back one instruction + break; + } + } + } + } + } + } + } + } }; } // end anonymous namespace Index: test/MC/Hexagon/align.s =================================================================== --- test/MC/Hexagon/align.s +++ test/MC/Hexagon/align.s @@ -0,0 +1,44 @@ +# RUN: llvm-mc -filetype=obj %s | llvm-objdump -d - | FileCheck %s + +# Verify that the .align directive emits the proper insn packets. + +{ r1 = sub(#1, r1) } +# CHECK: 76414021 { r1 = sub(#1, r1) +# CHECK-NEXT: 7f004000 nop +# CHECK-NEXT: 7f004000 nop +# CHECK-NEXT: 7f00c000 nop } + +.align 16 +{ r1 = sub(#1, r1) + r2 = sub(#1, r2) } +# CHECK: 76414021 { r1 = sub(#1, r1) +# CHECK-NEXT: 76424022 r2 = sub(#1, r2) +# CHECK-NEXT: 7f004000 nop +# CHECK-NEXT: 7f00c000 nop } + +.p2align 5 +{ r1 = sub(#1, r1) + r2 = sub(#1, r2) + r3 = sub(#1, r3) } +# CHECK: 76434023 r3 = sub(#1, r3) +# CHECK-NEXT: 7f00c000 nop } + +.align 16 +{ r1 = sub(#1, r1) + r2 = sub(#1, r2) + r3 = sub(#1, r3) + r4 = sub(#1, r4) } + +# Don't pad packets that can't be padded e.g. solo insts +# CHECK: 9200c020 { r0 = vextract(v0,r0) } +r0 = vextract(v0, r0) +.align 128 +# CHECK: 76414021 { r1 = sub(#1, r1) +# CHECK-NEXT: 7f00c000 nop } +{ r1 = sub(#1, r1) } + +.falign +.align 8 +{ r1 = sub(#1, r1) + r2 = sub(#1, r2) + r3 = sub(#1, r3) } \ No newline at end of file