commit 3dba5d858455149cb843c2f000109265cc523267
author Akshat Oke <76596238+optimisan@users.noreply.github.com> Mon Oct 14 14:29:09 2024 +0530
committer GitHub <noreply@github.com> Mon Oct 14 14:29:09 2024 +0530
tree 5c92b18dc8aa181b8071b1c19f37f2f8e0ddb362
parent dbfca24b99987111586551698253dba0afbec09b

[MIR] Add missing noteNewVirtualRegister callbacks (#111634)

The delegates' callback isn't invoked on parsing new virtual registers.

There are two places in the serialization where new virtual registers can be discovered: in register infos and in instructions.

llvm/lib/CodeGen/MIRParser/MIParser.cpp

diff --git a/llvm/lib/CodeGen/MIRParser/MIParser.cpp b/llvm/lib/CodeGen/MIRParser/MIParser.cpp
index f1d3ce9..7aaa0f4 100644
--- a/llvm/lib/CodeGen/MIRParser/MIParser.cpp
+++ b/llvm/lib/CodeGen/MIRParser/MIParser.cpp
@@ -1786,6 +1786,7 @@
 
         MRI.setRegClassOrRegBank(Reg, static_cast<RegisterBank *>(nullptr));
         MRI.setType(Reg, Ty);
+        MRI.noteNewVirtualRegister(Reg);
       }
     }
   } else if (consumeIfPresent(MIToken::lparen)) {

llvm/lib/CodeGen/MIRParser/MIRParser.cpp

diff --git a/llvm/lib/CodeGen/MIRParser/MIRParser.cpp b/llvm/lib/CodeGen/MIRParser/MIRParser.cpp
index 0c8a3eb..10d3cdc 100644
--- a/llvm/lib/CodeGen/MIRParser/MIRParser.cpp
+++ b/llvm/lib/CodeGen/MIRParser/MIRParser.cpp
@@ -705,6 +705,7 @@
                          FlagStringValue.Value + "'");
       Info.Flags.push_back(FlagValue);
     }
+    RegInfo.noteNewVirtualRegister(Info.VReg);
   }
 
   // Parse the liveins.