[ConstFold] Don't crash on ConstantExprs when folding get_active_lane_m.

Check if operands are ConstantInt to avoid crashing on constant
expression after #156659.

Author: fhahn

llvm/lib/Analysis/ConstantFolding.cpp

@@ -4253,9 +4253,9 @@ static Constant *ConstantFoldScalableVectorCall(
     return ConstantInt::getFalse(SVTy);
   }
   case Intrinsic::get_active_lane_mask: {
-    auto Op0 = cast<ConstantInt>(Operands[0])->getValue();
-    auto Op1 = cast<ConstantInt>(Operands[1])->getValue();
-    if (Op0.uge(Op1))
+    auto *Op0 = dyn_cast<ConstantInt>(Operands[0]);
+    auto *Op1 = dyn_cast<ConstantInt>(Operands[1]);
+    if (Op0 && Op1 && Op0->getValue().uge(Op1->getValue()))
       return ConstantVector::getNullValue(SVTy);
     break;
   }

llvm/test/Transforms/InstSimplify/ConstProp/active-lane-mask.ll

@@ -3,6 +3,8 @@
 
 target datalayout = "e-m:e-p:32:32-Fi8-i64:64-v128:64:128-a:0:32-n32-S64"
 
+@glob = external global i32
+
 define <16 x i1> @v16i1_0() {
 ; CHECK-LABEL: @v16i1_0(
 ; CHECK-NEXT:  entry:
@@ -337,6 +339,41 @@ entry:
   ret <vscale x 16 x i1> %mask
 }
 
+
+define <vscale x 16 x i1> @nxv16i1_0_constexpr() {
+; CHECK-LABEL: @nxv16i1_0_constexpr(
+; CHECK-NEXT:  entry:
+; CHECK-NEXT:    [[MASK:%.*]] = call <vscale x 16 x i1> @llvm.get.active.lane.mask.nxv16i1.i64(i64 0, i64 ptrtoint (ptr @glob to i64))
+; CHECK-NEXT:    ret <vscale x 16 x i1> [[MASK]]
+;
+entry:
+  %mask = call <vscale x 16 x i1> @llvm.get.active.lane.mask.nxv16i1.i64(i64 0, i64 ptrtoint (ptr @glob to i64))
+  ret <vscale x 16 x i1> %mask
+}
+
+define <vscale x 16 x i1> @nxv16i1_constexpr_0() {
+; CHECK-LABEL: @nxv16i1_constexpr_0(
+; CHECK-NEXT:  entry:
+; CHECK-NEXT:    [[MASK:%.*]] = call <vscale x 16 x i1> @llvm.get.active.lane.mask.nxv16i1.i64(i64 ptrtoint (ptr @glob to i64), i64 0)
+; CHECK-NEXT:    ret <vscale x 16 x i1> [[MASK]]
+;
+entry:
+  %mask = call <vscale x 16 x i1> @llvm.get.active.lane.mask.nxv16i1.i64(i64 ptrtoint (ptr @glob to i64), i64 0)
+  ret <vscale x 16 x i1> %mask
+}
+
+define <vscale x 16 x i1> @nxv16i1_constexpr_constexpr() {
+; CHECK-LABEL: @nxv16i1_constexpr_constexpr(
+; CHECK-NEXT:  entry:
+; CHECK-NEXT:    [[MASK:%.*]] = call <vscale x 16 x i1> @llvm.get.active.lane.mask.nxv16i1.i64(i64 ptrtoint (ptr @glob to i64), i64 ptrtoint (ptr getelementptr inbounds nuw (i8, ptr @glob, i64 2) to i64))
+; CHECK-NEXT:    ret <vscale x 16 x i1> [[MASK]]
+;
+entry:
+  %mask = call <vscale x 16 x i1> @llvm.get.active.lane.mask.nxv16i1.i64(i64 ptrtoint (ptr @glob to i64), i64 ptrtoint (ptr getelementptr inbounds nuw (i8, ptr @glob, i64 2) to i64))
+  ret <vscale x 16 x i1> %mask
+}
+
+
 declare <4 x i1> @llvm.get.active.lane.mask.v4i1.i32(i32, i32)
 declare <8 x i1> @llvm.get.active.lane.mask.v8i1.i32(i32, i32)
 declare <16 x i1> @llvm.get.active.lane.mask.v16i1.i32(i32, i32)