From f6fdfb8d484b4e52cdb24a94a31b0f00d41a5eb7 Mon Sep 17 00:00:00 2001 From: Richard-Qin-X Date: Sat, 15 Aug 2026 00:38:20 +0800 Subject: [GlobalISel] Match LLVM CSE builder result semantics --- backend/include/rcc/CodeGen/MachineIRBuilder.h | 146 ++++++++++++++++++------- 1 file changed, 104 insertions(+), 42 deletions(-) (limited to 'backend/include/rcc/CodeGen/MachineIRBuilder.h') diff --git a/backend/include/rcc/CodeGen/MachineIRBuilder.h b/backend/include/rcc/CodeGen/MachineIRBuilder.h index e830837..f87adfe 100644 --- a/backend/include/rcc/CodeGen/MachineIRBuilder.h +++ b/backend/include/rcc/CodeGen/MachineIRBuilder.h @@ -14,6 +14,7 @@ #include #include #include +#include #include #include #include @@ -24,7 +25,9 @@ class MachineBasicBlock; class MachineFunction; class MachineInstr; class MachineMemOperand; +class MachineRegisterInfo; class RegisterMask; +class GISelCSEInfo; class MachineInstrBuilder final { public: @@ -143,7 +146,32 @@ private: friend class MachineIRBuilder; }; -class MachineIRBuilder final { +class MachineIRBuilderResult final { +public: + enum class Kind : std::uint8_t { + Register, + Type, + }; + + MachineIRBuilderResult(Register Reg) noexcept : Reg(Reg), ResultKind(Kind::Register) {} + MachineIRBuilderResult(LowLevelType Type) noexcept : Type(Type), ResultKind(Kind::Type) {} + + [[nodiscard]] Kind kind() const noexcept { + return ResultKind; + } + + [[nodiscard]] Register reg() const noexcept; + [[nodiscard]] LowLevelType type() const noexcept; + [[nodiscard]] Expected type(const MachineRegisterInfo &MRI) const; + [[nodiscard]] Expected addDef(MachineRegisterInfo &MRI, MachineInstrBuilder &Builder) const; + +private: + Register Reg; + LowLevelType Type; + Kind ResultKind; +}; + +class MachineIRBuilder { public: MachineIRBuilder(MachineFunction &Function, const TargetDescription &Target) noexcept : Function(&Function), Target(&Target) {} @@ -151,12 +179,13 @@ public: MachineIRBuilder &operator=(const MachineIRBuilder &) = delete; MachineIRBuilder(MachineIRBuilder &&) = delete; MachineIRBuilder &operator=(MachineIRBuilder &&) = delete; + virtual ~MachineIRBuilder() = default; [[nodiscard]] MachineFunction &function() noexcept { return *Function; } - [[nodiscard]] const MachineFunction &function() const noexcept { + [[nodiscard]] MachineFunction &function() const noexcept { return *Function; } @@ -168,6 +197,22 @@ public: return InsertBlock; } + [[nodiscard]] MachineInstr *insertionInstruction() const noexcept { + return InsertBefore; + } + + [[nodiscard]] GISelCSEInfo *getCSEInfo() noexcept { + return CSEInfo; + } + + [[nodiscard]] GISelCSEInfo *getCSEInfo() const noexcept { + return CSEInfo; + } + + void setCSEInfo(GISelCSEInfo *Info) noexcept { + CSEInfo = Info; + } + [[nodiscard]] Expected verifyInsertPoint() const; [[nodiscard]] Expected setInsertPoint(MachineBasicBlock &Block); @@ -179,63 +224,79 @@ public: } [[nodiscard]] Expected buildInstr(Opcode InstructionOpcode) const; - [[nodiscard]] Expected buildInstr(Opcode InstructionOpcode, - std::initializer_list Destinations, - std::initializer_list Sources) const; - [[nodiscard]] Expected buildTrunc(Register Destination, Register Source) const; - [[nodiscard]] Expected buildCopy(Register Destination, Register Source) const; - [[nodiscard]] Expected buildImplicitDef(Register Destination) const; - [[nodiscard]] Expected buildAnyExt(Register Destination, Register Source) const; - [[nodiscard]] Expected buildZExt(Register Destination, Register Source) const; - [[nodiscard]] Expected buildSExt(Register Destination, Register Source) const; - [[nodiscard]] Expected buildUndef(Register Destination) const; - [[nodiscard]] Expected buildAnyExtOrTrunc(Register Destination, + [[nodiscard]] Expected + buildInstr(Opcode InstructionOpcode, std::initializer_list Destinations, + std::initializer_list Sources) const; + [[nodiscard]] virtual Expected + buildInstr(Opcode InstructionOpcode, std::span Destinations, + std::span Sources) const; + [[nodiscard]] Expected buildTrunc(MachineIRBuilderResult Destination, + Register Source) const; + [[nodiscard]] Expected buildCopy(MachineIRBuilderResult Destination, + Register Source) const; + [[nodiscard]] Expected buildImplicitDef(MachineIRBuilderResult Destination) const; + [[nodiscard]] Expected buildAnyExt(MachineIRBuilderResult Destination, + Register Source) const; + [[nodiscard]] Expected buildZExt(MachineIRBuilderResult Destination, + Register Source) const; + [[nodiscard]] Expected buildSExt(MachineIRBuilderResult Destination, + Register Source) const; + [[nodiscard]] Expected buildUndef(MachineIRBuilderResult Destination) const; + [[nodiscard]] Expected buildAnyExtOrTrunc(MachineIRBuilderResult Destination, Register Source) const; - [[nodiscard]] Expected buildSExtOrTrunc(Register Destination, + [[nodiscard]] Expected buildSExtOrTrunc(MachineIRBuilderResult Destination, Register Source) const; - [[nodiscard]] Expected buildConstant(Register Destination, APInt Value) const; - [[nodiscard]] Expected buildConstant(Register Destination, + [[nodiscard]] virtual Expected buildConstant(MachineIRBuilderResult Destination, + APInt Value) const; + [[nodiscard]] Expected buildConstant(MachineIRBuilderResult Destination, std::int64_t Value) const; - [[nodiscard]] Expected buildSExtInReg(Register Destination, Register Source, - std::uint32_t Width) const; - [[nodiscard]] Expected buildAnd(Register Destination, Register Left, + [[nodiscard]] virtual Expected + buildSExtInReg(MachineIRBuilderResult Destination, Register Source, std::uint32_t Width) const; + [[nodiscard]] Expected buildAnd(MachineIRBuilderResult Destination, Register Left, Register Right) const; - [[nodiscard]] Expected buildOr(Register Destination, Register Left, + [[nodiscard]] Expected buildOr(MachineIRBuilderResult Destination, Register Left, Register Right) const; - [[nodiscard]] Expected buildXor(Register Destination, Register Left, + [[nodiscard]] Expected buildXor(MachineIRBuilderResult Destination, Register Left, Register Right) const; - [[nodiscard]] Expected buildAdd(Register Destination, Register Left, + [[nodiscard]] Expected buildAdd(MachineIRBuilderResult Destination, Register Left, Register Right) const; - [[nodiscard]] Expected buildSub(Register Destination, Register Left, + [[nodiscard]] Expected buildSub(MachineIRBuilderResult Destination, Register Left, Register Right) const; - [[nodiscard]] Expected buildShl(Register Destination, Register Left, + [[nodiscard]] Expected buildShl(MachineIRBuilderResult Destination, Register Left, Register Right) const; - [[nodiscard]] Expected buildLShr(Register Destination, Register Left, - Register Right) const; - [[nodiscard]] Expected buildAShr(Register Destination, Register Left, - Register Right) const; - [[nodiscard]] Expected buildPtrToInt(Register Destination, Register Source) const; - [[nodiscard]] Expected buildIntToPtr(Register Destination, Register Source) const; - [[nodiscard]] Expected buildICmp(IntegerPredicate Predicate, Register Destination, + [[nodiscard]] Expected buildLShr(MachineIRBuilderResult Destination, + Register Left, Register Right) const; + [[nodiscard]] Expected buildAShr(MachineIRBuilderResult Destination, + Register Left, Register Right) const; + [[nodiscard]] Expected buildPtrToInt(MachineIRBuilderResult Destination, + Register Source) const; + [[nodiscard]] Expected buildIntToPtr(MachineIRBuilderResult Destination, + Register Source) const; + [[nodiscard]] Expected buildICmp(IntegerPredicate Predicate, + MachineIRBuilderResult Destination, Register Left, Register Right) const; [[nodiscard]] Expected - buildMergeLikeInstr(Register Destination, const std::vector &Sources) const; + buildMergeLikeInstr(MachineIRBuilderResult Destination, + const std::vector &Sources) const; [[nodiscard]] Expected - buildMergeValues(Register Destination, const std::vector &Sources) const; + buildMergeValues(MachineIRBuilderResult Destination, const std::vector &Sources) const; [[nodiscard]] Expected buildUnmergeValues(const std::vector &Destinations, Register Source) const; - [[nodiscard]] Expected buildExtract(Register Destination, Register Source, - std::uint64_t Offset) const; - [[nodiscard]] Expected buildInsert(Register Destination, Register Source, - Register Inserted, std::uint64_t Offset) const; + [[nodiscard]] virtual Expected + buildExtract(MachineIRBuilderResult Destination, Register Source, std::uint64_t Offset) const; + [[nodiscard]] Expected buildInsert(MachineIRBuilderResult Destination, + Register Source, Register Inserted, + std::uint64_t Offset) const; private: - [[nodiscard]] Expected buildUnary(Opcode InstructionOpcode, Register Destination, - Register Source) const; - [[nodiscard]] Expected buildBinary(Opcode InstructionOpcode, Register Destination, - Register Left, Register Right) const; [[nodiscard]] Expected - buildExtOrTrunc(Opcode ExtensionOpcode, Register Destination, Register Source) const; + buildUnary(Opcode InstructionOpcode, MachineIRBuilderResult Destination, Register Source) const; + [[nodiscard]] Expected buildBinary(Opcode InstructionOpcode, + MachineIRBuilderResult Destination, + Register Left, Register Right) const; + [[nodiscard]] Expected buildExtOrTrunc(Opcode ExtensionOpcode, + MachineIRBuilderResult Destination, + Register Source) const; [[nodiscard]] Expected registerType(Register Reg) const; [[nodiscard]] Expected validateBlock(const MachineBasicBlock &Block) const; [[nodiscard]] Expected validateInstruction(const MachineInstr &Instruction) const; @@ -244,6 +305,7 @@ private: const TargetDescription *Target; MachineBasicBlock *InsertBlock = nullptr; MachineInstr *InsertBefore = nullptr; + GISelCSEInfo *CSEInfo = nullptr; }; } // namespace rcc -- cgit v1.2.3