Alexander_KS/SAPFOR
4
2
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

moved

Browse Source
This commit is contained in:
Alexander
2025-03-12 12:37:19 +03:00
parent 1c851baa7e
commit 6a4040be3e
426 changed files with 0 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

184
Sapfor/_src/Distribution/DvmhDirective.h Normal file
View File

@@ -0,0 +1,184 @@
#pragma once
#include <string>
#include <vector>
#include <set>
#include <map>
#include "Array.h"
#include "DvmhDirectiveBase.h"
#include "../Utils/AstWrapper.h"
#include "../Utils/types.h"
#include "../Utils/utils.h"
extern int sharedMemoryParallelization;
struct LoopGraph;
namespace Distribution
{
template<typename vType, typename wType, typename attrType> class GraphCSR;
template<typename vType> class Arrays;
}
namespace DIST = Distribution;
struct AlignRule : public AlignRuleBase
{
public:
Statement* GenRule(File *file, const int variant) const;
};
struct DistrVariant : public DistrVariantBase
{
public:
DistrVariant(const std::vector<dist> &distRule) : DistrVariantBase(distRule) { }
void GenRule(File *file, Expression *arrayRef, const std::vector<int> &newOrder) const;
std::vector<Expression*> GenRuleSt(File *file, const std::vector<int> &newOrder) const;
};
struct DataDirective : Directive
{
public:
std::vector<AlignRule> alignRules;
std::vector<std::pair<DIST::Array*, std::vector<DistrVariant>>> distrRules;
public:
std::vector<std::pair<DIST::Array*, std::vector<DistrVariant>>> GetDistrRules() const { return distrRules; }
void createDirstributionVariants(const std::vector<DIST::Array*> &arraysToDist);
std::vector<std::string> GenRule(const std::vector<int> &rules) const;
std::vector<std::string> GenAlignsRules() const;
std::vector<AlignRule> GenAlignsRules(void*) { return alignRules; }
std::vector<Statement*> GenRule(File *file, const std::vector<int> &rules, const int variant) const;
std::vector<std::vector<dist>> GenRule(const std::vector<int> &rules, int) const;
std::vector<Statement*> GenAlignsRules(File *file, const int variant) const;
void UpdateLinks(const std::map<DIST::Array*, DIST::Array*> &oldNewArrays)
{
for (int i = 0; i < distrRules.size(); ++i)
distrRules[i].first = oldNewArrays.find(distrRules[i].first)->second;
}
~DataDirective()
{
for (auto &elem : distrRules)
{
if (elem.first && elem.first->IsTemplate())
{
delete elem.first;
elem.first = NULL;
}
}
}
};
struct CreatedDirective : Directive
{
std::string data;
std::vector<Expression*> sageData;
explicit CreatedDirective(const std::string& data, const std::vector<Expression*>& sageData) :
data(data), sageData(sageData)
{ }
explicit CreatedDirective(const std::string& data, const std::vector<Expression*>& sageData, int line) :
data(data), sageData(sageData)
{
this->line = line;
}
};
struct ParallelDirective : Directive
{
public:
std::vector<std::string> parallel;
// tuple of <varName, A, B> on A*x + B
std::vector<std::pair<std::string, std::pair<int, int>>> on;
DIST::Array *arrayRef; // template
DIST::Array *arrayRef2;// main array in loop
std::vector<std::pair<std::string, std::pair<int, int>>> on2;
std::string cloneOfTemplate; // if loop has realigns before
std::set<Symbol*> privates;
// origin_Name uniqName bounds
std::vector<std::pair<std::pair<std::string, std::string>, std::vector<std::pair<int, int>>>> shadowRenew;
std::vector<std::vector<std::pair<int, int>>> shadowRenewShifts;
// origin_Name uniqName bounds
std::vector<std::pair<std::pair<std::string, std::string>, std::vector<std::pair<int, int>>>> across;
std::vector<std::vector<std::pair<int, int>>> acrossShifts;
// origin_Name uniqName expression
std::map<std::pair<std::pair<std::string, std::string>, std::string>, Expression*> remoteAccess;
std::map<std::string, std::set<std::string>> reduction;
std::map<std::string, std::set<std::tuple<std::string, std::string, int>>> reductionLoc;
public:
ParallelDirective() : arrayRef(NULL), arrayRef2(NULL) { }
ParallelDirective(const ParallelDirective &copyFrom) : Directive(copyFrom)
{
parallel = copyFrom.parallel;
on = copyFrom.on;
on2 = copyFrom.on2;
arrayRef = copyFrom.arrayRef;
arrayRef2 = copyFrom.arrayRef2;
privates = copyFrom.privates;
shadowRenew = copyFrom.shadowRenew;
shadowRenewShifts = copyFrom.shadowRenewShifts;
across = copyFrom.across;
acrossShifts = copyFrom.acrossShifts;
remoteAccess = copyFrom.remoteAccess;
reduction = copyFrom.reduction;
reductionLoc = copyFrom.reductionLoc;
cloneOfTemplate = copyFrom.cloneOfTemplate;
}
//for F_LANG;
Directive*
genDirective(File *file, const std::vector<std::pair<DIST::Array*, const DistrVariant*>> &distribution,
LoopGraph* currLoop,
DIST::GraphCSR<int, double, attrType> &reducedG,
DIST::Arrays<int> &allArrays,
const uint64_t regionId,
const std::map<DIST::Array*, std::set<DIST::Array*>> &arrayLinksByFuncCalls);
friend ParallelDirective* operator+(const ParallelDirective &first, const ParallelDirective &second);
~ParallelDirective()
{
parallel.clear();
on.clear();
privates.clear();
shadowRenew.clear();
across.clear();
acrossShifts.clear();
reduction.clear();
reductionLoc.clear();
remoteAccess.clear();
}
private:
std::string genBounds(std::pair<std::pair<std::string, std::string>, std::vector<std::pair<int, int>>> &shadowOp,
std::vector<std::pair<int, int>> &shadowOpShift,
DIST::GraphCSR<int, double, attrType> &reducedG,
DIST::Arrays<int> &allArrays, DIST::Array* shadowArray,
std::map<DIST::Array*, std::vector<ArrayOp>>& remoteRegularReads,
const std::map<DIST::Array*, std::pair<std::vector<ArrayOp>, std::vector<bool>>> &readOps,
const bool isAcross, const uint64_t regionId,
const std::vector<std::pair<DIST::Array*, const DistrVariant*>> &distribution,
std::set<DIST::Array*> &arraysInAcross,
std::vector<std::map<std::pair<int, int>, int>> &shiftsByAccess,
const std::map<DIST::Array*, std::set<DIST::Array*>> &arrayLinksByFuncCalls) const;
std::string genBoundsNoDist(std::pair<std::pair<std::string, std::string>, std::vector<std::pair<int, int>>>& shadowOp,
std::vector<std::pair<int, int>>& shadowOpShift,
DIST::Array* currArray,
const std::map<DIST::Array*, std::pair<std::vector<ArrayOp>, std::vector<bool>>>& readOps,
std::set<DIST::Array*>& arraysInAcross,
std::vector<std::map<std::pair<int, int>, int>>& shiftsByAccess,
const std::map<DIST::Array*, std::set<DIST::Array*>>& arrayLinksByFuncCalls) const;
};
std::string genStringExpr(const std::string &letter, const std::pair<int, int> expr);
std::pair<std::string, std::string> convertDigitToPositive(const int digit);