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

Scheduler: Allow to select the fusion strategy 

llvm-svn: 149265
This commit is contained in:
Tobias Grosser 2012-01-30 19:38:50 +00:00
parent 42152ff3c7
commit b3ad85b03c
1 changed files with 18 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
polly/lib/ScheduleOptimizer.cpp
View File

@ -56,6 +56,11 @@ SimplifyDeps("polly-opt-simplify-deps",
cl::desc("Dependences should be simplified (yes/no)"), cl::desc("Dependences should be simplified (yes/no)"),
cl::Hidden, cl::init("yes")); cl::Hidden, cl::init("yes"));
static cl::opt<std::string>
FusionStrategy("polly-opt-fusion",
cl::desc("The fusion strategy to choose (min/max)"),
cl::Hidden, cl::init("max"));
namespace { namespace {
class IslScheduleOptimizer : public ScopPass { class IslScheduleOptimizer : public ScopPass {
@ -459,6 +464,19 @@ bool IslScheduleOptimizer::runOnScop(Scop &S) {
DEBUG(dbgs() << "Validity := "; isl_union_map_dump(validity); DEBUG(dbgs() << "Validity := "; isl_union_map_dump(validity);
dbgs() << ";\n"); dbgs() << ";\n");
int IslFusionStrategy;
if (FusionStrategy == "max") {
IslFusionStrategy = ISL_SCHEDULE_FUSE_MAX;
} else if (FusionStrategy == "min") {
IslFusionStrategy = ISL_SCHEDULE_FUSE_MIN;
} else {
errs() << "warning: Unknown fusion strategy. Falling back to maximal "
"fusion.\n";
IslFusionStrategy = ISL_SCHEDULE_FUSE_MAX;
}
isl_options_set_schedule_fuse(S.getIslCtx(), IslFusionStrategy);
isl_options_set_schedule_max_constant_term(S.getIslCtx(), CONSTANT_BOUND); isl_options_set_schedule_max_constant_term(S.getIslCtx(), CONSTANT_BOUND);
isl_options_set_schedule_maximize_band_depth(S.getIslCtx(), 1); isl_options_set_schedule_maximize_band_depth(S.getIslCtx(), 1);