vw_allreduce.h File Reference

#include "vw.h"
#include "allreduce.h"

Go to the source code of this file.

Functions
template<class T , void(*)(T &, const T &) f>
void	all_reduce (vw &all, T *buffer, const size_t n)

Function Documentation

all_reduce()

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

void all_reduce	(	vw &	all,
		T *	buffer,
		const size_t	n
	)

Definition at line 13 of file vw_allreduce.h.

References vw::all_reduce, vw::all_reduce_type, f, Socket, and Thread.

Referenced by vw::vw().

{
  switch (all.all_reduce_type)
  {
    case AllReduceType::Socket:
      ((AllReduceSockets*)all.all_reduce)->all_reduce<T, f>(buffer, n);
      break;

    case AllReduceType::Thread:
      ((AllReduceThreads*)all.all_reduce)->all_reduce<T, f>(buffer, n);
      break;
  }
}