[FIX] Allow non-constant assume operand bundle operands.

[jdoerfert]

Summary

Fixes PR49545

[lebedev.ri]

Looks good to me

[zequanwu]

LGTM.

[fhahn]

LGTM, thanks!

llvm/lib/Analysis/AssumeBundleQueries.cpp

if (bundleHasArgument(Bundles, ABA_WasOn))
       Key.first = getValueFromBundleOpInfo(Assume, Bundles, ABA_WasOn);
 
     if (Key.first == nullptr && Key.second == Attribute::None)
       continue;
     if (!bundleHasArgument(Bundles, ABA_Argument)) {
       Result[Key][&Assume] = {0, 0};
       continue;
     }
-    unsigned Val = cast<ConstantInt>(
-                       getValueFromBundleOpInfo(Assume, Bundles, ABA_Argument))
-                       ->getZExtValue();
+    auto *CI = dyn_cast<ConstantInt>(
+        getValueFromBundleOpInfo(Assume, Bundles, ABA_Argument));
+    if (!CI)
+      continue;
+    unsigned Val = CI->getZExtValue();
     auto Lookup = Result.find(Key);
     if (Lookup == Result.end() || !Lookup->second.count(&Assume)) {
       Result[Key][&Assume] = {Val, Val};
       continue;
     }
     Lookup->second[&Assume].Min = std::min(Val, Lookup->second[&Assume].Min);
     Lookup->second[&Assume].Max = std::max(Val, Lookup->second[&Assume].Max);
   }

llvm/unittests/Analysis/AssumeBundleQueriesTest.cpp

Tests.push_back(std::make_pair(
                                        "(nonnull|align|dereferenceable)"));
         Old->replaceAllUsesWith(New);
         Map.clear();
         fillMapFromAssume(*Assume, Map);
         ASSERT_TRUE(FindExactlyAttributes(Map, New,
                                        "(nonnull|align|dereferenceable)"));
         ASSERT_TRUE(FindExactlyAttributes(Map, Old, ""));
       }));
+  Tests.push_back(std::make_pair(
+      "call void @llvm.assume(i1 true) [\"align\"(i8* undef, i32 undef)]",
+      [](Instruction *I) {
+        // Don't crash but don't learn from undef.
+        RetainedKnowledgeMap Map;
+        fillMapFromAssume(*cast<IntrinsicInst>(I), Map);
+
+        ASSERT_TRUE(Map.empty());
+      }));
   RunTest(Head, Tail, Tests);
 }
 
 static void RunRandTest(uint64_t Seed, int Size, int MinCount, int MaxCount,
                         unsigned MaxValue) {
   LLVMContext C;
   SMDiagnostic Err;