[ValueTracking] Handle range assume bundles in computeConstantRange

Author: andjo403

llvm/lib/Analysis/ValueTracking.cpp

@@ -9913,11 +9913,17 @@ ConstantRange llvm::computeConstantRange(const Value *V, bool ForSigned,
     for (auto &AssumeVH : AC->assumptionsFor(V)) {
       if (!AssumeVH)
         continue;
-      CallInst *I = cast<CallInst>(AssumeVH);
+      AssumeInst *I = cast<AssumeInst>(AssumeVH);
       assert(I->getParent()->getParent() == CtxI->getParent()->getParent() &&
              "Got assumption for the wrong function!");
-      assert(I->getIntrinsicID() == Intrinsic::assume &&
-             "must be an assume intrinsic");
+
+      if (AssumeVH.Index != AssumptionCache::ExprResultIdx) {
+        if (std::optional<ConstantRange> Range = getRangeFromBundle(
+                *I, I->bundle_op_info_begin()[AssumeVH.Index]))
+          if (isValidAssumeForContext(I, CtxI, DT))
+            CR = CR.intersectWith(*Range);
+        continue;
+      }
 
       if (!isValidAssumeForContext(I, CtxI, DT))
         continue;

llvm/test/Analysis/BasicAA/range.ll

@@ -272,7 +272,7 @@ entry:
 }
 
 ; CHECK-LABEL: Function: range_assume
-; CHECK: MayAlias: i32* %gep1, i32* %gep2
+; CHECK: NoAlias: i32* %gep1, i32* %gep2
 define void @range_assume(ptr %s, ptr %q) {
   %in_array = load i32, ptr %q
   call void @llvm.assume(i1 true) ["range"(i32 %in_array, i32 0, i32 2)]