[llvm] [SCEV] Updates for getUDivExactExpr() to improve handling of MulExpr numerators/denominators (PR #195704)

[Ryan Buchner via llvm-commits]

================
@@ -3745,6 +3745,39 @@ const SCEV *ScalarEvolution::getUDivExactExpr(SCEVUse LHS, SCEVUse RHS) {
   // just deal with u/exact (multiply, constant). See SCEVDivision towards the
   // end of this file for inspiration.
 
+  if (LHS == RHS)
+    return getConstant(LHS->getType(), 1);
+
+  if (const SCEVMulExpr *RHSMul = dyn_cast<SCEVMulExpr>(RHS);
+      RHSMul && RHSMul->hasNoUnsignedWrap() &&
+      isa<SCEVMulExpr, SCEVConstant, SCEVUnknown>(LHS)) {
+    const SCEV *CurrSCEV = LHS;
+    SmallVector<SCEVUse, 2> Operands;
+    bool OnlyNonZero = true;
+    for (SCEVUse Op : RHSMul->operands()) {
+      if (!OnlyNonZero) {
+        Operands.push_back(Op);
+        continue;
+      }
+      const SCEV *NewSCEV = getUDivExactExpr(CurrSCEV, Op);
+      if (isa<SCEVUDivExpr>(NewSCEV)) {
+        Operands.push_back(Op);
+        continue;
+      }
+      if (!isa<SCEVConstant>(Op)) // Constant is guaranteed non-zero is MulExpr
+        OnlyNonZero = false;
----------------
bababuck wrote:

Thanks for the clarification!

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