Neko
0.9.0
A portable framework for high-order spectral element flow simulations
|
#include <stdlib.h>
#include <stdio.h>
#include <mpi.h>
#include <comm/comm.h>
#include "device_mpi_op.h"
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) |
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.
void device_mpi_allreduce_inplace | ( | void * | buf_d, |
int | count, | ||
int | nbytes, | ||
int | op | ||
) |