Fix pr4544. When remating, make sure the destination register fits the instruction definition. It may be mismatched due to sub-register coalescing.

No test case yet because the code doesn't trigger until 75408 is re-applied.


git-svn-id: https://llvm.org/svn/llvm-project/llvm/trunk@75572 91177308-0d34-0410-b5e6-96231b3b80d8

1 files changed, 11 insertions, 0 deletions

diff --git a/lib/CodeGen/SimpleRegisterCoalescing.cpp b/lib/CodeGen/SimpleRegisterCoalescing.cpp
index 7dcd4cc6ed..cb5d3f0daa 100644
--- a/lib/CodeGen/SimpleRegisterCoalescing.cpp
+++ b/lib/CodeGen/SimpleRegisterCoalescing.cpp
@@ -611,6 +611,17 @@ bool SimpleRegisterCoalescing::ReMaterializeTrivialDef(LiveInterval &SrcInt,
   bool SawStore = false;
   if (!DefMI->isSafeToMove(tii_, SawStore))
     return false;
+  if (TID.getNumDefs() != 1)
+    return false;
+  // Make sure the copy destination register class fits the instruction
+  // definition register class. The mismatch can happen as a result of earlier
+  // extract_subreg, insert_subreg, subreg_to_reg coalescing.
+  const TargetRegisterClass *RC = getInstrOperandRegClass(tri_, TID, 0);
+  if (TargetRegisterInfo::isVirtualRegister(DstReg)) {
+    if (mri_->getRegClass(DstReg) != RC)
+      return false;
+  } else if (!RC->contains(DstReg))
+    return false;
 
   unsigned DefIdx = li_->getDefIndex(CopyIdx);
   const LiveRange *DLR= li_->getInterval(DstReg).getLiveRangeContaining(DefIdx);