/// version of \p Ty, otherwise return \p Ty.
static Type *getReducedType(Value *V, Type *Ty) {
assert(Ty && !Ty->isVectorTy() && "Expect Scalar Type");
- if (auto *VTy = dyn_cast<VectorType>(V->getType())) {
- // FIXME: should this handle scalable vectors?
- return FixedVectorType::get(Ty,
- cast<FixedVectorType>(VTy)->getNumElements());
- }
+ if (auto *VTy = dyn_cast<VectorType>(V->getType()))
+ return VectorType::get(Ty, VTy->getElementCount());
return Ty;
}
declare i32 @use32(i32)
declare <2 x i32> @use32_vec(<2 x i32>)
+declare <vscale x 2 x i32> @use32_scale_vec(<vscale x 2 x i32>)
;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;
;; These tests check cases where expression dag post-dominated by TruncInst
call <2 x i32> @use32_vec(<2 x i32> %T)
ret void
}
+
+define void @const_expression_mul_scale_vec() {
+; CHECK-LABEL: @const_expression_mul_scale_vec(
+; CHECK-NEXT: [[TMP1:%.*]] = call <vscale x 2 x i32> @use32_scale_vec(<vscale x 2 x i32> zeroinitializer)
+; CHECK-NEXT: ret void
+;
+ %A = mul <vscale x 2 x i64> zeroinitializer, zeroinitializer
+ %T = trunc <vscale x 2 x i64> %A to <vscale x 2 x i32>
+ call <vscale x 2 x i32> @use32_scale_vec(<vscale x 2 x i32> %T)
+ ret void
+}
+
+define void @const_expression_zext_scale_vec() {
+; CHECK-LABEL: @const_expression_zext_scale_vec(
+; CHECK-NEXT: [[TMP1:%.*]] = call <vscale x 2 x i32> @use32_scale_vec(<vscale x 2 x i32> zeroinitializer)
+; CHECK-NEXT: ret void
+;
+ %A = zext <vscale x 2 x i32> zeroinitializer to <vscale x 2 x i64>
+ %T = trunc <vscale x 2 x i64> %A to <vscale x 2 x i32>
+ call <vscale x 2 x i32> @use32_scale_vec(<vscale x 2 x i32> %T)
+ ret void
+}
+
+define void @const_expression_trunc_scale_vec() {
+; CHECK-LABEL: @const_expression_trunc_scale_vec(
+; CHECK-NEXT: [[TMP1:%.*]] = call <vscale x 2 x i32> @use32_scale_vec(<vscale x 2 x i32> zeroinitializer)
+; CHECK-NEXT: ret void
+;
+ %T = trunc <vscale x 2 x i64> zeroinitializer to <vscale x 2 x i32>
+ call <vscale x 2 x i32> @use32_scale_vec(<vscale x 2 x i32> %T)
+ ret void
+}