gem5/src/mem/ruby/network/orion/Allocator/VCAllocator.cc
2011-06-02 14:36:35 -07:00

303 lines
11 KiB
C++

/*
* Copyright (c) 2009 Princeton University
* Copyright (c) 2009 The Regents of the University of California
* All rights reserved.
*
* Redistribution and use in source and binary forms, with or without
* modification, are permitted provided that the following conditions are
* met: redistributions of source code must retain the above copyright
* notice, this list of conditions and the following disclaimer;
* redistributions in binary form must reproduce the above copyright
* notice, this list of conditions and the following disclaimer in the
* documentation and/or other materials provided with the distribution;
* neither the name of the copyright holders nor the names of its
* contributors may be used to endorse or promote products derived from
* this software without specific prior written permission.
*
* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
* "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
* LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
* A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
* OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
* SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
* LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
* DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
* THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
* (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
* OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
*
* Authors: Hangsheng Wang (Orion 1.0, Princeton)
* Xinping Zhu (Orion 1.0, Princeton)
* Xuning Chen (Orion 1.0, Princeton)
* Bin Li (Orion 2.0, Princeton)
* Kambiz Samadi (Orion 2.0, UC San Diego)
*/
#include <cassert>
#include <cmath>
#include <cstdlib>
#include <iostream>
#include "mem/ruby/network/orion/Allocator/Arbiter.hh"
#include "mem/ruby/network/orion/Allocator/VCAllocator.hh"
#include "mem/ruby/network/orion/Buffer/Buffer.hh"
#include "mem/ruby/network/orion/OrionConfig.hh"
using namespace std;
VCAllocator::VCAllocator(
uint32_t num_in_port_,
uint32_t num_out_port_,
uint32_t num_vclass_,
uint32_t num_vchannel_,
const string& arb_model_str_,
const string& arb_ff_model_str_,
const TechParameter* tech_param_ptr_
)
{
assert(num_in_port_ == num_in_port_);
assert(num_out_port_ == num_out_port_);
assert(num_vclass_ == num_vclass_);
assert(num_vchannel_ == num_vchannel_);
m_va_model = ONE_STAGE_ARB;
m_num_in_port = num_in_port_;
m_num_out_port = num_out_port_;
m_num_vclass = num_vclass_;
m_num_vchannel = num_vchannel_;
m_local_arb_ptr = NULL;
m_global_arb_ptr = Arbiter::create_arbiter(
arb_model_str_, arb_ff_model_str_,
(m_num_in_port-1)*m_num_vchannel, 0, tech_param_ptr_);
m_vc_select_ptr = NULL;
}
VCAllocator::VCAllocator(
uint32_t num_in_port_,
uint32_t num_out_port_,
uint32_t num_vclass_,
uint32_t num_vchannel_,
const string& local_arb_model_str_,
const string& local_arb_ff_model_str_,
const string& global_arb_model_str_,
const string& global_arb_ff_model_str_,
const TechParameter* tech_param_ptr_
)
{
assert(num_in_port_ == num_in_port_);
assert(num_out_port_ == num_out_port_);
assert(num_vclass_ == num_vclass_);
assert(num_vchannel_ == num_vchannel_);
m_va_model = TWO_STAGE_ARB;
m_num_in_port = num_in_port_;
m_num_out_port = num_out_port_;
m_num_vclass = num_vclass_;
m_num_vchannel = num_vchannel_;
// first stage
m_local_arb_ptr = Arbiter::create_arbiter(
local_arb_model_str_, local_arb_ff_model_str_,
m_num_vchannel, 0, tech_param_ptr_);
// second stage
m_global_arb_ptr = Arbiter::create_arbiter(
global_arb_model_str_, global_arb_ff_model_str_,
(m_num_in_port-1)*m_num_vchannel, 0, tech_param_ptr_);
m_vc_select_ptr = NULL;
}
VCAllocator::VCAllocator(
uint32_t num_in_port_,
uint32_t num_out_port_,
uint32_t num_vclass_,
uint32_t num_vchannel_,
const string& vc_select_buf_model_str_,
const OrionConfig* orion_cfg_ptr_
)
{
assert(num_in_port_ == num_in_port_);
assert(num_out_port_ == num_out_port_);
assert(num_vclass_ == num_vclass_);
assert(num_vchannel_ == num_vchannel_);
m_va_model = VC_SELECT;
m_num_in_port = num_in_port_;
m_num_out_port = num_out_port_;
m_num_vclass = num_vclass_;
m_num_vchannel = num_vchannel_;
m_local_arb_ptr = NULL;
m_global_arb_ptr = NULL;
uint32_t vc_select_buf_num_set = m_num_vchannel;
uint32_t vc_select_buf_line_width = (uint32_t)ceil(log2(m_num_vchannel));
m_vc_select_ptr = new Buffer(vc_select_buf_model_str_, true, false,
vc_select_buf_num_set, vc_select_buf_line_width, 1, 1, orion_cfg_ptr_);
}
VCAllocator::~VCAllocator()
{
delete m_local_arb_ptr;
delete m_global_arb_ptr;
delete m_vc_select_ptr;
}
double VCAllocator::get_dynamic_energy_local_vc_arb(double num_req_, bool is_max_) const
{
double e_local_arb = 0;
switch(m_va_model)
{
case TWO_STAGE_ARB:
e_local_arb = m_local_arb_ptr->calc_dynamic_energy(num_req_, is_max_);
break;
case ONE_STAGE_ARB:
case VC_SELECT:
default:
e_local_arb = 0;
}
return e_local_arb;
}
double VCAllocator::get_dynamic_energy_global_vc_arb(double num_req_, bool is_max_) const
{
double e_global_arb = 0;
switch(m_va_model)
{
case ONE_STAGE_ARB:
case TWO_STAGE_ARB:
e_global_arb = m_global_arb_ptr->calc_dynamic_energy(num_req_, is_max_);
break;
case VC_SELECT:
default:
e_global_arb = 0;
}
return e_global_arb;
}
double VCAllocator::get_dynamic_energy_vc_select(bool is_read_, bool is_max_) const
{
double e_vc_select = 0;
switch(m_va_model)
{
case VC_SELECT:
e_vc_select = m_vc_select_ptr->get_dynamic_energy(is_read_, is_max_);
break;
case ONE_STAGE_ARB:
case TWO_STAGE_ARB:
default:
e_vc_select = 0;
}
return e_vc_select;
}
double VCAllocator::get_static_power() const
{
double p_va = 0;
switch(m_va_model)
{
case ONE_STAGE_ARB:
p_va = m_global_arb_ptr->get_static_power()*m_num_out_port*m_num_vclass*m_num_vchannel;
break;
case TWO_STAGE_ARB:
p_va += m_local_arb_ptr->get_static_power()*m_num_in_port*m_num_vclass*m_num_vchannel;
p_va += m_global_arb_ptr->get_static_power()*m_num_out_port*m_num_vclass*m_num_vchannel;
break;
case VC_SELECT:
p_va = m_vc_select_ptr->get_static_power()*m_num_out_port*m_num_vclass;
break;
default:
cerr << "ERROR: Invalid VA model" << endl;
exit(1);
}
return p_va;
}
void VCAllocator::print_all() const
{
switch(m_va_model)
{
case ONE_STAGE_ARB:
cout << "VCAllocator: ONE_STAGE_ARB" << endl;
for (uint32_t i = 0; i < (m_num_in_port-1)*m_num_vchannel; i++)
{
cout << "\t" << "Global arb (" << i << ") = " << get_dynamic_energy_global_vc_arb(i, false) << endl;
}
break;
case TWO_STAGE_ARB:
cout << "VCAllocator: TWO_STAGE_ARB" << endl;
for (uint32_t i = 0; i < m_num_vchannel; i++)
{
cout << "\t" << "Local arb (" << i << ") = " << get_dynamic_energy_local_vc_arb(i, false) << endl;
}
for (uint32_t i = 0; i < (m_num_in_port-1)*m_num_vchannel; i++)
{
cout << "\t" << "Global arb (" << i << ") = " << get_dynamic_energy_global_vc_arb(i, false) << endl;
}
break;
case VC_SELECT:
cout << "VCAllocator: VC_SELECT" << endl;
cout << "\t" << "Read = " << get_dynamic_energy_vc_select(true, false) << endl;
cout << "\t" << "Write = " << get_dynamic_energy_vc_select(false, false) << endl;
break;
default:
;
}
cout << "\t" << "Static power = " << get_static_power() << endl;
return;
}
VCAllocator* VCAllocator::create_vcallocator(
const string& vcalloc_model_str_,
uint32_t num_in_port_,
uint32_t num_out_port_,
uint32_t num_vclass_,
uint32_t num_vchannel_,
const OrionConfig* orion_cfg_ptr_
)
{
if (num_vchannel_ > 1)
{
if (vcalloc_model_str_ == string("ONE_STAGE_ARB"))
{
const string& arb_model_str = orion_cfg_ptr_->get<string>("VA_OUT_ARB_MODEL");
const string& arb_ff_model_str = orion_cfg_ptr_->get<string>("VA_OUT_ARB_FF_MODEL");
const TechParameter* tech_param_ptr = orion_cfg_ptr_->get_tech_param_ptr();
return new VCAllocator(num_in_port_, num_out_port_, num_vclass_, num_vchannel_,
arb_model_str, arb_ff_model_str, tech_param_ptr);
}
else if (vcalloc_model_str_ == string("TWO_STAGE_ARB"))
{
const string& local_arb_model_str = orion_cfg_ptr_->get<string>("VA_IN_ARB_MODEL");
const string& local_arb_ff_model_str = orion_cfg_ptr_->get<string>("VA_IN_ARB_FF_MODEL");
const string& global_arb_model_str = orion_cfg_ptr_->get<string>("VA_OUT_ARB_MODEL");
const string& global_arb_ff_model_str = orion_cfg_ptr_->get<string>("VA_OUT_ARB_FF_MODEL");
const TechParameter* tech_param_ptr = orion_cfg_ptr_->get_tech_param_ptr();
return new VCAllocator(num_in_port_, num_out_port_, num_vclass_, num_vchannel_,
local_arb_model_str, local_arb_ff_model_str,
global_arb_model_str, global_arb_ff_model_str,tech_param_ptr);
}
else if (vcalloc_model_str_ == string("VC_SELECT"))
{
const string& vc_select_buf_model_str = orion_cfg_ptr_->get<string>("VA_BUF_MODEL");
return new VCAllocator(num_in_port_, num_out_port_, num_vclass_, num_vchannel_,
vc_select_buf_model_str, orion_cfg_ptr_);
}
else
{
cerr << "WARNING: No VC allocator model" << endl;
return (VCAllocator*)NULL;
}
}
else
{
// reduce to a register
return new VCAllocator(num_in_port_, num_out_port_, num_vclass_, 1,
"REGISTER", orion_cfg_ptr_);
}
}