blob: 9e26179abed62b4249cf81a4dfc8b614bffe2130 [file] [log] [blame]
Tobias Grosser75805372011-04-29 06:27:02 +00001//===--------- ScopInfo.cpp - Create Scops from LLVM IR ------------------===//
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// Create a polyhedral description for a static control flow region.
11//
12// The pass creates a polyhedral description of the Scops detected by the Scop
13// detection derived from their LLVM-IR code.
14//
15// This represantation is shared among several tools in the polyhedral
16// community, which are e.g. Cloog, Pluto, Loopo, Graphite.
17//
18//===----------------------------------------------------------------------===//
19
20#include "polly/ScopInfo.h"
21
22#include "polly/TempScopInfo.h"
23#include "polly/LinkAllPasses.h"
24#include "polly/Support/GICHelper.h"
25#include "polly/Support/ScopHelper.h"
26
27#include "llvm/Analysis/LoopInfo.h"
28#include "llvm/Analysis/ScalarEvolutionExpressions.h"
29#include "llvm/Analysis/RegionIterator.h"
30#include "llvm/Assembly/Writer.h"
31#include "llvm/ADT/Statistic.h"
32#include "llvm/ADT/SetVector.h"
33#include "llvm/Support/CommandLine.h"
34
35#define DEBUG_TYPE "polly-scops"
36#include "llvm/Support/Debug.h"
37
38#include "isl/constraint.h"
39#include "isl/set.h"
40#include "isl/map.h"
Tobias Grosser33ba62ad2011-08-18 06:31:50 +000041#include "isl/aff.h"
42#include "isl/printer.h"
Tobias Grosserf5338802011-10-06 00:03:35 +000043#include "isl/local_space.h"
Tobias Grosser75805372011-04-29 06:27:02 +000044#include <sstream>
45#include <string>
46#include <vector>
47
48using namespace llvm;
49using namespace polly;
50
51STATISTIC(ScopFound, "Number of valid Scops");
52STATISTIC(RichScopFound, "Number of Scops containing a loop");
53
Tobias Grosser33ba62ad2011-08-18 06:31:50 +000054/// Convert an int into a string.
55static std::string convertInt(int number)
56{
57 if (number == 0)
58 return "0";
59 std::string temp = "";
60 std::string returnvalue = "";
61 while (number > 0)
62 {
63 temp += number % 10 + 48;
64 number /= 10;
65 }
66 for (unsigned i = 0; i < temp.length(); i++)
67 returnvalue+=temp[temp.length() - i - 1];
68 return returnvalue;
Tobias Grosser75805372011-04-29 06:27:02 +000069}
70
Tobias Grosser33ba62ad2011-08-18 06:31:50 +000071/// Translate a SCEVExpression into an isl_pw_aff object.
72struct SCEVAffinator : public SCEVVisitor<SCEVAffinator, isl_pw_aff*> {
73private:
74 isl_ctx *ctx;
Tobias Grosserf5338802011-10-06 00:03:35 +000075 int NbLoopSpaces;
Tobias Grosser33ba62ad2011-08-18 06:31:50 +000076 const Scop *scop;
77
78 /// baseAdress is set if we analyze a memory access. It holds the base address
79 /// of this memory access.
80 const Value *baseAddress;
81
82public:
83 static isl_pw_aff *getPwAff(const ScopStmt *stmt, const SCEV *scev,
84 const Value *baseAddress) {
85 SCEVAffinator Affinator(stmt, baseAddress);
86 return Affinator.visit(scev);
87 }
88
89 isl_pw_aff *visit(const SCEV *scev) {
90 // In case the scev is contained in our list of parameters, we do not
91 // further analyze this expression, but create a new parameter in the
92 // isl_pw_aff. This allows us to treat subexpressions that we cannot
93 // translate into an piecewise affine expression, as constant parameters of
94 // the piecewise affine expression.
95 int i = 0;
96 for (Scop::const_param_iterator PI = scop->param_begin(),
97 PE = scop->param_end(); PI != PE; ++PI) {
98 if (*PI == scev) {
99 isl_id *ID = isl_id_alloc(ctx, ("p" + convertInt(i)).c_str(),
100 (void *) scev);
Tobias Grosserf5338802011-10-06 00:03:35 +0000101 isl_space *Space = isl_space_set_alloc(ctx, 1, NbLoopSpaces);
102 Space = isl_space_set_dim_id(Space, isl_dim_param, 0, ID);
Tobias Grosser33ba62ad2011-08-18 06:31:50 +0000103
Tobias Grosserf5338802011-10-06 00:03:35 +0000104 isl_set *Domain = isl_set_universe(isl_space_copy(Space));
105 isl_aff *Affine = isl_aff_zero_on_domain(
106 isl_local_space_from_space(Space));
Tobias Grosser33ba62ad2011-08-18 06:31:50 +0000107 Affine = isl_aff_add_coefficient_si(Affine, isl_dim_param, 0, 1);
108
109 return isl_pw_aff_alloc(Domain, Affine);
110 }
111 i++;
112 }
113
114 return SCEVVisitor<SCEVAffinator, isl_pw_aff*>::visit(scev);
115 }
116
117 SCEVAffinator(const ScopStmt *stmt, const Value *baseAddress) :
Tobias Grosser3c69fab2011-10-06 00:03:54 +0000118 ctx(stmt->getIslCtx()),
Tobias Grosserf5338802011-10-06 00:03:35 +0000119 NbLoopSpaces(stmt->getNumIterators()),
Tobias Grosser33ba62ad2011-08-18 06:31:50 +0000120 scop(stmt->getParent()),
121 baseAddress(baseAddress) {};
122
123 __isl_give isl_pw_aff *visitConstant(const SCEVConstant *Constant) {
124 ConstantInt *Value = Constant->getValue();
125 isl_int v;
126 isl_int_init(v);
127
128 // LLVM does not define if an integer value is interpreted as a signed or
129 // unsigned value. Hence, without further information, it is unknown how
130 // this value needs to be converted to GMP. At the moment, we only support
131 // signed operations. So we just interpret it as signed. Later, there are
132 // two options:
133 //
134 // 1. We always interpret any value as signed and convert the values on
135 // demand.
136 // 2. We pass down the signedness of the calculation and use it to interpret
137 // this constant correctly.
138 MPZ_from_APInt(v, Value->getValue(), /* isSigned */ true);
139
Tobias Grosserf5338802011-10-06 00:03:35 +0000140 isl_space *Space = isl_space_set_alloc(ctx, 0, NbLoopSpaces);
141 isl_local_space *ls = isl_local_space_from_space(isl_space_copy(Space));
142 isl_aff *Affine = isl_aff_zero_on_domain(ls);
143 isl_set *Domain = isl_set_universe(Space);
Tobias Grosser33ba62ad2011-08-18 06:31:50 +0000144
145 Affine = isl_aff_add_constant(Affine, v);
146 isl_int_clear(v);
147
148 return isl_pw_aff_alloc(Domain, Affine);
149 }
150
151 __isl_give isl_pw_aff *visitTruncateExpr(const SCEVTruncateExpr* Expr) {
152 assert(0 && "Not yet supported");
153 }
154
155 __isl_give isl_pw_aff *visitZeroExtendExpr(const SCEVZeroExtendExpr * Expr) {
156 assert(0 && "Not yet supported");
157 }
158
159 __isl_give isl_pw_aff *visitSignExtendExpr(const SCEVSignExtendExpr* Expr) {
160 // Assuming the value is signed, a sign extension is basically a noop.
161 // TODO: Reconsider this as soon as we support unsigned values.
162 return visit(Expr->getOperand());
163 }
164
165 __isl_give isl_pw_aff *visitAddExpr(const SCEVAddExpr* Expr) {
166 isl_pw_aff *Sum = visit(Expr->getOperand(0));
167
168 for (int i = 1, e = Expr->getNumOperands(); i < e; ++i) {
169 isl_pw_aff *NextSummand = visit(Expr->getOperand(i));
170 Sum = isl_pw_aff_add(Sum, NextSummand);
171 }
172
173 // TODO: Check for NSW and NUW.
174
175 return Sum;
176 }
177
178 __isl_give isl_pw_aff *visitMulExpr(const SCEVMulExpr* Expr) {
179 isl_pw_aff *Product = visit(Expr->getOperand(0));
180
181 for (int i = 1, e = Expr->getNumOperands(); i < e; ++i) {
182 isl_pw_aff *NextOperand = visit(Expr->getOperand(i));
183
184 if (!isl_pw_aff_is_cst(Product) && !isl_pw_aff_is_cst(NextOperand)) {
185 isl_pw_aff_free(Product);
186 isl_pw_aff_free(NextOperand);
187 return NULL;
188 }
189
190 Product = isl_pw_aff_mul(Product, NextOperand);
191 }
192
193 // TODO: Check for NSW and NUW.
194 return Product;
195 }
196
197 __isl_give isl_pw_aff *visitUDivExpr(const SCEVUDivExpr* Expr) {
198 assert(0 && "Not yet supported");
199 }
200
201 int getLoopDepth(const Loop *L) {
202 Loop *outerLoop =
203 scop->getRegion().outermostLoopInRegion(const_cast<Loop*>(L));
204 return L->getLoopDepth() - outerLoop->getLoopDepth();
205 }
206
207 __isl_give isl_pw_aff *visitAddRecExpr(const SCEVAddRecExpr* Expr) {
208 assert(Expr->isAffine() && "Only affine AddRecurrences allowed");
209
210 isl_pw_aff *Start = visit(Expr->getStart());
211 isl_pw_aff *Step = visit(Expr->getOperand(1));
Tobias Grosserf5338802011-10-06 00:03:35 +0000212 isl_space *Space = isl_space_set_alloc(ctx, 0, NbLoopSpaces);
213 isl_local_space *LocalSpace = isl_local_space_from_space(Space);
Tobias Grosser33ba62ad2011-08-18 06:31:50 +0000214
215 int loopDimension = getLoopDepth(Expr->getLoop());
216
Tobias Grosserf5338802011-10-06 00:03:35 +0000217 isl_aff *LAff = isl_aff_set_coefficient_si(
218 isl_aff_zero_on_domain (LocalSpace), isl_dim_in, loopDimension, 1);
Tobias Grosser33ba62ad2011-08-18 06:31:50 +0000219 isl_pw_aff *LPwAff = isl_pw_aff_from_aff(LAff);
220
221 // TODO: Do we need to check for NSW and NUW?
222 return isl_pw_aff_add(Start, isl_pw_aff_mul(Step, LPwAff));
223 }
224
225 __isl_give isl_pw_aff *visitSMaxExpr(const SCEVSMaxExpr* Expr) {
226 isl_pw_aff *Max = visit(Expr->getOperand(0));
227
228 for (int i = 1, e = Expr->getNumOperands(); i < e; ++i) {
229 isl_pw_aff *NextOperand = visit(Expr->getOperand(i));
230 Max = isl_pw_aff_max(Max, NextOperand);
231 }
232
233 return Max;
234 }
235
236 __isl_give isl_pw_aff *visitUMaxExpr(const SCEVUMaxExpr* Expr) {
237 assert(0 && "Not yet supported");
238 }
239
240 __isl_give isl_pw_aff *visitUnknown(const SCEVUnknown* Expr) {
241 Value *Value = Expr->getValue();
242
Tobias Grosserf5338802011-10-06 00:03:35 +0000243 isl_space *Space;
Tobias Grosser33ba62ad2011-08-18 06:31:50 +0000244
245 /// If baseAddress is set, we ignore its Value object in the scev and do not
246 /// add it to the isl_pw_aff. This is because it is regarded as defining the
247 /// name of an array, in contrast to its array subscript.
248 if (baseAddress != Value) {
249 isl_id *ID = isl_id_alloc(ctx, Value->getNameStr().c_str(), Value);
Tobias Grosserf5338802011-10-06 00:03:35 +0000250 Space = isl_space_set_alloc(ctx, 1, NbLoopSpaces);
251 Space = isl_space_set_dim_id(Space, isl_dim_param, 0, ID);
Tobias Grosser33ba62ad2011-08-18 06:31:50 +0000252 } else {
Tobias Grosserf5338802011-10-06 00:03:35 +0000253 Space = isl_space_set_alloc(ctx, 0, NbLoopSpaces);
Tobias Grosser33ba62ad2011-08-18 06:31:50 +0000254 }
255
Tobias Grosserf5338802011-10-06 00:03:35 +0000256 isl_set *Domain = isl_set_universe(isl_space_copy(Space));
257 isl_aff *Affine = isl_aff_zero_on_domain(isl_local_space_from_space(Space));
Tobias Grosser33ba62ad2011-08-18 06:31:50 +0000258
259 if (baseAddress != Value)
260 Affine = isl_aff_add_coefficient_si(Affine, isl_dim_param, 0, 1);
261
262 return isl_pw_aff_alloc(Domain, Affine);
263 }
264};
265
Tobias Grosser75805372011-04-29 06:27:02 +0000266//===----------------------------------------------------------------------===//
267
268MemoryAccess::~MemoryAccess() {
Tobias Grosser54a86e62011-08-18 06:31:46 +0000269 isl_map_free(AccessRelation);
Raghesh Aloor129e8672011-08-15 02:33:39 +0000270 isl_map_free(newAccessRelation);
Tobias Grosser75805372011-04-29 06:27:02 +0000271}
272
273static void replace(std::string& str, const std::string& find,
274 const std::string& replace) {
275 size_t pos = 0;
276 while((pos = str.find(find, pos)) != std::string::npos)
277 {
278 str.replace(pos, find.length(), replace);
279 pos += replace.length();
280 }
281}
282
283static void makeIslCompatible(std::string& str) {
Tobias Grossereec4d56e2011-08-20 11:11:14 +0000284 str.erase(0, 1);
Tobias Grosser75805372011-04-29 06:27:02 +0000285 replace(str, ".", "_");
Tobias Grosser3b660f82011-08-03 00:12:11 +0000286 replace(str, "\"", "_");
Tobias Grosser75805372011-04-29 06:27:02 +0000287}
288
289void MemoryAccess::setBaseName() {
290 raw_string_ostream OS(BaseName);
291 WriteAsOperand(OS, getBaseAddr(), false);
292 BaseName = OS.str();
293
Tobias Grosser75805372011-04-29 06:27:02 +0000294 makeIslCompatible(BaseName);
295 BaseName = "MemRef_" + BaseName;
296}
297
298std::string MemoryAccess::getAccessFunctionStr() const {
299 return stringFromIslObj(getAccessFunction());
300}
301
302isl_basic_map *MemoryAccess::createBasicAccessMap(ScopStmt *Statement) {
Tobias Grosser3c69fab2011-10-06 00:03:54 +0000303 isl_space *Space = isl_space_alloc(Statement->getIslCtx(), 0,
Tobias Grosserf5338802011-10-06 00:03:35 +0000304 Statement->getNumIterators(), 1);
Tobias Grosser75805372011-04-29 06:27:02 +0000305 setBaseName();
306
Tobias Grosserf5338802011-10-06 00:03:35 +0000307 Space = isl_space_set_tuple_name(Space, isl_dim_out, getBaseName().c_str());
308 Space = isl_space_set_tuple_name(Space, isl_dim_in, Statement->getBaseName());
Tobias Grosser75805372011-04-29 06:27:02 +0000309
Tobias Grosserf5338802011-10-06 00:03:35 +0000310 return isl_basic_map_universe(Space);
Tobias Grosser75805372011-04-29 06:27:02 +0000311}
312
313MemoryAccess::MemoryAccess(const SCEVAffFunc &AffFunc, ScopStmt *Statement) {
Raghesh Aloor3cb66282011-07-12 17:14:03 +0000314 newAccessRelation = NULL;
Tobias Grosser75805372011-04-29 06:27:02 +0000315 BaseAddr = AffFunc.getBaseAddr();
316 Type = AffFunc.isRead() ? Read : Write;
317 statement = Statement;
318
319 setBaseName();
320
Tobias Grosser7d4cee42011-08-19 23:34:28 +0000321 isl_pw_aff *Affine = SCEVAffinator::getPwAff(Statement, AffFunc.OriginalSCEV,
322 AffFunc.getBaseAddr());
Tobias Grosser75805372011-04-29 06:27:02 +0000323
Tobias Grosser7d4cee42011-08-19 23:34:28 +0000324 // Devide the access function by the size of the elements in the array.
325 //
326 // A stride one array access in C expressed as A[i] is expressed in LLVM-IR
327 // as something like A[i * elementsize]. This hides the fact that two
328 // subsequent values of 'i' index two values that are stored next to each
329 // other in memory. By this devision we make this characteristic obvious
330 // again.
Tobias Grosser75805372011-04-29 06:27:02 +0000331 isl_int v;
332 isl_int_init(v);
Tobias Grosser75805372011-04-29 06:27:02 +0000333 isl_int_set_si(v, AffFunc.getElemSizeInBytes());
Tobias Grosser7d4cee42011-08-19 23:34:28 +0000334 Affine = isl_pw_aff_scale_down(Affine, v);
335 isl_int_clear(v);
Tobias Grosser75805372011-04-29 06:27:02 +0000336
Tobias Grosser7d4cee42011-08-19 23:34:28 +0000337 AccessRelation = isl_map_from_pw_aff(Affine);
338 AccessRelation = isl_map_set_tuple_name(AccessRelation, isl_dim_in,
339 Statement->getBaseName());
Tobias Grosser75805372011-04-29 06:27:02 +0000340 AccessRelation = isl_map_set_tuple_name(AccessRelation, isl_dim_out,
341 getBaseName().c_str());
Tobias Grosser30b8a092011-08-18 07:51:37 +0000342
Tobias Grosser37487052011-10-06 00:03:42 +0000343 isl_space *ParamSpace = Statement->getParent()->getParamSpace();
344 AccessRelation = isl_map_align_params(AccessRelation, ParamSpace);
Tobias Grosser75805372011-04-29 06:27:02 +0000345}
346
347MemoryAccess::MemoryAccess(const Value *BaseAddress, ScopStmt *Statement) {
Raghesh Aloor3cb66282011-07-12 17:14:03 +0000348 newAccessRelation = NULL;
Tobias Grosser75805372011-04-29 06:27:02 +0000349 BaseAddr = BaseAddress;
350 Type = Read;
351 statement = Statement;
352
353 isl_basic_map *BasicAccessMap = createBasicAccessMap(Statement);
354 AccessRelation = isl_map_from_basic_map(BasicAccessMap);
Tobias Grosser37487052011-10-06 00:03:42 +0000355 isl_space *ParamSpace = Statement->getParent()->getParamSpace();
356 AccessRelation = isl_map_align_params(AccessRelation, ParamSpace);
Tobias Grosser75805372011-04-29 06:27:02 +0000357}
358
359void MemoryAccess::print(raw_ostream &OS) const {
360 OS.indent(12) << (isRead() ? "Read" : "Write") << "Access := \n";
361 OS.indent(16) << getAccessFunctionStr() << ";\n";
362}
363
364void MemoryAccess::dump() const {
365 print(errs());
366}
367
368// Create a map in the size of the provided set domain, that maps from the
369// one element of the provided set domain to another element of the provided
370// set domain.
371// The mapping is limited to all points that are equal in all but the last
372// dimension and for which the last dimension of the input is strict smaller
373// than the last dimension of the output.
374//
375// getEqualAndLarger(set[i0, i1, ..., iX]):
376//
377// set[i0, i1, ..., iX] -> set[o0, o1, ..., oX]
378// : i0 = o0, i1 = o1, ..., i(X-1) = o(X-1), iX < oX
379//
Tobias Grosserf5338802011-10-06 00:03:35 +0000380static isl_map *getEqualAndLarger(isl_space *setDomain) {
381 isl_space *mapDomain = isl_space_map_from_set(setDomain);
Tobias Grosser75805372011-04-29 06:27:02 +0000382 isl_basic_map *bmap = isl_basic_map_universe(mapDomain);
Tobias Grosserf5338802011-10-06 00:03:35 +0000383 isl_local_space *MapLocalSpace = isl_local_space_from_space(mapDomain);
Tobias Grosser75805372011-04-29 06:27:02 +0000384
385 // Set all but the last dimension to be equal for the input and output
386 //
387 // input[i0, i1, ..., iX] -> output[o0, o1, ..., oX]
388 // : i0 = o0, i1 = o1, ..., i(X-1) = o(X-1)
389 for (unsigned i = 0; i < isl_basic_map_n_in(bmap) - 1; ++i) {
390 isl_int v;
391 isl_int_init(v);
Tobias Grosserf5338802011-10-06 00:03:35 +0000392 isl_constraint *c = isl_equality_alloc(isl_local_space_copy(MapLocalSpace));
Tobias Grosser75805372011-04-29 06:27:02 +0000393
394 isl_int_set_si(v, 1);
395 isl_constraint_set_coefficient(c, isl_dim_in, i, v);
396 isl_int_set_si(v, -1);
397 isl_constraint_set_coefficient(c, isl_dim_out, i, v);
398
399 bmap = isl_basic_map_add_constraint(bmap, c);
400
401 isl_int_clear(v);
402 }
403
404 // Set the last dimension of the input to be strict smaller than the
405 // last dimension of the output.
406 //
407 // input[?,?,?,...,iX] -> output[?,?,?,...,oX] : iX < oX
408 //
409 unsigned lastDimension = isl_basic_map_n_in(bmap) - 1;
410 isl_int v;
411 isl_int_init(v);
Tobias Grosserf5338802011-10-06 00:03:35 +0000412 isl_constraint *c = isl_inequality_alloc(isl_local_space_copy(MapLocalSpace));
Tobias Grosser75805372011-04-29 06:27:02 +0000413 isl_int_set_si(v, -1);
414 isl_constraint_set_coefficient(c, isl_dim_in, lastDimension, v);
415 isl_int_set_si(v, 1);
416 isl_constraint_set_coefficient(c, isl_dim_out, lastDimension, v);
417 isl_int_set_si(v, -1);
418 isl_constraint_set_constant(c, v);
419 isl_int_clear(v);
420
421 bmap = isl_basic_map_add_constraint(bmap, c);
422
423 return isl_map_from_basic_map(bmap);
424}
425
426isl_set *MemoryAccess::getStride(const isl_set *domainSubset) const {
427 isl_map *accessRelation = isl_map_copy(getAccessFunction());
428 isl_set *scatteringDomain = isl_set_copy(const_cast<isl_set*>(domainSubset));
Tobias Grossercf3942d2011-10-06 00:04:05 +0000429 isl_map *scattering = getStatement()->getScattering();
Tobias Grosser75805372011-04-29 06:27:02 +0000430
431 scattering = isl_map_reverse(scattering);
432 int difference = isl_map_n_in(scattering) - isl_set_n_dim(scatteringDomain);
433 scattering = isl_map_project_out(scattering, isl_dim_in,
434 isl_set_n_dim(scatteringDomain),
435 difference);
436
437 // Remove all names of the scattering dimensions, as the names may be lost
438 // anyways during the project. This leads to consistent results.
439 scattering = isl_map_set_tuple_name(scattering, isl_dim_in, "");
440 scatteringDomain = isl_set_set_tuple_name(scatteringDomain, "");
441
Tobias Grosserf5338802011-10-06 00:03:35 +0000442 isl_map *nextScatt = getEqualAndLarger(isl_set_get_space(scatteringDomain));
Tobias Grosser75805372011-04-29 06:27:02 +0000443 nextScatt = isl_map_lexmin(nextScatt);
444
445 scattering = isl_map_intersect_domain(scattering, scatteringDomain);
446
447 nextScatt = isl_map_apply_range(nextScatt, isl_map_copy(scattering));
448 nextScatt = isl_map_apply_range(nextScatt, isl_map_copy(accessRelation));
449 nextScatt = isl_map_apply_domain(nextScatt, scattering);
450 nextScatt = isl_map_apply_domain(nextScatt, accessRelation);
451
452 return isl_map_deltas(nextScatt);
453}
454
455bool MemoryAccess::isStrideZero(const isl_set *domainSubset) const {
456 isl_set *stride = getStride(domainSubset);
Tobias Grosserf5338802011-10-06 00:03:35 +0000457 isl_space *StrideSpace = isl_set_get_space(stride);
458 isl_local_space *StrideLS = isl_local_space_from_space(StrideSpace);
459 isl_constraint *c = isl_equality_alloc(StrideLS);
Tobias Grosser75805372011-04-29 06:27:02 +0000460
461 isl_int v;
462 isl_int_init(v);
463 isl_int_set_si(v, 1);
464 isl_constraint_set_coefficient(c, isl_dim_set, 0, v);
465 isl_int_set_si(v, 0);
466 isl_constraint_set_constant(c, v);
467 isl_int_clear(v);
468
Tobias Grosserf5338802011-10-06 00:03:35 +0000469 isl_basic_set *bset = isl_basic_set_universe(isl_set_get_space(stride));
Tobias Grosser75805372011-04-29 06:27:02 +0000470
471 bset = isl_basic_set_add_constraint(bset, c);
472 isl_set *strideZero = isl_set_from_basic_set(bset);
473
Tobias Grosserb76f38532011-08-20 11:11:25 +0000474 bool isStrideZero = isl_set_is_equal(stride, strideZero);
475
476 isl_set_free(strideZero);
477 isl_set_free(stride);
478
479 return isStrideZero;
Tobias Grosser75805372011-04-29 06:27:02 +0000480}
481
482bool MemoryAccess::isStrideOne(const isl_set *domainSubset) const {
483 isl_set *stride = getStride(domainSubset);
Tobias Grosserf5338802011-10-06 00:03:35 +0000484 isl_space *StrideSpace = isl_set_get_space(stride);
485 isl_local_space *StrideLSpace = isl_local_space_from_space(StrideSpace);
486 isl_constraint *c = isl_equality_alloc(StrideLSpace);
Tobias Grosser75805372011-04-29 06:27:02 +0000487
488 isl_int v;
489 isl_int_init(v);
490 isl_int_set_si(v, 1);
491 isl_constraint_set_coefficient(c, isl_dim_set, 0, v);
492 isl_int_set_si(v, -1);
493 isl_constraint_set_constant(c, v);
494 isl_int_clear(v);
495
Tobias Grosserf5338802011-10-06 00:03:35 +0000496 isl_basic_set *bset = isl_basic_set_universe(isl_set_get_space(stride));
Tobias Grosser75805372011-04-29 06:27:02 +0000497
498 bset = isl_basic_set_add_constraint(bset, c);
Tobias Grosserb76f38532011-08-20 11:11:25 +0000499 isl_set *strideOne = isl_set_from_basic_set(bset);
Tobias Grosser75805372011-04-29 06:27:02 +0000500
Tobias Grosserb76f38532011-08-20 11:11:25 +0000501 bool isStrideOne = isl_set_is_equal(stride, strideOne);
502
503 isl_set_free(strideOne);
504 isl_set_free(stride);
505
506 return isStrideOne;
Tobias Grosser75805372011-04-29 06:27:02 +0000507}
508
Raghesh Aloor7a04f4f2011-08-03 13:47:59 +0000509void MemoryAccess::setNewAccessFunction(isl_map *newAccess) {
Tobias Grosserb76f38532011-08-20 11:11:25 +0000510 isl_map_free(newAccessRelation);
Raghesh Aloor7a04f4f2011-08-03 13:47:59 +0000511 newAccessRelation = newAccess;
Raghesh Aloor3cb66282011-07-12 17:14:03 +0000512}
Tobias Grosser75805372011-04-29 06:27:02 +0000513
514//===----------------------------------------------------------------------===//
Tobias Grossercf3942d2011-10-06 00:04:05 +0000515
516isl_map *ScopStmt::getScattering() const {
517 return isl_map_copy(Scattering);
518}
519
520void ScopStmt::setScattering(isl_map *NewScattering) {
Tobias Grosserb76f38532011-08-20 11:11:25 +0000521 isl_map_free(Scattering);
Tobias Grossercf3942d2011-10-06 00:04:05 +0000522 Scattering = NewScattering;
Tobias Grosserb76f38532011-08-20 11:11:25 +0000523}
524
Tobias Grosser75805372011-04-29 06:27:02 +0000525void ScopStmt::buildScattering(SmallVectorImpl<unsigned> &Scatter) {
526 unsigned NumberOfIterators = getNumIterators();
Tobias Grosserf5338802011-10-06 00:03:35 +0000527 unsigned ScatSpace = Parent.getMaxLoopDepth() * 2 + 1;
Tobias Grosser3c69fab2011-10-06 00:03:54 +0000528 isl_space *Space = isl_space_alloc(getIslCtx(), 0, NumberOfIterators,
Tobias Grosserf5338802011-10-06 00:03:35 +0000529 ScatSpace);
530 Space = isl_space_set_tuple_name(Space, isl_dim_out, "scattering");
531 Space = isl_space_set_tuple_name(Space, isl_dim_in, getBaseName());
532 isl_local_space *LSpace = isl_local_space_from_space(isl_space_copy(Space));
533 isl_basic_map *bmap = isl_basic_map_universe(Space);
Tobias Grosser75805372011-04-29 06:27:02 +0000534 isl_int v;
535 isl_int_init(v);
536
537 // Loop dimensions.
538 for (unsigned i = 0; i < NumberOfIterators; ++i) {
Tobias Grosserf5338802011-10-06 00:03:35 +0000539 isl_constraint *c = isl_equality_alloc(isl_local_space_copy(LSpace));
Tobias Grosser75805372011-04-29 06:27:02 +0000540 isl_int_set_si(v, 1);
541 isl_constraint_set_coefficient(c, isl_dim_out, 2 * i + 1, v);
542 isl_int_set_si(v, -1);
543 isl_constraint_set_coefficient(c, isl_dim_in, i, v);
544
545 bmap = isl_basic_map_add_constraint(bmap, c);
546 }
547
548 // Constant dimensions
549 for (unsigned i = 0; i < NumberOfIterators + 1; ++i) {
Tobias Grosserf5338802011-10-06 00:03:35 +0000550 isl_constraint *c = isl_equality_alloc(isl_local_space_copy(LSpace));
Tobias Grosser75805372011-04-29 06:27:02 +0000551 isl_int_set_si(v, -1);
552 isl_constraint_set_coefficient(c, isl_dim_out, 2 * i, v);
553 isl_int_set_si(v, Scatter[i]);
554 isl_constraint_set_constant(c, v);
555
556 bmap = isl_basic_map_add_constraint(bmap, c);
557 }
558
559 // Fill scattering dimensions.
Tobias Grosserf5338802011-10-06 00:03:35 +0000560 for (unsigned i = 2 * NumberOfIterators + 1; i < ScatSpace ; ++i) {
561 isl_constraint *c = isl_equality_alloc(isl_local_space_copy(LSpace));
Tobias Grosser75805372011-04-29 06:27:02 +0000562 isl_int_set_si(v, 1);
563 isl_constraint_set_coefficient(c, isl_dim_out, i, v);
564 isl_int_set_si(v, 0);
565 isl_constraint_set_constant(c, v);
566
567 bmap = isl_basic_map_add_constraint(bmap, c);
568 }
569
570 isl_int_clear(v);
Tobias Grosser75805372011-04-29 06:27:02 +0000571 Scattering = isl_map_from_basic_map(bmap);
Tobias Grosser37487052011-10-06 00:03:42 +0000572 Scattering = isl_map_align_params(Scattering, Parent.getParamSpace());
Tobias Grosser75805372011-04-29 06:27:02 +0000573}
574
575void ScopStmt::buildAccesses(TempScop &tempScop, const Region &CurRegion) {
576 const AccFuncSetType *AccFuncs = tempScop.getAccessFunctions(BB);
577
578 for (AccFuncSetType::const_iterator I = AccFuncs->begin(),
579 E = AccFuncs->end(); I != E; ++I) {
580 MemAccs.push_back(new MemoryAccess(I->first, this));
581 InstructionToAccess[I->second] = MemAccs.back();
582 }
583}
584
Tobias Grosserf5338802011-10-06 00:03:35 +0000585isl_set *ScopStmt::toConditionSet(const Comparison &Comp,
586 isl_space *space) const {
Tobias Grosserd2795d02011-08-18 07:51:40 +0000587 isl_pw_aff *LHS = SCEVAffinator::getPwAff(this, Comp.getLHS()->OriginalSCEV,
588 0);
589 isl_pw_aff *RHS = SCEVAffinator::getPwAff(this, Comp.getRHS()->OriginalSCEV,
590 0);
Tobias Grosser75805372011-04-29 06:27:02 +0000591
Tobias Grosserd2795d02011-08-18 07:51:40 +0000592 isl_set *set;
Tobias Grosser75805372011-04-29 06:27:02 +0000593
Tobias Grosserd2795d02011-08-18 07:51:40 +0000594 switch (Comp.getPred()) {
Tobias Grosser75805372011-04-29 06:27:02 +0000595 case ICmpInst::ICMP_EQ:
Tobias Grosserd2795d02011-08-18 07:51:40 +0000596 set = isl_pw_aff_eq_set(LHS, RHS);
Tobias Grosser75805372011-04-29 06:27:02 +0000597 break;
598 case ICmpInst::ICMP_NE:
Tobias Grosserd2795d02011-08-18 07:51:40 +0000599 set = isl_pw_aff_ne_set(LHS, RHS);
Tobias Grosser75805372011-04-29 06:27:02 +0000600 break;
601 case ICmpInst::ICMP_SLT:
Tobias Grosserd2795d02011-08-18 07:51:40 +0000602 set = isl_pw_aff_lt_set(LHS, RHS);
Tobias Grosser75805372011-04-29 06:27:02 +0000603 break;
604 case ICmpInst::ICMP_SLE:
Tobias Grosserd2795d02011-08-18 07:51:40 +0000605 set = isl_pw_aff_le_set(LHS, RHS);
Tobias Grosser75805372011-04-29 06:27:02 +0000606 break;
Tobias Grosserd2795d02011-08-18 07:51:40 +0000607 case ICmpInst::ICMP_SGT:
608 set = isl_pw_aff_gt_set(LHS, RHS);
Tobias Grosser75805372011-04-29 06:27:02 +0000609 break;
610 case ICmpInst::ICMP_SGE:
Tobias Grosserd2795d02011-08-18 07:51:40 +0000611 set = isl_pw_aff_ge_set(LHS, RHS);
Tobias Grosser75805372011-04-29 06:27:02 +0000612 break;
Tobias Grosserd2795d02011-08-18 07:51:40 +0000613 case ICmpInst::ICMP_ULT:
614 case ICmpInst::ICMP_UGT:
615 case ICmpInst::ICMP_ULE:
Tobias Grosser75805372011-04-29 06:27:02 +0000616 case ICmpInst::ICMP_UGE:
Tobias Grosserd2795d02011-08-18 07:51:40 +0000617 llvm_unreachable("Unsigned comparisons not yet supported");
Tobias Grosser75805372011-04-29 06:27:02 +0000618 default:
619 llvm_unreachable("Non integer predicate not supported");
620 }
621
Tobias Grosserf5338802011-10-06 00:03:35 +0000622 set = isl_set_set_tuple_name(set, isl_space_get_tuple_name(space, isl_dim_set));
Tobias Grosserd2795d02011-08-18 07:51:40 +0000623
624 return set;
Tobias Grosser75805372011-04-29 06:27:02 +0000625}
626
Tobias Grosserf5338802011-10-06 00:03:35 +0000627isl_set *ScopStmt::toUpperLoopBound(const SCEVAffFunc &UpperBound,
628 isl_space *Space,
Tobias Grosser75805372011-04-29 06:27:02 +0000629 unsigned BoundedDimension) const {
Tobias Grosserd2795d02011-08-18 07:51:40 +0000630 // FIXME: We should choose a consistent scheme of when to name the dimensions.
Tobias Grosserf5338802011-10-06 00:03:35 +0000631 isl_space *UnnamedSpace = isl_space_copy(Space);
632 UnnamedSpace = isl_space_set_tuple_name(UnnamedSpace, isl_dim_set, 0);
633 isl_local_space *LocalSpace = isl_local_space_from_space(UnnamedSpace);
634 isl_aff *LAff = isl_aff_set_coefficient_si(isl_aff_zero_on_domain(LocalSpace),
635 isl_dim_in, BoundedDimension, 1);
636 isl_pw_aff *BoundedSpace = isl_pw_aff_from_aff(LAff);
Tobias Grosserd2795d02011-08-18 07:51:40 +0000637 isl_pw_aff *Bound = SCEVAffinator::getPwAff(this, UpperBound.OriginalSCEV, 0);
Tobias Grosserf5338802011-10-06 00:03:35 +0000638 isl_set *set = isl_pw_aff_le_set(BoundedSpace, Bound);
639 set = isl_set_set_tuple_name(set, isl_space_get_tuple_name(Space, isl_dim_set));
640 isl_space_free(Space);
Tobias Grosserd2795d02011-08-18 07:51:40 +0000641 return set;
Tobias Grosser75805372011-04-29 06:27:02 +0000642}
643
644void ScopStmt::buildIterationDomainFromLoops(TempScop &tempScop) {
Tobias Grosser3c69fab2011-10-06 00:03:54 +0000645 isl_space *Space = isl_space_set_alloc(getIslCtx(), 0, getNumIterators());
Tobias Grosserf5338802011-10-06 00:03:35 +0000646 Space = isl_space_set_tuple_name(Space, isl_dim_set, getBaseName());
Tobias Grosser75805372011-04-29 06:27:02 +0000647
Tobias Grosserf5338802011-10-06 00:03:35 +0000648 Domain = isl_set_universe(isl_space_copy(Space));
Tobias Grosser37487052011-10-06 00:03:42 +0000649 Domain = isl_set_align_params(Domain, Parent.getParamSpace());
Tobias Grosser75805372011-04-29 06:27:02 +0000650
651 isl_int v;
652 isl_int_init(v);
653
Tobias Grosserf5338802011-10-06 00:03:35 +0000654 isl_local_space *LocalSpace;
655 LocalSpace = isl_local_space_from_space(isl_space_copy(Space));
656
Tobias Grosser75805372011-04-29 06:27:02 +0000657 for (int i = 0, e = getNumIterators(); i != e; ++i) {
658 // Lower bound: IV >= 0.
Tobias Grosserf5338802011-10-06 00:03:35 +0000659 isl_basic_set *bset = isl_basic_set_universe(isl_space_copy(Space));
660 isl_constraint *c = isl_inequality_alloc(isl_local_space_copy(LocalSpace));
Tobias Grosser75805372011-04-29 06:27:02 +0000661 isl_int_set_si(v, 1);
662 isl_constraint_set_coefficient(c, isl_dim_set, i, v);
663 bset = isl_basic_set_add_constraint(bset, c);
664 Domain = isl_set_intersect(Domain, isl_set_from_basic_set(bset));
665
666 // Upper bound: IV <= NumberOfIterations.
Hongbin Zheng27f3afb2011-04-30 03:26:51 +0000667 const Loop *L = getLoopForDimension(i);
Tobias Grosser75805372011-04-29 06:27:02 +0000668 const SCEVAffFunc &UpperBound = tempScop.getLoopBound(L);
Tobias Grosserf5338802011-10-06 00:03:35 +0000669 isl_set *UpperBoundSet = toUpperLoopBound(UpperBound, isl_space_copy(Space),
670 i);
Tobias Grosser75805372011-04-29 06:27:02 +0000671 Domain = isl_set_intersect(Domain, UpperBoundSet);
672 }
673
Tobias Grosserf5338802011-10-06 00:03:35 +0000674 isl_local_space_free(LocalSpace);
675 isl_space_free(Space);
Tobias Grosser75805372011-04-29 06:27:02 +0000676 isl_int_clear(v);
677}
678
679void ScopStmt::addConditionsToDomain(TempScop &tempScop,
680 const Region &CurRegion) {
Tobias Grosserf5338802011-10-06 00:03:35 +0000681 isl_space *Space = isl_set_get_space(Domain);
Tobias Grosser75805372011-04-29 06:27:02 +0000682 const Region *TopR = tempScop.getMaxRegion().getParent(),
683 *CurR = &CurRegion;
684 const BasicBlock *CurEntry = BB;
685
686 // Build BB condition constrains, by traveling up the region tree.
687 do {
688 assert(CurR && "We exceed the top region?");
689 // Skip when multiple regions share the same entry.
690 if (CurEntry != CurR->getEntry()) {
691 if (const BBCond *Cnd = tempScop.getBBCond(CurEntry))
692 for (BBCond::const_iterator I = Cnd->begin(), E = Cnd->end();
693 I != E; ++I) {
Tobias Grosserf5338802011-10-06 00:03:35 +0000694 isl_set *c = toConditionSet(*I, Space);
Tobias Grosser75805372011-04-29 06:27:02 +0000695 Domain = isl_set_intersect(Domain, c);
696 }
697 }
698 CurEntry = CurR->getEntry();
699 CurR = CurR->getParent();
700 } while (TopR != CurR);
701
Tobias Grosserf5338802011-10-06 00:03:35 +0000702 isl_space_free(Space);
Tobias Grosser75805372011-04-29 06:27:02 +0000703}
704
705void ScopStmt::buildIterationDomain(TempScop &tempScop, const Region &CurRegion)
706{
707 buildIterationDomainFromLoops(tempScop);
708 addConditionsToDomain(tempScop, CurRegion);
709}
710
711ScopStmt::ScopStmt(Scop &parent, TempScop &tempScop,
712 const Region &CurRegion, BasicBlock &bb,
713 SmallVectorImpl<Loop*> &NestLoops,
714 SmallVectorImpl<unsigned> &Scatter)
715 : Parent(parent), BB(&bb), IVS(NestLoops.size()) {
716 // Setup the induction variables.
717 for (unsigned i = 0, e = NestLoops.size(); i < e; ++i) {
718 PHINode *PN = NestLoops[i]->getCanonicalInductionVariable();
719 assert(PN && "Non canonical IV in Scop!");
Hongbin Zheng27f3afb2011-04-30 03:26:51 +0000720 IVS[i] = std::make_pair(PN, NestLoops[i]);
Tobias Grosser75805372011-04-29 06:27:02 +0000721 }
722
723 raw_string_ostream OS(BaseName);
724 WriteAsOperand(OS, &bb, false);
725 BaseName = OS.str();
726
Tobias Grosser75805372011-04-29 06:27:02 +0000727 makeIslCompatible(BaseName);
728 BaseName = "Stmt_" + BaseName;
729
730 buildIterationDomain(tempScop, CurRegion);
731 buildScattering(Scatter);
732 buildAccesses(tempScop, CurRegion);
Tobias Grosser75805372011-04-29 06:27:02 +0000733}
734
735ScopStmt::ScopStmt(Scop &parent, SmallVectorImpl<unsigned> &Scatter)
736 : Parent(parent), BB(NULL), IVS(0) {
737
738 BaseName = "FinalRead";
739
740 // Build iteration domain.
741 std::string IterationDomainString = "{[i0] : i0 = 0}";
Tobias Grosser3c69fab2011-10-06 00:03:54 +0000742 Domain = isl_set_read_from_str(getIslCtx(), IterationDomainString.c_str());
Tobias Grosser75805372011-04-29 06:27:02 +0000743 Domain = isl_set_set_tuple_name(Domain, getBaseName());
Tobias Grosser37487052011-10-06 00:03:42 +0000744 Domain = isl_set_align_params(Domain, parent.getParamSpace());
Tobias Grosser75805372011-04-29 06:27:02 +0000745
746 // Build scattering.
Tobias Grosserf5338802011-10-06 00:03:35 +0000747 unsigned ScatSpace = Parent.getMaxLoopDepth() * 2 + 1;
Tobias Grosser3c69fab2011-10-06 00:03:54 +0000748 isl_space *Space = isl_space_alloc(getIslCtx(), 0, 1, ScatSpace);
Tobias Grosserf5338802011-10-06 00:03:35 +0000749 Space = isl_space_set_tuple_name(Space, isl_dim_out, "scattering");
750 Space = isl_space_set_tuple_name(Space, isl_dim_in, getBaseName());
751 isl_basic_map *bmap = isl_basic_map_universe(isl_space_copy(Space));
Tobias Grosser75805372011-04-29 06:27:02 +0000752 isl_int v;
753 isl_int_init(v);
754
Tobias Grosserf5338802011-10-06 00:03:35 +0000755 isl_constraint *c = isl_equality_alloc(isl_local_space_from_space(Space));
Tobias Grosser75805372011-04-29 06:27:02 +0000756 isl_int_set_si(v, -1);
757 isl_constraint_set_coefficient(c, isl_dim_out, 0, v);
758
759 // TODO: This is incorrect. We should not use a very large number to ensure
760 // that this statement is executed last.
761 isl_int_set_si(v, 200000000);
762 isl_constraint_set_constant(c, v);
763
764 bmap = isl_basic_map_add_constraint(bmap, c);
765 isl_int_clear(v);
766 Scattering = isl_map_from_basic_map(bmap);
Tobias Grosser37487052011-10-06 00:03:42 +0000767 Scattering = isl_map_align_params(Scattering, parent.getParamSpace());
Tobias Grosser75805372011-04-29 06:27:02 +0000768
769 // Build memory accesses, use SetVector to keep the order of memory accesses
770 // and prevent the same memory access inserted more than once.
771 SetVector<const Value*> BaseAddressSet;
772
773 for (Scop::const_iterator SI = Parent.begin(), SE = Parent.end(); SI != SE;
774 ++SI) {
775 ScopStmt *Stmt = *SI;
776
777 for (MemoryAccessVec::const_iterator I = Stmt->memacc_begin(),
778 E = Stmt->memacc_end(); I != E; ++I)
779 BaseAddressSet.insert((*I)->getBaseAddr());
780 }
781
782 for (SetVector<const Value*>::iterator BI = BaseAddressSet.begin(),
783 BE = BaseAddressSet.end(); BI != BE; ++BI)
784 MemAccs.push_back(new MemoryAccess(*BI, this));
Tobias Grosser75805372011-04-29 06:27:02 +0000785}
786
787std::string ScopStmt::getDomainStr() const {
Tobias Grosser4da8d9f2011-10-06 00:03:59 +0000788 return stringFromIslObj(Domain);
Tobias Grosser75805372011-04-29 06:27:02 +0000789}
790
791std::string ScopStmt::getScatteringStr() const {
Tobias Grossercf3942d2011-10-06 00:04:05 +0000792 return stringFromIslObj(Scattering);
Tobias Grosser75805372011-04-29 06:27:02 +0000793}
794
795unsigned ScopStmt::getNumParams() const {
796 return Parent.getNumParams();
797}
798
799unsigned ScopStmt::getNumIterators() const {
800 // The final read has one dimension with one element.
801 if (!BB)
802 return 1;
803
804 return IVS.size();
805}
806
807unsigned ScopStmt::getNumScattering() const {
808 return isl_map_dim(Scattering, isl_dim_out);
809}
810
811const char *ScopStmt::getBaseName() const { return BaseName.c_str(); }
812
813const PHINode *ScopStmt::getInductionVariableForDimension(unsigned Dimension)
814 const {
Hongbin Zheng27f3afb2011-04-30 03:26:51 +0000815 return IVS[Dimension].first;
816}
817
818const Loop *ScopStmt::getLoopForDimension(unsigned Dimension) const {
819 return IVS[Dimension].second;
Tobias Grosser75805372011-04-29 06:27:02 +0000820}
821
822const SCEVAddRecExpr *ScopStmt::getSCEVForDimension(unsigned Dimension)
823 const {
Hongbin Zheng27f3afb2011-04-30 03:26:51 +0000824 PHINode *PN =
825 const_cast<PHINode*>(getInductionVariableForDimension(Dimension));
Tobias Grosser75805372011-04-29 06:27:02 +0000826 return cast<SCEVAddRecExpr>(getParent()->getSE()->getSCEV(PN));
827}
828
Tobias Grosser3c69fab2011-10-06 00:03:54 +0000829isl_ctx *ScopStmt::getIslCtx() const {
830 return Parent.getIslCtx();
Tobias Grosser75805372011-04-29 06:27:02 +0000831}
832
Tobias Grosserd5a7bfc2011-05-06 19:52:19 +0000833isl_set *ScopStmt::getDomain() const {
834 return isl_set_copy(Domain);
835}
836
Tobias Grosser75805372011-04-29 06:27:02 +0000837ScopStmt::~ScopStmt() {
838 while (!MemAccs.empty()) {
839 delete MemAccs.back();
840 MemAccs.pop_back();
841 }
842
843 isl_set_free(Domain);
844 isl_map_free(Scattering);
845}
846
847void ScopStmt::print(raw_ostream &OS) const {
848 OS << "\t" << getBaseName() << "\n";
849
850 OS.indent(12) << "Domain :=\n";
851
852 if (Domain) {
853 OS.indent(16) << getDomainStr() << ";\n";
854 } else
855 OS.indent(16) << "n/a\n";
856
857 OS.indent(12) << "Scattering :=\n";
858
859 if (Domain) {
860 OS.indent(16) << getScatteringStr() << ";\n";
861 } else
862 OS.indent(16) << "n/a\n";
863
864 for (MemoryAccessVec::const_iterator I = MemAccs.begin(), E = MemAccs.end();
865 I != E; ++I)
866 (*I)->print(OS);
867}
868
869void ScopStmt::dump() const { print(dbgs()); }
870
871//===----------------------------------------------------------------------===//
872/// Scop class implement
Tobias Grosser75805372011-04-29 06:27:02 +0000873
Tobias Grosser0e27e242011-10-06 00:03:48 +0000874void Scop::buildContext(isl_ctx *IslCtx, ParamSetType *ParamSet) {
875 isl_space *Space = isl_space_params_alloc(IslCtx, ParamSet->size());
Tobias Grosser75805372011-04-29 06:27:02 +0000876
Tobias Grosser30b8a092011-08-18 07:51:37 +0000877 int i = 0;
Tobias Grosser0e27e242011-10-06 00:03:48 +0000878 for (ParamSetType::iterator PI = ParamSet->begin(), PE = ParamSet->end();
Tobias Grosser30b8a092011-08-18 07:51:37 +0000879 PI != PE; ++PI) {
Tobias Grosser0e27e242011-10-06 00:03:48 +0000880 const SCEV *Parameter = *PI;
881 Parameters.push_back(Parameter);
882 std::string ParameterName = "p" + convertInt(i);
883 isl_id *id = isl_id_alloc(IslCtx, ParameterName.c_str(),
884 (void *) Parameter);
Tobias Grosserf5338802011-10-06 00:03:35 +0000885 Space = isl_space_set_dim_id(Space, isl_dim_param, i, id);
Tobias Grosser30b8a092011-08-18 07:51:37 +0000886 i++;
887 }
888
Tobias Grosser75805372011-04-29 06:27:02 +0000889 // TODO: Insert relations between parameters.
890 // TODO: Insert constraints on parameters.
Tobias Grosserf5338802011-10-06 00:03:35 +0000891 Context = isl_set_universe (Space);
Tobias Grosser0e27e242011-10-06 00:03:48 +0000892}
893
894Scop::Scop(TempScop &tempScop, LoopInfo &LI, ScalarEvolution &ScalarEvolution,
895 isl_ctx *Context)
896 : SE(&ScalarEvolution), R(tempScop.getMaxRegion()),
897 MaxLoopDepth(tempScop.getMaxLoopDepth()) {
898 buildContext(Context, &tempScop.getParamSet());
Tobias Grosser75805372011-04-29 06:27:02 +0000899
900 SmallVector<Loop*, 8> NestLoops;
901 SmallVector<unsigned, 8> Scatter;
902
903 Scatter.assign(MaxLoopDepth + 1, 0);
904
905 // Build the iteration domain, access functions and scattering functions
906 // traversing the region tree.
907 buildScop(tempScop, getRegion(), NestLoops, Scatter, LI);
908 Stmts.push_back(new ScopStmt(*this, Scatter));
909
Tobias Grosser75805372011-04-29 06:27:02 +0000910 assert(NestLoops.empty() && "NestLoops not empty at top level!");
911}
912
913Scop::~Scop() {
914 isl_set_free(Context);
915
916 // Free the statements;
917 for (iterator I = begin(), E = end(); I != E; ++I)
918 delete *I;
Tobias Grosser75805372011-04-29 06:27:02 +0000919}
920
921std::string Scop::getContextStr() const {
Tobias Grosser4da8d9f2011-10-06 00:03:59 +0000922 return stringFromIslObj(Context);
Tobias Grosser75805372011-04-29 06:27:02 +0000923}
924
925std::string Scop::getNameStr() const {
926 std::string ExitName, EntryName;
927 raw_string_ostream ExitStr(ExitName);
928 raw_string_ostream EntryStr(EntryName);
929
930 WriteAsOperand(EntryStr, R.getEntry(), false);
931 EntryStr.str();
932
933 if (R.getExit()) {
934 WriteAsOperand(ExitStr, R.getExit(), false);
935 ExitStr.str();
936 } else
937 ExitName = "FunctionExit";
938
939 return EntryName + "---" + ExitName;
940}
941
Tobias Grosser4da8d9f2011-10-06 00:03:59 +0000942__isl_give isl_set *Scop::getContext() const {
943 return isl_set_copy(Context);
944}
Tobias Grosser37487052011-10-06 00:03:42 +0000945__isl_give isl_space *Scop::getParamSpace() const {
946 return isl_set_get_space(this->Context);
947}
948
Tobias Grosser75805372011-04-29 06:27:02 +0000949void Scop::printContext(raw_ostream &OS) const {
950 OS << "Context:\n";
951
952 if (!Context) {
953 OS.indent(4) << "n/a\n\n";
954 return;
955 }
956
957 OS.indent(4) << getContextStr() << "\n";
958}
959
960void Scop::printStatements(raw_ostream &OS) const {
961 OS << "Statements {\n";
962
963 for (const_iterator SI = begin(), SE = end();SI != SE; ++SI)
964 OS.indent(4) << (**SI);
965
966 OS.indent(4) << "}\n";
967}
968
969
970void Scop::print(raw_ostream &OS) const {
971 printContext(OS.indent(4));
972 printStatements(OS.indent(4));
973}
974
975void Scop::dump() const { print(dbgs()); }
976
Tobias Grosser3c69fab2011-10-06 00:03:54 +0000977isl_ctx *Scop::getIslCtx() const { return isl_set_get_ctx(Context); }
Tobias Grosser75805372011-04-29 06:27:02 +0000978
979ScalarEvolution *Scop::getSE() const { return SE; }
980
981bool Scop::isTrivialBB(BasicBlock *BB, TempScop &tempScop) {
982 if (tempScop.getAccessFunctions(BB))
983 return false;
984
985 return true;
986}
987
988void Scop::buildScop(TempScop &tempScop,
989 const Region &CurRegion,
990 SmallVectorImpl<Loop*> &NestLoops,
991 SmallVectorImpl<unsigned> &Scatter,
992 LoopInfo &LI) {
993 Loop *L = castToLoop(CurRegion, LI);
994
995 if (L)
996 NestLoops.push_back(L);
997
998 unsigned loopDepth = NestLoops.size();
999 assert(Scatter.size() > loopDepth && "Scatter not big enough!");
1000
1001 for (Region::const_element_iterator I = CurRegion.element_begin(),
1002 E = CurRegion.element_end(); I != E; ++I)
1003 if (I->isSubRegion())
1004 buildScop(tempScop, *(I->getNodeAs<Region>()), NestLoops, Scatter, LI);
1005 else {
1006 BasicBlock *BB = I->getNodeAs<BasicBlock>();
1007
1008 if (isTrivialBB(BB, tempScop))
1009 continue;
1010
1011 Stmts.push_back(new ScopStmt(*this, tempScop, CurRegion, *BB, NestLoops,
1012 Scatter));
1013
1014 // Increasing the Scattering function is OK for the moment, because
1015 // we are using a depth first iterator and the program is well structured.
1016 ++Scatter[loopDepth];
1017 }
1018
1019 if (!L)
1020 return;
1021
1022 // Exiting a loop region.
1023 Scatter[loopDepth] = 0;
1024 NestLoops.pop_back();
1025 ++Scatter[loopDepth-1];
1026}
1027
1028//===----------------------------------------------------------------------===//
Tobias Grosserb76f38532011-08-20 11:11:25 +00001029ScopInfo::ScopInfo() : RegionPass(ID), scop(0) {
1030 ctx = isl_ctx_alloc();
1031}
1032
1033ScopInfo::~ScopInfo() {
1034 clear();
1035 isl_ctx_free(ctx);
1036}
1037
1038
Tobias Grosser75805372011-04-29 06:27:02 +00001039
1040void ScopInfo::getAnalysisUsage(AnalysisUsage &AU) const {
1041 AU.addRequired<LoopInfo>();
1042 AU.addRequired<RegionInfo>();
1043 AU.addRequired<ScalarEvolution>();
1044 AU.addRequired<TempScopInfo>();
1045 AU.setPreservesAll();
1046}
1047
1048bool ScopInfo::runOnRegion(Region *R, RGPassManager &RGM) {
1049 LoopInfo &LI = getAnalysis<LoopInfo>();
1050 ScalarEvolution &SE = getAnalysis<ScalarEvolution>();
1051
1052 TempScop *tempScop = getAnalysis<TempScopInfo>().getTempScop(R);
1053
1054 // This region is no Scop.
1055 if (!tempScop) {
1056 scop = 0;
1057 return false;
1058 }
1059
1060 // Statistics.
1061 ++ScopFound;
1062 if (tempScop->getMaxLoopDepth() > 0) ++RichScopFound;
1063
Tobias Grosserb76f38532011-08-20 11:11:25 +00001064 scop = new Scop(*tempScop, LI, SE, ctx);
Tobias Grosser75805372011-04-29 06:27:02 +00001065
1066 return false;
1067}
1068
1069char ScopInfo::ID = 0;
1070
1071
1072static RegisterPass<ScopInfo>
1073X("polly-scops", "Polly - Create polyhedral description of Scops");
1074
1075Pass *polly::createScopInfoPass() {
1076 return new ScopInfo();
1077}