blob: d88563b39244d0a90a3fe5861b4171cabe825964 [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
Tobias Grosser4dca4392011-11-22 19:40:19 +000034#define DEBUG_TYPE "polly-opt-isl"
Tobias Grosser30aa24c2011-05-14 19:02:06 +000035#include "llvm/Support/Debug.h"
36
37using namespace llvm;
38using namespace polly;
39
Tobias Grosser4d96c8d2013-03-23 01:05:07 +000040namespace polly { bool DisablePollyTiling; }
Tobias Grossere602a072013-05-07 07:30:56 +000041static cl::opt<bool, true>
42DisableTiling("polly-no-tiling", cl::desc("Disable tiling in the scheduler"),
Tobias Grosser637bd632013-05-07 07:31:10 +000043 cl::location(polly::DisablePollyTiling), cl::init(false),
44 cl::cat(PollyCategory));
Tobias Grosser353a2682011-10-23 20:59:26 +000045
Tobias Grossera26db472012-01-30 19:38:43 +000046static cl::opt<std::string>
Tobias Grosser1deda292012-02-14 14:02:48 +000047OptimizeDeps("polly-opt-optimize-only",
48 cl::desc("Only a certain kind of dependences (all/raw)"),
Tobias Grosser637bd632013-05-07 07:31:10 +000049 cl::Hidden, cl::init("all"), cl::cat(PollyCategory));
Tobias Grosser1deda292012-02-14 14:02:48 +000050
51static cl::opt<std::string>
Tobias Grossera26db472012-01-30 19:38:43 +000052SimplifyDeps("polly-opt-simplify-deps",
Tobias Grosser4d96c8d2013-03-23 01:05:07 +000053 cl::desc("Dependences should be simplified (yes/no)"), cl::Hidden,
Tobias Grosser637bd632013-05-07 07:31:10 +000054 cl::init("yes"), cl::cat(PollyCategory));
Tobias Grossera26db472012-01-30 19:38:43 +000055
Tobias Grosser992e60c2012-02-20 08:41:15 +000056static cl::opt<int>
57MaxConstantTerm("polly-opt-max-constant-term",
58 cl::desc("The maximal constant term allowed (-1 is unlimited)"),
Tobias Grosser637bd632013-05-07 07:31:10 +000059 cl::Hidden, cl::init(20), cl::cat(PollyCategory));
Tobias Grosser992e60c2012-02-20 08:41:15 +000060
Tobias Grosser92f54802012-02-20 08:41:47 +000061static cl::opt<int>
62MaxCoefficient("polly-opt-max-coefficient",
63 cl::desc("The maximal coefficient allowed (-1 is unlimited)"),
Tobias Grosser637bd632013-05-07 07:31:10 +000064 cl::Hidden, cl::init(20), cl::cat(PollyCategory));
Tobias Grosser92f54802012-02-20 08:41:47 +000065
Tobias Grossere602a072013-05-07 07:30:56 +000066static cl::opt<std::string>
67FusionStrategy("polly-opt-fusion",
68 cl::desc("The fusion strategy to choose (min/max)"), cl::Hidden,
Tobias Grosser637bd632013-05-07 07:31:10 +000069 cl::init("min"), cl::cat(PollyCategory));
Tobias Grosserb3ad85b2012-01-30 19:38:50 +000070
Tobias Grossere602a072013-05-07 07:30:56 +000071static cl::opt<std::string>
72MaximizeBandDepth("polly-opt-maximize-bands",
73 cl::desc("Maximize the band depth (yes/no)"), cl::Hidden,
Tobias Grosser637bd632013-05-07 07:31:10 +000074 cl::init("yes"), cl::cat(PollyCategory));
Tobias Grosser95e860c2012-01-30 19:38:54 +000075
Tobias Grosser30aa24c2011-05-14 19:02:06 +000076namespace {
77
Tobias Grosser4d96c8d2013-03-23 01:05:07 +000078class IslScheduleOptimizer : public ScopPass {
Tobias Grosser30aa24c2011-05-14 19:02:06 +000079
Tobias Grosser4d96c8d2013-03-23 01:05:07 +000080public:
81 static char ID;
82 explicit IslScheduleOptimizer() : ScopPass(ID) { LastSchedule = NULL; }
Tobias Grosser28781422012-10-16 07:29:19 +000083
Tobias Grosser4d96c8d2013-03-23 01:05:07 +000084 ~IslScheduleOptimizer() { isl_schedule_free(LastSchedule); }
Tobias Grosser30aa24c2011-05-14 19:02:06 +000085
Tobias Grosser4d96c8d2013-03-23 01:05:07 +000086 virtual bool runOnScop(Scop &S);
87 void printScop(llvm::raw_ostream &OS) const;
88 void getAnalysisUsage(AnalysisUsage &AU) const;
Tobias Grosser460e9a42012-04-25 13:22:43 +000089
Tobias Grosser4d96c8d2013-03-23 01:05:07 +000090private:
91 isl_schedule *LastSchedule;
Tobias Grosser28781422012-10-16 07:29:19 +000092
Tobias Grosser4d96c8d2013-03-23 01:05:07 +000093 static void extendScattering(Scop &S, unsigned NewDimensions);
Tobias Grosser460e9a42012-04-25 13:22:43 +000094
Tobias Grosser4d96c8d2013-03-23 01:05:07 +000095 /// @brief Create a map that describes a n-dimensonal tiling.
96 ///
97 /// getTileMap creates a map from a n-dimensional scattering space into an
98 /// 2*n-dimensional scattering space. The map describes a rectangular
99 /// tiling.
100 ///
101 /// Example:
102 /// scheduleDimensions = 2, parameterDimensions = 1, tileSize = 32
103 ///
104 /// tileMap := [p0] -> {[s0, s1] -> [t0, t1, s0, s1]:
105 /// t0 % 32 = 0 and t0 <= s0 < t0 + 32 and
106 /// t1 % 32 = 0 and t1 <= s1 < t1 + 32}
107 ///
108 /// Before tiling:
109 ///
110 /// for (i = 0; i < N; i++)
111 /// for (j = 0; j < M; j++)
112 /// S(i,j)
113 ///
114 /// After tiling:
115 ///
116 /// for (t_i = 0; t_i < N; i+=32)
117 /// for (t_j = 0; t_j < M; j+=32)
118 /// for (i = t_i; i < min(t_i + 32, N); i++) | Unknown that N % 32 = 0
119 /// for (j = t_j; j < t_j + 32; j++) | Known that M % 32 = 0
120 /// S(i,j)
121 ///
122 static isl_basic_map *getTileMap(isl_ctx *ctx, int scheduleDimensions,
123 isl_space *SpaceModel, int tileSize = 32);
Tobias Grosser460e9a42012-04-25 13:22:43 +0000124
Tobias Grosser4d96c8d2013-03-23 01:05:07 +0000125 /// @brief Get the schedule for this band.
126 ///
127 /// Polly applies transformations like tiling on top of the isl calculated
128 /// value. This can influence the number of scheduling dimension. The
129 /// number of schedule dimensions is returned in the parameter 'Dimension'.
130 static isl_union_map *getScheduleForBand(isl_band *Band, int *Dimensions);
Tobias Grosser460e9a42012-04-25 13:22:43 +0000131
Tobias Grosser4d96c8d2013-03-23 01:05:07 +0000132 /// @brief Create a map that pre-vectorizes one scheduling dimension.
133 ///
134 /// getPrevectorMap creates a map that maps each input dimension to the same
135 /// output dimension, except for the dimension DimToVectorize.
136 /// DimToVectorize is strip mined by 'VectorWidth' and the newly created
137 /// point loop of DimToVectorize is moved to the innermost level.
138 ///
139 /// Example (DimToVectorize=0, ScheduleDimensions=2, VectorWidth=4):
140 ///
141 /// | Before transformation
142 /// |
143 /// | A[i,j] -> [i,j]
144 /// |
145 /// | for (i = 0; i < 128; i++)
146 /// | for (j = 0; j < 128; j++)
147 /// | A(i,j);
148 ///
149 /// Prevector map:
150 /// [i,j] -> [it,j,ip] : it % 4 = 0 and it <= ip <= it + 3 and i = ip
151 ///
152 /// | After transformation:
153 /// |
154 /// | A[i,j] -> [it,j,ip] : it % 4 = 0 and it <= ip <= it + 3 and i = ip
155 /// |
156 /// | for (it = 0; it < 128; it+=4)
157 /// | for (j = 0; j < 128; j++)
158 /// | for (ip = max(0,it); ip < min(128, it + 3); ip++)
159 /// | A(ip,j);
160 ///
161 /// The goal of this transformation is to create a trivially vectorizable
162 /// loop. This means a parallel loop at the innermost level that has a
163 /// constant number of iterations corresponding to the target vector width.
164 ///
165 /// This transformation creates a loop at the innermost level. The loop has
166 /// a constant number of iterations, if the number of loop iterations at
167 /// DimToVectorize can be divided by VectorWidth. The default VectorWidth is
168 /// currently constant and not yet target specific. This function does not
169 /// reason about parallelism.
170 static isl_map *getPrevectorMap(isl_ctx *ctx, int DimToVectorize,
171 int ScheduleDimensions, int VectorWidth = 4);
Tobias Grosser460e9a42012-04-25 13:22:43 +0000172
Tobias Grosser4d96c8d2013-03-23 01:05:07 +0000173 /// @brief Get the scheduling map for a list of bands.
174 ///
175 /// Walk recursively the forest of bands to combine the schedules of the
176 /// individual bands to the overall schedule. In case tiling is requested,
177 /// the individual bands are tiled.
178 static isl_union_map *getScheduleForBandList(isl_band_list *BandList);
Tobias Grosser460e9a42012-04-25 13:22:43 +0000179
Tobias Grosser4d96c8d2013-03-23 01:05:07 +0000180 static isl_union_map *getScheduleMap(isl_schedule *Schedule);
Tobias Grosser28781422012-10-16 07:29:19 +0000181
Tobias Grosser4d96c8d2013-03-23 01:05:07 +0000182 bool doFinalization() {
183 isl_schedule_free(LastSchedule);
184 LastSchedule = NULL;
185 return true;
186 }
187};
Tobias Grosser30aa24c2011-05-14 19:02:06 +0000188
189}
190
Tobias Grosser73600b82011-10-08 00:30:40 +0000191char IslScheduleOptimizer::ID = 0;
Tobias Grosser30aa24c2011-05-14 19:02:06 +0000192
Tobias Grosser460e9a42012-04-25 13:22:43 +0000193void IslScheduleOptimizer::extendScattering(Scop &S, unsigned NewDimensions) {
Tobias Grosser30aa24c2011-05-14 19:02:06 +0000194 for (Scop::iterator SI = S.begin(), SE = S.end(); SI != SE; ++SI) {
Tobias Grossercf3942d2011-10-06 00:04:05 +0000195 ScopStmt *Stmt = *SI;
Tobias Grossercf3942d2011-10-06 00:04:05 +0000196 unsigned OldDimensions = Stmt->getNumScattering();
197 isl_space *Space;
Tobias Grosser29666112012-05-22 10:47:31 +0000198 isl_map *Map, *New;
Tobias Grossercf3942d2011-10-06 00:04:05 +0000199
200 Space = isl_space_alloc(Stmt->getIslCtx(), 0, OldDimensions, NewDimensions);
Tobias Grosser29666112012-05-22 10:47:31 +0000201 Map = isl_map_universe(Space);
Tobias Grosser30aa24c2011-05-14 19:02:06 +0000202
Tobias Grosser29666112012-05-22 10:47:31 +0000203 for (unsigned i = 0; i < OldDimensions; i++)
204 Map = isl_map_equate(Map, isl_dim_in, i, isl_dim_out, i);
Tobias Grosser30aa24c2011-05-14 19:02:06 +0000205
Tobias Grosser29666112012-05-22 10:47:31 +0000206 for (unsigned i = OldDimensions; i < NewDimensions; i++)
207 Map = isl_map_fix_si(Map, isl_dim_out, i, 0);
Tobias Grosser30aa24c2011-05-14 19:02:06 +0000208
Tobias Grosser29666112012-05-22 10:47:31 +0000209 Map = isl_map_align_params(Map, S.getParamSpace());
210 New = isl_map_apply_range(Stmt->getScattering(), Map);
211 Stmt->setScattering(New);
Tobias Grosser30aa24c2011-05-14 19:02:06 +0000212 }
213}
214
Tobias Grossere602a072013-05-07 07:30:56 +0000215isl_basic_map *IslScheduleOptimizer::getTileMap(isl_ctx *ctx,
216 int scheduleDimensions,
217 isl_space *SpaceModel,
218 int tileSize) {
Tobias Grosserde68cc92011-06-30 20:01:02 +0000219 // We construct
220 //
221 // tileMap := [p0] -> {[s0, s1] -> [t0, t1, p0, p1, a0, a1]:
222 // s0 = a0 * 32 and s0 = p0 and t0 <= p0 < t0 + 32 and
223 // s1 = a1 * 32 and s1 = p1 and t1 <= p1 < t1 + 32}
224 //
225 // and project out the auxilary dimensions a0 and a1.
Tobias Grosser4d96c8d2013-03-23 01:05:07 +0000226 isl_space *Space =
227 isl_space_alloc(ctx, 0, scheduleDimensions, scheduleDimensions * 3);
Tobias Grosserf5338802011-10-06 00:03:35 +0000228 isl_basic_map *tileMap = isl_basic_map_universe(isl_space_copy(Space));
229
230 isl_local_space *LocalSpace = isl_local_space_from_space(Space);
Tobias Grosser30aa24c2011-05-14 19:02:06 +0000231
Tobias Grosserde68cc92011-06-30 20:01:02 +0000232 for (int x = 0; x < scheduleDimensions; x++) {
233 int sX = x;
234 int tX = x;
235 int pX = scheduleDimensions + x;
236 int aX = 2 * scheduleDimensions + x;
Tobias Grosser30aa24c2011-05-14 19:02:06 +0000237
Tobias Grosserde68cc92011-06-30 20:01:02 +0000238 isl_constraint *c;
Tobias Grosser30aa24c2011-05-14 19:02:06 +0000239
Tobias Grosserde68cc92011-06-30 20:01:02 +0000240 // sX = aX * tileSize;
Tobias Grosserf5338802011-10-06 00:03:35 +0000241 c = isl_equality_alloc(isl_local_space_copy(LocalSpace));
Tobias Grosserde68cc92011-06-30 20:01:02 +0000242 isl_constraint_set_coefficient_si(c, isl_dim_out, sX, 1);
243 isl_constraint_set_coefficient_si(c, isl_dim_out, aX, -tileSize);
244 tileMap = isl_basic_map_add_constraint(tileMap, c);
Tobias Grosser30aa24c2011-05-14 19:02:06 +0000245
Tobias Grosserde68cc92011-06-30 20:01:02 +0000246 // pX = sX;
Tobias Grosserf5338802011-10-06 00:03:35 +0000247 c = isl_equality_alloc(isl_local_space_copy(LocalSpace));
Tobias Grosserde68cc92011-06-30 20:01:02 +0000248 isl_constraint_set_coefficient_si(c, isl_dim_out, pX, 1);
249 isl_constraint_set_coefficient_si(c, isl_dim_in, sX, -1);
250 tileMap = isl_basic_map_add_constraint(tileMap, c);
Tobias Grosser30aa24c2011-05-14 19:02:06 +0000251
Tobias Grosserde68cc92011-06-30 20:01:02 +0000252 // tX <= pX
Tobias Grosserf5338802011-10-06 00:03:35 +0000253 c = isl_inequality_alloc(isl_local_space_copy(LocalSpace));
Tobias Grosserde68cc92011-06-30 20:01:02 +0000254 isl_constraint_set_coefficient_si(c, isl_dim_out, pX, 1);
255 isl_constraint_set_coefficient_si(c, isl_dim_out, tX, -1);
256 tileMap = isl_basic_map_add_constraint(tileMap, c);
257
258 // pX <= tX + (tileSize - 1)
Tobias Grosserf5338802011-10-06 00:03:35 +0000259 c = isl_inequality_alloc(isl_local_space_copy(LocalSpace));
Tobias Grosserde68cc92011-06-30 20:01:02 +0000260 isl_constraint_set_coefficient_si(c, isl_dim_out, tX, 1);
261 isl_constraint_set_coefficient_si(c, isl_dim_out, pX, -1);
Tobias Grosser30aa24c2011-05-14 19:02:06 +0000262 isl_constraint_set_constant_si(c, tileSize - 1);
Tobias Grosserde68cc92011-06-30 20:01:02 +0000263 tileMap = isl_basic_map_add_constraint(tileMap, c);
Tobias Grosser30aa24c2011-05-14 19:02:06 +0000264 }
265
Tobias Grosserde68cc92011-06-30 20:01:02 +0000266 // Project out auxilary dimensions.
Tobias Grosser30aa24c2011-05-14 19:02:06 +0000267 //
Tobias Grosserde68cc92011-06-30 20:01:02 +0000268 // The auxilary dimensions are transformed into existentially quantified ones.
269 // This reduces the number of visible scattering dimensions and allows Cloog
270 // to produces better code.
Tobias Grosser4d96c8d2013-03-23 01:05:07 +0000271 tileMap = isl_basic_map_project_out(
272 tileMap, isl_dim_out, 2 * scheduleDimensions, scheduleDimensions);
Tobias Grosserf5338802011-10-06 00:03:35 +0000273 isl_local_space_free(LocalSpace);
Tobias Grosserde68cc92011-06-30 20:01:02 +0000274 return tileMap;
275}
Tobias Grosser30aa24c2011-05-14 19:02:06 +0000276
Tobias Grossere602a072013-05-07 07:30:56 +0000277isl_union_map *IslScheduleOptimizer::getScheduleForBand(isl_band *Band,
278 int *Dimensions) {
Tobias Grosser1ae9a602011-11-17 12:56:03 +0000279 isl_union_map *PartialSchedule;
Tobias Grosserde68cc92011-06-30 20:01:02 +0000280 isl_ctx *ctx;
Tobias Grosserf5338802011-10-06 00:03:35 +0000281 isl_space *Space;
Tobias Grosser1ae9a602011-11-17 12:56:03 +0000282 isl_basic_map *TileMap;
283 isl_union_map *TileUMap;
Tobias Grosserde68cc92011-06-30 20:01:02 +0000284
Tobias Grosser1ae9a602011-11-17 12:56:03 +0000285 PartialSchedule = isl_band_get_partial_schedule(Band);
Tobias Grosserb6033392011-12-08 13:02:58 +0000286 *Dimensions = isl_band_n_member(Band);
Tobias Grosserde68cc92011-06-30 20:01:02 +0000287
Tobias Grosser79b30202011-11-17 12:56:00 +0000288 if (DisableTiling)
Tobias Grosser1ae9a602011-11-17 12:56:03 +0000289 return PartialSchedule;
Tobias Grosser353a2682011-10-23 20:59:26 +0000290
Tobias Grosserb6033392011-12-08 13:02:58 +0000291 // It does not make any sense to tile a band with just one dimension.
292 if (*Dimensions == 1)
293 return PartialSchedule;
294
Tobias Grosser1ae9a602011-11-17 12:56:03 +0000295 ctx = isl_union_map_get_ctx(PartialSchedule);
296 Space = isl_union_map_get_space(PartialSchedule);
Tobias Grosserde68cc92011-06-30 20:01:02 +0000297
Tobias Grosserb6033392011-12-08 13:02:58 +0000298 TileMap = getTileMap(ctx, *Dimensions, Space);
Tobias Grosser1ae9a602011-11-17 12:56:03 +0000299 TileUMap = isl_union_map_from_map(isl_map_from_basic_map(TileMap));
300 TileUMap = isl_union_map_align_params(TileUMap, Space);
Tobias Grosserb6033392011-12-08 13:02:58 +0000301 *Dimensions = 2 * *Dimensions;
302
Tobias Grosser1ae9a602011-11-17 12:56:03 +0000303 return isl_union_map_apply_range(PartialSchedule, TileUMap);
Tobias Grosserde68cc92011-06-30 20:01:02 +0000304}
305
Tobias Grossere602a072013-05-07 07:30:56 +0000306isl_map *IslScheduleOptimizer::getPrevectorMap(isl_ctx *ctx, int DimToVectorize,
307 int ScheduleDimensions,
308 int VectorWidth) {
Tobias Grosser2493e922011-12-07 07:42:57 +0000309 isl_space *Space;
310 isl_local_space *LocalSpace, *LocalSpaceRange;
311 isl_set *Modulo;
312 isl_map *TilingMap;
Tobias Grosserc6699b72011-06-30 20:29:13 +0000313 isl_constraint *c;
Tobias Grosser2493e922011-12-07 07:42:57 +0000314 isl_aff *Aff;
315 int PointDimension; /* ip */
316 int TileDimension; /* it */
Tobias Grosseredab1352013-06-21 06:41:31 +0000317 isl_val *VectorWidthMP;
Tobias Grosserc6699b72011-06-30 20:29:13 +0000318
Tobias Grosser4d96c8d2013-03-23 01:05:07 +0000319 assert(0 <= DimToVectorize && DimToVectorize < ScheduleDimensions);
Tobias Grosserf5338802011-10-06 00:03:35 +0000320
Tobias Grosser2493e922011-12-07 07:42:57 +0000321 Space = isl_space_alloc(ctx, 0, ScheduleDimensions, ScheduleDimensions + 1);
322 TilingMap = isl_map_universe(isl_space_copy(Space));
323 LocalSpace = isl_local_space_from_space(Space);
324 PointDimension = ScheduleDimensions;
325 TileDimension = DimToVectorize;
326
327 // Create an identity map for everything except DimToVectorize and map
328 // DimToVectorize to the point loop at the innermost dimension.
329 for (int i = 0; i < ScheduleDimensions; i++) {
Tobias Grosserf5338802011-10-06 00:03:35 +0000330 c = isl_equality_alloc(isl_local_space_copy(LocalSpace));
Tobias Grosserc6699b72011-06-30 20:29:13 +0000331 isl_constraint_set_coefficient_si(c, isl_dim_in, i, -1);
Tobias Grosser2493e922011-12-07 07:42:57 +0000332
333 if (i == DimToVectorize)
334 isl_constraint_set_coefficient_si(c, isl_dim_out, PointDimension, 1);
335 else
336 isl_constraint_set_coefficient_si(c, isl_dim_out, i, 1);
337
338 TilingMap = isl_map_add_constraint(TilingMap, c);
Tobias Grosserc6699b72011-06-30 20:29:13 +0000339 }
340
Tobias Grosser2493e922011-12-07 07:42:57 +0000341 // it % 'VectorWidth' = 0
342 LocalSpaceRange = isl_local_space_range(isl_local_space_copy(LocalSpace));
343 Aff = isl_aff_zero_on_domain(LocalSpaceRange);
344 Aff = isl_aff_set_constant_si(Aff, VectorWidth);
345 Aff = isl_aff_set_coefficient_si(Aff, isl_dim_in, TileDimension, 1);
Tobias Grosseredab1352013-06-21 06:41:31 +0000346 VectorWidthMP = isl_val_int_from_si(ctx, VectorWidth);
347 Aff = isl_aff_mod_val(Aff, VectorWidthMP);
Tobias Grosser2493e922011-12-07 07:42:57 +0000348 Modulo = isl_pw_aff_zero_set(isl_pw_aff_from_aff(Aff));
349 TilingMap = isl_map_intersect_range(TilingMap, Modulo);
Tobias Grosserc6699b72011-06-30 20:29:13 +0000350
Tobias Grosser2493e922011-12-07 07:42:57 +0000351 // it <= ip
Tobias Grosserf5338802011-10-06 00:03:35 +0000352 c = isl_inequality_alloc(isl_local_space_copy(LocalSpace));
Tobias Grosser2493e922011-12-07 07:42:57 +0000353 isl_constraint_set_coefficient_si(c, isl_dim_out, TileDimension, -1);
354 isl_constraint_set_coefficient_si(c, isl_dim_out, PointDimension, 1);
355 TilingMap = isl_map_add_constraint(TilingMap, c);
Tobias Grosserc6699b72011-06-30 20:29:13 +0000356
Tobias Grosser2493e922011-12-07 07:42:57 +0000357 // ip <= it + ('VectorWidth' - 1)
Tobias Grosserf5338802011-10-06 00:03:35 +0000358 c = isl_inequality_alloc(LocalSpace);
Tobias Grosser2493e922011-12-07 07:42:57 +0000359 isl_constraint_set_coefficient_si(c, isl_dim_out, TileDimension, 1);
360 isl_constraint_set_coefficient_si(c, isl_dim_out, PointDimension, -1);
361 isl_constraint_set_constant_si(c, VectorWidth - 1);
362 TilingMap = isl_map_add_constraint(TilingMap, c);
Tobias Grosserc6699b72011-06-30 20:29:13 +0000363
Tobias Grosser2493e922011-12-07 07:42:57 +0000364 return TilingMap;
Tobias Grosserc6699b72011-06-30 20:29:13 +0000365}
366
Tobias Grosser4d96c8d2013-03-23 01:05:07 +0000367isl_union_map *
368IslScheduleOptimizer::getScheduleForBandList(isl_band_list *BandList) {
Tobias Grosser1ae9a602011-11-17 12:56:03 +0000369 int NumBands;
370 isl_union_map *Schedule;
371 isl_ctx *ctx;
Tobias Grosserde68cc92011-06-30 20:01:02 +0000372
Tobias Grosser1ae9a602011-11-17 12:56:03 +0000373 ctx = isl_band_list_get_ctx(BandList);
374 NumBands = isl_band_list_n_band(BandList);
Tobias Grosser62872012011-11-17 12:56:04 +0000375 Schedule = isl_union_map_empty(isl_space_params_alloc(ctx, 0));
Tobias Grosserde68cc92011-06-30 20:01:02 +0000376
Tobias Grosser1ae9a602011-11-17 12:56:03 +0000377 for (int i = 0; i < NumBands; i++) {
378 isl_band *Band;
379 isl_union_map *PartialSchedule;
380 int ScheduleDimensions;
381 isl_space *Space;
Tobias Grosser44f19ac2011-07-05 22:15:53 +0000382
Tobias Grosser1ae9a602011-11-17 12:56:03 +0000383 Band = isl_band_list_get_band(BandList, i);
Tobias Grosserb6033392011-12-08 13:02:58 +0000384 PartialSchedule = getScheduleForBand(Band, &ScheduleDimensions);
Tobias Grosser1ae9a602011-11-17 12:56:03 +0000385 Space = isl_union_map_get_space(PartialSchedule);
Tobias Grosserde68cc92011-06-30 20:01:02 +0000386
Tobias Grosser1ae9a602011-11-17 12:56:03 +0000387 if (isl_band_has_children(Band)) {
388 isl_band_list *Children;
389 isl_union_map *SuffixSchedule;
390
391 Children = isl_band_get_children(Band);
392 SuffixSchedule = getScheduleForBandList(Children);
Tobias Grosser4d96c8d2013-03-23 01:05:07 +0000393 PartialSchedule =
394 isl_union_map_flat_range_product(PartialSchedule, SuffixSchedule);
Tobias Grosser1ae9a602011-11-17 12:56:03 +0000395 isl_band_list_free(Children);
Hongbin Zheng68794212012-05-06 10:22:19 +0000396 } else if (PollyVectorizerChoice != VECTORIZER_NONE) {
Tobias Grosser4d96c8d2013-03-23 01:05:07 +0000397 for (int j = 0; j < isl_band_n_member(Band); j++) {
398 if (isl_band_member_is_zero_distance(Band, j)) {
Tobias Grosser1ae9a602011-11-17 12:56:03 +0000399 isl_map *TileMap;
400 isl_union_map *TileUMap;
Tobias Grosserc6699b72011-06-30 20:29:13 +0000401
Tobias Grosser4d96c8d2013-03-23 01:05:07 +0000402 TileMap = getPrevectorMap(ctx, ScheduleDimensions - j - 1,
Tobias Grosser216ea582012-04-16 11:06:06 +0000403 ScheduleDimensions);
Tobias Grosser4d96c8d2013-03-23 01:05:07 +0000404 TileUMap = isl_union_map_from_map(TileMap);
405 TileUMap =
406 isl_union_map_align_params(TileUMap, isl_space_copy(Space));
407 PartialSchedule =
408 isl_union_map_apply_range(PartialSchedule, TileUMap);
409 break;
410 }
Tobias Grosser7c5ba832011-06-30 20:29:20 +0000411 }
Tobias Grosserde68cc92011-06-30 20:01:02 +0000412 }
413
Tobias Grosser62872012011-11-17 12:56:04 +0000414 Schedule = isl_union_map_union(Schedule, PartialSchedule);
Tobias Grosserde68cc92011-06-30 20:01:02 +0000415
Tobias Grosser1ae9a602011-11-17 12:56:03 +0000416 isl_band_free(Band);
Tobias Grosserf5338802011-10-06 00:03:35 +0000417 isl_space_free(Space);
Tobias Grosserde68cc92011-06-30 20:01:02 +0000418 }
419
Tobias Grosser1ae9a602011-11-17 12:56:03 +0000420 return Schedule;
Tobias Grosserde68cc92011-06-30 20:01:02 +0000421}
422
Tobias Grosser460e9a42012-04-25 13:22:43 +0000423isl_union_map *IslScheduleOptimizer::getScheduleMap(isl_schedule *Schedule) {
Tobias Grosser1ae9a602011-11-17 12:56:03 +0000424 isl_band_list *BandList = isl_schedule_get_band_forest(Schedule);
425 isl_union_map *ScheduleMap = getScheduleForBandList(BandList);
426 isl_band_list_free(BandList);
427 return ScheduleMap;
Tobias Grosser30aa24c2011-05-14 19:02:06 +0000428}
429
Tobias Grosser73600b82011-10-08 00:30:40 +0000430bool IslScheduleOptimizer::runOnScop(Scop &S) {
Tobias Grosser30aa24c2011-05-14 19:02:06 +0000431 Dependences *D = &getAnalysis<Dependences>();
432
Tobias Grosser28781422012-10-16 07:29:19 +0000433 isl_schedule_free(LastSchedule);
434 LastSchedule = NULL;
435
Tobias Grosser30aa24c2011-05-14 19:02:06 +0000436 // Build input data.
Tobias Grosser4d96c8d2013-03-23 01:05:07 +0000437 int ValidityKinds =
438 Dependences::TYPE_RAW | Dependences::TYPE_WAR | Dependences::TYPE_WAW;
Tobias Grosser1deda292012-02-14 14:02:48 +0000439 int ProximityKinds;
440
441 if (OptimizeDeps == "all")
Tobias Grosser4d96c8d2013-03-23 01:05:07 +0000442 ProximityKinds =
443 Dependences::TYPE_RAW | Dependences::TYPE_WAR | Dependences::TYPE_WAW;
Tobias Grosser1deda292012-02-14 14:02:48 +0000444 else if (OptimizeDeps == "raw")
Tobias Grosser1e03ad72012-02-15 09:58:42 +0000445 ProximityKinds = Dependences::TYPE_RAW;
Tobias Grosser1deda292012-02-14 14:02:48 +0000446 else {
447 errs() << "Do not know how to optimize for '" << OptimizeDeps << "'"
Tobias Grosser4d96c8d2013-03-23 01:05:07 +0000448 << " Falling back to optimizing all dependences.\n";
449 ProximityKinds =
450 Dependences::TYPE_RAW | Dependences::TYPE_WAR | Dependences::TYPE_WAW;
Tobias Grosser1deda292012-02-14 14:02:48 +0000451 }
452
Tobias Grosser5f9a7622012-02-14 14:02:40 +0000453 isl_union_set *Domain = S.getDomains();
Tobias Grosser30aa24c2011-05-14 19:02:06 +0000454
Tobias Grosser98610ee2012-02-13 23:31:39 +0000455 if (!Domain)
Tobias Grosser30aa24c2011-05-14 19:02:06 +0000456 return false;
457
Tobias Grosser8a507022012-03-16 11:51:41 +0000458 isl_union_map *Validity = D->getDependences(ValidityKinds);
459 isl_union_map *Proximity = D->getDependences(ProximityKinds);
460
Tobias Grossera26db472012-01-30 19:38:43 +0000461 // Simplify the dependences by removing the constraints introduced by the
462 // domains. This can speed up the scheduling time significantly, as large
463 // constant coefficients will be removed from the dependences. The
464 // introduction of some additional dependences reduces the possible
465 // transformations, but in most cases, such transformation do not seem to be
466 // interesting anyway. In some cases this option may stop the scheduler to
467 // find any schedule.
468 if (SimplifyDeps == "yes") {
Tobias Grosser00383a72012-02-14 14:02:44 +0000469 Validity = isl_union_map_gist_domain(Validity, isl_union_set_copy(Domain));
470 Validity = isl_union_map_gist_range(Validity, isl_union_set_copy(Domain));
Tobias Grosser4d96c8d2013-03-23 01:05:07 +0000471 Proximity =
472 isl_union_map_gist_domain(Proximity, isl_union_set_copy(Domain));
Tobias Grosser00383a72012-02-14 14:02:44 +0000473 Proximity = isl_union_map_gist_range(Proximity, isl_union_set_copy(Domain));
Tobias Grossera26db472012-01-30 19:38:43 +0000474 } else if (SimplifyDeps != "no") {
475 errs() << "warning: Option -polly-opt-simplify-deps should either be 'yes' "
476 "or 'no'. Falling back to default: 'yes'\n";
477 }
478
Tobias Grosser30aa24c2011-05-14 19:02:06 +0000479 DEBUG(dbgs() << "\n\nCompute schedule from: ");
Tobias Grosser98610ee2012-02-13 23:31:39 +0000480 DEBUG(dbgs() << "Domain := "; isl_union_set_dump(Domain); dbgs() << ";\n");
481 DEBUG(dbgs() << "Proximity := "; isl_union_map_dump(Proximity);
Tobias Grosser30aa24c2011-05-14 19:02:06 +0000482 dbgs() << ";\n");
Tobias Grosser98610ee2012-02-13 23:31:39 +0000483 DEBUG(dbgs() << "Validity := "; isl_union_map_dump(Validity);
Tobias Grosser30aa24c2011-05-14 19:02:06 +0000484 dbgs() << ";\n");
485
Tobias Grosserb3ad85b2012-01-30 19:38:50 +0000486 int IslFusionStrategy;
487
488 if (FusionStrategy == "max") {
489 IslFusionStrategy = ISL_SCHEDULE_FUSE_MAX;
490 } else if (FusionStrategy == "min") {
491 IslFusionStrategy = ISL_SCHEDULE_FUSE_MIN;
492 } else {
493 errs() << "warning: Unknown fusion strategy. Falling back to maximal "
494 "fusion.\n";
495 IslFusionStrategy = ISL_SCHEDULE_FUSE_MAX;
496 }
497
Tobias Grosser95e860c2012-01-30 19:38:54 +0000498 int IslMaximizeBands;
499
Tobias Grossera4ea90b2012-01-30 22:43:56 +0000500 if (MaximizeBandDepth == "yes") {
Tobias Grosser95e860c2012-01-30 19:38:54 +0000501 IslMaximizeBands = 1;
Tobias Grossera4ea90b2012-01-30 22:43:56 +0000502 } else if (MaximizeBandDepth == "no") {
Tobias Grosser95e860c2012-01-30 19:38:54 +0000503 IslMaximizeBands = 0;
504 } else {
505 errs() << "warning: Option -polly-opt-maximize-bands should either be 'yes'"
506 " or 'no'. Falling back to default: 'yes'\n";
507 IslMaximizeBands = 1;
508 }
509
Tobias Grosserb3ad85b2012-01-30 19:38:50 +0000510 isl_options_set_schedule_fuse(S.getIslCtx(), IslFusionStrategy);
Tobias Grosser95e860c2012-01-30 19:38:54 +0000511 isl_options_set_schedule_maximize_band_depth(S.getIslCtx(), IslMaximizeBands);
Tobias Grosser992e60c2012-02-20 08:41:15 +0000512 isl_options_set_schedule_max_constant_term(S.getIslCtx(), MaxConstantTerm);
Tobias Grosser92f54802012-02-20 08:41:47 +0000513 isl_options_set_schedule_max_coefficient(S.getIslCtx(), MaxCoefficient);
Tobias Grosser42152ff2012-01-30 19:38:47 +0000514
515 isl_options_set_on_error(S.getIslCtx(), ISL_ON_ERROR_CONTINUE);
Tobias Grosser00383a72012-02-14 14:02:44 +0000516 isl_schedule *Schedule;
Tobias Grosser4d96c8d2013-03-23 01:05:07 +0000517 Schedule = isl_union_set_compute_schedule(Domain, Validity, Proximity);
Tobias Grosser42152ff2012-01-30 19:38:47 +0000518 isl_options_set_on_error(S.getIslCtx(), ISL_ON_ERROR_ABORT);
519
520 // In cases the scheduler is not able to optimize the code, we just do not
521 // touch the schedule.
Tobias Grosser98610ee2012-02-13 23:31:39 +0000522 if (!Schedule)
Tobias Grosser42152ff2012-01-30 19:38:47 +0000523 return false;
Tobias Grosser30aa24c2011-05-14 19:02:06 +0000524
Tobias Grosser4d96c8d2013-03-23 01:05:07 +0000525 DEBUG(dbgs() << "Schedule := "; isl_schedule_dump(Schedule); dbgs() << ";\n");
Tobias Grosser4d63b9d2012-02-20 08:41:21 +0000526
Tobias Grosser98610ee2012-02-13 23:31:39 +0000527 isl_union_map *ScheduleMap = getScheduleMap(Schedule);
Tobias Grosser30aa24c2011-05-14 19:02:06 +0000528
Tobias Grosserde68cc92011-06-30 20:01:02 +0000529 for (Scop::iterator SI = S.begin(), SE = S.end(); SI != SE; ++SI) {
Tobias Grosser98610ee2012-02-13 23:31:39 +0000530 ScopStmt *Stmt = *SI;
Tobias Grosser249c4b12013-04-11 05:55:13 +0000531 isl_map *StmtSchedule;
Tobias Grosser98610ee2012-02-13 23:31:39 +0000532 isl_set *Domain = Stmt->getDomain();
533 isl_union_map *StmtBand;
534 StmtBand = isl_union_map_intersect_domain(isl_union_map_copy(ScheduleMap),
Tobias Grosser4d96c8d2013-03-23 01:05:07 +0000535 isl_union_set_from_set(Domain));
Tobias Grosser249c4b12013-04-11 05:55:13 +0000536 if (isl_union_map_is_empty(StmtBand)) {
537 // Statements with an empty iteration domain may not have a schedule
538 // assigned by the isl schedule optimizer. As Polly expects each statement
539 // to have a schedule, we keep the old schedule for this statement. As
540 // there are zero iterations to execute, the content of the schedule does
541 // not matter.
542 //
543 // TODO: Consider removing such statements when constructing the scop.
Tobias Grosserf242b802013-04-10 22:48:08 +0000544 StmtSchedule = Stmt->getScattering();
545 StmtSchedule = isl_map_set_tuple_id(StmtSchedule, isl_dim_out, NULL);
Tobias Grosser249c4b12013-04-11 05:55:13 +0000546 isl_union_map_free(StmtBand);
547 } else {
548 assert(isl_union_map_n_map(StmtBand) == 1);
549 StmtSchedule = isl_map_from_union_map(StmtBand);
Tobias Grosserf242b802013-04-10 22:48:08 +0000550 }
551
Tobias Grosser98610ee2012-02-13 23:31:39 +0000552 Stmt->setScattering(StmtSchedule);
Tobias Grosser30aa24c2011-05-14 19:02:06 +0000553 }
554
Tobias Grosser1ae9a602011-11-17 12:56:03 +0000555 isl_union_map_free(ScheduleMap);
Tobias Grosser28781422012-10-16 07:29:19 +0000556 LastSchedule = Schedule;
Tobias Grosserde68cc92011-06-30 20:01:02 +0000557
Tobias Grosser98610ee2012-02-13 23:31:39 +0000558 unsigned MaxScatDims = 0;
Tobias Grosser30aa24c2011-05-14 19:02:06 +0000559
560 for (Scop::iterator SI = S.begin(), SE = S.end(); SI != SE; ++SI)
Tobias Grosser98610ee2012-02-13 23:31:39 +0000561 MaxScatDims = std::max((*SI)->getNumScattering(), MaxScatDims);
Tobias Grosser30aa24c2011-05-14 19:02:06 +0000562
Tobias Grosser98610ee2012-02-13 23:31:39 +0000563 extendScattering(S, MaxScatDims);
Tobias Grosser30aa24c2011-05-14 19:02:06 +0000564 return false;
565}
566
Tobias Grosser73600b82011-10-08 00:30:40 +0000567void IslScheduleOptimizer::printScop(raw_ostream &OS) const {
Tobias Grosser28781422012-10-16 07:29:19 +0000568 isl_printer *p;
569 char *ScheduleStr;
570
571 OS << "Calculated schedule:\n";
572
573 if (!LastSchedule) {
574 OS << "n/a\n";
575 return;
576 }
577
578 p = isl_printer_to_str(isl_schedule_get_ctx(LastSchedule));
579 p = isl_printer_print_schedule(p, LastSchedule);
580 ScheduleStr = isl_printer_get_str(p);
581 isl_printer_free(p);
582
583 OS << ScheduleStr << "\n";
Tobias Grosser30aa24c2011-05-14 19:02:06 +0000584}
585
Tobias Grosser73600b82011-10-08 00:30:40 +0000586void IslScheduleOptimizer::getAnalysisUsage(AnalysisUsage &AU) const {
Tobias Grosser30aa24c2011-05-14 19:02:06 +0000587 ScopPass::getAnalysisUsage(AU);
588 AU.addRequired<Dependences>();
589}
590
Tobias Grosser4d96c8d2013-03-23 01:05:07 +0000591Pass *polly::createIslScheduleOptimizerPass() {
Tobias Grosser73600b82011-10-08 00:30:40 +0000592 return new IslScheduleOptimizer();
Tobias Grosser30aa24c2011-05-14 19:02:06 +0000593}
Tobias Grosser4d96c8d2013-03-23 01:05:07 +0000594
595INITIALIZE_PASS_BEGIN(IslScheduleOptimizer, "polly-opt-isl",
596 "Polly - Optimize schedule of SCoP", false, false);
597INITIALIZE_PASS_DEPENDENCY(Dependences);
598INITIALIZE_PASS_DEPENDENCY(ScopInfo);
599INITIALIZE_PASS_END(IslScheduleOptimizer, "polly-opt-isl",
600 "Polly - Optimize schedule of SCoP", false, false)