1 files changed, 13 insertions, 10 deletions
diff --git a/lib/CodeGen/MachineCSE.cpp b/lib/CodeGen/MachineCSE.cpp
index 68407c88d8..e65d3bcaf2 100644
--- a/lib/CodeGen/MachineCSE.cpp
+++ b/lib/CodeGen/MachineCSE.cpp
@@ -93,16 +93,19 @@ bool MachineCSE::PerformTrivialCoalescing(MachineInstr *MI,
     if (TII->isMoveInstr(*DefMI, SrcReg, DstReg, SrcSubIdx, DstSubIdx) &&
         TargetRegisterInfo::isVirtualRegister(SrcReg) &&
         !SrcSubIdx && !DstSubIdx) {
-      const TargetRegisterClass *SRC = MRI->getRegClass(SrcReg);
-      const TargetRegisterClass *RC  = MRI->getRegClass(Reg);
-      if (SRC == RC || RC->hasSubClass(SRC)) {
-        DEBUG(dbgs() << "Coalescing: " << *DefMI);
-        DEBUG(dbgs() << "*** to: " << *MI);
-        MO.setReg(SrcReg);
-        DefMI->eraseFromParent();
-        ++NumCoalesces;
-        Changed = true;
-      }
+      const TargetRegisterClass *SRC   = MRI->getRegClass(SrcReg);
+      const TargetRegisterClass *RC    = MRI->getRegClass(Reg);
+      const TargetRegisterClass *NewRC = getCommonSubClass(RC, SRC);
+      if (!NewRC)
+        continue;
+      DEBUG(dbgs() << "Coalescing: " << *DefMI);
+      DEBUG(dbgs() << "*** to: " << *MI);
+      MO.setReg(SrcReg);
+      if (NewRC != SRC)
+        MRI->setRegClass(SrcReg, NewRC);
+      DefMI->eraseFromParent();
+      ++NumCoalesces;
+      Changed = true;
     }
   }