device_mpi_reduce.c File Reference

#include <stdlib.h>
#include <stdio.h>
#include <mpi.h>
#include <comm/comm.h>
#include "device_mpi_op.h"

Include dependency graph for device_mpi_reduce.c:

Go to the source code of this file.

Functions
void	device_mpi_allreduce (void *buf_d, void *buf, int count, int nbytes, int op)
void	device_mpi_allreduce_inplace (void *buf_d, int count, int nbytes, int op)

Function Documentation

device_mpi_allreduce()

void device_mpi_allreduce	(	void *	buf_d,
		void *	buf,
		int	count,
		int	nbytes,
		int	op
	)

C wrapper for MPI calls, until we integrate with NCCL/RCCL

Note

We use MPI_COMM_WORLD which should be equal to NEKO_COMM.

Definition at line 47 of file device_mpi_reduce.c.

Here is the caller graph for this function:

device_mpi_allreduce_inplace()

void device_mpi_allreduce_inplace	(	void *	buf_d,
		int	count,
		int	nbytes,
		int	op
	)

Definition at line 75 of file device_mpi_reduce.c.

Here is the caller graph for this function: