blob: 06ac8dcc476a3d2a5c333a2733707b39093d35be [file] [log] [blame]
Tobias Grosser30aa24c2011-05-14 19:02:06 +00001//===- Schedule.cpp - Calculate an optimized schedule ---------------------===//
2//
3// The LLVM Compiler Infrastructure
4//
5// This file is distributed under the University of Illinois Open Source
6// License. See LICENSE.TXT for details.
7//
8//===----------------------------------------------------------------------===//
9//
10// This pass the isl to calculate a schedule that is optimized for parallelism
11// and tileablility. The algorithm used in isl is an optimized version of the
12// algorithm described in following paper:
13//
14// U. Bondhugula, A. Hartono, J. Ramanujam, and P. Sadayappan.
15// A Practical Automatic Polyhedral Parallelizer and Locality Optimizer.
16// In Proceedings of the 2008 ACM SIGPLAN Conference On Programming Language
17// Design and Implementation, PLDI ’08, pages 101–113. ACM, 2008.
18//===----------------------------------------------------------------------===//
19
Tobias Grosser967239c2011-10-23 20:59:44 +000020#include "polly/ScheduleOptimizer.h"
Tobias Grosser2493e922011-12-07 07:42:57 +000021#include "isl/aff.h"
Tobias Grosserde68cc92011-06-30 20:01:02 +000022#include "isl/band.h"
Tobias Grosser8ad6bc32012-01-31 13:26:29 +000023#include "isl/constraint.h"
24#include "isl/map.h"
Tobias Grosser42152ff2012-01-30 19:38:47 +000025#include "isl/options.h"
Tobias Grosser8ad6bc32012-01-31 13:26:29 +000026#include "isl/schedule.h"
27#include "isl/space.h"
Tobias Grosser83628182013-05-07 08:11:54 +000028#include "polly/CodeGen/CodeGeneration.h"
29#include "polly/Dependences.h"
30#include "polly/LinkAllPasses.h"
31#include "polly/Options.h"
32#include "polly/ScopInfo.h"
Tobias Grosser30aa24c2011-05-14 19:02:06 +000033#include "llvm/Support/Debug.h"
34
35using namespace llvm;
36using namespace polly;
37
Chandler Carruth95fef942014-04-22 03:30:19 +000038#define DEBUG_TYPE "polly-opt-isl"
39
Tobias Grosser58032cb2013-06-23 01:29:29 +000040namespace polly {
41bool DisablePollyTiling;
42}
Tobias Grossere602a072013-05-07 07:30:56 +000043static cl::opt<bool, true>
44DisableTiling("polly-no-tiling", cl::desc("Disable tiling in the scheduler"),
Tobias Grosser637bd632013-05-07 07:31:10 +000045 cl::location(polly::DisablePollyTiling), cl::init(false),
Tobias Grosser64e8e372014-03-13 23:37:43 +000046 cl::ZeroOrMore, cl::cat(PollyCategory));
Tobias Grosser353a2682011-10-23 20:59:26 +000047
Tobias Grossera26db472012-01-30 19:38:43 +000048static cl::opt<std::string>
Tobias Grosser1deda292012-02-14 14:02:48 +000049OptimizeDeps("polly-opt-optimize-only",
50 cl::desc("Only a certain kind of dependences (all/raw)"),
Tobias Grosser64e8e372014-03-13 23:37:43 +000051 cl::Hidden, cl::init("all"), cl::ZeroOrMore,
52 cl::cat(PollyCategory));
Tobias Grosser1deda292012-02-14 14:02:48 +000053
54static cl::opt<std::string>
Tobias Grossera26db472012-01-30 19:38:43 +000055SimplifyDeps("polly-opt-simplify-deps",
Tobias Grosser4d96c8d2013-03-23 01:05:07 +000056 cl::desc("Dependences should be simplified (yes/no)"), cl::Hidden,
Tobias Grosser64e8e372014-03-13 23:37:43 +000057 cl::init("yes"), cl::ZeroOrMore, cl::cat(PollyCategory));
Tobias Grossera26db472012-01-30 19:38:43 +000058
Tobias Grosser992e60c2012-02-20 08:41:15 +000059static cl::opt<int>
60MaxConstantTerm("polly-opt-max-constant-term",
61 cl::desc("The maximal constant term allowed (-1 is unlimited)"),
Tobias Grosser64e8e372014-03-13 23:37:43 +000062 cl::Hidden, cl::init(20), cl::ZeroOrMore,
63 cl::cat(PollyCategory));
Tobias Grosser992e60c2012-02-20 08:41:15 +000064
Tobias Grosser92f54802012-02-20 08:41:47 +000065static cl::opt<int>
66MaxCoefficient("polly-opt-max-coefficient",
67 cl::desc("The maximal coefficient allowed (-1 is unlimited)"),
Tobias Grosser64e8e372014-03-13 23:37:43 +000068 cl::Hidden, cl::init(20), cl::ZeroOrMore,
69 cl::cat(PollyCategory));
Tobias Grosser92f54802012-02-20 08:41:47 +000070
Tobias Grossere602a072013-05-07 07:30:56 +000071static cl::opt<std::string>
72FusionStrategy("polly-opt-fusion",
73 cl::desc("The fusion strategy to choose (min/max)"), cl::Hidden,
Tobias Grosser64e8e372014-03-13 23:37:43 +000074 cl::init("min"), cl::ZeroOrMore, cl::cat(PollyCategory));
Tobias Grosserb3ad85b2012-01-30 19:38:50 +000075
Tobias Grossere602a072013-05-07 07:30:56 +000076static cl::opt<std::string>
77MaximizeBandDepth("polly-opt-maximize-bands",
78 cl::desc("Maximize the band depth (yes/no)"), cl::Hidden,
Tobias Grosser64e8e372014-03-13 23:37:43 +000079 cl::init("yes"), cl::ZeroOrMore, cl::cat(PollyCategory));
Tobias Grosser95e860c2012-01-30 19:38:54 +000080
Tobias Grosser30aa24c2011-05-14 19:02:06 +000081namespace {
82
Tobias Grosser4d96c8d2013-03-23 01:05:07 +000083class IslScheduleOptimizer : public ScopPass {
Tobias Grosser4d96c8d2013-03-23 01:05:07 +000084public:
85 static char ID;
Tobias Grosser5a56cbf2014-04-16 07:33:47 +000086 explicit IslScheduleOptimizer() : ScopPass(ID) { LastSchedule = nullptr; }
Tobias Grosser28781422012-10-16 07:29:19 +000087
Tobias Grosser4d96c8d2013-03-23 01:05:07 +000088 ~IslScheduleOptimizer() { isl_schedule_free(LastSchedule); }
Tobias Grosser30aa24c2011-05-14 19:02:06 +000089
Tobias Grosser4d96c8d2013-03-23 01:05:07 +000090 virtual bool runOnScop(Scop &S);
91 void printScop(llvm::raw_ostream &OS) const;
92 void getAnalysisUsage(AnalysisUsage &AU) const;
Tobias Grosser460e9a42012-04-25 13:22:43 +000093
Tobias Grosser4d96c8d2013-03-23 01:05:07 +000094private:
95 isl_schedule *LastSchedule;
Tobias Grosser28781422012-10-16 07:29:19 +000096
Tobias Grosser4d96c8d2013-03-23 01:05:07 +000097 static void extendScattering(Scop &S, unsigned NewDimensions);
Tobias Grosser460e9a42012-04-25 13:22:43 +000098
Tobias Grosser4d96c8d2013-03-23 01:05:07 +000099 /// @brief Create a map that describes a n-dimensonal tiling.
100 ///
101 /// getTileMap creates a map from a n-dimensional scattering space into an
102 /// 2*n-dimensional scattering space. The map describes a rectangular
103 /// tiling.
104 ///
105 /// Example:
106 /// scheduleDimensions = 2, parameterDimensions = 1, tileSize = 32
107 ///
108 /// tileMap := [p0] -> {[s0, s1] -> [t0, t1, s0, s1]:
109 /// t0 % 32 = 0 and t0 <= s0 < t0 + 32 and
110 /// t1 % 32 = 0 and t1 <= s1 < t1 + 32}
111 ///
112 /// Before tiling:
113 ///
114 /// for (i = 0; i < N; i++)
115 /// for (j = 0; j < M; j++)
116 /// S(i,j)
117 ///
118 /// After tiling:
119 ///
120 /// for (t_i = 0; t_i < N; i+=32)
121 /// for (t_j = 0; t_j < M; j+=32)
122 /// for (i = t_i; i < min(t_i + 32, N); i++) | Unknown that N % 32 = 0
123 /// for (j = t_j; j < t_j + 32; j++) | Known that M % 32 = 0
124 /// S(i,j)
125 ///
126 static isl_basic_map *getTileMap(isl_ctx *ctx, int scheduleDimensions,
127 isl_space *SpaceModel, int tileSize = 32);
Tobias Grosser460e9a42012-04-25 13:22:43 +0000128
Tobias Grosser4d96c8d2013-03-23 01:05:07 +0000129 /// @brief Get the schedule for this band.
130 ///
131 /// Polly applies transformations like tiling on top of the isl calculated
132 /// value. This can influence the number of scheduling dimension. The
133 /// number of schedule dimensions is returned in the parameter 'Dimension'.
134 static isl_union_map *getScheduleForBand(isl_band *Band, int *Dimensions);
Tobias Grosser460e9a42012-04-25 13:22:43 +0000135
Tobias Grosser4d96c8d2013-03-23 01:05:07 +0000136 /// @brief Create a map that pre-vectorizes one scheduling dimension.
137 ///
138 /// getPrevectorMap creates a map that maps each input dimension to the same
139 /// output dimension, except for the dimension DimToVectorize.
140 /// DimToVectorize is strip mined by 'VectorWidth' and the newly created
141 /// point loop of DimToVectorize is moved to the innermost level.
142 ///
143 /// Example (DimToVectorize=0, ScheduleDimensions=2, VectorWidth=4):
144 ///
145 /// | Before transformation
146 /// |
147 /// | A[i,j] -> [i,j]
148 /// |
149 /// | for (i = 0; i < 128; i++)
150 /// | for (j = 0; j < 128; j++)
151 /// | A(i,j);
152 ///
153 /// Prevector map:
154 /// [i,j] -> [it,j,ip] : it % 4 = 0 and it <= ip <= it + 3 and i = ip
155 ///
156 /// | After transformation:
157 /// |
158 /// | A[i,j] -> [it,j,ip] : it % 4 = 0 and it <= ip <= it + 3 and i = ip
159 /// |
160 /// | for (it = 0; it < 128; it+=4)
161 /// | for (j = 0; j < 128; j++)
162 /// | for (ip = max(0,it); ip < min(128, it + 3); ip++)
163 /// | A(ip,j);
164 ///
165 /// The goal of this transformation is to create a trivially vectorizable
166 /// loop. This means a parallel loop at the innermost level that has a
167 /// constant number of iterations corresponding to the target vector width.
168 ///
169 /// This transformation creates a loop at the innermost level. The loop has
170 /// a constant number of iterations, if the number of loop iterations at
171 /// DimToVectorize can be divided by VectorWidth. The default VectorWidth is
172 /// currently constant and not yet target specific. This function does not
173 /// reason about parallelism.
174 static isl_map *getPrevectorMap(isl_ctx *ctx, int DimToVectorize,
175 int ScheduleDimensions, int VectorWidth = 4);
Tobias Grosser460e9a42012-04-25 13:22:43 +0000176
Tobias Grosser4d96c8d2013-03-23 01:05:07 +0000177 /// @brief Get the scheduling map for a list of bands.
178 ///
179 /// Walk recursively the forest of bands to combine the schedules of the
180 /// individual bands to the overall schedule. In case tiling is requested,
181 /// the individual bands are tiled.
182 static isl_union_map *getScheduleForBandList(isl_band_list *BandList);
Tobias Grosser460e9a42012-04-25 13:22:43 +0000183
Tobias Grosser4d96c8d2013-03-23 01:05:07 +0000184 static isl_union_map *getScheduleMap(isl_schedule *Schedule);
Tobias Grosser28781422012-10-16 07:29:19 +0000185
Tobias Grosser20532b82014-04-11 17:56:49 +0000186 using llvm::Pass::doFinalization;
187
188 virtual bool doFinalization() {
Tobias Grosser4d96c8d2013-03-23 01:05:07 +0000189 isl_schedule_free(LastSchedule);
Tobias Grosser5a56cbf2014-04-16 07:33:47 +0000190 LastSchedule = nullptr;
Tobias Grosser4d96c8d2013-03-23 01:05:07 +0000191 return true;
192 }
193};
Tobias Grosser30aa24c2011-05-14 19:02:06 +0000194}
195
Tobias Grosser73600b82011-10-08 00:30:40 +0000196char IslScheduleOptimizer::ID = 0;
Tobias Grosser30aa24c2011-05-14 19:02:06 +0000197
Tobias Grosser460e9a42012-04-25 13:22:43 +0000198void IslScheduleOptimizer::extendScattering(Scop &S, unsigned NewDimensions) {
Tobias Grosser30aa24c2011-05-14 19:02:06 +0000199 for (Scop::iterator SI = S.begin(), SE = S.end(); SI != SE; ++SI) {
Tobias Grossercf3942d2011-10-06 00:04:05 +0000200 ScopStmt *Stmt = *SI;
Tobias Grossercf3942d2011-10-06 00:04:05 +0000201 unsigned OldDimensions = Stmt->getNumScattering();
202 isl_space *Space;
Tobias Grosser29666112012-05-22 10:47:31 +0000203 isl_map *Map, *New;
Tobias Grossercf3942d2011-10-06 00:04:05 +0000204
205 Space = isl_space_alloc(Stmt->getIslCtx(), 0, OldDimensions, NewDimensions);
Tobias Grosser29666112012-05-22 10:47:31 +0000206 Map = isl_map_universe(Space);
Tobias Grosser30aa24c2011-05-14 19:02:06 +0000207
Tobias Grosser29666112012-05-22 10:47:31 +0000208 for (unsigned i = 0; i < OldDimensions; i++)
209 Map = isl_map_equate(Map, isl_dim_in, i, isl_dim_out, i);
Tobias Grosser30aa24c2011-05-14 19:02:06 +0000210
Tobias Grosser29666112012-05-22 10:47:31 +0000211 for (unsigned i = OldDimensions; i < NewDimensions; i++)
212 Map = isl_map_fix_si(Map, isl_dim_out, i, 0);
Tobias Grosser30aa24c2011-05-14 19:02:06 +0000213
Tobias Grosser29666112012-05-22 10:47:31 +0000214 Map = isl_map_align_params(Map, S.getParamSpace());
215 New = isl_map_apply_range(Stmt->getScattering(), Map);
216 Stmt->setScattering(New);
Tobias Grosser30aa24c2011-05-14 19:02:06 +0000217 }
218}
219
Tobias Grossere602a072013-05-07 07:30:56 +0000220isl_basic_map *IslScheduleOptimizer::getTileMap(isl_ctx *ctx,
221 int scheduleDimensions,
222 isl_space *SpaceModel,
223 int tileSize) {
Tobias Grosserde68cc92011-06-30 20:01:02 +0000224 // We construct
225 //
226 // tileMap := [p0] -> {[s0, s1] -> [t0, t1, p0, p1, a0, a1]:
227 // s0 = a0 * 32 and s0 = p0 and t0 <= p0 < t0 + 32 and
228 // s1 = a1 * 32 and s1 = p1 and t1 <= p1 < t1 + 32}
229 //
230 // and project out the auxilary dimensions a0 and a1.
Tobias Grosser4d96c8d2013-03-23 01:05:07 +0000231 isl_space *Space =
232 isl_space_alloc(ctx, 0, scheduleDimensions, scheduleDimensions * 3);
Tobias Grosserf5338802011-10-06 00:03:35 +0000233 isl_basic_map *tileMap = isl_basic_map_universe(isl_space_copy(Space));
234
235 isl_local_space *LocalSpace = isl_local_space_from_space(Space);
Tobias Grosser30aa24c2011-05-14 19:02:06 +0000236
Tobias Grosserde68cc92011-06-30 20:01:02 +0000237 for (int x = 0; x < scheduleDimensions; x++) {
238 int sX = x;
239 int tX = x;
240 int pX = scheduleDimensions + x;
241 int aX = 2 * scheduleDimensions + x;
Tobias Grosser30aa24c2011-05-14 19:02:06 +0000242
Tobias Grosserde68cc92011-06-30 20:01:02 +0000243 isl_constraint *c;
Tobias Grosser30aa24c2011-05-14 19:02:06 +0000244
Tobias Grosserde68cc92011-06-30 20:01:02 +0000245 // sX = aX * tileSize;
Tobias Grosserf5338802011-10-06 00:03:35 +0000246 c = isl_equality_alloc(isl_local_space_copy(LocalSpace));
Tobias Grosserde68cc92011-06-30 20:01:02 +0000247 isl_constraint_set_coefficient_si(c, isl_dim_out, sX, 1);
248 isl_constraint_set_coefficient_si(c, isl_dim_out, aX, -tileSize);
249 tileMap = isl_basic_map_add_constraint(tileMap, c);
Tobias Grosser30aa24c2011-05-14 19:02:06 +0000250
Tobias Grosserde68cc92011-06-30 20:01:02 +0000251 // pX = sX;
Tobias Grosserf5338802011-10-06 00:03:35 +0000252 c = isl_equality_alloc(isl_local_space_copy(LocalSpace));
Tobias Grosserde68cc92011-06-30 20:01:02 +0000253 isl_constraint_set_coefficient_si(c, isl_dim_out, pX, 1);
254 isl_constraint_set_coefficient_si(c, isl_dim_in, sX, -1);
255 tileMap = isl_basic_map_add_constraint(tileMap, c);
Tobias Grosser30aa24c2011-05-14 19:02:06 +0000256
Tobias Grosserde68cc92011-06-30 20:01:02 +0000257 // tX <= pX
Tobias Grosserf5338802011-10-06 00:03:35 +0000258 c = isl_inequality_alloc(isl_local_space_copy(LocalSpace));
Tobias Grosserde68cc92011-06-30 20:01:02 +0000259 isl_constraint_set_coefficient_si(c, isl_dim_out, pX, 1);
260 isl_constraint_set_coefficient_si(c, isl_dim_out, tX, -1);
261 tileMap = isl_basic_map_add_constraint(tileMap, c);
262
263 // pX <= tX + (tileSize - 1)
Tobias Grosserf5338802011-10-06 00:03:35 +0000264 c = isl_inequality_alloc(isl_local_space_copy(LocalSpace));
Tobias Grosserde68cc92011-06-30 20:01:02 +0000265 isl_constraint_set_coefficient_si(c, isl_dim_out, tX, 1);
266 isl_constraint_set_coefficient_si(c, isl_dim_out, pX, -1);
Tobias Grosser30aa24c2011-05-14 19:02:06 +0000267 isl_constraint_set_constant_si(c, tileSize - 1);
Tobias Grosserde68cc92011-06-30 20:01:02 +0000268 tileMap = isl_basic_map_add_constraint(tileMap, c);
Tobias Grosser30aa24c2011-05-14 19:02:06 +0000269 }
270
Tobias Grosserde68cc92011-06-30 20:01:02 +0000271 // Project out auxilary dimensions.
Tobias Grosser30aa24c2011-05-14 19:02:06 +0000272 //
Tobias Grosserde68cc92011-06-30 20:01:02 +0000273 // The auxilary dimensions are transformed into existentially quantified ones.
274 // This reduces the number of visible scattering dimensions and allows Cloog
275 // to produces better code.
Tobias Grosser4d96c8d2013-03-23 01:05:07 +0000276 tileMap = isl_basic_map_project_out(
277 tileMap, isl_dim_out, 2 * scheduleDimensions, scheduleDimensions);
Tobias Grosserf5338802011-10-06 00:03:35 +0000278 isl_local_space_free(LocalSpace);
Tobias Grosserde68cc92011-06-30 20:01:02 +0000279 return tileMap;
280}
Tobias Grosser30aa24c2011-05-14 19:02:06 +0000281
Tobias Grossere602a072013-05-07 07:30:56 +0000282isl_union_map *IslScheduleOptimizer::getScheduleForBand(isl_band *Band,
283 int *Dimensions) {
Tobias Grosser1ae9a602011-11-17 12:56:03 +0000284 isl_union_map *PartialSchedule;
Tobias Grosserde68cc92011-06-30 20:01:02 +0000285 isl_ctx *ctx;
Tobias Grosserf5338802011-10-06 00:03:35 +0000286 isl_space *Space;
Tobias Grosser1ae9a602011-11-17 12:56:03 +0000287 isl_basic_map *TileMap;
288 isl_union_map *TileUMap;
Tobias Grosserde68cc92011-06-30 20:01:02 +0000289
Tobias Grosser1ae9a602011-11-17 12:56:03 +0000290 PartialSchedule = isl_band_get_partial_schedule(Band);
Tobias Grosserb6033392011-12-08 13:02:58 +0000291 *Dimensions = isl_band_n_member(Band);
Tobias Grosserde68cc92011-06-30 20:01:02 +0000292
Tobias Grosser79b30202011-11-17 12:56:00 +0000293 if (DisableTiling)
Tobias Grosser1ae9a602011-11-17 12:56:03 +0000294 return PartialSchedule;
Tobias Grosser353a2682011-10-23 20:59:26 +0000295
Tobias Grosserb6033392011-12-08 13:02:58 +0000296 // It does not make any sense to tile a band with just one dimension.
297 if (*Dimensions == 1)
298 return PartialSchedule;
299
Tobias Grosser1ae9a602011-11-17 12:56:03 +0000300 ctx = isl_union_map_get_ctx(PartialSchedule);
301 Space = isl_union_map_get_space(PartialSchedule);
Tobias Grosserde68cc92011-06-30 20:01:02 +0000302
Tobias Grosserb6033392011-12-08 13:02:58 +0000303 TileMap = getTileMap(ctx, *Dimensions, Space);
Tobias Grosser1ae9a602011-11-17 12:56:03 +0000304 TileUMap = isl_union_map_from_map(isl_map_from_basic_map(TileMap));
305 TileUMap = isl_union_map_align_params(TileUMap, Space);
Tobias Grosserb6033392011-12-08 13:02:58 +0000306 *Dimensions = 2 * *Dimensions;
307
Tobias Grosser1ae9a602011-11-17 12:56:03 +0000308 return isl_union_map_apply_range(PartialSchedule, TileUMap);
Tobias Grosserde68cc92011-06-30 20:01:02 +0000309}
310
Tobias Grossere602a072013-05-07 07:30:56 +0000311isl_map *IslScheduleOptimizer::getPrevectorMap(isl_ctx *ctx, int DimToVectorize,
312 int ScheduleDimensions,
313 int VectorWidth) {
Tobias Grosser2493e922011-12-07 07:42:57 +0000314 isl_space *Space;
315 isl_local_space *LocalSpace, *LocalSpaceRange;
316 isl_set *Modulo;
317 isl_map *TilingMap;
Tobias Grosserc6699b72011-06-30 20:29:13 +0000318 isl_constraint *c;
Tobias Grosser2493e922011-12-07 07:42:57 +0000319 isl_aff *Aff;
320 int PointDimension; /* ip */
321 int TileDimension; /* it */
Tobias Grosseredab1352013-06-21 06:41:31 +0000322 isl_val *VectorWidthMP;
Tobias Grosserc6699b72011-06-30 20:29:13 +0000323
Tobias Grosser4d96c8d2013-03-23 01:05:07 +0000324 assert(0 <= DimToVectorize && DimToVectorize < ScheduleDimensions);
Tobias Grosserf5338802011-10-06 00:03:35 +0000325
Tobias Grosser2493e922011-12-07 07:42:57 +0000326 Space = isl_space_alloc(ctx, 0, ScheduleDimensions, ScheduleDimensions + 1);
327 TilingMap = isl_map_universe(isl_space_copy(Space));
328 LocalSpace = isl_local_space_from_space(Space);
329 PointDimension = ScheduleDimensions;
330 TileDimension = DimToVectorize;
331
332 // Create an identity map for everything except DimToVectorize and map
333 // DimToVectorize to the point loop at the innermost dimension.
334 for (int i = 0; i < ScheduleDimensions; i++) {
Tobias Grosserf5338802011-10-06 00:03:35 +0000335 c = isl_equality_alloc(isl_local_space_copy(LocalSpace));
Tobias Grosserc6699b72011-06-30 20:29:13 +0000336 isl_constraint_set_coefficient_si(c, isl_dim_in, i, -1);
Tobias Grosser2493e922011-12-07 07:42:57 +0000337
338 if (i == DimToVectorize)
339 isl_constraint_set_coefficient_si(c, isl_dim_out, PointDimension, 1);
340 else
341 isl_constraint_set_coefficient_si(c, isl_dim_out, i, 1);
342
343 TilingMap = isl_map_add_constraint(TilingMap, c);
Tobias Grosserc6699b72011-06-30 20:29:13 +0000344 }
345
Tobias Grosser2493e922011-12-07 07:42:57 +0000346 // it % 'VectorWidth' = 0
347 LocalSpaceRange = isl_local_space_range(isl_local_space_copy(LocalSpace));
348 Aff = isl_aff_zero_on_domain(LocalSpaceRange);
349 Aff = isl_aff_set_constant_si(Aff, VectorWidth);
350 Aff = isl_aff_set_coefficient_si(Aff, isl_dim_in, TileDimension, 1);
Tobias Grosseredab1352013-06-21 06:41:31 +0000351 VectorWidthMP = isl_val_int_from_si(ctx, VectorWidth);
352 Aff = isl_aff_mod_val(Aff, VectorWidthMP);
Tobias Grosser2493e922011-12-07 07:42:57 +0000353 Modulo = isl_pw_aff_zero_set(isl_pw_aff_from_aff(Aff));
354 TilingMap = isl_map_intersect_range(TilingMap, Modulo);
Tobias Grosserc6699b72011-06-30 20:29:13 +0000355
Tobias Grosser2493e922011-12-07 07:42:57 +0000356 // it <= ip
Tobias Grosserf5338802011-10-06 00:03:35 +0000357 c = isl_inequality_alloc(isl_local_space_copy(LocalSpace));
Tobias Grosser2493e922011-12-07 07:42:57 +0000358 isl_constraint_set_coefficient_si(c, isl_dim_out, TileDimension, -1);
359 isl_constraint_set_coefficient_si(c, isl_dim_out, PointDimension, 1);
360 TilingMap = isl_map_add_constraint(TilingMap, c);
Tobias Grosserc6699b72011-06-30 20:29:13 +0000361
Tobias Grosser2493e922011-12-07 07:42:57 +0000362 // ip <= it + ('VectorWidth' - 1)
Tobias Grosserf5338802011-10-06 00:03:35 +0000363 c = isl_inequality_alloc(LocalSpace);
Tobias Grosser2493e922011-12-07 07:42:57 +0000364 isl_constraint_set_coefficient_si(c, isl_dim_out, TileDimension, 1);
365 isl_constraint_set_coefficient_si(c, isl_dim_out, PointDimension, -1);
366 isl_constraint_set_constant_si(c, VectorWidth - 1);
367 TilingMap = isl_map_add_constraint(TilingMap, c);
Tobias Grosserc6699b72011-06-30 20:29:13 +0000368
Tobias Grosser2493e922011-12-07 07:42:57 +0000369 return TilingMap;
Tobias Grosserc6699b72011-06-30 20:29:13 +0000370}
371
Tobias Grosser4d96c8d2013-03-23 01:05:07 +0000372isl_union_map *
373IslScheduleOptimizer::getScheduleForBandList(isl_band_list *BandList) {
Tobias Grosser1ae9a602011-11-17 12:56:03 +0000374 int NumBands;
375 isl_union_map *Schedule;
376 isl_ctx *ctx;
Tobias Grosserde68cc92011-06-30 20:01:02 +0000377
Tobias Grosser1ae9a602011-11-17 12:56:03 +0000378 ctx = isl_band_list_get_ctx(BandList);
379 NumBands = isl_band_list_n_band(BandList);
Tobias Grosser62872012011-11-17 12:56:04 +0000380 Schedule = isl_union_map_empty(isl_space_params_alloc(ctx, 0));
Tobias Grosserde68cc92011-06-30 20:01:02 +0000381
Tobias Grosser1ae9a602011-11-17 12:56:03 +0000382 for (int i = 0; i < NumBands; i++) {
383 isl_band *Band;
384 isl_union_map *PartialSchedule;
385 int ScheduleDimensions;
386 isl_space *Space;
Tobias Grosser44f19ac2011-07-05 22:15:53 +0000387
Tobias Grosser1ae9a602011-11-17 12:56:03 +0000388 Band = isl_band_list_get_band(BandList, i);
Tobias Grosserb6033392011-12-08 13:02:58 +0000389 PartialSchedule = getScheduleForBand(Band, &ScheduleDimensions);
Tobias Grosser1ae9a602011-11-17 12:56:03 +0000390 Space = isl_union_map_get_space(PartialSchedule);
Tobias Grosserde68cc92011-06-30 20:01:02 +0000391
Tobias Grosser1ae9a602011-11-17 12:56:03 +0000392 if (isl_band_has_children(Band)) {
393 isl_band_list *Children;
394 isl_union_map *SuffixSchedule;
395
396 Children = isl_band_get_children(Band);
397 SuffixSchedule = getScheduleForBandList(Children);
Tobias Grosser4d96c8d2013-03-23 01:05:07 +0000398 PartialSchedule =
399 isl_union_map_flat_range_product(PartialSchedule, SuffixSchedule);
Tobias Grosser1ae9a602011-11-17 12:56:03 +0000400 isl_band_list_free(Children);
Hongbin Zheng68794212012-05-06 10:22:19 +0000401 } else if (PollyVectorizerChoice != VECTORIZER_NONE) {
Tobias Grosser4ba60fe2014-03-11 06:27:36 +0000402 // In case we are at the innermost band, we try to prepare for
403 // vectorization. This means, we look for the innermost parallel loop
404 // and strip mine this loop to the innermost level using a strip-mine
405 // factor corresponding to the number of vector iterations.
406 int NumDims = isl_band_n_member(Band);
407 for (int j = NumDims - 1; j >= 0; j--) {
Tobias Grossera38c9242014-01-26 19:36:28 +0000408 if (isl_band_member_is_coincident(Band, j)) {
Tobias Grosser1ae9a602011-11-17 12:56:03 +0000409 isl_map *TileMap;
410 isl_union_map *TileUMap;
Tobias Grosserc6699b72011-06-30 20:29:13 +0000411
Tobias Grosser4ba60fe2014-03-11 06:27:36 +0000412 TileMap = getPrevectorMap(ctx, ScheduleDimensions - NumDims + j,
Tobias Grosser216ea582012-04-16 11:06:06 +0000413 ScheduleDimensions);
Tobias Grosser4d96c8d2013-03-23 01:05:07 +0000414 TileUMap = isl_union_map_from_map(TileMap);
415 TileUMap =
416 isl_union_map_align_params(TileUMap, isl_space_copy(Space));
417 PartialSchedule =
418 isl_union_map_apply_range(PartialSchedule, TileUMap);
419 break;
420 }
Tobias Grosser7c5ba832011-06-30 20:29:20 +0000421 }
Tobias Grosserde68cc92011-06-30 20:01:02 +0000422 }
423
Tobias Grosser62872012011-11-17 12:56:04 +0000424 Schedule = isl_union_map_union(Schedule, PartialSchedule);
Tobias Grosserde68cc92011-06-30 20:01:02 +0000425
Tobias Grosser1ae9a602011-11-17 12:56:03 +0000426 isl_band_free(Band);
Tobias Grosserf5338802011-10-06 00:03:35 +0000427 isl_space_free(Space);
Tobias Grosserde68cc92011-06-30 20:01:02 +0000428 }
429
Tobias Grosser1ae9a602011-11-17 12:56:03 +0000430 return Schedule;
Tobias Grosserde68cc92011-06-30 20:01:02 +0000431}
432
Tobias Grosser460e9a42012-04-25 13:22:43 +0000433isl_union_map *IslScheduleOptimizer::getScheduleMap(isl_schedule *Schedule) {
Tobias Grosser1ae9a602011-11-17 12:56:03 +0000434 isl_band_list *BandList = isl_schedule_get_band_forest(Schedule);
435 isl_union_map *ScheduleMap = getScheduleForBandList(BandList);
436 isl_band_list_free(BandList);
437 return ScheduleMap;
Tobias Grosser30aa24c2011-05-14 19:02:06 +0000438}
439
Tobias Grosser73600b82011-10-08 00:30:40 +0000440bool IslScheduleOptimizer::runOnScop(Scop &S) {
Tobias Grosser30aa24c2011-05-14 19:02:06 +0000441 Dependences *D = &getAnalysis<Dependences>();
442
Tobias Grosser38c36ea2014-02-23 15:15:44 +0000443 if (!D->hasValidDependences())
444 return false;
445
Tobias Grosser28781422012-10-16 07:29:19 +0000446 isl_schedule_free(LastSchedule);
Tobias Grosser5a56cbf2014-04-16 07:33:47 +0000447 LastSchedule = nullptr;
Tobias Grosser28781422012-10-16 07:29:19 +0000448
Tobias Grosser30aa24c2011-05-14 19:02:06 +0000449 // Build input data.
Tobias Grosser4d96c8d2013-03-23 01:05:07 +0000450 int ValidityKinds =
451 Dependences::TYPE_RAW | Dependences::TYPE_WAR | Dependences::TYPE_WAW;
Tobias Grosser1deda292012-02-14 14:02:48 +0000452 int ProximityKinds;
453
454 if (OptimizeDeps == "all")
Tobias Grosser4d96c8d2013-03-23 01:05:07 +0000455 ProximityKinds =
456 Dependences::TYPE_RAW | Dependences::TYPE_WAR | Dependences::TYPE_WAW;
Tobias Grosser1deda292012-02-14 14:02:48 +0000457 else if (OptimizeDeps == "raw")
Tobias Grosser1e03ad72012-02-15 09:58:42 +0000458 ProximityKinds = Dependences::TYPE_RAW;
Tobias Grosser1deda292012-02-14 14:02:48 +0000459 else {
460 errs() << "Do not know how to optimize for '" << OptimizeDeps << "'"
Tobias Grosser4d96c8d2013-03-23 01:05:07 +0000461 << " Falling back to optimizing all dependences.\n";
462 ProximityKinds =
463 Dependences::TYPE_RAW | Dependences::TYPE_WAR | Dependences::TYPE_WAW;
Tobias Grosser1deda292012-02-14 14:02:48 +0000464 }
465
Tobias Grosser5f9a7622012-02-14 14:02:40 +0000466 isl_union_set *Domain = S.getDomains();
Tobias Grosser30aa24c2011-05-14 19:02:06 +0000467
Tobias Grosser98610ee2012-02-13 23:31:39 +0000468 if (!Domain)
Tobias Grosser30aa24c2011-05-14 19:02:06 +0000469 return false;
470
Tobias Grosser8a507022012-03-16 11:51:41 +0000471 isl_union_map *Validity = D->getDependences(ValidityKinds);
472 isl_union_map *Proximity = D->getDependences(ProximityKinds);
473
Tobias Grossera26db472012-01-30 19:38:43 +0000474 // Simplify the dependences by removing the constraints introduced by the
475 // domains. This can speed up the scheduling time significantly, as large
476 // constant coefficients will be removed from the dependences. The
477 // introduction of some additional dependences reduces the possible
478 // transformations, but in most cases, such transformation do not seem to be
479 // interesting anyway. In some cases this option may stop the scheduler to
480 // find any schedule.
481 if (SimplifyDeps == "yes") {
Tobias Grosser00383a72012-02-14 14:02:44 +0000482 Validity = isl_union_map_gist_domain(Validity, isl_union_set_copy(Domain));
483 Validity = isl_union_map_gist_range(Validity, isl_union_set_copy(Domain));
Tobias Grosser4d96c8d2013-03-23 01:05:07 +0000484 Proximity =
485 isl_union_map_gist_domain(Proximity, isl_union_set_copy(Domain));
Tobias Grosser00383a72012-02-14 14:02:44 +0000486 Proximity = isl_union_map_gist_range(Proximity, isl_union_set_copy(Domain));
Tobias Grossera26db472012-01-30 19:38:43 +0000487 } else if (SimplifyDeps != "no") {
488 errs() << "warning: Option -polly-opt-simplify-deps should either be 'yes' "
489 "or 'no'. Falling back to default: 'yes'\n";
490 }
491
Tobias Grosser30aa24c2011-05-14 19:02:06 +0000492 DEBUG(dbgs() << "\n\nCompute schedule from: ");
Tobias Grosser98610ee2012-02-13 23:31:39 +0000493 DEBUG(dbgs() << "Domain := "; isl_union_set_dump(Domain); dbgs() << ";\n");
494 DEBUG(dbgs() << "Proximity := "; isl_union_map_dump(Proximity);
Tobias Grosser30aa24c2011-05-14 19:02:06 +0000495 dbgs() << ";\n");
Tobias Grosser98610ee2012-02-13 23:31:39 +0000496 DEBUG(dbgs() << "Validity := "; isl_union_map_dump(Validity);
Tobias Grosser30aa24c2011-05-14 19:02:06 +0000497 dbgs() << ";\n");
498
Tobias Grosserb3ad85b2012-01-30 19:38:50 +0000499 int IslFusionStrategy;
500
501 if (FusionStrategy == "max") {
502 IslFusionStrategy = ISL_SCHEDULE_FUSE_MAX;
503 } else if (FusionStrategy == "min") {
504 IslFusionStrategy = ISL_SCHEDULE_FUSE_MIN;
505 } else {
506 errs() << "warning: Unknown fusion strategy. Falling back to maximal "
507 "fusion.\n";
508 IslFusionStrategy = ISL_SCHEDULE_FUSE_MAX;
509 }
510
Tobias Grosser95e860c2012-01-30 19:38:54 +0000511 int IslMaximizeBands;
512
Tobias Grossera4ea90b2012-01-30 22:43:56 +0000513 if (MaximizeBandDepth == "yes") {
Tobias Grosser95e860c2012-01-30 19:38:54 +0000514 IslMaximizeBands = 1;
Tobias Grossera4ea90b2012-01-30 22:43:56 +0000515 } else if (MaximizeBandDepth == "no") {
Tobias Grosser95e860c2012-01-30 19:38:54 +0000516 IslMaximizeBands = 0;
517 } else {
518 errs() << "warning: Option -polly-opt-maximize-bands should either be 'yes'"
519 " or 'no'. Falling back to default: 'yes'\n";
520 IslMaximizeBands = 1;
521 }
522
Tobias Grosserb3ad85b2012-01-30 19:38:50 +0000523 isl_options_set_schedule_fuse(S.getIslCtx(), IslFusionStrategy);
Tobias Grosser95e860c2012-01-30 19:38:54 +0000524 isl_options_set_schedule_maximize_band_depth(S.getIslCtx(), IslMaximizeBands);
Tobias Grosser992e60c2012-02-20 08:41:15 +0000525 isl_options_set_schedule_max_constant_term(S.getIslCtx(), MaxConstantTerm);
Tobias Grosser92f54802012-02-20 08:41:47 +0000526 isl_options_set_schedule_max_coefficient(S.getIslCtx(), MaxCoefficient);
Tobias Grosser42152ff2012-01-30 19:38:47 +0000527
528 isl_options_set_on_error(S.getIslCtx(), ISL_ON_ERROR_CONTINUE);
Tobias Grossera38c9242014-01-26 19:36:28 +0000529
530 isl_schedule_constraints *ScheduleConstraints;
531 ScheduleConstraints = isl_schedule_constraints_on_domain(Domain);
532 ScheduleConstraints =
533 isl_schedule_constraints_set_proximity(ScheduleConstraints, Proximity);
534 ScheduleConstraints = isl_schedule_constraints_set_validity(
535 ScheduleConstraints, isl_union_map_copy(Validity));
536 ScheduleConstraints =
537 isl_schedule_constraints_set_coincidence(ScheduleConstraints, Validity);
Tobias Grosser00383a72012-02-14 14:02:44 +0000538 isl_schedule *Schedule;
Tobias Grossera38c9242014-01-26 19:36:28 +0000539 Schedule = isl_schedule_constraints_compute_schedule(ScheduleConstraints);
Tobias Grosser42152ff2012-01-30 19:38:47 +0000540 isl_options_set_on_error(S.getIslCtx(), ISL_ON_ERROR_ABORT);
541
542 // In cases the scheduler is not able to optimize the code, we just do not
543 // touch the schedule.
Tobias Grosser98610ee2012-02-13 23:31:39 +0000544 if (!Schedule)
Tobias Grosser42152ff2012-01-30 19:38:47 +0000545 return false;
Tobias Grosser30aa24c2011-05-14 19:02:06 +0000546
Tobias Grosser4d96c8d2013-03-23 01:05:07 +0000547 DEBUG(dbgs() << "Schedule := "; isl_schedule_dump(Schedule); dbgs() << ";\n");
Tobias Grosser4d63b9d2012-02-20 08:41:21 +0000548
Tobias Grosser98610ee2012-02-13 23:31:39 +0000549 isl_union_map *ScheduleMap = getScheduleMap(Schedule);
Tobias Grosser30aa24c2011-05-14 19:02:06 +0000550
Tobias Grosserde68cc92011-06-30 20:01:02 +0000551 for (Scop::iterator SI = S.begin(), SE = S.end(); SI != SE; ++SI) {
Tobias Grosser98610ee2012-02-13 23:31:39 +0000552 ScopStmt *Stmt = *SI;
Tobias Grosser249c4b12013-04-11 05:55:13 +0000553 isl_map *StmtSchedule;
Tobias Grosser98610ee2012-02-13 23:31:39 +0000554 isl_set *Domain = Stmt->getDomain();
555 isl_union_map *StmtBand;
556 StmtBand = isl_union_map_intersect_domain(isl_union_map_copy(ScheduleMap),
Tobias Grosser4d96c8d2013-03-23 01:05:07 +0000557 isl_union_set_from_set(Domain));
Tobias Grosser249c4b12013-04-11 05:55:13 +0000558 if (isl_union_map_is_empty(StmtBand)) {
Tobias Grosser34f06132014-02-21 20:51:36 +0000559 StmtSchedule = isl_map_from_domain(isl_set_empty(Stmt->getDomainSpace()));
Tobias Grosser249c4b12013-04-11 05:55:13 +0000560 isl_union_map_free(StmtBand);
561 } else {
562 assert(isl_union_map_n_map(StmtBand) == 1);
563 StmtSchedule = isl_map_from_union_map(StmtBand);
Tobias Grosserf242b802013-04-10 22:48:08 +0000564 }
565
Tobias Grosser98610ee2012-02-13 23:31:39 +0000566 Stmt->setScattering(StmtSchedule);
Tobias Grosser30aa24c2011-05-14 19:02:06 +0000567 }
568
Tobias Grosser1ae9a602011-11-17 12:56:03 +0000569 isl_union_map_free(ScheduleMap);
Tobias Grosser28781422012-10-16 07:29:19 +0000570 LastSchedule = Schedule;
Tobias Grosserde68cc92011-06-30 20:01:02 +0000571
Tobias Grosser98610ee2012-02-13 23:31:39 +0000572 unsigned MaxScatDims = 0;
Tobias Grosser30aa24c2011-05-14 19:02:06 +0000573
574 for (Scop::iterator SI = S.begin(), SE = S.end(); SI != SE; ++SI)
Tobias Grosser98610ee2012-02-13 23:31:39 +0000575 MaxScatDims = std::max((*SI)->getNumScattering(), MaxScatDims);
Tobias Grosser30aa24c2011-05-14 19:02:06 +0000576
Tobias Grosser98610ee2012-02-13 23:31:39 +0000577 extendScattering(S, MaxScatDims);
Tobias Grosser30aa24c2011-05-14 19:02:06 +0000578 return false;
579}
580
Tobias Grosser73600b82011-10-08 00:30:40 +0000581void IslScheduleOptimizer::printScop(raw_ostream &OS) const {
Tobias Grosser28781422012-10-16 07:29:19 +0000582 isl_printer *p;
583 char *ScheduleStr;
584
585 OS << "Calculated schedule:\n";
586
587 if (!LastSchedule) {
588 OS << "n/a\n";
589 return;
590 }
591
592 p = isl_printer_to_str(isl_schedule_get_ctx(LastSchedule));
593 p = isl_printer_print_schedule(p, LastSchedule);
594 ScheduleStr = isl_printer_get_str(p);
595 isl_printer_free(p);
596
597 OS << ScheduleStr << "\n";
Tobias Grosser30aa24c2011-05-14 19:02:06 +0000598}
599
Tobias Grosser73600b82011-10-08 00:30:40 +0000600void IslScheduleOptimizer::getAnalysisUsage(AnalysisUsage &AU) const {
Tobias Grosser30aa24c2011-05-14 19:02:06 +0000601 ScopPass::getAnalysisUsage(AU);
602 AU.addRequired<Dependences>();
603}
604
Tobias Grosser4d96c8d2013-03-23 01:05:07 +0000605Pass *polly::createIslScheduleOptimizerPass() {
Tobias Grosser73600b82011-10-08 00:30:40 +0000606 return new IslScheduleOptimizer();
Tobias Grosser30aa24c2011-05-14 19:02:06 +0000607}
Tobias Grosser4d96c8d2013-03-23 01:05:07 +0000608
609INITIALIZE_PASS_BEGIN(IslScheduleOptimizer, "polly-opt-isl",
610 "Polly - Optimize schedule of SCoP", false, false);
611INITIALIZE_PASS_DEPENDENCY(Dependences);
612INITIALIZE_PASS_DEPENDENCY(ScopInfo);
613INITIALIZE_PASS_END(IslScheduleOptimizer, "polly-opt-isl",
614 "Polly - Optimize schedule of SCoP", false, false)