[llvm] [SCEV] Simplify SCEVExpr for PHI to SCEV for operand if operands are identical (PR #115945)

[Akshay Deodhar via llvm-commits]

================
@@ -6019,6 +6019,34 @@ const SCEV *ScalarEvolution::createNodeFromSelectLikePHI(PHINode *PN) {
   return nullptr;
 }
 
+/// Returns SCEV for the first operand of a phi if all phi operands have
+/// identical opcodes and operands
+/// eg.
+/// a: %add = %a + %b
+///    br %c
+/// b: %add1 = %a + %b
+///    br %c
+/// c: %phi = phi [%add, a], [%add1, b]
+/// scev(%phi) => scev(%add)
+const SCEV *
+ScalarEvolution::createNodeForPHIWithIdenticalOperands(PHINode *PN) {
+  BinaryOperator *CommonInst = nullptr;
+  for (Value *Incoming : PN->incoming_values()) {
+    BinaryOperator *IncomingInst = dyn_cast<BinaryOperator>(Incoming);
+    if (CommonInst) {
+      if (!(IncomingInst && CommonInst->isIdenticalTo(IncomingInst)))
+        return nullptr; // Not identical, give up
+    } else if (IncomingInst) {
+      // Remember binary operator
+      CommonInst = IncomingInst;
+    } else {
+      // Not a binary operator, give up
+      return nullptr;
+    }
+  }
+  return CommonInst ? getSCEV(CommonInst) : nullptr;
----------------
akshayrdeodhar wrote:

Restored the original loop, and added an `all_of` check for SCEV exprs of the incoming values being identical.

https://github.com/llvm/llvm-project/pull/115945