Neko 0.9.99
A portable framework for high-order spectral element flow simulations
Loading...
Searching...
No Matches
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

Definition at line 46 of file device_mpi_reduce.c.

Here is the call graph for this function:
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 74 of file device_mpi_reduce.c.

Here is the call graph for this function:
Here is the caller graph for this function: