[llvm] [SCCP] Support structure in freeze (PR #226537)

[via llvm-commits]

================
@@ -1724,27 +1724,28 @@ void SCCPInstVisitor::visitUnaryOperator(Instruction &I) {
 }
 
 void SCCPInstVisitor::visitFreezeInst(FreezeInst &I) {
-  // If this freeze returns a struct, just mark the result overdefined.
-  // TODO: We could do a lot better than this.
-  if (I.getType()->isStructTy())
-    return (void)markOverdefined(&I);
-
-  ValueLatticeElement V0State = getValueState(I.getOperand(0));
-  ValueLatticeElement &IV = ValueState[&I];
   // resolvedUndefsIn might mark I as overdefined. Bail out, even if we would
   // discover a concrete value later.
-  if (IV.isOverdefined())
+  if (isInstFullyOverDefined(I))
     return (void)markOverdefined(&I);
 
-  // If something is unknown/undef, wait for it to resolve.
-  if (V0State.isUnknownOrUndef())
-    return;
+  forEachLatticeElement(&I, [&](auto GetValueState) {
+    ValueLatticeElement V0State = GetValueState(I.getOperand(0));
+    ValueLatticeElement &IV = GetValueState(&I);
+    // If something is unknown/undef, wait for it to resolve.
+    if (V0State.isUnknownOrUndef())
+      return;
 
-  if (SCCPSolver::isConstant(V0State) &&
-      isGuaranteedNotToBeUndefOrPoison(getConstant(V0State, I.getType())))
-    return (void)markConstant(IV, &I, getConstant(V0State, I.getType()));
+    // Freeze is a no-op on a constant that is guaranteed not to be undef or
+    // poison. Integer constants are represented as single-element ranges and
+    // can never be undef or poison.
----------------
aokblast wrote:

Oh. I think you are right. `getConstantOrNull` doesn't get per-element information if any elements in the structure is undefined. This breaks what freeze want us to have, that is, all of the users should see the same value.

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