OSchip
/
llvm-project
mirror of https://github.com/THU-DSP-LAB/llvm-project.git
13
0
Fork 2
Code Issues Packages Projects Releases Wiki Activity

[ConstraintElimination] Bail out for GEPs with scalable vectors. 

This fixes a crash with scalable vectors, thanks @nikic for spotting
this!
This commit is contained in:
Florian Hahn 2022-10-10 16:01:20 +01:00
parent b920407cf5
commit 80e49f49e4
No known key found for this signature in database
GPG Key ID: CF59919C6547A668
2 changed files with 54 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
llvm/lib/Transforms/Scalar/ConstraintElimination.cpp
View File

@ -212,7 +212,11 @@ decompose(Value *V, SmallVector<PreconditionTy, 4> &Preconditions,
Value *Op0, *Op1;
ConstantInt *CI;
// Bail out for scalable vectors for now.
auto GTI = gep_type_begin(GEP);
if (isa<ScalableVectorType>(GTI.getIndexedType()))
return {};
int64_t Scale = static_cast<int64_t>(
DL.getTypeAllocSize(GTI.getIndexedType()).getFixedSize());
int64_t MulRes;

50
llvm/test/Transforms/ConstraintElimination/gep-arithmetic-different-types.ll
View File

@ -404,3 +404,53 @@ define i1 @gep_add_nsw_positive_index_struct(ptr %A, ptr %upper, i8 %idx) {
ret i1 %res.3
}
define i1 @gep_constant_positive_index_fixed_vector_ty(ptr %A, ptr %upper) {
; CHECK-LABEL: @gep_constant_positive_index_fixed_vector_ty(
; CHECK-NEXT: [[ADD_I8_4:%.*]] = getelementptr inbounds <4 x i8>, ptr [[A:%.*]], i64 4
; CHECK-NEXT: [[C_0:%.*]] = icmp ult ptr [[ADD_I8_4]], [[UPPER:%.*]]
; CHECK-NEXT: call void @llvm.assume(i1 [[C_0]])
; CHECK-NEXT: [[ADD_I16_4:%.*]] = getelementptr inbounds <4 x i16>, ptr [[A]], i64 4
; CHECK-NEXT: [[C_1:%.*]] = icmp ult ptr [[ADD_I16_4]], [[UPPER]]
; CHECK-NEXT: [[ADD_I16_2:%.*]] = getelementptr inbounds <4 x i16>, ptr [[A]], i64 2
; CHECK-NEXT: [[T_1:%.*]] = icmp ult ptr [[ADD_I16_2]], [[UPPER]]
; CHECK-NEXT: [[RES_1:%.*]] = xor i1 [[C_1]], true
; CHECK-NEXT: ret i1 [[RES_1]]
;
%add.i8.4 = getelementptr inbounds <4 x i8>, ptr %A, i64 4
%c.0 = icmp ult ptr %add.i8.4, %upper
call void @llvm.assume(i1 %c.0)
%add.i16.4 = getelementptr inbounds <4 x i16>, ptr %A, i64 4
%c.1 = icmp ult ptr %add.i16.4, %upper
%add.i16.2 = getelementptr inbounds <4 x i16>, ptr %A, i64 2
%t.1 = icmp ult ptr %add.i16.2, %upper
%res.1 = xor i1 %c.1, %t.1
ret i1 %res.1
}
define i1 @gep_constant_positive_index_scalable_vector_ty(ptr %A, ptr %upper) {
; CHECK-LABEL: @gep_constant_positive_index_scalable_vector_ty(
; CHECK-NEXT: [[ADD_I8_4:%.*]] = getelementptr inbounds <vscale x 4 x i8>, ptr [[A:%.*]], i64 4
; CHECK-NEXT: [[C_0:%.*]] = icmp ult ptr [[ADD_I8_4]], [[UPPER:%.*]]
; CHECK-NEXT: call void @llvm.assume(i1 [[C_0]])
; CHECK-NEXT: [[ADD_I16_4:%.*]] = getelementptr inbounds <vscale x 4 x i16>, ptr [[A]], i64 4
; CHECK-NEXT: [[C_1:%.*]] = icmp ult ptr [[ADD_I16_4]], [[UPPER]]
; CHECK-NEXT: [[ADD_I16_2:%.*]] = getelementptr inbounds <vscale x 4 x i16>, ptr [[A]], i64 2
; CHECK-NEXT: [[T_1:%.*]] = icmp ult ptr [[ADD_I16_2]], [[UPPER]]
; CHECK-NEXT: [[RES_1:%.*]] = xor i1 [[C_1]], [[T_1]]
; CHECK-NEXT: ret i1 [[RES_1]]
;
%add.i8.4 = getelementptr inbounds <vscale x 4 x i8>, ptr %A, i64 4
%c.0 = icmp ult ptr %add.i8.4, %upper
call void @llvm.assume(i1 %c.0)
%add.i16.4 = getelementptr inbounds <vscale x 4 x i16>, ptr %A, i64 4
%c.1 = icmp ult ptr %add.i16.4, %upper
%add.i16.2 = getelementptr inbounds <vscale x 4 x i16>, ptr %A, i64 2
%t.1 = icmp ult ptr %add.i16.2, %upper
%res.1 = xor i1 %c.1, %t.1
ret i1 %res.1
}