gem5/ext/dsent/model/electrical/SeparableAllocator.cc
Nilay Vaish e8ed7b1d1b ext: add the source code for DSENT
This patch adds a tool called DSENT to the ext/ directory.  DSENT
is a tool that models power and area for on-chip networks.  The next
patch adds a script for using the tool.
2014-10-11 15:02:23 -05:00

270 lines
13 KiB
C++

#include "model/electrical/SeparableAllocator.h"
#include "model/ModelGen.h"
#include "model/timing_graph/ElectricalNet.h"
namespace DSENT
{
SeparableAllocator::SeparableAllocator(const String& instance_name_, const TechModel* tech_model_)
: ElectricalModel(instance_name_, tech_model_)
{
initParameters();
initProperties();
}
SeparableAllocator::~SeparableAllocator()
{}
void SeparableAllocator::initParameters()
{
addParameterName("NumberRequesters");
addParameterName("NumberResources");
addParameterName("IsRequesterFirst", true);
addParameterName("Stage1->ArbiterModel");
addParameterName("Stage2->ArbiterModel");
return;
}
void SeparableAllocator::initProperties()
{
addPropertyName("P(Request)");
addPropertyName("Act(Request)");
addPropertyName("P(CK)");
addPropertyName("Act(CK)");
return;
}
SeparableAllocator* SeparableAllocator::clone() const
{
// TODO
return NULL;
}
void SeparableAllocator::constructModel()
{
// Get parameters
unsigned int number_requesters = getParameter("NumberRequesters").toUInt();
unsigned int number_resources = getParameter("NumberResources").toUInt();
bool is_requester_first = getParameter("IsRequesterFirst").toBool();
const String& stage1_arbiter_model = getParameter("Stage1->ArbiterModel");
const String& stage2_arbiter_model = getParameter("Stage2->ArbiterModel");
ASSERT(number_requesters > 0, "[Error] " + getInstanceName() +
" -> Number of requesters must be > 0!");
ASSERT(number_resources > 0, "[Error] " + getInstanceName() +
" -> Number of resources must be > 0!");
// Create area, power, and event results
createElectricalResults();
addEventResult(new Result("Allocate"));
// Create ports
createInputPort("CK");
for(unsigned int i = 0; i < number_requesters; ++i)
{
createInputPort("Request" + (String)i, makeNetIndex(0, number_resources-1));
createOutputPort("Grant" + (String)i, makeNetIndex(0, number_resources-1));
}
// If is_requester_first is set, requests from the same requester will be arbitrate
// on stage 1
if(is_requester_first)
{
// Init stage 1 arbiters
for(unsigned int i = 0; i < number_requesters; ++i)
{
ElectricalModel* arb = (ElectricalModel*)ModelGen::createModel(stage1_arbiter_model, "Stage1Arb" + (String)i, getTechModel());
arb->setParameter("NumberRequests", number_resources);
arb->construct();
addSubInstances(arb, 1.0);
addElectricalSubResults(arb, 1.0);
getEventResult("Allocate")->addSubResult(arb->getEventResult("Arbitrate"), arb->getInstanceName(), 1.0);
createNet("Stage1Arb_In" + (String)i, makeNetIndex(0, number_resources-1));
createNet("Stage1Arb_Out" + (String)i, makeNetIndex(0, number_resources-1));
portConnect(arb, "CK", "CK");
assign("Stage1Arb_In" + (String)i, "Request" + (String)i);
for(unsigned int j = 0; j < number_resources; ++j)
{
portConnect(arb, "Request" + (String)j, "Stage1Arb_In" + (String)i, makeNetIndex(j));
portConnect(arb, "Grant" + (String)j, "Stage1Arb_Out" + (String)i, makeNetIndex(j));
}
}
// Init stage 2 arbiters
for(unsigned int i = 0; i < number_resources; ++i)
{
ElectricalModel* arb = (ElectricalModel*)ModelGen::createModel(stage2_arbiter_model, "Stage2Arb" + (String)i, getTechModel());
arb->setParameter("NumberRequests", number_requesters);
arb->construct();
addSubInstances(arb, 1.0);
addElectricalSubResults(arb, 1.0);
getEventResult("Allocate")->addSubResult(arb->getEventResult("Arbitrate"), arb->getInstanceName(), 1.0);
createNet("Stage2Arb_In" + (String)i, makeNetIndex(0, number_requesters-1));
createNet("Stage2Arb_Out" + (String)i, makeNetIndex(0, number_requesters-1));
portConnect(arb, "CK", "CK");
for(unsigned int j = 0; j < number_requesters; ++j)
{
assign("Stage2Arb_In" + (String)i, makeNetIndex(j), "Stage1Arb_Out" + (String)j, makeNetIndex(i));
portConnect(arb, "Request" + (String)j, "Stage2Arb_In" + (String)i, makeNetIndex(j));
portConnect(arb, "Grant" + (String)j, "Stage2Arb_Out" + (String)i, makeNetIndex(j));
assign("Grant" + (String)j, makeNetIndex(i), "Stage2Arb_Out" + (String)i, makeNetIndex(j));
}
}
}
else
{
// Init stage 1 arbiters
for(unsigned int i = 0; i < number_resources; ++i)
{
ElectricalModel* arb = (ElectricalModel*)ModelGen::createModel(stage1_arbiter_model, "Stage1Arb" + (String)i, getTechModel());
arb->setParameter("NumberRequests", number_requesters);
arb->construct();
addSubInstances(arb, 1.0);
addElectricalSubResults(arb, 1.0);
getEventResult("Allocate")->addSubResult(arb->getEventResult("Arbitrate"), arb->getInstanceName(), 1.0);
createNet("Stage1Arb_In" + (String)i, makeNetIndex(0, number_requesters-1));
createNet("Stage1Arb_Out" + (String)i, makeNetIndex(0, number_requesters-1));
portConnect(arb, "CK", "CK");
for(unsigned int j = 0; j < number_requesters; ++j)
{
assign("Stage1Arb_In" + (String)i, makeNetIndex(j), "Request" + (String)j, makeNetIndex(i));
portConnect(arb, "Request" + (String)j, "Stage1Arb_In" + (String)i, makeNetIndex(j));
portConnect(arb, "Grant" + (String)j, "Stage1Arb_Out" + (String)i, makeNetIndex(j));
}
}
// Init stage 2 arbiters
for(unsigned int i = 0; i < number_requesters; ++i)
{
ElectricalModel* arb = (ElectricalModel*)ModelGen::createModel(stage2_arbiter_model, "Stage2Arb" + (String)i, getTechModel());
arb->setParameter("NumberRequests", number_requesters);
arb->construct();
addSubInstances(arb, 1.0);
addElectricalSubResults(arb, 1.0);
getEventResult("Allocate")->addSubResult(arb->getEventResult("Arbitrate"), arb->getInstanceName(), 1.0);
createNet("Stage2Arb_In" + (String)i, makeNetIndex(0, number_resources-1));
createNet("Stage2Arb_Out" + (String)i, makeNetIndex(0, number_resources-1));
portConnect(arb, "CK", "CK");
for(unsigned int j = 0; j < number_resources; ++j)
{
assign("Stage2Arb_In" + (String)i, makeNetIndex(j), "Stage1Arb_Out" + (String)j, makeNetIndex(i));
portConnect(arb, "Request" + (String)j, "Stage2Arb_In", makeNetIndex(j));
portConnect(arb, "Grant" + (String)j, "Stage2Arb_Out", makeNetIndex(j));
}
assign("Grant" + (String)i, "Stage2Arb_Out" + (String)i);
}
}
return;
}
void SeparableAllocator::updateModel()
{
// Get parameters
unsigned int number_requesters = getParameter("NumberRequesters").toUInt();
unsigned int number_resources = getParameter("NumberResources").toUInt();
bool is_requester_first = getParameter("IsRequesterFirst").toBool();
// Get probabilities from inputs
const String& P_request = getProperty("P(Request)");
const String& act_request = getProperty("Act(Request)");
const String& P_CK = getProperty("P(CK)");
const String& act_CK = getProperty("Act(CK)");
const vector<double>& P_request_vector = LibUtil::castStringVector<double>(P_request.split("[,]"));
const vector<double>& act_request_vector = LibUtil::castStringVector<double>(act_request.split("[,]"));
ASSERT(P_request_vector.size() == (number_requesters * number_resources), "[Error] " + getInstanceName() +
" -> Expecting " + (String)(number_requesters * number_resources) +
" request probabilities, but got " + P_request);
ASSERT(act_request_vector.size() == (number_requesters * number_resources), "[Error] " + getInstanceName() +
" -> Expecting " + (String)(number_requesters * number_resources) +
" request actvities multiplier, but got " + act_request);
vector<double> P_int_request_vector(number_requesters * number_resources, 0.0);
vector<double> act_int_request_vector(number_requesters * number_resources, 0.0);
vector<double> P_out_request_vector(number_requesters * number_resources, 0.0);
vector<double> act_out_request_vector(number_requesters * number_resources, 0.0);
if(is_requester_first)
{
// Update stage1 arbiter
for(unsigned int i = 0; i < number_requesters; ++i)
{
vector<double> P_arb_request_vector(number_resources, 0.0);
vector<double> act_arb_request_vector(number_resources, 0.0);
for(unsigned int j = 0; j < number_resources; ++j)
{
P_arb_request_vector[j] = P_request_vector[i * number_resources + j];
act_arb_request_vector[j] = act_request_vector[i * number_resources + j];
}
Model* arb = getSubInstance("Stage1Arb" + (String)i);
arb->setProperty("P(Request)", LibUtil::vectorToString(P_arb_request_vector));
arb->setProperty("Act(Request)", LibUtil::vectorToString(act_arb_request_vector));
arb->setProperty("P(CK)", P_CK);
arb->setProperty("Act(CK)", act_CK);
arb->update();
const vector<double>& P_arb_out_request_vector = LibUtil::castStringVector<double>(arb->getGenProperties()->get("P(Grant)").split("[,]"));
const vector<double>& act_arb_out_request_vector = LibUtil::castStringVector<double>(arb->getGenProperties()->get("Act(Grant)").split("[,]"));
for(unsigned int j = 0; j < number_resources; ++j)
{
P_int_request_vector[i * number_resources + j] = P_arb_out_request_vector[j];
act_int_request_vector[i * number_resources + j] = act_arb_out_request_vector[j];
}
}
// Update stage2 arbiter
for(unsigned int i = 0; i < number_resources; ++i)
{
vector<double> P_arb_request_vector(number_requesters, 0.0);
vector<double> act_arb_request_vector(number_requesters, 0.0);
for(unsigned int j = 0; j < number_requesters; ++j)
{
P_arb_request_vector[j] = P_int_request_vector[j * number_resources + i];
act_arb_request_vector[j] = act_int_request_vector[j * number_resources + i];
}
Model* arb = getSubInstance("Stage2Arb" + (String)i);
arb->setProperty("P(Request)", LibUtil::vectorToString(P_arb_request_vector));
arb->setProperty("Act(Request)", LibUtil::vectorToString(act_arb_request_vector));
arb->setProperty("P(CK)", P_CK);
arb->setProperty("Act(CK)", act_CK);
arb->update();
const vector<double>& P_arb_out_request_vector = LibUtil::castStringVector<double>(arb->getGenProperties()->get("P(Grant)").split("[,]"));
const vector<double>& act_arb_out_request_vector = LibUtil::castStringVector<double>(arb->getGenProperties()->get("Act(Grant)").split("[,]"));
for(unsigned int j = 0; j < number_requesters; ++j)
{
P_out_request_vector[j * number_resources + i] = P_arb_out_request_vector[j];
act_out_request_vector[j * number_resources + i] = act_arb_out_request_vector[j];
}
}
}
else
{
}
// Update output probabilities
getGenProperties()->set("P(Grant)", LibUtil::vectorToString(P_out_request_vector));
getGenProperties()->set("Act(Grant)", LibUtil::vectorToString(act_out_request_vector));
return;
}
} // namespace DSENT