Fix bug in DAGBuilder for getelementptr with expanded vector.

Replacing the usage of MVT with EVT in case the vector type is expanded.
Differential Revision: https://reviews.llvm.org/D23306

llvm-svn: 278913
This commit is contained in:
Ayman Musa 2016-08-17 07:52:15 +00:00
parent b4447ed5f1
commit 6426d4ee9c
2 changed files with 26 additions and 1 deletions

View File

@ -3329,8 +3329,9 @@ void SelectionDAGBuilder::visitGetElementPtr(const User &I) {
if (CI->isZero())
continue;
APInt Offs = ElementSize * CI->getValue().sextOrTrunc(PtrSize);
LLVMContext &Context = *DAG.getContext();
SDValue OffsVal = VectorWidth ?
DAG.getConstant(Offs, dl, MVT::getVectorVT(PtrTy, VectorWidth)) :
DAG.getConstant(Offs, dl, EVT::getVectorVT(Context, PtrTy, VectorWidth)) :
DAG.getConstant(Offs, dl, PtrTy);
// In an inbouds GEP with an offset that is nonnegative even when

View File

@ -0,0 +1,24 @@
; RUN: llc < %s -O2 -mattr=avx512f -mtriple=x86_64-unknown | FileCheck %s
%struct.S1 = type { %struct.S1*, %struct.S1* }
define %struct.S1** @malloc_init_state(<64 x %struct.S1**> %tmp, i32 %ind) {
entry:
%Vec = getelementptr inbounds %struct.S1*, <64 x %struct.S1**> %tmp , i64 2
%ptr = extractelement <64 x %struct.S1**> %Vec, i32 %ind
ret %struct.S1** %ptr
}
; CHECK: .LCPI0_0:
; CHECK: .quad 16
; CHECK: vpbroadcastq .LCPI0_0(%rip), [[Z1:%zmm[0-9]]]
; CHECK-NEXT: vpaddq [[Z1]], [[Z2:%zmm[0-9]]], [[Z2]]
; CHECK-NEXT: vpaddq [[Z1]], [[Z3:%zmm[0-9]]], [[Z3]]
; CHECK-NEXT: vpaddq [[Z1]], [[Z4:%zmm[0-9]]], [[Z4]]
; CHECK-NEXT: vpaddq [[Z1]], [[Z5:%zmm[0-9]]], [[Z5]]
; CHECK-NEXT: vpaddq [[Z1]], [[Z6:%zmm[0-9]]], [[Z6]]
; CHECK-NEXT: vpaddq [[Z1]], [[Z7:%zmm[0-9]]], [[Z7]]
; CHECK-NEXT: vpaddq [[Z1]], [[Z8:%zmm[0-9]]], [[Z8]]
; CHECK-NEXT: vpaddq [[Z1]], [[Z9:%zmm[0-9]]], [[Z9]]