From 09e8ef3e0dae29be7aa21a6e22a118a3f545945f Mon Sep 17 00:00:00 2001
From: Richard-Qin-X <richard.qin.001@gmail.com>
Date: Wed, 12 Aug 2026 19:45:09 +0800
Subject: [Tooling] Complete the LLVM MIR command-line contract

---
 backend/lib/Tooling/MIRFuzzer.cpp | 18 +++++++++++++++++-
 1 file changed, 17 insertions(+), 1 deletion(-)

(limited to 'backend/lib/Tooling/MIRFuzzer.cpp')

diff --git a/backend/lib/Tooling/MIRFuzzer.cpp b/backend/lib/Tooling/MIRFuzzer.cpp
index ed5b2fa..653b9fd 100644
--- a/backend/lib/Tooling/MIRFuzzer.cpp
+++ b/backend/lib/Tooling/MIRFuzzer.cpp
@@ -225,7 +225,23 @@ private:
 }
 
 [[nodiscard]] BackendOutputKind outputForStage(BackendStage Stage) noexcept {
-  return static_cast<BackendOutputKind>(static_cast<std::uint8_t>(Stage));
+  switch (Stage) {
+  case BackendStage::PreLegalizerMIR:
+    return BackendOutputKind::PreLegalizerMIR;
+  case BackendStage::LegalizedMIR:
+    return BackendOutputKind::LegalizedMIR;
+  case BackendStage::RegisterBankMIR:
+    return BackendOutputKind::RegisterBankMIR;
+  case BackendStage::SelectedMIR:
+    return BackendOutputKind::SelectedMIR;
+  case BackendStage::PostSSAMIR:
+    return BackendOutputKind::PostSSAMIR;
+  case BackendStage::PostRegisterAllocationMIR:
+    return BackendOutputKind::PostRegisterAllocationMIR;
+  case BackendStage::FinalMIR:
+    return BackendOutputKind::FinalMIR;
+  }
+  return BackendOutputKind::FinalMIR;
 }
 
 [[nodiscard]] Expected<bool> propertyCase(std::string_view Canonical, DeterministicRandom &Random,
-- 
cgit v1.2.3