xref: /libCEED/backends/hip-gen/ceed-hip-gen.h (revision 9ba83ac0e4b1fca39d6fa6737a318a9f0cbc172d)
1 // Copyright (c) 2017-2026, 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 #pragma once
8 
9 #include <ceed.h>
10 #include <ceed/backend.h>
11 #include <ceed/jit-source/hip/hip-types.h>
12 #include <hip/hip_runtime.h>
13 
14 typedef struct {
15   bool          use_fallback, use_assembly_fallback;
16   CeedInt       dim;
17   CeedInt       Q, Q_1d;
18   CeedInt       max_P_1d;
19   CeedInt       thread_1d;
20   hipStream_t   streams[CEED_COMPOSITE_MAX];
21   hipModule_t   module, module_assemble_full, module_assemble_diagonal, module_assemble_qfunction;
22   hipFunction_t op, assemble_full, assemble_diagonal, assemble_qfunction;
23   FieldsInt_Hip indices;
24   Fields_Hip    fields;
25   Fields_Hip    B;
26   Fields_Hip    G;
27   CeedScalar   *W;
28   Points_Hip    points;
29 } CeedOperator_Hip_gen;
30 
31 typedef struct {
32   const char *qfunction_name;
33   void       *d_c;
34 } CeedQFunction_Hip_gen;
35 
36 CEED_INTERN int CeedQFunctionCreate_Hip_gen(CeedQFunction qf);
37 
38 CEED_INTERN int CeedOperatorCreate_Hip_gen(CeedOperator op);
39