Loading [MathJax]/extensions/tex2jax.js
Neko 0.9.99
A portable framework for high-order spectral element flow simulations
All Classes Namespaces Files Functions Variables Typedefs Enumerator Macros Pages
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 dirichlet_apply_scalar_kernel(const int *__restrict__ msk, T *__restrict__ x, const T g, const int m)