llvm-reduce: Fix mangling types of generic registers

This commit is contained in:
Matt Arsenault 2022-04-19 14:59:32 -04:00 committed by Matt Arsenault
parent 2d0c897212
commit e617d1a1d7
2 changed files with 9 additions and 7 deletions

View File

@ -12,13 +12,13 @@
# CHECK-INTERESTINGNESS: G_IMPLICIT_DEF
# CHECK-INTERESTINGNESS: G_STORE
# RESULT: %v0:vgpr(s32) = COPY $vgpr0, implicit-def %9(p1), implicit-def %10(s64), implicit-def %11(s64)
# RESULT-NEXT: %arst:_(<2 x s32>) = G_IMPLICIT_DEF
# RESULT-NEXT: %aoeu:_(s64) = G_BITCAST %arst(<2 x s32>)
# RESULT: %v0:vgpr(s32) = COPY $vgpr0, implicit-def %9(<2 x s16>), implicit-def %10(s64), implicit-def %11(s64), implicit-def %12(<2 x s32>)
# RESULT-NEXT: %unused_load_ptr:sgpr(p1) = G_IMPLICIT_DEF
# RESULT-NEXT: %aoeu:_(s64) = G_BITCAST %12(<2 x s32>)
# RESULT-NEXT: %add:_(s64) = G_ADD %aoeu, %aoeu
# RESULT-NEXT: %ptr:_(p1) = G_IMPLICIT_DEF
# RESULT-NEXT: G_STORE %v0(s32), %ptr(p1) :: (store (s32), addrspace 1)
# RESULT-NEXT: S_ENDPGM 0, implicit %add(s64), implicit %v0(s32), implicit %11(s64)
# RESULT-NEXT: S_ENDPGM 0, implicit %add(s64), implicit %9(<2 x s16>), implicit %11(s64)
---
name: f

View File

@ -23,7 +23,7 @@ using namespace llvm;
static Register getPrevDefOfRCInMBB(MachineBasicBlock &MBB,
MachineBasicBlock::reverse_iterator &RI,
const RegClassOrRegBank &RC,
const RegClassOrRegBank &RC, LLT Ty,
SetVector<MachineInstr *> &ExcludeMIs) {
auto MRI = &MBB.getParent()->getRegInfo();
for (MachineBasicBlock::reverse_instr_iterator E = MBB.instr_rend(); RI != E;
@ -37,7 +37,7 @@ static Register getPrevDefOfRCInMBB(MachineBasicBlock &MBB,
if (Register::isPhysicalRegister(Reg))
continue;
if (MRI->getRegClassOrRegBank(Reg) == RC &&
if (MRI->getRegClassOrRegBank(Reg) == RC && MRI->getType(Reg) == Ty &&
!ExcludeMIs.count(MO.getParent()))
return Reg;
}
@ -81,6 +81,8 @@ static void extractInstrFromModule(Oracle &O, MachineFunction &MF) {
auto UE = MRI->use_end();
const auto &RegRC = MRI->getRegClassOrRegBank(Reg);
LLT RegTy = MRI->getType(Reg);
Register NewReg = 0;
// If this is not a physical register and there are some uses.
if (UI != UE) {
@ -88,7 +90,7 @@ static void extractInstrFromModule(Oracle &O, MachineFunction &MF) {
MachineBasicBlock *BB = MI->getParent();
++RI;
while (NewReg == 0 && BB) {
NewReg = getPrevDefOfRCInMBB(*BB, RI, RegRC, ToDelete);
NewReg = getPrevDefOfRCInMBB(*BB, RI, RegRC, RegTy, ToDelete);
// Prepare for idom(BB).
if (auto *IDM = MDT.getNode(BB)->getIDom()) {
BB = IDM->getBlock();