[MIR-Canon] Add support for rewriting VRegs that are typed but don't have an RC.

[plotfi]

Summary

So it appears that in a lot of cases before registerbanks are selected, we dont have Register classes and MIR-Canon is crashing. This attempts to fix this by creating new override registers that use the generic type rather than the register class.

[compnerd]

llvm/lib/CodeGen/MIRCanonicalizerPass.cpp

320

I think that the do is superfluous

325

I think that this can just be return MRI.getRegClassOrNull(VReg0) || MRI.getType(VReg0) == MRI.getType(VReg1);

516–518

Why not sink the get into the conditional?

if (auto RC = MRI.getRegClassOrNull(VReg))
  return MRI.createVirtualRegister(RC, OS.str());

[bogner]

lgtm once you've addressed compnerd's comments

[plotfi]

Going to leave out the vregTypesMatch part as that changes some of the behavior in a way that causes llvm/test/CodeGen/MIR/AMDGPU/mir-canon-multi.mir to fail, and I'd like to understand why that is happening. But for now this change makes the cursor create vregs based on the vreg type if there is no register class to base off of.

llvm/lib/CodeGen/MIRCanonicalizerPass.cpp

   return Changed;
 }
 
+static bool vregTypesMatch(const MachineRegisterInfo &MRI, unsigned VReg0,
+                           unsigned VReg1) {
+  return (MRI.getRegClassOrNull(VReg0) &&
+          (MRI.getRegClassOrNull(VReg0) == MRI.getRegClassOrNull(VReg1))) ||
+         (MRI.getType(VReg0) == MRI.getType(VReg1));
+}
+
 static bool propagateLocalCopies(MachineBasicBlock *MBB) {
   bool Changed = false;
   MachineRegisterInfo &MRI = MBB->getParent()->getRegInfo();

       continue;
     if (!TargetRegisterInfo::isVirtualRegister(Src))
       continue;
-    if (MRI.getRegClass(Dst) != MRI.getRegClass(Src))
+    if (!vregTypesMatch(MRI, Dst, Src))
       continue;
 
     for (auto UI = MRI.use_begin(Dst); UI != MRI.use_end(); ++UI) {

     return virtualVRegNumber;
   }
 
-  unsigned createVirtualRegister(const TargetRegisterClass *RC) {
+  unsigned createVirtualRegister(MachineRegisterInfo &MRI, unsigned VReg) {
     std::string S;
     raw_string_ostream OS(S);
     OS << "namedVReg" << (virtualVRegNumber & ~0x80000000);
     OS.flush();
     virtualVRegNumber++;
-
+    if (auto RC = MRI.getRegClassOrNull(VReg))
       return MRI.createVirtualRegister(RC, OS.str());
+    return MRI.createGenericVirtualRegister(MRI.getType(VReg), OS.str());
   }
 };
 } // namespace

       continue;
     }
 
-    auto Rename = NVC.createVirtualRegister(MRI.getRegClass(Reg));
+    auto Rename = NVC.createVirtualRegister(MRI, Reg);
 
     if (VRegRenameMap.find(Reg) == VRegRenameMap.end()) {
       LLVM_DEBUG(dbgs() << "Mapping vreg ";);

     MachineInstr &MI = *MII++;
     Changed = true;
     unsigned vRegToRename = MI.getOperand(0).getReg();
-    auto Rename = NVC.createVirtualRegister(MRI.getRegClass(vRegToRename));
+    auto Rename = NVC.createVirtualRegister(MRI, vRegToRename);
 
     std::vector<MachineOperand *> RenameMOs;
     for (auto &MO : MRI.reg_operands(vRegToRename)) {

llvm/test/CodeGen/MIR/AArch64/addrspace-memoperands.mir

 # RUN: llc -mtriple=aarch64-none-linux-gnu -run-pass none -o - %s | FileCheck %s
+# RUN: llc -mtriple=aarch64-none-linux-gnu -run-pass mir-canonicalizer -o - %s
 
 --- |