[llvm] r313394 - [ConstantFold] Return the correct type when folding a GEP with vector indices.
Davide Italiano via llvm-commits
llvm-commits at lists.llvm.org
Fri Sep 15 13:53:05 PDT 2017
Author: davide
Date: Fri Sep 15 13:53:05 2017
New Revision: 313394
URL: http://llvm.org/viewvc/llvm-project?rev=313394&view=rev
Log:
[ConstantFold] Return the correct type when folding a GEP with vector indices.
As Eli pointed out (and I got wrong in the first place), langref says: "The
getelementptr returns a vector of pointers, instead of a single address, when one
or more of its arguments is a vector. In such cases, all vector arguments should
have the same number of elements, and every scalar argument will be effectively
broadcast into a vector during address calculation."
Costantfold for gep doesn't really take in account this paragraph, returning a
pointer instead of a vector of pointer which triggers an assertion in RAUW,
as we're trying to replace values with mistmatching types.
Differential Revision: https://reviews.llvm.org/D37928
Added:
llvm/trunk/test/Transforms/InstCombine/pr34627.ll
Modified:
llvm/trunk/lib/IR/ConstantFold.cpp
Modified: llvm/trunk/lib/IR/ConstantFold.cpp
URL: http://llvm.org/viewvc/llvm-project/llvm/trunk/lib/IR/ConstantFold.cpp?rev=313394&r1=313393&r2=313394&view=diff
==============================================================================
--- llvm/trunk/lib/IR/ConstantFold.cpp (original)
+++ llvm/trunk/lib/IR/ConstantFold.cpp Fri Sep 15 13:53:05 2017
@@ -2062,9 +2062,20 @@ Constant *llvm::ConstantFoldGetElementPt
Type *Ty = GetElementPtrInst::getIndexedType(PointeeTy, Idxs);
assert(Ty && "Invalid indices for GEP!");
+ Type *OrigGEPTy = PointerType::get(Ty, PtrTy->getAddressSpace());
Type *GEPTy = PointerType::get(Ty, PtrTy->getAddressSpace());
if (VectorType *VT = dyn_cast<VectorType>(C->getType()))
- GEPTy = VectorType::get(GEPTy, VT->getNumElements());
+ GEPTy = VectorType::get(OrigGEPTy, VT->getNumElements());
+
+ // The GEP returns a vector of pointers when one of more of
+ // its arguments is a vector.
+ for (unsigned i = 0, e = Idxs.size(); i != e; ++i) {
+ if (auto *VT = dyn_cast<VectorType>(Idxs[i]->getType())) {
+ GEPTy = VectorType::get(OrigGEPTy, VT->getNumElements());
+ break;
+ }
+ }
+
return Constant::getNullValue(GEPTy);
}
}
Added: llvm/trunk/test/Transforms/InstCombine/pr34627.ll
URL: http://llvm.org/viewvc/llvm-project/llvm/trunk/test/Transforms/InstCombine/pr34627.ll?rev=313394&view=auto
==============================================================================
--- llvm/trunk/test/Transforms/InstCombine/pr34627.ll (added)
+++ llvm/trunk/test/Transforms/InstCombine/pr34627.ll Fri Sep 15 13:53:05 2017
@@ -0,0 +1,11 @@
+; NOTE: Assertions have been autogenerated by utils/update_test_checks.py
+; RUN: opt -S -instcombine %s |FileCheck %s
+
+define <2 x i16> @patatino() {
+; CHECK-LABEL: @patatino(
+; CHECK-NEXT: ret <2 x i16> zeroinitializer
+;
+ %tmp2 = getelementptr inbounds [1 x i16], [1 x i16]* null, i16 0, <2 x i16> undef
+ %tmp3 = ptrtoint <2 x i16*> %tmp2 to <2 x i16>
+ ret <2 x i16> %tmp3
+}
More information about the llvm-commits
mailing list