@@ -86,8 +86,8 @@ TEST(ADLocalCounterStatisticsTest, GlobalCounterStatsWorks){
8686
8787
8888struct ADLocalCounterStatisticsWrapper : public ADLocalCounterStatistics {
89- static std::pair<size_t , size_t > updateGlobalStatisticsTest (ADNetClient &net_client, const std::string &l_stats, int step){
90- return ADLocalCounterStatistics::updateGlobalStatistics (net_client, l_stats, step);
89+ static std::pair<size_t , size_t > updateGlobalStatisticsTest (ADNetClient &net_client, const std::string &l_stats, int step, int rank, std::string pserver_addr ){
90+ return ADLocalCounterStatistics::updateGlobalStatistics (net_client, l_stats, step, rank, pserver_addr );
9191 }
9292};
9393
@@ -125,7 +125,7 @@ TEST(ADLocalCounterStatisticsTest, UpdateGlobalStatisticsWithMockPS){
125125 net_client.connect_ps (0 , 0 , sname);
126126 barrier2.wait ();
127127 std::cout << " AD thread updating local stats" << std::endl;
128- ADLocalCounterStatisticsWrapper::updateGlobalStatisticsTest (net_client, " test" ,0 );
128+ ADLocalCounterStatisticsWrapper::updateGlobalStatisticsTest (net_client, " test" ,0 , 0 ,sname );
129129 barrier2.wait ();
130130 std::cout << " AD thread terminating connection" << std::endl;
131131 net_client.disconnect_ps ();
@@ -187,9 +187,9 @@ TEST(ADLocalCounterStatisticsTest, UpdateGlobalStatisticsWithRealPS){
187187 net_client.connect_ps (0 , 0 , sname);
188188
189189 cs.setStats (nm, a);
190- cs.updateGlobalStatistics (net_client);
190+ cs.updateGlobalStatistics (net_client, 0 , sname );
191191 cs.setStats (nm, b);
192- cs.updateGlobalStatistics (net_client);
192+ cs.updateGlobalStatistics (net_client, 0 , sname );
193193
194194 std::cout << " AD thread terminating connection" << std::endl;
195195 net_client.disconnect_ps ();
0 commit comments