forked from OSchip/llvm-project
383 lines
13 KiB
C++
383 lines
13 KiB
C++
//===-- JSONExporter.cpp - Export Scops as JSON -------------------------===//
|
|
//
|
|
// The LLVM Compiler Infrastructure
|
|
//
|
|
// This file is distributed under the University of Illinois Open Source
|
|
// License. See LICENSE.TXT for details.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
//
|
|
// Export the Scops build by ScopInfo pass as a JSON file.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include "polly/LinkAllPasses.h"
|
|
#include "polly/DependenceInfo.h"
|
|
#include "polly/Options.h"
|
|
#include "polly/ScopInfo.h"
|
|
#include "polly/ScopPass.h"
|
|
|
|
#include "llvm/ADT/Statistic.h"
|
|
#include "llvm/Analysis/RegionInfo.h"
|
|
#include "llvm/IR/Module.h"
|
|
#include "llvm/Support/FileSystem.h"
|
|
#include "llvm/Support/MemoryBuffer.h"
|
|
#include "llvm/Support/ToolOutputFile.h"
|
|
|
|
#include "json/reader.h"
|
|
#include "json/writer.h"
|
|
|
|
#include "isl/set.h"
|
|
#include "isl/map.h"
|
|
#include "isl/constraint.h"
|
|
#include "isl/printer.h"
|
|
|
|
#include <memory>
|
|
#include <string>
|
|
#include <system_error>
|
|
|
|
using namespace llvm;
|
|
using namespace polly;
|
|
|
|
#define DEBUG_TYPE "polly-import-jscop"
|
|
|
|
STATISTIC(NewAccessMapFound, "Number of updated access functions");
|
|
|
|
namespace {
|
|
static cl::opt<std::string>
|
|
ImportDir("polly-import-jscop-dir",
|
|
cl::desc("The directory to import the .jscop files from."),
|
|
cl::Hidden, cl::value_desc("Directory path"), cl::ValueRequired,
|
|
cl::init("."), cl::cat(PollyCategory));
|
|
|
|
static cl::opt<std::string>
|
|
ImportPostfix("polly-import-jscop-postfix",
|
|
cl::desc("Postfix to append to the import .jsop files."),
|
|
cl::Hidden, cl::value_desc("File postfix"), cl::ValueRequired,
|
|
cl::init(""), cl::cat(PollyCategory));
|
|
|
|
struct JSONExporter : public ScopPass {
|
|
static char ID;
|
|
explicit JSONExporter() : ScopPass(ID) {}
|
|
|
|
std::string getFileName(Scop &S) const;
|
|
Json::Value getJSON(Scop &S) const;
|
|
virtual bool runOnScop(Scop &S);
|
|
void printScop(raw_ostream &OS, Scop &S) const;
|
|
void getAnalysisUsage(AnalysisUsage &AU) const;
|
|
};
|
|
|
|
struct JSONImporter : public ScopPass {
|
|
static char ID;
|
|
std::vector<std::string> newAccessStrings;
|
|
explicit JSONImporter() : ScopPass(ID) {}
|
|
|
|
std::string getFileName(Scop &S) const;
|
|
virtual bool runOnScop(Scop &S);
|
|
void printScop(raw_ostream &OS, Scop &S) const;
|
|
void getAnalysisUsage(AnalysisUsage &AU) const;
|
|
};
|
|
}
|
|
|
|
char JSONExporter::ID = 0;
|
|
std::string JSONExporter::getFileName(Scop &S) const {
|
|
std::string FunctionName = S.getRegion().getEntry()->getParent()->getName();
|
|
std::string FileName = FunctionName + "___" + S.getNameStr() + ".jscop";
|
|
return FileName;
|
|
}
|
|
|
|
void JSONExporter::printScop(raw_ostream &OS, Scop &S) const { S.print(OS); }
|
|
|
|
Json::Value JSONExporter::getJSON(Scop &S) const {
|
|
Json::Value root;
|
|
|
|
root["name"] = S.getRegion().getNameStr();
|
|
root["context"] = S.getContextStr();
|
|
root["statements"];
|
|
|
|
for (Scop::iterator SI = S.begin(), SE = S.end(); SI != SE; ++SI) {
|
|
ScopStmt *Stmt = *SI;
|
|
|
|
Json::Value statement;
|
|
|
|
statement["name"] = Stmt->getBaseName();
|
|
statement["domain"] = Stmt->getDomainStr();
|
|
statement["schedule"] = Stmt->getScatteringStr();
|
|
statement["accesses"];
|
|
|
|
for (MemoryAccess *MA : *Stmt) {
|
|
Json::Value access;
|
|
|
|
access["kind"] = MA->isRead() ? "read" : "write";
|
|
access["relation"] = MA->getOriginalAccessRelationStr();
|
|
|
|
statement["accesses"].append(access);
|
|
}
|
|
|
|
root["statements"].append(statement);
|
|
}
|
|
|
|
return root;
|
|
}
|
|
|
|
bool JSONExporter::runOnScop(Scop &S) {
|
|
Region &R = S.getRegion();
|
|
|
|
std::string FileName = ImportDir + "/" + getFileName(S);
|
|
|
|
Json::Value jscop = getJSON(S);
|
|
Json::StyledWriter writer;
|
|
std::string fileContent = writer.write(jscop);
|
|
|
|
// Write to file.
|
|
std::error_code EC;
|
|
tool_output_file F(FileName, EC, llvm::sys::fs::F_Text);
|
|
|
|
std::string FunctionName = R.getEntry()->getParent()->getName();
|
|
errs() << "Writing JScop '" << R.getNameStr() << "' in function '"
|
|
<< FunctionName << "' to '" << FileName << "'.\n";
|
|
|
|
if (!EC) {
|
|
F.os() << fileContent;
|
|
F.os().close();
|
|
if (!F.os().has_error()) {
|
|
errs() << "\n";
|
|
F.keep();
|
|
return false;
|
|
}
|
|
}
|
|
|
|
errs() << " error opening file for writing!\n";
|
|
F.os().clear_error();
|
|
|
|
return false;
|
|
}
|
|
|
|
void JSONExporter::getAnalysisUsage(AnalysisUsage &AU) const {
|
|
AU.setPreservesAll();
|
|
AU.addRequired<ScopInfo>();
|
|
}
|
|
|
|
Pass *polly::createJSONExporterPass() { return new JSONExporter(); }
|
|
|
|
char JSONImporter::ID = 0;
|
|
std::string JSONImporter::getFileName(Scop &S) const {
|
|
std::string FunctionName = S.getRegion().getEntry()->getParent()->getName();
|
|
std::string FileName = FunctionName + "___" + S.getNameStr() + ".jscop";
|
|
|
|
if (ImportPostfix != "")
|
|
FileName += "." + ImportPostfix;
|
|
|
|
return FileName;
|
|
}
|
|
|
|
void JSONImporter::printScop(raw_ostream &OS, Scop &S) const {
|
|
S.print(OS);
|
|
for (std::vector<std::string>::const_iterator I = newAccessStrings.begin(),
|
|
E = newAccessStrings.end();
|
|
I != E; I++)
|
|
OS << "New access function '" << *I << "'detected in JSCOP file\n";
|
|
}
|
|
|
|
typedef Dependences::StatementToIslMapTy StatementToIslMapTy;
|
|
|
|
bool JSONImporter::runOnScop(Scop &S) {
|
|
Region &R = S.getRegion();
|
|
const Dependences &D = getAnalysis<DependenceInfo>().getDependences();
|
|
const DataLayout &DL =
|
|
S.getRegion().getEntry()->getParent()->getParent()->getDataLayout();
|
|
|
|
std::string FileName = ImportDir + "/" + getFileName(S);
|
|
|
|
std::string FunctionName = R.getEntry()->getParent()->getName();
|
|
errs() << "Reading JScop '" << R.getNameStr() << "' in function '"
|
|
<< FunctionName << "' from '" << FileName << "'.\n";
|
|
ErrorOr<std::unique_ptr<MemoryBuffer>> result =
|
|
MemoryBuffer::getFile(FileName);
|
|
std::error_code ec = result.getError();
|
|
|
|
if (ec) {
|
|
errs() << "File could not be read: " << ec.message() << "\n";
|
|
return false;
|
|
}
|
|
|
|
Json::Reader reader;
|
|
Json::Value jscop;
|
|
|
|
bool parsingSuccessful = reader.parse(result.get()->getBufferStart(), jscop);
|
|
|
|
if (!parsingSuccessful) {
|
|
errs() << "JSCoP file could not be parsed\n";
|
|
return false;
|
|
}
|
|
|
|
isl_set *OldContext = S.getContext();
|
|
isl_set *NewContext =
|
|
isl_set_read_from_str(S.getIslCtx(), jscop["context"].asCString());
|
|
|
|
for (unsigned i = 0; i < isl_set_dim(OldContext, isl_dim_param); i++) {
|
|
isl_id *id = isl_set_get_dim_id(OldContext, isl_dim_param, i);
|
|
NewContext = isl_set_set_dim_id(NewContext, isl_dim_param, i, id);
|
|
}
|
|
|
|
isl_set_free(OldContext);
|
|
S.setContext(NewContext);
|
|
|
|
StatementToIslMapTy &NewScattering = *(new StatementToIslMapTy());
|
|
|
|
int index = 0;
|
|
|
|
for (Scop::iterator SI = S.begin(), SE = S.end(); SI != SE; ++SI) {
|
|
Json::Value schedule = jscop["statements"][index]["schedule"];
|
|
isl_map *m = isl_map_read_from_str(S.getIslCtx(), schedule.asCString());
|
|
isl_space *Space = (*SI)->getDomainSpace();
|
|
|
|
// Copy the old tuple id. This is necessary to retain the user pointer,
|
|
// that stores the reference to the ScopStmt this scattering belongs to.
|
|
m = isl_map_set_tuple_id(m, isl_dim_in,
|
|
isl_space_get_tuple_id(Space, isl_dim_set));
|
|
isl_space_free(Space);
|
|
NewScattering[*SI] = m;
|
|
index++;
|
|
}
|
|
|
|
if (!D.isValidScattering(S, &NewScattering)) {
|
|
errs() << "JScop file contains a scattering that changes the "
|
|
<< "dependences. Use -disable-polly-legality to continue anyways\n";
|
|
for (StatementToIslMapTy::iterator SI = NewScattering.begin(),
|
|
SE = NewScattering.end();
|
|
SI != SE; ++SI)
|
|
isl_map_free(SI->second);
|
|
return false;
|
|
}
|
|
|
|
for (Scop::iterator SI = S.begin(), SE = S.end(); SI != SE; ++SI) {
|
|
ScopStmt *Stmt = *SI;
|
|
|
|
if (NewScattering.find(Stmt) != NewScattering.end())
|
|
Stmt->setScattering(NewScattering[Stmt]);
|
|
}
|
|
|
|
int statementIdx = 0;
|
|
for (Scop::iterator SI = S.begin(), SE = S.end(); SI != SE; ++SI) {
|
|
ScopStmt *Stmt = *SI;
|
|
|
|
int memoryAccessIdx = 0;
|
|
for (MemoryAccess *MA : *Stmt) {
|
|
Json::Value accesses = jscop["statements"][statementIdx]["accesses"]
|
|
[memoryAccessIdx]["relation"];
|
|
isl_map *newAccessMap =
|
|
isl_map_read_from_str(S.getIslCtx(), accesses.asCString());
|
|
isl_map *currentAccessMap = MA->getAccessRelation();
|
|
|
|
if (isl_map_dim(newAccessMap, isl_dim_param) !=
|
|
isl_map_dim(currentAccessMap, isl_dim_param)) {
|
|
errs() << "JScop file changes the number of parameter dimensions\n";
|
|
isl_map_free(currentAccessMap);
|
|
isl_map_free(newAccessMap);
|
|
return false;
|
|
}
|
|
|
|
isl_id *OutId = isl_map_get_tuple_id(currentAccessMap, isl_dim_out);
|
|
newAccessMap = isl_map_set_tuple_id(newAccessMap, isl_dim_out, OutId);
|
|
|
|
// We keep the old alignment, thus we cannot allow accesses to memory
|
|
// locations that were not accessed before if the alignment of the access
|
|
// is not the default alignment.
|
|
bool SpecialAlignment = true;
|
|
if (LoadInst *LoadI = dyn_cast<LoadInst>(MA->getAccessInstruction())) {
|
|
SpecialAlignment =
|
|
DL.getABITypeAlignment(LoadI->getType()) != LoadI->getAlignment();
|
|
} else if (StoreInst *StoreI =
|
|
dyn_cast<StoreInst>(MA->getAccessInstruction())) {
|
|
SpecialAlignment =
|
|
DL.getABITypeAlignment(StoreI->getValueOperand()->getType()) !=
|
|
StoreI->getAlignment();
|
|
}
|
|
|
|
if (SpecialAlignment) {
|
|
isl_set *newAccessSet = isl_map_range(isl_map_copy(newAccessMap));
|
|
isl_set *currentAccessSet =
|
|
isl_map_range(isl_map_copy(currentAccessMap));
|
|
bool isSubset = isl_set_is_subset(newAccessSet, currentAccessSet);
|
|
isl_set_free(newAccessSet);
|
|
isl_set_free(currentAccessSet);
|
|
|
|
if (!isSubset) {
|
|
errs() << "JScop file changes the accessed memory\n";
|
|
isl_map_free(currentAccessMap);
|
|
isl_map_free(newAccessMap);
|
|
return false;
|
|
}
|
|
}
|
|
|
|
// We need to copy the isl_ids for the parameter dimensions to the new
|
|
// map. Without doing this the current map would have different
|
|
// ids then the new one, even though both are named identically.
|
|
for (unsigned i = 0; i < isl_map_dim(currentAccessMap, isl_dim_param);
|
|
i++) {
|
|
isl_id *id = isl_map_get_dim_id(currentAccessMap, isl_dim_param, i);
|
|
newAccessMap = isl_map_set_dim_id(newAccessMap, isl_dim_param, i, id);
|
|
}
|
|
|
|
// Copy the old tuple id. This is necessary to retain the user pointer,
|
|
// that stores the reference to the ScopStmt this access belongs to.
|
|
isl_id *Id = isl_map_get_tuple_id(currentAccessMap, isl_dim_in);
|
|
newAccessMap = isl_map_set_tuple_id(newAccessMap, isl_dim_in, Id);
|
|
|
|
if (!isl_map_has_equal_space(currentAccessMap, newAccessMap)) {
|
|
errs() << "JScop file contains access function with incompatible "
|
|
<< "dimensions\n";
|
|
isl_map_free(currentAccessMap);
|
|
isl_map_free(newAccessMap);
|
|
return false;
|
|
}
|
|
if (isl_map_dim(newAccessMap, isl_dim_out) != 1) {
|
|
errs() << "New access map in JScop file should be single dimensional\n";
|
|
isl_map_free(currentAccessMap);
|
|
isl_map_free(newAccessMap);
|
|
return false;
|
|
}
|
|
if (!isl_map_is_equal(newAccessMap, currentAccessMap)) {
|
|
// Statistics.
|
|
++NewAccessMapFound;
|
|
newAccessStrings.push_back(accesses.asCString());
|
|
MA->setNewAccessRelation(newAccessMap);
|
|
} else {
|
|
isl_map_free(newAccessMap);
|
|
}
|
|
isl_map_free(currentAccessMap);
|
|
memoryAccessIdx++;
|
|
}
|
|
statementIdx++;
|
|
}
|
|
|
|
return false;
|
|
}
|
|
|
|
void JSONImporter::getAnalysisUsage(AnalysisUsage &AU) const {
|
|
ScopPass::getAnalysisUsage(AU);
|
|
AU.addRequired<DependenceInfo>();
|
|
}
|
|
Pass *polly::createJSONImporterPass() { return new JSONImporter(); }
|
|
|
|
INITIALIZE_PASS_BEGIN(JSONExporter, "polly-export-jscop",
|
|
"Polly - Export Scops as JSON"
|
|
" (Writes a .jscop file for each Scop)",
|
|
false, false);
|
|
INITIALIZE_PASS_DEPENDENCY(DependenceInfo)
|
|
INITIALIZE_PASS_END(JSONExporter, "polly-export-jscop",
|
|
"Polly - Export Scops as JSON"
|
|
" (Writes a .jscop file for each Scop)",
|
|
false, false)
|
|
|
|
INITIALIZE_PASS_BEGIN(JSONImporter, "polly-import-jscop",
|
|
"Polly - Import Scops from JSON"
|
|
" (Reads a .jscop file for each Scop)",
|
|
false, false);
|
|
INITIALIZE_PASS_DEPENDENCY(DependenceInfo)
|
|
INITIALIZE_PASS_END(JSONImporter, "polly-import-jscop",
|
|
"Polly - Import Scops from JSON"
|
|
" (Reads a .jscop file for each Scop)",
|
|
false, false)
|