[MIR] Create virtual registers referenced by live-ins

3 files changed, 14 insertions, 6 deletions

diff --git a/backend/lib/MIR/MIRDocumentParser.cpp b/backend/lib/MIR/MIRDocumentParser.cpp
index 954d0d8..9c4728a 100644
--- a/backend/lib/MIR/MIRDocumentParser.cpp
+++ b/backend/lib/MIR/MIRDocumentParser.cpp
@@ -172,15 +172,22 @@ bool MIRDocumentParser::physicalRegister(const yaml::Node &Node, PhysicalRegiste
 }
 
 bool MIRDocumentParser::virtualRegister(std::string_view Text, const TextRange &Range,
-                                        const MIParserContext &Context, Register &Reg) {
+                                        MachineFunction &Function, MIParserContext &Context,
+                                        Register &Reg) {
   const auto ID = parsePrefixedID(Text, "%");
   if (!ID)
     return fail(Range, "virtual register must use the '%N' MIR spelling");
-  const auto Found =
+  auto Found =
       std::ranges::find_if(Context.VirtualRegisters,
                            [ID](const MIVirtualRegisterSlot &Slot) { return Slot.TextID == *ID; });
-  if (Found == Context.VirtualRegisters.end())
-    return fail(Range, "reference to undefined virtual register '%" + std::to_string(*ID) + "'");
+  if (Found == Context.VirtualRegisters.end()) {
+    auto Created = Function.registerInfo().createVirtualRegister();
+    if (!Created)
+      return importError(Range, Created.error());
+    Context.VirtualRegisters.push_back({*ID, *Created});
+    Reg = *Created;
+    return false;
+  }
   Reg = Found->Reg;
   return false;
 }
diff --git a/backend/lib/MIR/MIRDocumentParser.h b/backend/lib/MIR/MIRDocumentParser.h
index 3fade39..4d1d1ea 100644
--- a/backend/lib/MIR/MIRDocumentParser.h
+++ b/backend/lib/MIR/MIRDocumentParser.h
@@ -70,7 +70,8 @@ private:
   [[nodiscard]] bool physicalRegister(std::string_view Text, const TextRange &Range,
                                       PhysicalRegister &Reg);
   [[nodiscard]] bool virtualRegister(std::string_view Text, const TextRange &Range,
-                                     const MIParserContext &Context, Register &Reg);
+                                     MachineFunction &Function, MIParserContext &Context,
+                                     Register &Reg);
   [[nodiscard]] Expected<MachineMetadataReference>
   resolveMetadataReference(const MachineFunction &Function, std::uint32_t Slot) const;
   [[nodiscard]] bool parseVariableDebugInfo(const yaml::Node &Entry, MachineFunction &Function,
diff --git a/backend/lib/MIR/MIRFunctionParser.cpp b/backend/lib/MIR/MIRFunctionParser.cpp
index c8bba70..7e8821b 100644
--- a/backend/lib/MIR/MIRFunctionParser.cpp
+++ b/backend/lib/MIR/MIRFunctionParser.cpp
@@ -348,7 +348,7 @@ bool MIRDocumentParser::parseRegisters(const yaml::Node &Root, MachineFunction &
           return true;
         if (!trim(Text).empty()) {
           Register Virtual;
-          if (virtualRegister(Text, VirtualNode->range(), Context, Virtual))
+          if (virtualRegister(Text, VirtualNode->range(), Function, Context, Virtual))
             return true;
           Initial = Virtual;
         }