maxjhandsome
/
llvm-project
forked from OSchip/llvm-project
1
0
Fork 0
Code Issues Pull Requests Packages Releases Wiki Activity

Allow BBVectorize to fuse compare instructions. 

llvm-svn: 159088
This commit is contained in:
Hal Finkel 2012-06-23 21:52:50 +00:00
parent 7e9415220a
commit 4b06b1a0ee
3 changed files with 37 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
llvm/include/llvm/Transforms/Vectorize.h
View File

@ -49,6 +49,9 @@ struct VectorizeConfig {
/// @brief Vectorize select instructions. /// @brief Vectorize select instructions.
bool VectorizeSelect; bool VectorizeSelect;
/// @brief Vectorize comparison instructions.
bool VectorizeCmp;
/// @brief Vectorize getelementptr instructions. /// @brief Vectorize getelementptr instructions.
bool VectorizeGEP; bool VectorizeGEP;

8
llvm/lib/Transforms/Vectorize/BBVectorize.cpp
View File

@ -104,6 +104,10 @@ static cl::opt<bool>
NoSelect("bb-vectorize-no-select", cl::init(false), cl::Hidden, NoSelect("bb-vectorize-no-select", cl::init(false), cl::Hidden,
cl::desc("Don't try to vectorize select instructions")); cl::desc("Don't try to vectorize select instructions"));
static cl::opt<bool>
NoCmp("bb-vectorize-no-cmp", cl::init(false), cl::Hidden,
cl::desc("Don't try to vectorize comparison instructions"));
static cl::opt<bool> static cl::opt<bool>
NoGEP("bb-vectorize-no-gep", cl::init(false), cl::Hidden, NoGEP("bb-vectorize-no-gep", cl::init(false), cl::Hidden,
cl::desc("Don't try to vectorize getelementptr instructions")); cl::desc("Don't try to vectorize getelementptr instructions"));
@ -570,6 +574,9 @@ namespace {
} else if (isa<SelectInst>(I)) { } else if (isa<SelectInst>(I)) {
if (!Config.VectorizeSelect) if (!Config.VectorizeSelect)
return false; return false;
} else if (isa<CmpInst>(I)) {
if (!Config.VectorizeCmp)
return false;
} else if (GetElementPtrInst *G = dyn_cast<GetElementPtrInst>(I)) { } else if (GetElementPtrInst *G = dyn_cast<GetElementPtrInst>(I)) {
if (!Config.VectorizeGEP) if (!Config.VectorizeGEP)
return false; return false;
@ -1990,6 +1997,7 @@ VectorizeConfig::VectorizeConfig() {
VectorizeMath = !::NoMath; VectorizeMath = !::NoMath;
VectorizeFMA = !::NoFMA; VectorizeFMA = !::NoFMA;
VectorizeSelect = !::NoSelect; VectorizeSelect = !::NoSelect;
VectorizeCmp = !::NoCmp;
VectorizeGEP = !::NoGEP; VectorizeGEP = !::NoGEP;
VectorizeMemOps = !::NoMemOps; VectorizeMemOps = !::NoMemOps;
AlignedOnly = ::AlignedOnly; AlignedOnly = ::AlignedOnly;

26
llvm/test/Transforms/BBVectorize/simple-sel.ll
View File

@ -27,4 +27,30 @@ define double @test1(double %A1, double %A2, double %B1, double %B2, i1 %C1, i1
; CHECK: ret double %R ; CHECK: ret double %R
} }
; Basic depth-3 chain with select (and vect. compare)
define double @test2(double %A1, double %A2, double %B1, double %B2) {
; CHECK: @test2
; CHECK: %X1.v.i1.1 = insertelement <2 x double> undef, double %B1, i32 0
; CHECK: %X1.v.i0.1 = insertelement <2 x double> undef, double %A1, i32 0
; CHECK: %X1.v.i1.2 = insertelement <2 x double> %X1.v.i1.1, double %B2, i32 1
; CHECK: %X1.v.i0.2 = insertelement <2 x double> %X1.v.i0.1, double %A2, i32 1
%X1 = fsub double %A1, %B1
%X2 = fsub double %A2, %B2
; CHECK: %X1 = fsub <2 x double> %X1.v.i0.2, %X1.v.i1.2
%Y1 = fmul double %X1, %A1
%Y2 = fmul double %X2, %A2
; CHECK: %Y1 = fmul <2 x double> %X1, %X1.v.i0.2
%C1 = fcmp ogt double %X1, %A1
%C2 = fcmp ogt double %X2, %A2
; CHECK: %C1 = fcmp ogt <2 x double> %X1, %X1.v.i0.2
%Z1 = select i1 %C1, double %Y1, double %B1
%Z2 = select i1 %C2, double %Y2, double %B2
; CHECK: %Z1 = select <2 x i1> %C1, <2 x double> %Y1, <2 x double> %X1.v.i1.2
%R = fmul double %Z1, %Z2
; CHECK: %Z1.v.r1 = extractelement <2 x double> %Z1, i32 0
; CHECK: %Z1.v.r2 = extractelement <2 x double> %Z1, i32 1
; CHECK: %R = fmul double %Z1.v.r1, %Z1.v.r2
ret double %R
; CHECK: ret double %R
}