blob: 686beb27ce138fce9a824f045ce96564c20e778a [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,
Tobias Grosser9b13d3d2011-10-06 22:32:58 +000084 const Value *baseAddress = 0) {
Tobias Grosser33ba62ad2011-08-18 06:31:50 +000085 SCEVAffinator Affinator(stmt, baseAddress);
86 return Affinator.visit(scev);
87 }
88
89 isl_pw_aff *visit(const SCEV *scev) {
Tobias Grosser76c2e322011-11-07 12:58:59 +000090 // In case the scev is a valid parameter, we do not further analyze this
91 // expression, but create a new parameter in the isl_pw_aff. This allows us
92 // to treat subexpressions that we cannot translate into an piecewise affine
93 // expression, as constant parameters of the piecewise affine expression.
94 if (isl_id *Id = scop->getIdForParam(scev)) {
95 isl_space *Space = isl_space_set_alloc(ctx, 1, NbLoopSpaces);
96 Space = isl_space_set_dim_id(Space, isl_dim_param, 0, Id);
Tobias Grosser33ba62ad2011-08-18 06:31:50 +000097
Tobias Grosser76c2e322011-11-07 12:58:59 +000098 isl_set *Domain = isl_set_universe(isl_space_copy(Space));
99 isl_aff *Affine = isl_aff_zero_on_domain(
100 isl_local_space_from_space(Space));
101 Affine = isl_aff_add_coefficient_si(Affine, isl_dim_param, 0, 1);
Tobias Grosser33ba62ad2011-08-18 06:31:50 +0000102
Tobias Grosser76c2e322011-11-07 12:58:59 +0000103 return isl_pw_aff_alloc(Domain, Affine);
Tobias Grosser33ba62ad2011-08-18 06:31:50 +0000104 }
105
106 return SCEVVisitor<SCEVAffinator, isl_pw_aff*>::visit(scev);
107 }
108
109 SCEVAffinator(const ScopStmt *stmt, const Value *baseAddress) :
Tobias Grosser3c69fab2011-10-06 00:03:54 +0000110 ctx(stmt->getIslCtx()),
Tobias Grosserf5338802011-10-06 00:03:35 +0000111 NbLoopSpaces(stmt->getNumIterators()),
Tobias Grosser33ba62ad2011-08-18 06:31:50 +0000112 scop(stmt->getParent()),
113 baseAddress(baseAddress) {};
114
115 __isl_give isl_pw_aff *visitConstant(const SCEVConstant *Constant) {
116 ConstantInt *Value = Constant->getValue();
117 isl_int v;
118 isl_int_init(v);
119
120 // LLVM does not define if an integer value is interpreted as a signed or
121 // unsigned value. Hence, without further information, it is unknown how
122 // this value needs to be converted to GMP. At the moment, we only support
123 // signed operations. So we just interpret it as signed. Later, there are
124 // two options:
125 //
126 // 1. We always interpret any value as signed and convert the values on
127 // demand.
128 // 2. We pass down the signedness of the calculation and use it to interpret
129 // this constant correctly.
130 MPZ_from_APInt(v, Value->getValue(), /* isSigned */ true);
131
Tobias Grosserf5338802011-10-06 00:03:35 +0000132 isl_space *Space = isl_space_set_alloc(ctx, 0, NbLoopSpaces);
133 isl_local_space *ls = isl_local_space_from_space(isl_space_copy(Space));
134 isl_aff *Affine = isl_aff_zero_on_domain(ls);
135 isl_set *Domain = isl_set_universe(Space);
Tobias Grosser33ba62ad2011-08-18 06:31:50 +0000136
137 Affine = isl_aff_add_constant(Affine, v);
138 isl_int_clear(v);
139
140 return isl_pw_aff_alloc(Domain, Affine);
141 }
142
143 __isl_give isl_pw_aff *visitTruncateExpr(const SCEVTruncateExpr* Expr) {
144 assert(0 && "Not yet supported");
145 }
146
147 __isl_give isl_pw_aff *visitZeroExtendExpr(const SCEVZeroExtendExpr * Expr) {
148 assert(0 && "Not yet supported");
149 }
150
151 __isl_give isl_pw_aff *visitSignExtendExpr(const SCEVSignExtendExpr* Expr) {
152 // Assuming the value is signed, a sign extension is basically a noop.
153 // TODO: Reconsider this as soon as we support unsigned values.
154 return visit(Expr->getOperand());
155 }
156
157 __isl_give isl_pw_aff *visitAddExpr(const SCEVAddExpr* Expr) {
158 isl_pw_aff *Sum = visit(Expr->getOperand(0));
159
160 for (int i = 1, e = Expr->getNumOperands(); i < e; ++i) {
161 isl_pw_aff *NextSummand = visit(Expr->getOperand(i));
162 Sum = isl_pw_aff_add(Sum, NextSummand);
163 }
164
165 // TODO: Check for NSW and NUW.
166
167 return Sum;
168 }
169
170 __isl_give isl_pw_aff *visitMulExpr(const SCEVMulExpr* Expr) {
171 isl_pw_aff *Product = visit(Expr->getOperand(0));
172
173 for (int i = 1, e = Expr->getNumOperands(); i < e; ++i) {
174 isl_pw_aff *NextOperand = visit(Expr->getOperand(i));
175
176 if (!isl_pw_aff_is_cst(Product) && !isl_pw_aff_is_cst(NextOperand)) {
177 isl_pw_aff_free(Product);
178 isl_pw_aff_free(NextOperand);
179 return NULL;
180 }
181
182 Product = isl_pw_aff_mul(Product, NextOperand);
183 }
184
185 // TODO: Check for NSW and NUW.
186 return Product;
187 }
188
189 __isl_give isl_pw_aff *visitUDivExpr(const SCEVUDivExpr* Expr) {
190 assert(0 && "Not yet supported");
191 }
192
193 int getLoopDepth(const Loop *L) {
194 Loop *outerLoop =
195 scop->getRegion().outermostLoopInRegion(const_cast<Loop*>(L));
Tobias Grosser7b0ee0e2011-11-04 10:08:03 +0000196 assert(outerLoop && "Scop does not contain this loop");
Tobias Grosser33ba62ad2011-08-18 06:31:50 +0000197 return L->getLoopDepth() - outerLoop->getLoopDepth();
198 }
199
200 __isl_give isl_pw_aff *visitAddRecExpr(const SCEVAddRecExpr* Expr) {
201 assert(Expr->isAffine() && "Only affine AddRecurrences allowed");
Tobias Grosser7b0ee0e2011-11-04 10:08:03 +0000202 assert(scop->getRegion().contains(Expr->getLoop())
203 && "Scop does not contain the loop referenced in this AddRec");
Tobias Grosser33ba62ad2011-08-18 06:31:50 +0000204
205 isl_pw_aff *Start = visit(Expr->getStart());
206 isl_pw_aff *Step = visit(Expr->getOperand(1));
Tobias Grosserf5338802011-10-06 00:03:35 +0000207 isl_space *Space = isl_space_set_alloc(ctx, 0, NbLoopSpaces);
208 isl_local_space *LocalSpace = isl_local_space_from_space(Space);
Tobias Grosser33ba62ad2011-08-18 06:31:50 +0000209
210 int loopDimension = getLoopDepth(Expr->getLoop());
211
Tobias Grosserf5338802011-10-06 00:03:35 +0000212 isl_aff *LAff = isl_aff_set_coefficient_si(
213 isl_aff_zero_on_domain (LocalSpace), isl_dim_in, loopDimension, 1);
Tobias Grosser33ba62ad2011-08-18 06:31:50 +0000214 isl_pw_aff *LPwAff = isl_pw_aff_from_aff(LAff);
215
216 // TODO: Do we need to check for NSW and NUW?
217 return isl_pw_aff_add(Start, isl_pw_aff_mul(Step, LPwAff));
218 }
219
220 __isl_give isl_pw_aff *visitSMaxExpr(const SCEVSMaxExpr* Expr) {
221 isl_pw_aff *Max = visit(Expr->getOperand(0));
222
223 for (int i = 1, e = Expr->getNumOperands(); i < e; ++i) {
224 isl_pw_aff *NextOperand = visit(Expr->getOperand(i));
225 Max = isl_pw_aff_max(Max, NextOperand);
226 }
227
228 return Max;
229 }
230
231 __isl_give isl_pw_aff *visitUMaxExpr(const SCEVUMaxExpr* Expr) {
232 assert(0 && "Not yet supported");
233 }
234
235 __isl_give isl_pw_aff *visitUnknown(const SCEVUnknown* Expr) {
236 Value *Value = Expr->getValue();
237
Tobias Grosserf5338802011-10-06 00:03:35 +0000238 isl_space *Space;
Tobias Grosser33ba62ad2011-08-18 06:31:50 +0000239
240 /// If baseAddress is set, we ignore its Value object in the scev and do not
241 /// add it to the isl_pw_aff. This is because it is regarded as defining the
242 /// name of an array, in contrast to its array subscript.
243 if (baseAddress != Value) {
244 isl_id *ID = isl_id_alloc(ctx, Value->getNameStr().c_str(), Value);
Tobias Grosserf5338802011-10-06 00:03:35 +0000245 Space = isl_space_set_alloc(ctx, 1, NbLoopSpaces);
246 Space = isl_space_set_dim_id(Space, isl_dim_param, 0, ID);
Tobias Grosser33ba62ad2011-08-18 06:31:50 +0000247 } else {
Tobias Grosserf5338802011-10-06 00:03:35 +0000248 Space = isl_space_set_alloc(ctx, 0, NbLoopSpaces);
Tobias Grosser33ba62ad2011-08-18 06:31:50 +0000249 }
250
Tobias Grosserf5338802011-10-06 00:03:35 +0000251 isl_set *Domain = isl_set_universe(isl_space_copy(Space));
252 isl_aff *Affine = isl_aff_zero_on_domain(isl_local_space_from_space(Space));
Tobias Grosser33ba62ad2011-08-18 06:31:50 +0000253
254 if (baseAddress != Value)
255 Affine = isl_aff_add_coefficient_si(Affine, isl_dim_param, 0, 1);
256
257 return isl_pw_aff_alloc(Domain, Affine);
258 }
259};
260
Tobias Grosser75805372011-04-29 06:27:02 +0000261//===----------------------------------------------------------------------===//
262
263MemoryAccess::~MemoryAccess() {
Tobias Grosser54a86e62011-08-18 06:31:46 +0000264 isl_map_free(AccessRelation);
Raghesh Aloor129e8672011-08-15 02:33:39 +0000265 isl_map_free(newAccessRelation);
Tobias Grosser75805372011-04-29 06:27:02 +0000266}
267
268static void replace(std::string& str, const std::string& find,
269 const std::string& replace) {
270 size_t pos = 0;
271 while((pos = str.find(find, pos)) != std::string::npos)
272 {
273 str.replace(pos, find.length(), replace);
274 pos += replace.length();
275 }
276}
277
278static void makeIslCompatible(std::string& str) {
Tobias Grossereec4d56e2011-08-20 11:11:14 +0000279 str.erase(0, 1);
Tobias Grosser75805372011-04-29 06:27:02 +0000280 replace(str, ".", "_");
Tobias Grosser3b660f82011-08-03 00:12:11 +0000281 replace(str, "\"", "_");
Tobias Grosser75805372011-04-29 06:27:02 +0000282}
283
284void MemoryAccess::setBaseName() {
285 raw_string_ostream OS(BaseName);
286 WriteAsOperand(OS, getBaseAddr(), false);
287 BaseName = OS.str();
288
Tobias Grosser75805372011-04-29 06:27:02 +0000289 makeIslCompatible(BaseName);
290 BaseName = "MemRef_" + BaseName;
291}
292
Tobias Grosser5d453812011-10-06 00:04:11 +0000293isl_map *MemoryAccess::getAccessRelation() const {
294 return isl_map_copy(AccessRelation);
295}
296
297std::string MemoryAccess::getAccessRelationStr() const {
298 return stringFromIslObj(AccessRelation);
299}
300
301isl_map *MemoryAccess::getNewAccessRelation() const {
302 return isl_map_copy(newAccessRelation);
Tobias Grosser75805372011-04-29 06:27:02 +0000303}
304
305isl_basic_map *MemoryAccess::createBasicAccessMap(ScopStmt *Statement) {
Tobias Grosser3c69fab2011-10-06 00:03:54 +0000306 isl_space *Space = isl_space_alloc(Statement->getIslCtx(), 0,
Tobias Grosserf5338802011-10-06 00:03:35 +0000307 Statement->getNumIterators(), 1);
Tobias Grosser75805372011-04-29 06:27:02 +0000308 setBaseName();
309
Tobias Grosserf5338802011-10-06 00:03:35 +0000310 Space = isl_space_set_tuple_name(Space, isl_dim_out, getBaseName().c_str());
311 Space = isl_space_set_tuple_name(Space, isl_dim_in, Statement->getBaseName());
Tobias Grosser75805372011-04-29 06:27:02 +0000312
Tobias Grosserf5338802011-10-06 00:03:35 +0000313 return isl_basic_map_universe(Space);
Tobias Grosser75805372011-04-29 06:27:02 +0000314}
315
316MemoryAccess::MemoryAccess(const SCEVAffFunc &AffFunc, ScopStmt *Statement) {
Raghesh Aloor3cb66282011-07-12 17:14:03 +0000317 newAccessRelation = NULL;
Tobias Grosser75805372011-04-29 06:27:02 +0000318 BaseAddr = AffFunc.getBaseAddr();
319 Type = AffFunc.isRead() ? Read : Write;
320 statement = Statement;
321
322 setBaseName();
323
Tobias Grosser7d4cee42011-08-19 23:34:28 +0000324 isl_pw_aff *Affine = SCEVAffinator::getPwAff(Statement, AffFunc.OriginalSCEV,
325 AffFunc.getBaseAddr());
Tobias Grosser75805372011-04-29 06:27:02 +0000326
Tobias Grosser7d4cee42011-08-19 23:34:28 +0000327 // Devide the access function by the size of the elements in the array.
328 //
329 // A stride one array access in C expressed as A[i] is expressed in LLVM-IR
330 // as something like A[i * elementsize]. This hides the fact that two
331 // subsequent values of 'i' index two values that are stored next to each
332 // other in memory. By this devision we make this characteristic obvious
333 // again.
Tobias Grosser75805372011-04-29 06:27:02 +0000334 isl_int v;
335 isl_int_init(v);
Tobias Grosser75805372011-04-29 06:27:02 +0000336 isl_int_set_si(v, AffFunc.getElemSizeInBytes());
Tobias Grosser7d4cee42011-08-19 23:34:28 +0000337 Affine = isl_pw_aff_scale_down(Affine, v);
338 isl_int_clear(v);
Tobias Grosser75805372011-04-29 06:27:02 +0000339
Tobias Grosser7d4cee42011-08-19 23:34:28 +0000340 AccessRelation = isl_map_from_pw_aff(Affine);
341 AccessRelation = isl_map_set_tuple_name(AccessRelation, isl_dim_in,
342 Statement->getBaseName());
Tobias Grosser75805372011-04-29 06:27:02 +0000343 AccessRelation = isl_map_set_tuple_name(AccessRelation, isl_dim_out,
344 getBaseName().c_str());
Tobias Grosser8cae72f2011-11-08 15:41:08 +0000345}
Tobias Grosser30b8a092011-08-18 07:51:37 +0000346
Tobias Grosser8cae72f2011-11-08 15:41:08 +0000347void MemoryAccess::realignParams() {
348 isl_space *ParamSpace = statement->getParent()->getParamSpace();
Tobias Grosser37487052011-10-06 00:03:42 +0000349 AccessRelation = isl_map_align_params(AccessRelation, ParamSpace);
Tobias Grosser75805372011-04-29 06:27:02 +0000350}
351
352MemoryAccess::MemoryAccess(const Value *BaseAddress, ScopStmt *Statement) {
Raghesh Aloor3cb66282011-07-12 17:14:03 +0000353 newAccessRelation = NULL;
Tobias Grosser75805372011-04-29 06:27:02 +0000354 BaseAddr = BaseAddress;
355 Type = Read;
356 statement = Statement;
357
358 isl_basic_map *BasicAccessMap = createBasicAccessMap(Statement);
359 AccessRelation = isl_map_from_basic_map(BasicAccessMap);
Tobias Grosser37487052011-10-06 00:03:42 +0000360 isl_space *ParamSpace = Statement->getParent()->getParamSpace();
361 AccessRelation = isl_map_align_params(AccessRelation, ParamSpace);
Tobias Grosser75805372011-04-29 06:27:02 +0000362}
363
364void MemoryAccess::print(raw_ostream &OS) const {
365 OS.indent(12) << (isRead() ? "Read" : "Write") << "Access := \n";
Tobias Grosser5d453812011-10-06 00:04:11 +0000366 OS.indent(16) << getAccessRelationStr() << ";\n";
Tobias Grosser75805372011-04-29 06:27:02 +0000367}
368
369void MemoryAccess::dump() const {
370 print(errs());
371}
372
373// Create a map in the size of the provided set domain, that maps from the
374// one element of the provided set domain to another element of the provided
375// set domain.
376// The mapping is limited to all points that are equal in all but the last
377// dimension and for which the last dimension of the input is strict smaller
378// than the last dimension of the output.
379//
380// getEqualAndLarger(set[i0, i1, ..., iX]):
381//
382// set[i0, i1, ..., iX] -> set[o0, o1, ..., oX]
383// : i0 = o0, i1 = o1, ..., i(X-1) = o(X-1), iX < oX
384//
Tobias Grosserf5338802011-10-06 00:03:35 +0000385static isl_map *getEqualAndLarger(isl_space *setDomain) {
386 isl_space *mapDomain = isl_space_map_from_set(setDomain);
Tobias Grosser23b36662011-10-17 08:32:36 +0000387 isl_basic_map *bmap = isl_basic_map_universe(isl_space_copy(mapDomain));
Tobias Grosserf5338802011-10-06 00:03:35 +0000388 isl_local_space *MapLocalSpace = isl_local_space_from_space(mapDomain);
Tobias Grosser75805372011-04-29 06:27:02 +0000389
390 // Set all but the last dimension to be equal for the input and output
391 //
392 // input[i0, i1, ..., iX] -> output[o0, o1, ..., oX]
393 // : i0 = o0, i1 = o1, ..., i(X-1) = o(X-1)
394 for (unsigned i = 0; i < isl_basic_map_n_in(bmap) - 1; ++i) {
395 isl_int v;
396 isl_int_init(v);
Tobias Grosserf5338802011-10-06 00:03:35 +0000397 isl_constraint *c = isl_equality_alloc(isl_local_space_copy(MapLocalSpace));
Tobias Grosser75805372011-04-29 06:27:02 +0000398
399 isl_int_set_si(v, 1);
400 isl_constraint_set_coefficient(c, isl_dim_in, i, v);
401 isl_int_set_si(v, -1);
402 isl_constraint_set_coefficient(c, isl_dim_out, i, v);
403
404 bmap = isl_basic_map_add_constraint(bmap, c);
405
406 isl_int_clear(v);
407 }
408
409 // Set the last dimension of the input to be strict smaller than the
410 // last dimension of the output.
411 //
412 // input[?,?,?,...,iX] -> output[?,?,?,...,oX] : iX < oX
413 //
414 unsigned lastDimension = isl_basic_map_n_in(bmap) - 1;
415 isl_int v;
416 isl_int_init(v);
Tobias Grosserf5338802011-10-06 00:03:35 +0000417 isl_constraint *c = isl_inequality_alloc(isl_local_space_copy(MapLocalSpace));
Tobias Grosser75805372011-04-29 06:27:02 +0000418 isl_int_set_si(v, -1);
419 isl_constraint_set_coefficient(c, isl_dim_in, lastDimension, v);
420 isl_int_set_si(v, 1);
421 isl_constraint_set_coefficient(c, isl_dim_out, lastDimension, v);
422 isl_int_set_si(v, -1);
423 isl_constraint_set_constant(c, v);
424 isl_int_clear(v);
425
426 bmap = isl_basic_map_add_constraint(bmap, c);
427
Tobias Grosser23b36662011-10-17 08:32:36 +0000428 isl_local_space_free(MapLocalSpace);
Tobias Grosser75805372011-04-29 06:27:02 +0000429 return isl_map_from_basic_map(bmap);
430}
431
432isl_set *MemoryAccess::getStride(const isl_set *domainSubset) const {
Tobias Grosser5d453812011-10-06 00:04:11 +0000433 isl_map *accessRelation = getAccessRelation();
Tobias Grosser75805372011-04-29 06:27:02 +0000434 isl_set *scatteringDomain = isl_set_copy(const_cast<isl_set*>(domainSubset));
Tobias Grossercf3942d2011-10-06 00:04:05 +0000435 isl_map *scattering = getStatement()->getScattering();
Tobias Grosser75805372011-04-29 06:27:02 +0000436
437 scattering = isl_map_reverse(scattering);
438 int difference = isl_map_n_in(scattering) - isl_set_n_dim(scatteringDomain);
439 scattering = isl_map_project_out(scattering, isl_dim_in,
440 isl_set_n_dim(scatteringDomain),
441 difference);
442
443 // Remove all names of the scattering dimensions, as the names may be lost
444 // anyways during the project. This leads to consistent results.
445 scattering = isl_map_set_tuple_name(scattering, isl_dim_in, "");
446 scatteringDomain = isl_set_set_tuple_name(scatteringDomain, "");
447
Tobias Grosserf5338802011-10-06 00:03:35 +0000448 isl_map *nextScatt = getEqualAndLarger(isl_set_get_space(scatteringDomain));
Tobias Grosser75805372011-04-29 06:27:02 +0000449 nextScatt = isl_map_lexmin(nextScatt);
450
451 scattering = isl_map_intersect_domain(scattering, scatteringDomain);
452
453 nextScatt = isl_map_apply_range(nextScatt, isl_map_copy(scattering));
454 nextScatt = isl_map_apply_range(nextScatt, isl_map_copy(accessRelation));
455 nextScatt = isl_map_apply_domain(nextScatt, scattering);
456 nextScatt = isl_map_apply_domain(nextScatt, accessRelation);
457
458 return isl_map_deltas(nextScatt);
459}
460
461bool MemoryAccess::isStrideZero(const isl_set *domainSubset) const {
462 isl_set *stride = getStride(domainSubset);
Tobias Grosserf5338802011-10-06 00:03:35 +0000463 isl_space *StrideSpace = isl_set_get_space(stride);
464 isl_local_space *StrideLS = isl_local_space_from_space(StrideSpace);
465 isl_constraint *c = isl_equality_alloc(StrideLS);
Tobias Grosser75805372011-04-29 06:27:02 +0000466
467 isl_int v;
468 isl_int_init(v);
469 isl_int_set_si(v, 1);
470 isl_constraint_set_coefficient(c, isl_dim_set, 0, v);
471 isl_int_set_si(v, 0);
472 isl_constraint_set_constant(c, v);
473 isl_int_clear(v);
474
Tobias Grosserf5338802011-10-06 00:03:35 +0000475 isl_basic_set *bset = isl_basic_set_universe(isl_set_get_space(stride));
Tobias Grosser75805372011-04-29 06:27:02 +0000476
477 bset = isl_basic_set_add_constraint(bset, c);
478 isl_set *strideZero = isl_set_from_basic_set(bset);
479
Tobias Grosserb76f38532011-08-20 11:11:25 +0000480 bool isStrideZero = isl_set_is_equal(stride, strideZero);
481
482 isl_set_free(strideZero);
483 isl_set_free(stride);
484
485 return isStrideZero;
Tobias Grosser75805372011-04-29 06:27:02 +0000486}
487
488bool MemoryAccess::isStrideOne(const isl_set *domainSubset) const {
489 isl_set *stride = getStride(domainSubset);
Tobias Grosserf5338802011-10-06 00:03:35 +0000490 isl_space *StrideSpace = isl_set_get_space(stride);
491 isl_local_space *StrideLSpace = isl_local_space_from_space(StrideSpace);
492 isl_constraint *c = isl_equality_alloc(StrideLSpace);
Tobias Grosser75805372011-04-29 06:27:02 +0000493
494 isl_int v;
495 isl_int_init(v);
496 isl_int_set_si(v, 1);
497 isl_constraint_set_coefficient(c, isl_dim_set, 0, v);
498 isl_int_set_si(v, -1);
499 isl_constraint_set_constant(c, v);
500 isl_int_clear(v);
501
Tobias Grosserf5338802011-10-06 00:03:35 +0000502 isl_basic_set *bset = isl_basic_set_universe(isl_set_get_space(stride));
Tobias Grosser75805372011-04-29 06:27:02 +0000503
504 bset = isl_basic_set_add_constraint(bset, c);
Tobias Grosserb76f38532011-08-20 11:11:25 +0000505 isl_set *strideOne = isl_set_from_basic_set(bset);
Tobias Grosser75805372011-04-29 06:27:02 +0000506
Tobias Grosserb76f38532011-08-20 11:11:25 +0000507 bool isStrideOne = isl_set_is_equal(stride, strideOne);
508
509 isl_set_free(strideOne);
510 isl_set_free(stride);
511
512 return isStrideOne;
Tobias Grosser75805372011-04-29 06:27:02 +0000513}
514
Tobias Grosser5d453812011-10-06 00:04:11 +0000515void MemoryAccess::setNewAccessRelation(isl_map *newAccess) {
Tobias Grosserb76f38532011-08-20 11:11:25 +0000516 isl_map_free(newAccessRelation);
Raghesh Aloor7a04f4f2011-08-03 13:47:59 +0000517 newAccessRelation = newAccess;
Raghesh Aloor3cb66282011-07-12 17:14:03 +0000518}
Tobias Grosser75805372011-04-29 06:27:02 +0000519
520//===----------------------------------------------------------------------===//
Tobias Grossercf3942d2011-10-06 00:04:05 +0000521
522isl_map *ScopStmt::getScattering() const {
523 return isl_map_copy(Scattering);
524}
525
526void ScopStmt::setScattering(isl_map *NewScattering) {
Tobias Grosserb76f38532011-08-20 11:11:25 +0000527 isl_map_free(Scattering);
Tobias Grossercf3942d2011-10-06 00:04:05 +0000528 Scattering = NewScattering;
Tobias Grosserb76f38532011-08-20 11:11:25 +0000529}
530
Tobias Grosser75805372011-04-29 06:27:02 +0000531void ScopStmt::buildScattering(SmallVectorImpl<unsigned> &Scatter) {
532 unsigned NumberOfIterators = getNumIterators();
Tobias Grosserf5338802011-10-06 00:03:35 +0000533 unsigned ScatSpace = Parent.getMaxLoopDepth() * 2 + 1;
Tobias Grosser3c69fab2011-10-06 00:03:54 +0000534 isl_space *Space = isl_space_alloc(getIslCtx(), 0, NumberOfIterators,
Tobias Grosserf5338802011-10-06 00:03:35 +0000535 ScatSpace);
536 Space = isl_space_set_tuple_name(Space, isl_dim_out, "scattering");
537 Space = isl_space_set_tuple_name(Space, isl_dim_in, getBaseName());
538 isl_local_space *LSpace = isl_local_space_from_space(isl_space_copy(Space));
539 isl_basic_map *bmap = isl_basic_map_universe(Space);
Tobias Grosser75805372011-04-29 06:27:02 +0000540 isl_int v;
541 isl_int_init(v);
542
543 // Loop dimensions.
544 for (unsigned i = 0; i < NumberOfIterators; ++i) {
Tobias Grosserf5338802011-10-06 00:03:35 +0000545 isl_constraint *c = isl_equality_alloc(isl_local_space_copy(LSpace));
Tobias Grosser75805372011-04-29 06:27:02 +0000546 isl_int_set_si(v, 1);
547 isl_constraint_set_coefficient(c, isl_dim_out, 2 * i + 1, v);
548 isl_int_set_si(v, -1);
549 isl_constraint_set_coefficient(c, isl_dim_in, i, v);
550
551 bmap = isl_basic_map_add_constraint(bmap, c);
552 }
553
554 // Constant dimensions
555 for (unsigned i = 0; i < NumberOfIterators + 1; ++i) {
Tobias Grosserf5338802011-10-06 00:03:35 +0000556 isl_constraint *c = isl_equality_alloc(isl_local_space_copy(LSpace));
Tobias Grosser75805372011-04-29 06:27:02 +0000557 isl_int_set_si(v, -1);
558 isl_constraint_set_coefficient(c, isl_dim_out, 2 * i, v);
559 isl_int_set_si(v, Scatter[i]);
560 isl_constraint_set_constant(c, v);
561
562 bmap = isl_basic_map_add_constraint(bmap, c);
563 }
564
565 // Fill scattering dimensions.
Tobias Grosserf5338802011-10-06 00:03:35 +0000566 for (unsigned i = 2 * NumberOfIterators + 1; i < ScatSpace ; ++i) {
567 isl_constraint *c = isl_equality_alloc(isl_local_space_copy(LSpace));
Tobias Grosser75805372011-04-29 06:27:02 +0000568 isl_int_set_si(v, 1);
569 isl_constraint_set_coefficient(c, isl_dim_out, i, v);
570 isl_int_set_si(v, 0);
571 isl_constraint_set_constant(c, v);
572
573 bmap = isl_basic_map_add_constraint(bmap, c);
574 }
575
576 isl_int_clear(v);
Tobias Grosser75805372011-04-29 06:27:02 +0000577 Scattering = isl_map_from_basic_map(bmap);
Tobias Grosser37487052011-10-06 00:03:42 +0000578 Scattering = isl_map_align_params(Scattering, Parent.getParamSpace());
Tobias Grosser0ad4caa2011-10-08 00:35:17 +0000579 isl_local_space_free(LSpace);
Tobias Grosser75805372011-04-29 06:27:02 +0000580}
581
582void ScopStmt::buildAccesses(TempScop &tempScop, const Region &CurRegion) {
583 const AccFuncSetType *AccFuncs = tempScop.getAccessFunctions(BB);
584
585 for (AccFuncSetType::const_iterator I = AccFuncs->begin(),
586 E = AccFuncs->end(); I != E; ++I) {
587 MemAccs.push_back(new MemoryAccess(I->first, this));
588 InstructionToAccess[I->second] = MemAccs.back();
589 }
590}
591
Tobias Grosser8cae72f2011-11-08 15:41:08 +0000592void ScopStmt::realignParams() {
593 for (memacc_iterator MI = memacc_begin(), ME = memacc_end(); MI != ME; ++MI)
594 (*MI)->realignParams();
595
596 Domain = isl_set_align_params(Domain, Parent.getParamSpace());
597 Scattering = isl_map_align_params(Scattering, Parent.getParamSpace());
598}
599
Tobias Grosser65b00582011-11-08 15:41:19 +0000600__isl_give isl_set *ScopStmt::buildConditionSet(const Comparison &Comp) {
Tobias Grossere19661e2011-10-07 08:46:57 +0000601
Tobias Grosser048c8792011-10-23 20:59:20 +0000602 isl_pw_aff *L = SCEVAffinator::getPwAff(this, Comp.getLHS()->OriginalSCEV, 0);
603 isl_pw_aff *R = SCEVAffinator::getPwAff(this, Comp.getRHS()->OriginalSCEV, 0);
Tobias Grosser75805372011-04-29 06:27:02 +0000604
Tobias Grosserd2795d02011-08-18 07:51:40 +0000605 switch (Comp.getPred()) {
Tobias Grosser75805372011-04-29 06:27:02 +0000606 case ICmpInst::ICMP_EQ:
Tobias Grosser048c8792011-10-23 20:59:20 +0000607 return isl_pw_aff_eq_set(L, R);
Tobias Grosser75805372011-04-29 06:27:02 +0000608 case ICmpInst::ICMP_NE:
Tobias Grosser048c8792011-10-23 20:59:20 +0000609 return isl_pw_aff_ne_set(L, R);
Tobias Grosser75805372011-04-29 06:27:02 +0000610 case ICmpInst::ICMP_SLT:
Tobias Grosser048c8792011-10-23 20:59:20 +0000611 return isl_pw_aff_lt_set(L, R);
Tobias Grosser75805372011-04-29 06:27:02 +0000612 case ICmpInst::ICMP_SLE:
Tobias Grosser048c8792011-10-23 20:59:20 +0000613 return isl_pw_aff_le_set(L, R);
Tobias Grosserd2795d02011-08-18 07:51:40 +0000614 case ICmpInst::ICMP_SGT:
Tobias Grosser048c8792011-10-23 20:59:20 +0000615 return isl_pw_aff_gt_set(L, R);
Tobias Grosser75805372011-04-29 06:27:02 +0000616 case ICmpInst::ICMP_SGE:
Tobias Grosser048c8792011-10-23 20:59:20 +0000617 return isl_pw_aff_ge_set(L, R);
Tobias Grosserd2795d02011-08-18 07:51:40 +0000618 case ICmpInst::ICMP_ULT:
619 case ICmpInst::ICMP_UGT:
620 case ICmpInst::ICMP_ULE:
Tobias Grosser75805372011-04-29 06:27:02 +0000621 case ICmpInst::ICMP_UGE:
Tobias Grosserd2795d02011-08-18 07:51:40 +0000622 llvm_unreachable("Unsigned comparisons not yet supported");
Tobias Grosser75805372011-04-29 06:27:02 +0000623 default:
624 llvm_unreachable("Non integer predicate not supported");
625 }
Tobias Grosser75805372011-04-29 06:27:02 +0000626}
627
Tobias Grossere19661e2011-10-07 08:46:57 +0000628__isl_give isl_set *ScopStmt::addLoopBoundsToDomain(__isl_take isl_set *Domain,
629 TempScop &tempScop) const {
630 isl_space *Space;
631 isl_local_space *LocalSpace;
Tobias Grosser75805372011-04-29 06:27:02 +0000632
Tobias Grossere19661e2011-10-07 08:46:57 +0000633 Space = isl_set_get_space(Domain);
634 LocalSpace = isl_local_space_from_space(Space);
Tobias Grosserf5338802011-10-06 00:03:35 +0000635
Tobias Grosser75805372011-04-29 06:27:02 +0000636 for (int i = 0, e = getNumIterators(); i != e; ++i) {
Tobias Grosser9b13d3d2011-10-06 22:32:58 +0000637 isl_aff *Zero = isl_aff_zero_on_domain(isl_local_space_copy(LocalSpace));
638 isl_pw_aff *IV = isl_pw_aff_from_aff(
639 isl_aff_set_coefficient_si(Zero, isl_dim_in, i, 1));
Tobias Grosser75805372011-04-29 06:27:02 +0000640
Tobias Grosser9b13d3d2011-10-06 22:32:58 +0000641 // 0 <= IV.
642 isl_set *LowerBound = isl_pw_aff_nonneg_set(isl_pw_aff_copy(IV));
643 Domain = isl_set_intersect(Domain, LowerBound);
644
645 // IV <= LatchExecutions.
Hongbin Zheng27f3afb2011-04-30 03:26:51 +0000646 const Loop *L = getLoopForDimension(i);
Tobias Grosser1179afa2011-11-02 21:37:51 +0000647 const SCEV *LatchExecutions = tempScop.getLoopBound(L);
Tobias Grosser9b13d3d2011-10-06 22:32:58 +0000648 isl_pw_aff *UpperBound = SCEVAffinator::getPwAff(this, LatchExecutions);
649 isl_set *UpperBoundSet = isl_pw_aff_le_set(IV, UpperBound);
Tobias Grosser75805372011-04-29 06:27:02 +0000650 Domain = isl_set_intersect(Domain, UpperBoundSet);
651 }
652
Tobias Grosserf5338802011-10-06 00:03:35 +0000653 isl_local_space_free(LocalSpace);
Tobias Grossere19661e2011-10-07 08:46:57 +0000654 return Domain;
Tobias Grosser75805372011-04-29 06:27:02 +0000655}
656
Tobias Grossere19661e2011-10-07 08:46:57 +0000657__isl_give isl_set *ScopStmt::addConditionsToDomain(__isl_take isl_set *Domain,
658 TempScop &tempScop,
Tobias Grosser65b00582011-11-08 15:41:19 +0000659 const Region &CurRegion) {
Tobias Grossere19661e2011-10-07 08:46:57 +0000660 const Region *TopRegion = tempScop.getMaxRegion().getParent(),
661 *CurrentRegion = &CurRegion;
662 const BasicBlock *BranchingBB = BB;
Tobias Grosser75805372011-04-29 06:27:02 +0000663
Tobias Grosser75805372011-04-29 06:27:02 +0000664 do {
Tobias Grossere19661e2011-10-07 08:46:57 +0000665 if (BranchingBB != CurrentRegion->getEntry()) {
666 if (const BBCond *Condition = tempScop.getBBCond(BranchingBB))
667 for (BBCond::const_iterator CI = Condition->begin(),
668 CE = Condition->end(); CI != CE; ++CI) {
Tobias Grosser048c8792011-10-23 20:59:20 +0000669 isl_set *ConditionSet = buildConditionSet(*CI);
Tobias Grossere19661e2011-10-07 08:46:57 +0000670 Domain = isl_set_intersect(Domain, ConditionSet);
Tobias Grosser75805372011-04-29 06:27:02 +0000671 }
672 }
Tobias Grossere19661e2011-10-07 08:46:57 +0000673 BranchingBB = CurrentRegion->getEntry();
674 CurrentRegion = CurrentRegion->getParent();
675 } while (TopRegion != CurrentRegion);
Tobias Grosser75805372011-04-29 06:27:02 +0000676
Tobias Grossere19661e2011-10-07 08:46:57 +0000677 return Domain;
Tobias Grosser75805372011-04-29 06:27:02 +0000678}
679
Tobias Grossere19661e2011-10-07 08:46:57 +0000680__isl_give isl_set *ScopStmt::buildDomain(TempScop &tempScop,
Tobias Grosser65b00582011-11-08 15:41:19 +0000681 const Region &CurRegion) {
Tobias Grossere19661e2011-10-07 08:46:57 +0000682 isl_space *Space;
683 isl_set *Domain;
684
685 Space = isl_space_set_alloc(getIslCtx(), 0, getNumIterators());
686
687 Domain = isl_set_universe(Space);
Tobias Grossere19661e2011-10-07 08:46:57 +0000688 Domain = addLoopBoundsToDomain(Domain, tempScop);
689 Domain = addConditionsToDomain(Domain, tempScop, CurRegion);
690 Domain = isl_set_set_tuple_name(Domain, getBaseName());
691
692 return Domain;
Tobias Grosser75805372011-04-29 06:27:02 +0000693}
694
695ScopStmt::ScopStmt(Scop &parent, TempScop &tempScop,
696 const Region &CurRegion, BasicBlock &bb,
697 SmallVectorImpl<Loop*> &NestLoops,
698 SmallVectorImpl<unsigned> &Scatter)
699 : Parent(parent), BB(&bb), IVS(NestLoops.size()) {
700 // Setup the induction variables.
701 for (unsigned i = 0, e = NestLoops.size(); i < e; ++i) {
702 PHINode *PN = NestLoops[i]->getCanonicalInductionVariable();
703 assert(PN && "Non canonical IV in Scop!");
Hongbin Zheng27f3afb2011-04-30 03:26:51 +0000704 IVS[i] = std::make_pair(PN, NestLoops[i]);
Tobias Grosser75805372011-04-29 06:27:02 +0000705 }
706
707 raw_string_ostream OS(BaseName);
708 WriteAsOperand(OS, &bb, false);
709 BaseName = OS.str();
710
Tobias Grosser75805372011-04-29 06:27:02 +0000711 makeIslCompatible(BaseName);
712 BaseName = "Stmt_" + BaseName;
713
Tobias Grossere19661e2011-10-07 08:46:57 +0000714 Domain = buildDomain(tempScop, CurRegion);
Tobias Grosser75805372011-04-29 06:27:02 +0000715 buildScattering(Scatter);
716 buildAccesses(tempScop, CurRegion);
Tobias Grosser75805372011-04-29 06:27:02 +0000717}
718
719ScopStmt::ScopStmt(Scop &parent, SmallVectorImpl<unsigned> &Scatter)
720 : Parent(parent), BB(NULL), IVS(0) {
721
722 BaseName = "FinalRead";
723
724 // Build iteration domain.
725 std::string IterationDomainString = "{[i0] : i0 = 0}";
Tobias Grosser3c69fab2011-10-06 00:03:54 +0000726 Domain = isl_set_read_from_str(getIslCtx(), IterationDomainString.c_str());
Tobias Grosser75805372011-04-29 06:27:02 +0000727 Domain = isl_set_set_tuple_name(Domain, getBaseName());
728
729 // Build scattering.
Tobias Grosserf5338802011-10-06 00:03:35 +0000730 unsigned ScatSpace = Parent.getMaxLoopDepth() * 2 + 1;
Tobias Grosser3c69fab2011-10-06 00:03:54 +0000731 isl_space *Space = isl_space_alloc(getIslCtx(), 0, 1, ScatSpace);
Tobias Grosserf5338802011-10-06 00:03:35 +0000732 Space = isl_space_set_tuple_name(Space, isl_dim_out, "scattering");
733 Space = isl_space_set_tuple_name(Space, isl_dim_in, getBaseName());
734 isl_basic_map *bmap = isl_basic_map_universe(isl_space_copy(Space));
Tobias Grosser75805372011-04-29 06:27:02 +0000735 isl_int v;
736 isl_int_init(v);
737
Tobias Grosserf5338802011-10-06 00:03:35 +0000738 isl_constraint *c = isl_equality_alloc(isl_local_space_from_space(Space));
Tobias Grosser75805372011-04-29 06:27:02 +0000739 isl_int_set_si(v, -1);
740 isl_constraint_set_coefficient(c, isl_dim_out, 0, v);
741
742 // TODO: This is incorrect. We should not use a very large number to ensure
743 // that this statement is executed last.
744 isl_int_set_si(v, 200000000);
745 isl_constraint_set_constant(c, v);
746
747 bmap = isl_basic_map_add_constraint(bmap, c);
748 isl_int_clear(v);
749 Scattering = isl_map_from_basic_map(bmap);
750
751 // Build memory accesses, use SetVector to keep the order of memory accesses
752 // and prevent the same memory access inserted more than once.
753 SetVector<const Value*> BaseAddressSet;
754
755 for (Scop::const_iterator SI = Parent.begin(), SE = Parent.end(); SI != SE;
756 ++SI) {
757 ScopStmt *Stmt = *SI;
758
759 for (MemoryAccessVec::const_iterator I = Stmt->memacc_begin(),
760 E = Stmt->memacc_end(); I != E; ++I)
761 BaseAddressSet.insert((*I)->getBaseAddr());
762 }
763
764 for (SetVector<const Value*>::iterator BI = BaseAddressSet.begin(),
765 BE = BaseAddressSet.end(); BI != BE; ++BI)
766 MemAccs.push_back(new MemoryAccess(*BI, this));
Tobias Grosser75805372011-04-29 06:27:02 +0000767}
768
769std::string ScopStmt::getDomainStr() const {
Tobias Grosser4da8d9f2011-10-06 00:03:59 +0000770 return stringFromIslObj(Domain);
Tobias Grosser75805372011-04-29 06:27:02 +0000771}
772
773std::string ScopStmt::getScatteringStr() const {
Tobias Grossercf3942d2011-10-06 00:04:05 +0000774 return stringFromIslObj(Scattering);
Tobias Grosser75805372011-04-29 06:27:02 +0000775}
776
777unsigned ScopStmt::getNumParams() const {
778 return Parent.getNumParams();
779}
780
781unsigned ScopStmt::getNumIterators() const {
782 // The final read has one dimension with one element.
783 if (!BB)
784 return 1;
785
786 return IVS.size();
787}
788
789unsigned ScopStmt::getNumScattering() const {
790 return isl_map_dim(Scattering, isl_dim_out);
791}
792
793const char *ScopStmt::getBaseName() const { return BaseName.c_str(); }
794
795const PHINode *ScopStmt::getInductionVariableForDimension(unsigned Dimension)
796 const {
Hongbin Zheng27f3afb2011-04-30 03:26:51 +0000797 return IVS[Dimension].first;
798}
799
800const Loop *ScopStmt::getLoopForDimension(unsigned Dimension) const {
801 return IVS[Dimension].second;
Tobias Grosser75805372011-04-29 06:27:02 +0000802}
803
804const SCEVAddRecExpr *ScopStmt::getSCEVForDimension(unsigned Dimension)
805 const {
Hongbin Zheng27f3afb2011-04-30 03:26:51 +0000806 PHINode *PN =
807 const_cast<PHINode*>(getInductionVariableForDimension(Dimension));
Tobias Grosser75805372011-04-29 06:27:02 +0000808 return cast<SCEVAddRecExpr>(getParent()->getSE()->getSCEV(PN));
809}
810
Tobias Grosser3c69fab2011-10-06 00:03:54 +0000811isl_ctx *ScopStmt::getIslCtx() const {
812 return Parent.getIslCtx();
Tobias Grosser75805372011-04-29 06:27:02 +0000813}
814
Tobias Grosserd5a7bfc2011-05-06 19:52:19 +0000815isl_set *ScopStmt::getDomain() const {
816 return isl_set_copy(Domain);
817}
818
Tobias Grosser75805372011-04-29 06:27:02 +0000819ScopStmt::~ScopStmt() {
820 while (!MemAccs.empty()) {
821 delete MemAccs.back();
822 MemAccs.pop_back();
823 }
824
825 isl_set_free(Domain);
826 isl_map_free(Scattering);
827}
828
829void ScopStmt::print(raw_ostream &OS) const {
830 OS << "\t" << getBaseName() << "\n";
831
832 OS.indent(12) << "Domain :=\n";
833
834 if (Domain) {
835 OS.indent(16) << getDomainStr() << ";\n";
836 } else
837 OS.indent(16) << "n/a\n";
838
839 OS.indent(12) << "Scattering :=\n";
840
841 if (Domain) {
842 OS.indent(16) << getScatteringStr() << ";\n";
843 } else
844 OS.indent(16) << "n/a\n";
845
846 for (MemoryAccessVec::const_iterator I = MemAccs.begin(), E = MemAccs.end();
847 I != E; ++I)
848 (*I)->print(OS);
849}
850
851void ScopStmt::dump() const { print(dbgs()); }
852
853//===----------------------------------------------------------------------===//
854/// Scop class implement
Tobias Grosser9a38ab82011-11-08 15:41:03 +0000855__isl_give isl_id *Scop::getIdForParam(const SCEV *Parameter) const {
856 ParamIdType::const_iterator IdIter = ParameterIds.find(Parameter);
Tobias Grosser76c2e322011-11-07 12:58:59 +0000857
Tobias Grosser9a38ab82011-11-08 15:41:03 +0000858 if (IdIter == ParameterIds.end())
859 return NULL;
Tobias Grosser76c2e322011-11-07 12:58:59 +0000860
Tobias Grosser9a38ab82011-11-08 15:41:03 +0000861 std::string ParameterName = "p" + convertInt(IdIter->second);
862 return isl_id_alloc(getIslCtx(), ParameterName.c_str(), (void *) Parameter);
Tobias Grosser76c2e322011-11-07 12:58:59 +0000863}
Tobias Grosser75805372011-04-29 06:27:02 +0000864
Tobias Grosser6be480c2011-11-08 15:41:13 +0000865void Scop::initializeParameters(ParamSetType *ParamSet) {
Tobias Grosser30b8a092011-08-18 07:51:37 +0000866 int i = 0;
Tobias Grosser0e27e242011-10-06 00:03:48 +0000867 for (ParamSetType::iterator PI = ParamSet->begin(), PE = ParamSet->end();
Tobias Grosser30b8a092011-08-18 07:51:37 +0000868 PI != PE; ++PI) {
Tobias Grosser0e27e242011-10-06 00:03:48 +0000869 const SCEV *Parameter = *PI;
870 Parameters.push_back(Parameter);
Tobias Grosser9a38ab82011-11-08 15:41:03 +0000871 ParameterIds.insert(std::pair<const SCEV*, int>(Parameter, i));
Tobias Grosser30b8a092011-08-18 07:51:37 +0000872 i++;
873 }
Tobias Grosser6be480c2011-11-08 15:41:13 +0000874}
Tobias Grosser30b8a092011-08-18 07:51:37 +0000875
Tobias Grosser6be480c2011-11-08 15:41:13 +0000876void Scop::buildContext() {
877 isl_space *Space = isl_space_params_alloc(IslCtx, 0);
Tobias Grosserf5338802011-10-06 00:03:35 +0000878 Context = isl_set_universe (Space);
Tobias Grosser0e27e242011-10-06 00:03:48 +0000879}
880
Tobias Grosser8cae72f2011-11-08 15:41:08 +0000881void Scop::realignParams() {
Tobias Grosser6be480c2011-11-08 15:41:13 +0000882 // Add all parameters into a common model.
883 isl_space *Space = isl_space_params_alloc(IslCtx, Parameters.size());
884
885 for (ParamIdType::iterator PI = ParameterIds.begin(), PE = ParameterIds.end();
886 PI != PE; ++PI) {
887 const SCEV *Parameter = PI->first;
888 isl_id *id = getIdForParam(Parameter);
889 Space = isl_space_set_dim_id(Space, isl_dim_param, PI->second, id);
890 }
891
892 // Align the parameters of all data structures to the model.
893 Context = isl_set_align_params(Context, Space);
894
Tobias Grosser8cae72f2011-11-08 15:41:08 +0000895 for (iterator I = begin(), E = end(); I != E; ++I)
896 (*I)->realignParams();
897}
898
Tobias Grosser0e27e242011-10-06 00:03:48 +0000899Scop::Scop(TempScop &tempScop, LoopInfo &LI, ScalarEvolution &ScalarEvolution,
900 isl_ctx *Context)
901 : SE(&ScalarEvolution), R(tempScop.getMaxRegion()),
902 MaxLoopDepth(tempScop.getMaxLoopDepth()) {
Tobias Grosser9a38ab82011-11-08 15:41:03 +0000903 IslCtx = Context;
Tobias Grosser6be480c2011-11-08 15:41:13 +0000904 initializeParameters(&tempScop.getParamSet());
905 buildContext();
Tobias Grosser75805372011-04-29 06:27:02 +0000906
907 SmallVector<Loop*, 8> NestLoops;
908 SmallVector<unsigned, 8> Scatter;
909
910 Scatter.assign(MaxLoopDepth + 1, 0);
911
912 // Build the iteration domain, access functions and scattering functions
913 // traversing the region tree.
914 buildScop(tempScop, getRegion(), NestLoops, Scatter, LI);
915 Stmts.push_back(new ScopStmt(*this, Scatter));
916
Tobias Grosser8cae72f2011-11-08 15:41:08 +0000917 realignParams();
918
Tobias Grosser75805372011-04-29 06:27:02 +0000919 assert(NestLoops.empty() && "NestLoops not empty at top level!");
920}
921
922Scop::~Scop() {
923 isl_set_free(Context);
924
925 // Free the statements;
926 for (iterator I = begin(), E = end(); I != E; ++I)
927 delete *I;
Tobias Grosser75805372011-04-29 06:27:02 +0000928}
929
930std::string Scop::getContextStr() const {
Tobias Grosser4da8d9f2011-10-06 00:03:59 +0000931 return stringFromIslObj(Context);
Tobias Grosser75805372011-04-29 06:27:02 +0000932}
933
934std::string Scop::getNameStr() const {
935 std::string ExitName, EntryName;
936 raw_string_ostream ExitStr(ExitName);
937 raw_string_ostream EntryStr(EntryName);
938
939 WriteAsOperand(EntryStr, R.getEntry(), false);
940 EntryStr.str();
941
942 if (R.getExit()) {
943 WriteAsOperand(ExitStr, R.getExit(), false);
944 ExitStr.str();
945 } else
946 ExitName = "FunctionExit";
947
948 return EntryName + "---" + ExitName;
949}
950
Tobias Grosser4da8d9f2011-10-06 00:03:59 +0000951__isl_give isl_set *Scop::getContext() const {
952 return isl_set_copy(Context);
953}
Tobias Grosser37487052011-10-06 00:03:42 +0000954__isl_give isl_space *Scop::getParamSpace() const {
955 return isl_set_get_space(this->Context);
956}
957
Tobias Grosser75805372011-04-29 06:27:02 +0000958void Scop::printContext(raw_ostream &OS) const {
959 OS << "Context:\n";
960
961 if (!Context) {
962 OS.indent(4) << "n/a\n\n";
963 return;
964 }
965
966 OS.indent(4) << getContextStr() << "\n";
967}
968
969void Scop::printStatements(raw_ostream &OS) const {
970 OS << "Statements {\n";
971
972 for (const_iterator SI = begin(), SE = end();SI != SE; ++SI)
973 OS.indent(4) << (**SI);
974
975 OS.indent(4) << "}\n";
976}
977
978
979void Scop::print(raw_ostream &OS) const {
980 printContext(OS.indent(4));
981 printStatements(OS.indent(4));
982}
983
984void Scop::dump() const { print(dbgs()); }
985
Tobias Grosser9a38ab82011-11-08 15:41:03 +0000986isl_ctx *Scop::getIslCtx() const { return IslCtx; }
Tobias Grosser75805372011-04-29 06:27:02 +0000987
988ScalarEvolution *Scop::getSE() const { return SE; }
989
990bool Scop::isTrivialBB(BasicBlock *BB, TempScop &tempScop) {
991 if (tempScop.getAccessFunctions(BB))
992 return false;
993
994 return true;
995}
996
997void Scop::buildScop(TempScop &tempScop,
998 const Region &CurRegion,
999 SmallVectorImpl<Loop*> &NestLoops,
1000 SmallVectorImpl<unsigned> &Scatter,
1001 LoopInfo &LI) {
1002 Loop *L = castToLoop(CurRegion, LI);
1003
1004 if (L)
1005 NestLoops.push_back(L);
1006
1007 unsigned loopDepth = NestLoops.size();
1008 assert(Scatter.size() > loopDepth && "Scatter not big enough!");
1009
1010 for (Region::const_element_iterator I = CurRegion.element_begin(),
1011 E = CurRegion.element_end(); I != E; ++I)
1012 if (I->isSubRegion())
1013 buildScop(tempScop, *(I->getNodeAs<Region>()), NestLoops, Scatter, LI);
1014 else {
1015 BasicBlock *BB = I->getNodeAs<BasicBlock>();
1016
1017 if (isTrivialBB(BB, tempScop))
1018 continue;
1019
1020 Stmts.push_back(new ScopStmt(*this, tempScop, CurRegion, *BB, NestLoops,
1021 Scatter));
1022
1023 // Increasing the Scattering function is OK for the moment, because
1024 // we are using a depth first iterator and the program is well structured.
1025 ++Scatter[loopDepth];
1026 }
1027
1028 if (!L)
1029 return;
1030
1031 // Exiting a loop region.
1032 Scatter[loopDepth] = 0;
1033 NestLoops.pop_back();
1034 ++Scatter[loopDepth-1];
1035}
1036
1037//===----------------------------------------------------------------------===//
Tobias Grosserb76f38532011-08-20 11:11:25 +00001038ScopInfo::ScopInfo() : RegionPass(ID), scop(0) {
1039 ctx = isl_ctx_alloc();
1040}
1041
1042ScopInfo::~ScopInfo() {
1043 clear();
1044 isl_ctx_free(ctx);
1045}
1046
1047
Tobias Grosser75805372011-04-29 06:27:02 +00001048
1049void ScopInfo::getAnalysisUsage(AnalysisUsage &AU) const {
1050 AU.addRequired<LoopInfo>();
1051 AU.addRequired<RegionInfo>();
1052 AU.addRequired<ScalarEvolution>();
1053 AU.addRequired<TempScopInfo>();
1054 AU.setPreservesAll();
1055}
1056
1057bool ScopInfo::runOnRegion(Region *R, RGPassManager &RGM) {
1058 LoopInfo &LI = getAnalysis<LoopInfo>();
1059 ScalarEvolution &SE = getAnalysis<ScalarEvolution>();
1060
1061 TempScop *tempScop = getAnalysis<TempScopInfo>().getTempScop(R);
1062
1063 // This region is no Scop.
1064 if (!tempScop) {
1065 scop = 0;
1066 return false;
1067 }
1068
1069 // Statistics.
1070 ++ScopFound;
1071 if (tempScop->getMaxLoopDepth() > 0) ++RichScopFound;
1072
Tobias Grosserb76f38532011-08-20 11:11:25 +00001073 scop = new Scop(*tempScop, LI, SE, ctx);
Tobias Grosser75805372011-04-29 06:27:02 +00001074
1075 return false;
1076}
1077
1078char ScopInfo::ID = 0;
1079
Tobias Grosser73600b82011-10-08 00:30:40 +00001080INITIALIZE_PASS_BEGIN(ScopInfo, "polly-scops",
1081 "Polly - Create polyhedral description of Scops", false,
1082 false)
1083INITIALIZE_PASS_DEPENDENCY(LoopInfo)
1084INITIALIZE_PASS_DEPENDENCY(RegionInfo)
1085INITIALIZE_PASS_DEPENDENCY(ScalarEvolution)
1086INITIALIZE_PASS_DEPENDENCY(TempScopInfo)
1087INITIALIZE_PASS_END(ScopInfo, "polly-scops",
1088 "Polly - Create polyhedral description of Scops", false,
1089 false)
Tobias Grosser75805372011-04-29 06:27:02 +00001090
1091Pass *polly::createScopInfoPass() {
1092 return new ScopInfo();
1093}