xref: /libCEED/backends/cuda-shared/ceed-cuda-shared.h (revision 94c017353f70642ff53584f4c2920d59bb75d965)
1 // Copyright (c) 2017-2022, Lawrence Livermore National Security, LLC and other CEED contributors.
2 // All Rights Reserved. See the top-level LICENSE and NOTICE files for details.
3 //
4 // SPDX-License-Identifier: BSD-2-Clause
5 //
6 // This file is part of CEED:  http://github.com/ceed
7 
8 #ifndef _ceed_cuda_shared_h
9 #define _ceed_cuda_shared_h
10 
11 #include <ceed/backend.h>
12 #include <ceed/ceed.h>
13 #include <cuda.h>
14 
15 #include "../cuda/ceed-cuda-common.h"
16 
17 typedef struct {
18   CUmodule    module;
19   CUfunction  Interp;
20   CUfunction  InterpTranspose;
21   CUfunction  Grad;
22   CUfunction  GradTranspose;
23   CUfunction  Weight;
24   CeedScalar *d_interp_1d;
25   CeedScalar *d_grad_1d;
26   CeedScalar *d_collo_grad_1d;
27   CeedScalar *d_q_weight_1d;
28   CeedScalar *c_B;
29   CeedScalar *c_G;
30 } CeedBasis_Cuda_shared;
31 
32 CEED_INTERN int CeedBasisCreateTensorH1_Cuda_shared(CeedInt dim, CeedInt P_1d, CeedInt Q_1d, const CeedScalar *interp_1d, const CeedScalar *grad_1d,
33                                                     const CeedScalar *q_ref_1d, const CeedScalar *q_weight_1d, CeedBasis basis);
34 
35 #endif  // _ceed_cuda_shared_h
36