1 // Copyright (c) 2017-2024, 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 <hip/hip_runtime.h> 12 13 typedef struct { 14 hipModule_t module; 15 hipFunction_t Interp; 16 hipFunction_t InterpTranspose; 17 hipFunction_t Grad; 18 hipFunction_t GradTranspose; 19 hipFunction_t Weight; 20 CeedInt block_sizes[3]; // interp, grad, weight thread block sizes 21 CeedScalar *d_interp_1d; 22 CeedScalar *d_grad_1d; 23 CeedScalar *d_collo_grad_1d; 24 CeedScalar *d_q_weight_1d; 25 } CeedBasis_Hip_shared; 26 27 CEED_INTERN int CeedBasisCreateTensorH1_Hip_shared(CeedInt dim, CeedInt P_1d, CeedInt Q_1d, const CeedScalar *interp_1d, const CeedScalar *grad_1d, 28 const CeedScalar *q_ref_1d, const CeedScalar *q_weight_1d, CeedBasis basis); 29