[Pipeliner] Fix incorrect phi values in the epilog and kernel

[bcahoon]

Summary

The code that generates the loop definition operand for phis in the epilog and kernel is incorrect in some cases.

In the kernel, when a phi refers to another phi, the code that updates PhiOp2 needs to include the stage difference between the two phis.

In the epilog, the check for using the loop definition instead of the phi definition uses the StageDiffAdj value (the difference between the phi stage and the loop definition stage), but the adjustment is not needed to determine if the current stage contains an iteration with the loop definition.

[bcahoon]

Fixed test case. Needed an extra llc option to work correctly.

[xbolva00]

llvm/lib/CodeGen/MachinePipeliner.cpp

2728

static_cast ?

+ below too

[bcahoon]

Changed code in the patch to use static_cast. I'll submit a separate patch to fix the other occurrences of a cast.

llvm/lib/CodeGen/MachinePipeliner.cpp

for (unsigned np = 0; np < NumPhis; ++np) {
             VRMap[PrevStage - StageDiffAdj].count(LoopVal))
           PhiOp2 = VRMap[PrevStage - StageDiffAdj][LoopVal];
         // Use the value defined by the Phi. We add one because we switch
         // from looking at the loop value to the Phi definition.
         else if (np > 0 && PrevStage == LastStageNum &&
                  VRMap[PrevStage - np + 1].count(Def))
           PhiOp2 = VRMap[PrevStage - np + 1][Def];
         // Use the loop value defined in the kernel.
-        else if ((unsigned)LoopValStage + StageDiffAdj > PrologStage + 1 &&
+        else if ((unsigned)LoopValStage > PrologStage + 1 &&
                  VRMap[PrevStage - StageDiffAdj - np].count(LoopVal))
           PhiOp2 = VRMap[PrevStage - StageDiffAdj - np][LoopVal];
         // Use the value defined by the Phi, unless we're generating the first
         // epilog and the Phi refers to a Phi in a different stage.
         else if (VRMap[PrevStage - np].count(Def) &&
                  (!LoopDefIsPhi || PrevStage != LastStageNum))
           PhiOp2 = VRMap[PrevStage - np][Def];
       }
 
       // Check if we can reuse an existing Phi. This occurs when a Phi
       // references another Phi, and the other Phi is scheduled in an
       // earlier stage. We can try to reuse an existing Phi up until the last
       // stage of the current Phi.
-      if (LoopDefIsPhi && (int)(PrologStage - np) >= StageScheduled) {
+      if (LoopDefIsPhi) {
+        if ((int)(PrologStage - np) >= StageScheduled) {
           int LVNumStages = Schedule.getStagesForPhi(LoopVal);
           int StageDiff = (StageScheduled - LoopValStage);
           LVNumStages -= StageDiff;
           // Make sure the loop value Phi has been processed already.
           if (LVNumStages > (int)np && VRMap[CurStageNum].count(LoopVal)) {
             NewReg = PhiOp2;
             unsigned ReuseStage = CurStageNum;
             if (Schedule.isLoopCarried(this, *PhiInst))
               ReuseStage -= LVNumStages;
             // Check if the Phi to reuse has been generated yet. If not, then
             // there is nothing to reuse.
             if (VRMap[ReuseStage - np].count(LoopVal)) {
               NewReg = VRMap[ReuseStage - np][LoopVal];
 
               rewriteScheduledInstr(NewBB, Schedule, InstrMap, CurStageNum, np,
                                     &*BBI, Def, NewReg);
               // Update the map with the new Phi name.
               VRMap[CurStageNum - np][Def] = NewReg;
               PhiOp2 = NewReg;
               if (VRMap[LastStageNum - np - 1].count(LoopVal))
                 PhiOp2 = VRMap[LastStageNum - np - 1][LoopVal];
 
               if (IsLast && np == NumPhis - 1)
                 replaceRegUsesAfterLoop(Def, NewReg, BB, MRI, LIS);
               continue;
             }
-        } else if (InKernel && StageDiff > 0 &&
-                   VRMap[CurStageNum - StageDiff - np].count(LoopVal))
+          }
+        }
+        if (InKernel && StageDiff > 0 &&
           PhiOp2 = VRMap[CurStageNum - StageDiff - np][LoopVal];
       }
 
       const TargetRegisterClass *RC = MRI.getRegClass(Def);
       NewReg = MRI.createVirtualRegister(RC);
 
       MachineInstrBuilder NewPhi =
           BuildMI(*NewBB, NewBB->getFirstNonPHI(), DebugLoc(),

llvm/test/CodeGen/Hexagon/swp-epilog-phi11.ll

+; RUN: llc -mtriple=hexagon-unknown-elf -mcpu=hexagonv55 < %s | FileCheck %s
+
+; Test that the pipeliner correctly generates the operands in the
+; epilog.
+
+; CHECK: loop0
+; CHECK: r{{[0-9]+}} = sfsub([[REG0:r([0-9]+)]],[[REG1:r([0-9]+)]])
+; CHECK: endloop0
+; CHECK: r{{[0-9]+}} = sfsub([[REG0]],[[REG1]])
+; CHECK: r{{[0-9]+}} = sfsub([[REG0]],r{{[0-9]+}})
+
+define dso_local void @test(i32 %m) local_unnamed_addr #0 {
+entry:
+  %div = sdiv i32 %m, 2
+  %sub = add nsw i32 %div, -1
+  br label %for.body.prol
+
+for.body.prol:
+  %i.0106.prol = phi i32 [ undef, %for.body.prol ], [ %sub, %entry ]
+  %sr.prol = phi float [ %0, %for.body.prol ], [ undef, %entry ]
+  %sr109.prol = phi float [ %sr.prol, %for.body.prol ], [ undef, %entry ]
+  %prol.iter = phi i32 [ %prol.iter.sub, %for.body.prol ], [ undef, %entry ]
+  %0 = load float, float* undef, align 4
+  %sub7.prol = fsub contract float %sr109.prol, %0
+  store float %sub7.prol, float* null, align 4
+  %prol.iter.sub = add i32 %prol.iter, -1
+  %prol.iter.cmp = icmp eq i32 %prol.iter.sub, 0
+  br i1 %prol.iter.cmp, label %for.body.prol.loopexit, label %for.body.prol
+
+for.body.prol.loopexit:
+  unreachable
+}
+