From 10b7e69448f866bcd6893b21d7f78d766252eb11 Mon Sep 17 00:00:00 2001 From: Richard-Qin-X Date: Sat, 8 Aug 2026 02:58:50 +0800 Subject: [CodeGen] Implement RV64GC legalization and instruction selection --- .../include/rcc/CodeGen/MachinePipelinePasses.h | 63 +++++++++++++++ backend/include/rcc/CodeGen/TargetCodeGen.h | 94 ++++++++++++++++++++++ backend/include/rcc/Target/RISCV/RISCVCodeGen.h | 16 ++++ 3 files changed, 173 insertions(+) create mode 100644 backend/include/rcc/CodeGen/MachinePipelinePasses.h create mode 100644 backend/include/rcc/CodeGen/TargetCodeGen.h create mode 100644 backend/include/rcc/Target/RISCV/RISCVCodeGen.h (limited to 'backend/include') diff --git a/backend/include/rcc/CodeGen/MachinePipelinePasses.h b/backend/include/rcc/CodeGen/MachinePipelinePasses.h new file mode 100644 index 0000000..c5bdac0 --- /dev/null +++ b/backend/include/rcc/CodeGen/MachinePipelinePasses.h @@ -0,0 +1,63 @@ +// SPDX-License-Identifier: GPL-3.0-or-later + +#ifndef RCC_CODEGEN_MACHINEPIPELINEPASSES_H +#define RCC_CODEGEN_MACHINEPIPELINEPASSES_H + +#include "rcc/CodeGen/MachinePass.h" +#include "rcc/CodeGen/TargetCodeGen.h" + +namespace rcc { + +class LegalizerPass final : public MachineFunctionPass { +public: + explicit LegalizerPass(const TargetLegalizerInfo *Info) : Info(Info) {} + explicit LegalizerPass(const TargetLegalizerInfo &Info) : Info(&Info) {} + + [[nodiscard]] const MachinePassContract &contract() const noexcept override; + [[nodiscard]] Expected run(MachineFunction &Function, + const TargetDescription &Target, + MachineAnalysisManager &Analyses) override; + +private: + [[nodiscard]] Expected runImplementation(MachineFunction &Function, + const TargetDescription &Target, + MachineAnalysisManager &Analyses); + + const TargetLegalizerInfo *Info; +}; + +class RegisterBankSelectPass final : public MachineFunctionPass { +public: + explicit RegisterBankSelectPass(const TargetRegisterBankInfo *Info) : Info(Info) {} + explicit RegisterBankSelectPass(const TargetRegisterBankInfo &Info) : Info(&Info) {} + + [[nodiscard]] const MachinePassContract &contract() const noexcept override; + [[nodiscard]] Expected run(MachineFunction &Function, + const TargetDescription &Target, + MachineAnalysisManager &Analyses) override; + +private: + const TargetRegisterBankInfo *Info; +}; + +class InstructionSelectPass final : public MachineFunctionPass { +public: + explicit InstructionSelectPass(const TargetInstructionSelector *Info) : Info(Info) {} + explicit InstructionSelectPass(const TargetInstructionSelector &Info) : Info(&Info) {} + + [[nodiscard]] const MachinePassContract &contract() const noexcept override; + [[nodiscard]] Expected run(MachineFunction &Function, + const TargetDescription &Target, + MachineAnalysisManager &Analyses) override; + +private: + [[nodiscard]] Expected runImplementation(MachineFunction &Function, + const TargetDescription &Target, + MachineAnalysisManager &Analyses); + + const TargetInstructionSelector *Info; +}; + +} // namespace rcc + +#endif // RCC_CODEGEN_MACHINEPIPELINEPASSES_H diff --git a/backend/include/rcc/CodeGen/TargetCodeGen.h b/backend/include/rcc/CodeGen/TargetCodeGen.h new file mode 100644 index 0000000..b1be1f6 --- /dev/null +++ b/backend/include/rcc/CodeGen/TargetCodeGen.h @@ -0,0 +1,94 @@ +// SPDX-License-Identifier: GPL-3.0-or-later + +#ifndef RCC_CODEGEN_TARGETCODEGEN_H +#define RCC_CODEGEN_TARGETCODEGEN_H + +#include "rcc/CodeGen/MachineIRTypes.h" +#include "rcc/CodeGen/Register.h" +#include "rcc/Support/Error.h" + +#include +#include + +namespace rcc { + +class MachineFunction; +class MachineInstr; +class MachineRegisterInfo; +class TargetDescription; +struct VirtualRegisterInfo; + +enum class LegalizationAction : std::uint8_t { + Legal, + Unsupported, + Custom, + Lower, + Libcall, + WidenScalar, + NarrowScalar, + MoreElements, + FewerElements, + Bitcast, + ChangeElementType, + ChangeAddressSpace, +}; + +class TargetLegalizerInfo { +public: + virtual ~TargetLegalizerInfo() = default; + + [[nodiscard]] virtual Expected + classify(const MachineInstr &Instruction, const MachineFunction &Function, + const TargetDescription &Target) const = 0; + + [[nodiscard]] virtual Expected lower(MachineInstr &Instruction, MachineFunction &Function, + const TargetDescription &Target, + LegalizationAction Action) const = 0; + + // This hook performs read-only target checks before a lowering mutation is committed. + [[nodiscard]] virtual Expected validateLowering(const MachineInstr &, + const MachineFunction &, + const TargetDescription &, + LegalizationAction) const { + return {}; + } +}; + +class TargetRegisterBankInfo { +public: + virtual ~TargetRegisterBankInfo() = default; + + [[nodiscard]] virtual Expected + selectBank(const MachineFunction &Function, Register Reg, const VirtualRegisterInfo &Info, + const MachineInstr *Context, const TargetDescription &Target) const = 0; + + [[nodiscard]] virtual Expected + constrainInstruction(MachineInstr &Instruction, MachineFunction &Function, + const TargetDescription &Target) const = 0; + + // This hook must not mutate MachineIR. It allows the pass to reject a bad mapping + // before publishing any bank or class assignment. + [[nodiscard]] virtual Expected validateInstruction(const MachineInstr &, + const MachineFunction &, + const TargetDescription &) const { + return {}; + } +}; + +class TargetInstructionSelector { +public: + virtual ~TargetInstructionSelector() = default; + + [[nodiscard]] virtual Expected select(MachineInstr &Instruction, MachineFunction &Function, + const TargetDescription &Target) const = 0; + + // Selection matching is read-only and must complete before select() mutates MachineIR. + [[nodiscard]] virtual Expected canSelect(const MachineInstr &, const MachineFunction &, + const TargetDescription &) const { + return {}; + } +}; + +} // namespace rcc + +#endif // RCC_CODEGEN_TARGETCODEGEN_H diff --git a/backend/include/rcc/Target/RISCV/RISCVCodeGen.h b/backend/include/rcc/Target/RISCV/RISCVCodeGen.h new file mode 100644 index 0000000..b06f20d --- /dev/null +++ b/backend/include/rcc/Target/RISCV/RISCVCodeGen.h @@ -0,0 +1,16 @@ +// SPDX-License-Identifier: GPL-3.0-or-later + +#ifndef RCC_TARGET_RISCV_RISCVCODEGEN_H +#define RCC_TARGET_RISCV_RISCVCODEGEN_H + +#include "rcc/CodeGen/TargetCodeGen.h" + +namespace rcc::riscv { + +[[nodiscard]] const TargetLegalizerInfo &legalizerInfo() noexcept; +[[nodiscard]] const TargetRegisterBankInfo ®isterBankInfo() noexcept; +[[nodiscard]] const TargetInstructionSelector &instructionSelector() noexcept; + +} // namespace rcc::riscv + +#endif // RCC_TARGET_RISCV_RISCVCODEGEN_H -- cgit v1.2.3