allreduce.h File Reference

#include <string>
#include <algorithm>
#include <sys/socket.h>
#include <netinet/in.h>
#include <netinet/tcp.h>
#include <netdb.h>
#include <stdlib.h>
#include <stdio.h>
#include <unistd.h>
#include <string.h>
#include <future>
#include "vw_exception.h"
#include <cassert>

Go to the source code of this file.

Classes
struct	node_socks
class	AllReduce
struct	Data
class	AllReduceSync
class	AllReduceThreads
class	AllReduceSockets

Macros
#define	CLOSESOCK   close

Typedefs
typedef int	socket_t

Functions
template<class T , void(*)(T &, const T &) f>
void	addbufs (T *buf1, const T *buf2, const size_t n)

Variables
constexpr size_t	ar_buf_size = 1 << 16

Macro Definition Documentation

CLOSESOCK

#define CLOSESOCK   close

Definition at line 43 of file allreduce.h.

Referenced by AllReduceSockets::all_reduce_init(), VW::SpanningTree::Run(), VW::SpanningTree::Stop(), and node_socks::~node_socks().

Typedef Documentation

socket_t

typedef int socket_t

Definition at line 42 of file allreduce.h.

Function Documentation

addbufs()

template<class T , void(*)(T &, const T &) f>

void addbufs	(	T *	buf1,
		const T *	buf2,
		const size_t	n
	)

Definition at line 72 of file allreduce.h.

References f.

{
  for (size_t i = 0; i < n; i++) f(buf1[i], buf2[i]);
}

Variable Documentation

ar_buf_size

constexpr size_t ar_buf_size = 1 << 16

Definition at line 49 of file allreduce.h.

Referenced by AllReduceSockets::broadcast(), AllReduceSockets::pass_down(), AllReduceSockets::pass_up(), and AllReduceSockets::reduce().