diff --git a/lib/Transforms/Vectorize/VecUtils.cpp b/lib/Transforms/Vectorize/VecUtils.cpp index 21e6cdde307..658162aa9f6 100644 --- a/lib/Transforms/Vectorize/VecUtils.cpp +++ b/lib/Transforms/Vectorize/VecUtils.cpp @@ -225,7 +225,7 @@ Value *BoUpSLP::isUnsafeToSink(Instruction *Src, Instruction *Dst) { return 0; } -void BoUpSLP::vectorizeArith(ArrayRef Operands) { +Value *BoUpSLP::vectorizeArith(ArrayRef Operands) { int LastIdx = getLastIndex(Operands, Operands.size()); Instruction *Loc = getInsertionPoint(LastIdx); Builder.SetInsertPoint(Loc); @@ -241,6 +241,8 @@ void BoUpSLP::vectorizeArith(ArrayRef Operands) { Value *S = Builder.CreateExtractElement(Vec, Builder.getInt32(i)); Operands[i]->replaceAllUsesWith(S); } + + return Vec; } int BoUpSLP::getTreeCost(ArrayRef VL) { diff --git a/lib/Transforms/Vectorize/VecUtils.h b/lib/Transforms/Vectorize/VecUtils.h index d41d2ed63c8..e0d87f71d41 100644 --- a/lib/Transforms/Vectorize/VecUtils.h +++ b/lib/Transforms/Vectorize/VecUtils.h @@ -72,7 +72,7 @@ struct BoUpSLP { bool vectorizeStores(ArrayRef Stores, int costThreshold); /// \brief Vectorize a group of scalars into a vector tree. - void vectorizeArith(ArrayRef Operands); + Value *vectorizeArith(ArrayRef Operands); /// \returns the list of new instructions that were added in order to collect /// scalars into vectors. This list can be used to further optimize the gather