From 495c675771f43cfa12ab7a426a538be72fec1805 Mon Sep 17 00:00:00 2001
From: Richard-Qin-X <richard.qin.001@gmail.com>
Date: Sun, 9 Aug 2026 17:03:19 +0800
Subject: [CodeGen] Implement machine optimization and scheduling

---
 .../include/rcc/CodeGen/ScheduleHazardRecognizer.h | 68 ++++++++++++++++++++++
 1 file changed, 68 insertions(+)
 create mode 100644 backend/include/rcc/CodeGen/ScheduleHazardRecognizer.h

(limited to 'backend/include/rcc/CodeGen/ScheduleHazardRecognizer.h')

diff --git a/backend/include/rcc/CodeGen/ScheduleHazardRecognizer.h b/backend/include/rcc/CodeGen/ScheduleHazardRecognizer.h
new file mode 100644
index 0000000..947db5e
--- /dev/null
+++ b/backend/include/rcc/CodeGen/ScheduleHazardRecognizer.h
@@ -0,0 +1,68 @@
+// SPDX-License-Identifier: GPL-3.0-or-later
+
+#ifndef RCC_CODEGEN_SCHEDULEHAZARDRECOGNIZER_H
+#define RCC_CODEGEN_SCHEDULEHAZARDRECOGNIZER_H
+
+#include "rcc/CodeGen/TargetSchedulingModel.h"
+#include "rcc/Support/Error.h"
+
+#include <cstddef>
+#include <cstdint>
+#include <memory>
+#include <unordered_map>
+#include <vector>
+
+namespace rcc {
+
+struct ScheduleUnit;
+
+class ScheduleHazardRecognizer final {
+public:
+  [[nodiscard]] static Expected<ScheduleHazardRecognizer>
+  create(const TargetSchedulingModel &Model,
+         ScheduleDirection Direction = ScheduleDirection::TopDown);
+
+  ScheduleHazardRecognizer(const ScheduleHazardRecognizer &Other);
+  ScheduleHazardRecognizer &operator=(const ScheduleHazardRecognizer &Other);
+  ScheduleHazardRecognizer(ScheduleHazardRecognizer &&) noexcept = default;
+  ScheduleHazardRecognizer &operator=(ScheduleHazardRecognizer &&) noexcept = default;
+  ~ScheduleHazardRecognizer() = default;
+
+  [[nodiscard]] std::uint64_t currentCycle() const noexcept {
+    return CurrentCycle;
+  }
+
+  [[nodiscard]] Expected<bool> canIssue(const ScheduleUnit &Unit, std::uint64_t Cycle) const;
+  [[nodiscard]] Expected<std::uint64_t> earliestIssueCycle(const ScheduleUnit &Unit,
+                                                           std::uint64_t Earliest) const;
+  [[nodiscard]] Expected<void> issue(const ScheduleUnit &Unit, std::uint64_t Cycle);
+
+private:
+  struct ResourceState final {
+    ProcessorResourceID ID;
+    std::vector<std::uint64_t> AvailableAt;
+  };
+
+  ScheduleHazardRecognizer(const TargetSchedulingModel &Model, ScheduleDirection Direction) noexcept
+      : Model(&Model), Direction(Direction) {}
+
+  [[nodiscard]] Expected<void> reset();
+  [[nodiscard]] Expected<void> moveToCycle(std::uint64_t Cycle);
+  [[nodiscard]] Expected<void> issueImplementation(const ScheduleUnit &Unit, std::uint64_t Cycle);
+  [[nodiscard]] Expected<void> reserveResource(ProcessorResourceID Resource, std::uint32_t Cycles,
+                                               std::uint64_t Cycle);
+
+  const TargetSchedulingModel *Model;
+  ScheduleDirection Direction = ScheduleDirection::TopDown;
+  std::unique_ptr<TargetScheduleHazardRecognizer> TargetHazards;
+  std::vector<ResourceState> Resources;
+  std::unordered_map<ProcessorResourceID, std::size_t> ResourceIndexes;
+  std::uint64_t CurrentCycle = 0;
+  std::uint64_t DispatchAvailableAt = 0;
+  std::uint32_t IssuedMicroOps = 0;
+  std::uint32_t DispatchedUnits = 0;
+};
+
+} // namespace rcc
+
+#endif // RCC_CODEGEN_SCHEDULEHAZARDRECOGNIZER_H
-- 
cgit v1.2.3