Neko 1.99.5
A portable framework for high-order spectral element flow simulations
Loading...
Searching...
No Matches
comm_nccl.h
Go to the documentation of this file.
1#ifndef __COMM_NCCL_H
2#define __COMM_NCCL_H
3
4#if defined(HAVE_NCCL) || defined(HAVE_RCCL)
5
6#ifdef HAVE_NCCL
7#include <nccl.h>
8#endif
9
10#ifdef HAVE_RCCL
11#include <rccl.h>
12#endif
13
18
19
20#endif
21
22#endif
__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)