Neko 1.99.5
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
ale_add_kinematics_kernel
__global__ void ale_add_kinematics_kernel(const int n, T *__restrict__ wx, T *__restrict__ wy, T *__restrict__ wz, const T *__restrict__ x_ref, const T *__restrict__ y_ref, const T *__restrict__ z_ref, const T *__restrict__ phi, const T *__restrict__ x, const T *__restrict__ y, const T *__restrict__ z, const kinematics_params_t kin_params)
Definition ale_kinematics_kernel.h:52
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