f4e201e09f
NonowningIslPtr<isl_X> was used as types of function parameters when the function does not consume the isl object, i.e. an __isl_keep parameter. The alternatives are: 1. IslPtr<isl_X> This has additional calls to isl_X_copy and isl_X_free to increase/decrease the reference counter even though not needed. The caller already owns a reference to the isl object. 2. const IslPtr<isl_X>& This does not change the reference counter, but requires an additional load to get the pointer to the isl object (instead of just passing the pointer itself). Moreover, the compiler cannot rely on the constness of the pointer and has to reload the pointer every time it writes to memory (unless alias analysis such as TBAA says it is not possible). The isl C++ bindings currently in development do not have an equivalent to NonowningIslPtr and adding one would make the binding more complicated and its advantage in performance is small. In order to simplify the transition to these C++ bindings, remove NonowningIslPtr. Change every former use of it to alternative 2 mentioned aboce (const IslPtr<isl_X>&). llvm-svn: 295998
109 lines
3.4 KiB
C++
109 lines
3.4 KiB
C++
//===------ FlattenSchedule.cpp --------------------------------*- C++ -*-===//
|
|
//
|
|
// The LLVM Compiler Infrastructure
|
|
//
|
|
// This file is distributed under the University of Illinois Open Source
|
|
// License. See LICENSE.TXT for details.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
//
|
|
// Try to reduce the number of scatter dimension. Useful to make isl_union_map
|
|
// schedules more understandable. This is only intended for debugging and
|
|
// unittests, not for production use.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include "polly/FlattenSchedule.h"
|
|
#include "polly/FlattenAlgo.h"
|
|
#include "polly/ScopInfo.h"
|
|
#include "polly/ScopPass.h"
|
|
#define DEBUG_TYPE "polly-flatten-schedule"
|
|
|
|
using namespace polly;
|
|
using namespace llvm;
|
|
|
|
namespace {
|
|
|
|
/// Print a schedule to @p OS.
|
|
///
|
|
/// Prints the schedule for each statements on a new line.
|
|
void printSchedule(raw_ostream &OS, const IslPtr<isl_union_map> &Schedule,
|
|
int indent) {
|
|
foreachElt(Schedule, [&OS, indent](IslPtr<isl_map> Map) {
|
|
OS.indent(indent) << Map << "\n";
|
|
});
|
|
}
|
|
|
|
/// Flatten the schedule stored in an polly::Scop.
|
|
class FlattenSchedule : public ScopPass {
|
|
private:
|
|
FlattenSchedule(const FlattenSchedule &) = delete;
|
|
const FlattenSchedule &operator=(const FlattenSchedule &) = delete;
|
|
|
|
std::shared_ptr<isl_ctx> IslCtx;
|
|
IslPtr<isl_union_map> OldSchedule;
|
|
|
|
public:
|
|
static char ID;
|
|
explicit FlattenSchedule() : ScopPass(ID) {}
|
|
|
|
virtual void getAnalysisUsage(AnalysisUsage &AU) const override {
|
|
AU.addRequiredTransitive<ScopInfoRegionPass>();
|
|
AU.setPreservesAll();
|
|
}
|
|
|
|
virtual bool runOnScop(Scop &S) override {
|
|
// Keep a reference to isl_ctx to ensure that it is not freed before we free
|
|
// OldSchedule.
|
|
IslCtx = S.getSharedIslCtx();
|
|
|
|
DEBUG(dbgs() << "Going to flatten old schedule:\n");
|
|
OldSchedule = give(S.getSchedule());
|
|
DEBUG(printSchedule(dbgs(), OldSchedule, 2));
|
|
|
|
auto Domains = give(S.getDomains());
|
|
auto RestrictedOldSchedule = give(
|
|
isl_union_map_intersect_domain(OldSchedule.copy(), Domains.copy()));
|
|
DEBUG(dbgs() << "Old schedule with domains:\n");
|
|
DEBUG(printSchedule(dbgs(), RestrictedOldSchedule, 2));
|
|
|
|
auto NewSchedule = flattenSchedule(RestrictedOldSchedule);
|
|
|
|
DEBUG(dbgs() << "Flattened new schedule:\n");
|
|
DEBUG(printSchedule(dbgs(), NewSchedule, 2));
|
|
|
|
NewSchedule =
|
|
give(isl_union_map_gist_domain(NewSchedule.take(), Domains.take()));
|
|
DEBUG(dbgs() << "Gisted, flattened new schedule:\n");
|
|
DEBUG(printSchedule(dbgs(), NewSchedule, 2));
|
|
|
|
S.setSchedule(NewSchedule.take());
|
|
return false;
|
|
}
|
|
|
|
virtual void printScop(raw_ostream &OS, Scop &S) const override {
|
|
OS << "Schedule before flattening {\n";
|
|
printSchedule(OS, OldSchedule, 4);
|
|
OS << "}\n\n";
|
|
|
|
OS << "Schedule after flattening {\n";
|
|
printSchedule(OS, give(S.getSchedule()), 4);
|
|
OS << "}\n";
|
|
}
|
|
|
|
virtual void releaseMemory() override {
|
|
OldSchedule = nullptr;
|
|
IslCtx.reset();
|
|
}
|
|
};
|
|
|
|
char FlattenSchedule::ID;
|
|
} // anonymous namespace
|
|
|
|
Pass *polly::createFlattenSchedulePass() { return new FlattenSchedule(); }
|
|
|
|
INITIALIZE_PASS_BEGIN(FlattenSchedule, "polly-flatten-schedule",
|
|
"Polly - Flatten schedule", false, false)
|
|
INITIALIZE_PASS_END(FlattenSchedule, "polly-flatten-schedule",
|
|
"Polly - Flatten schedule", false, false)
|