Neko 1.99.1
A portable framework for high-order spectral element flow simulations
Loading...
Searching...
No Matches
device_nccl_reduce.h
Go to the documentation of this file.
1#ifndef __MATH_DEVICE_NCCL_REDUCE_H__
2#define __MATH_DEVICE_NCCL_REDUCE_H__
3
8extern "C" {
9void device_nccl_allreduce(void *sbuf_d, void *rbuf_d, int count,
10 int nbytes, int op, void *stream);
11}
12
13#endif
dirichlet_apply_scalar_kernel
__global__ void dirichlet_apply_scalar_kernel(const int *__restrict__ msk, T *__restrict__ x, const T g, const int m)
Definition dirichlet_kernel.h:42
device_nccl_allreduce
void device_nccl_allreduce(void *sbuf_d, void *rbuf_d, int count, int nbytes, int op, void *stream)
Definition device_nccl_reduce.c:45