cyp0633
/
ns3-datacenter
mirror of https://github.com/cyp0633/ns3-datacenter.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

update patched timely

This commit is contained in:
cyp0633 2024-04-10 15:59:18 +08:00
parent 5f8dea9833
commit 27cd3bddd8
Signed by: cyp0633
GPG Key ID: CF90D09FB1FDCE45
2 changed files with 18 additions and 18 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

25
simulator/ns-3.39/src/point-to-point/model/rdma-hw.cc
View File

@ -1797,10 +1797,7 @@ RdmaHw::HandleAckPatchedTimely(Ptr<RdmaQueuePair> qp, Ptr<Packet> p, CustomHeade
}
void
RdmaHw::UpdateRatePatchedTimely(Ptr<RdmaQueuePair> qp,
Ptr<Packet> p,
CustomHeader& ch,
bool us) const
RdmaHw::UpdateRatePatchedTimely(Ptr<RdmaQueuePair> qp, Ptr<Packet> p, CustomHeader& ch, bool us)
{
uint32_t next_seq = qp->snd_nxt;
uint64_t rtt = Simulator::Now().GetTimeStep() - ch.ack.ih.ts;
@ -1810,7 +1807,7 @@ RdmaHw::UpdateRatePatchedTimely(Ptr<RdmaQueuePair> qp,
int64_t new_rtt_diff = (int64_t)rtt - (int64_t)qp->tmly.lastRtt;
double rtt_diff = (1 - m_tmly_alpha) * qp->tmly.rttDiff + m_tmly_alpha * new_rtt_diff;
double gradient = rtt_diff / m_tmly_minRtt;
int weight;
double weight;
double error;
#if PRINT_LOG
if (print)
@ -1825,31 +1822,33 @@ RdmaHw::UpdateRatePatchedTimely(Ptr<RdmaQueuePair> qp,
if (rtt < m_tmly_TLow) // newRTT < Tlow
{
// rate = rate + rai
qp->m_rate = qp->tmly.m_curRate + m_rai;
ChangeRate(qp, qp->tmly.m_curRate + m_rai);
}
else if (rtt > m_tmly_THigh) // newRTT > Thigh
{
// rate = rate * (1 - beta(1 - Thigh / new_rtt))
qp->m_rate = qp->tmly.m_curRate * (1 - m_tmly_beta * (1 - (double)m_tmly_THigh / rtt));
ChangeRate(qp,
qp->tmly.m_curRate * (1 - m_tmly_beta * (1 - (double)m_tmly_THigh / rtt)));
}
else
{
// weight = w(rttGradient)
if (gradient <= -0.25)
{
weight = 0;
weight = 0.0;
}
else if (gradient >= 0.25)
{
weight = 1;
weight = 1.0;
}
else
{
weight = 2 * gradient + 0.5;
weight = 2.0 * gradient + 0.5;
}
error = (rtt - m_ptmly_RttRef) * 1.0 / m_ptmly_RttRef;
qp->m_rate =
m_rai * (1 - weight) + qp->tmly.m_curRate * (1 - m_ptmly_beta * error * weight);
ChangeRate(qp,
m_rai * (1 - weight) +
qp->tmly.m_curRate * (1 - m_ptmly_beta * error * weight));
}
qp->tmly.m_curRate = std::max(m_minRate, std::min(qp->m_max_rate, qp->m_rate));
qp->tmly.rttDiff = rtt_diff;
@ -1866,6 +1865,8 @@ RdmaHw::UpdateRatePatchedTimely(Ptr<RdmaQueuePair> qp,
// update
qp->tmly.lastRtt = rtt;
}
std::cout << "[PTMLY] node:" << m_node->GetId() << " rate:" << qp->m_rate << " RTT:" << rtt
<< std::endl;
}
void

11
simulator/ns-3.39/src/point-to-point/model/rdma-hw.h
View File

@ -182,10 +182,7 @@ class RdmaHw : public Object
double m_ptmly_beta;
uint64_t m_ptmly_RttRef;
void HandleAckPatchedTimely(Ptr<RdmaQueuePair> qp, Ptr<Packet> p, CustomHeader& ch);
void UpdateRatePatchedTimely(Ptr<RdmaQueuePair> qp,
Ptr<Packet> p,
CustomHeader& ch,
bool us) const;
void UpdateRatePatchedTimely(Ptr<RdmaQueuePair> qp, Ptr<Packet> p, CustomHeader& ch, bool us);
void FastReactPatchedTimely(Ptr<RdmaQueuePair> qp, Ptr<Packet> p, CustomHeader& ch);
/**********************
@ -232,8 +229,10 @@ class RdmaHw : public Object
********************/
uint64_t rtt_qcn_tmin; // max rtt value to generate no ecn
uint64_t rtt_qcn_tmax; // min rtt value to always generate ecn
double rtt_qcn_alpha; // additive increase when cwnd < 1; use original value, don't multiply by mtu
double rtt_qcn_beta; // multiplicative decrease when cwnd > 1; use original value, don't multiply by mtu
double
rtt_qcn_alpha; // additive increase when cwnd < 1; use original value, don't multiply by mtu
double rtt_qcn_beta; // multiplicative decrease when cwnd > 1; use original value, don't
// multiply by mtu
void HandleAckRttQcn(Ptr<RdmaQueuePair> qp, Ptr<Packet> p, CustomHeader& ch) const;
};