blob: 5020aea56b2f9ba58f568829574f7cd57a3b52a8 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
|
/*
Copyright (c) 2011 Yahoo! Inc. All rights reserved. The copyrights
embodied in the content of this file are licensed under the BSD
(revised) open source license
This implements various accumulate functions building on top of allreduce.
*/
#ifndef ACCUMULATE_H
#define ACCUMULATE_H
#include "allreduce.h"
#include "global_data.h"
using namespace std;
void accumulate(string master_location, regressor& reg, size_t o);
float accumulate_scalar(string master_location, float local_sum);
void accumulate_weighted_avg(string master_location, regressor& reg);
void accumulate_avg(string master_location, regressor& reg, size_t o);
double get_comm_time();
#endif
|
