Neko 1.99.1
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
17extern ncclComm_t NEKO_COMM_NCCL;
18
19
20#endif
21
22#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