[CodeGen] Implement the greedy register allocator pipeline

1 files changed, 31 insertions, 0 deletions

diff --git a/backend/lib/CodeGen/RegAllocGreedy.h b/backend/lib/CodeGen/RegAllocGreedy.h
new file mode 100644
index 0000000..1f25a5c
--- /dev/null
+++ b/backend/lib/CodeGen/RegAllocGreedy.h
@@ -0,0 +1,31 @@
+// SPDX-License-Identifier: GPL-3.0-or-later
+
+// The private header has no stable include-root-relative guard name.
+// NOLINTNEXTLINE(llvm-header-guard)
+#ifndef RCC_LIB_CODEGEN_REGALLOCGREEDY_H
+#define RCC_LIB_CODEGEN_REGALLOCGREEDY_H
+
+#include "rcc/CodeGen/RegAllocCommon.h"
+#include "rcc/CodeGen/TargetRegisterAllocationInfo.h"
+
+namespace rcc {
+
+class GreedyAllocatorPass final : public MachineFunctionPass {
+public:
+  explicit GreedyAllocatorPass(const TargetRegisterAllocationInfo &Info,
+                               RegisterAllocationOptions Options = {}) noexcept
+      : Info(&Info), Options(Options) {}
+
+  [[nodiscard]] const MachinePassContract &contract() const noexcept override;
+  [[nodiscard]] Expected<MachinePassResult> run(MachineFunction &Function,
+                                                const TargetDescription &Target,
+                                                MachineFunctionAnalysisManager &Analyses) override;
+
+private:
+  const TargetRegisterAllocationInfo *Info;
+  RegisterAllocationOptions Options;
+};
+
+} // namespace rcc
+
+#endif // RCC_LIB_CODEGEN_REGALLOCGREEDY_H