xref: /libCEED/backends/hip-shared/ceed-hip-shared.h (revision db2becc9f302fe8eb3a32ace50ce3f3a5d42e6c4)
15aed82e4SJeremy L Thompson // Copyright (c) 2017-2024, Lawrence Livermore National Security, LLC and other CEED contributors.
23d8e8822SJeremy L Thompson // All Rights Reserved. See the top-level LICENSE and NOTICE files for details.
37d8d0e25Snbeams //
43d8e8822SJeremy L Thompson // SPDX-License-Identifier: BSD-2-Clause
57d8d0e25Snbeams //
63d8e8822SJeremy L Thompson // This file is part of CEED:  http://github.com/ceed
7509d4af6SJeremy L Thompson #pragma once
83d576824SJeremy L Thompson 
949aac155SJeremy L Thompson #include <ceed.h>
10ec3da8bcSJed Brown #include <ceed/backend.h>
113d576824SJeremy L Thompson #include <hip/hip_runtime.h>
122b730f8bSJeremy L Thompson 
137d8d0e25Snbeams typedef struct {
147d8d0e25Snbeams   hipModule_t   module;
15437930d1SJeremy L Thompson   hipFunction_t Interp;
169e201c85SYohann   hipFunction_t InterpTranspose;
17*db2becc9SJeremy L Thompson   hipFunction_t InterpTransposeAdd;
18437930d1SJeremy L Thompson   hipFunction_t Grad;
199e201c85SYohann   hipFunction_t GradTranspose;
20*db2becc9SJeremy L Thompson   hipFunction_t GradTransposeAdd;
21437930d1SJeremy L Thompson   hipFunction_t Weight;
221dda9c1aSJeremy L Thompson   hipModule_t   moduleAtPoints;
231dda9c1aSJeremy L Thompson   CeedInt       num_points;
241dda9c1aSJeremy L Thompson   hipFunction_t InterpAtPoints;
251dda9c1aSJeremy L Thompson   hipFunction_t GradAtPoints;
26437930d1SJeremy L Thompson   CeedInt       block_sizes[3];  // interp, grad, weight thread block sizes
27437930d1SJeremy L Thompson   CeedScalar   *d_interp_1d;
28437930d1SJeremy L Thompson   CeedScalar   *d_grad_1d;
29437930d1SJeremy L Thompson   CeedScalar   *d_collo_grad_1d;
30437930d1SJeremy L Thompson   CeedScalar   *d_q_weight_1d;
311dda9c1aSJeremy L Thompson   CeedScalar   *d_chebyshev_interp_1d;
327d8d0e25Snbeams } CeedBasis_Hip_shared;
337d8d0e25Snbeams 
346574a04fSJeremy L Thompson CEED_INTERN int CeedBasisCreateTensorH1_Hip_shared(CeedInt dim, CeedInt P_1d, CeedInt Q_1d, const CeedScalar *interp_1d, const CeedScalar *grad_1d,
356574a04fSJeremy L Thompson                                                    const CeedScalar *q_ref_1d, const CeedScalar *q_weight_1d, CeedBasis basis);
36