xref: /libCEED/include/ceed/jit-source/hip/hip-ref-qfunction.h (revision a0154adecfab8547cdc0febbbf40ac009dbe9d1d)
1*a0154adeSJed Brown // Copyright (c) 2017-2022, Lawrence Livermore National Security, LLC and other CEED contributors.
2*a0154adeSJed Brown // All Rights Reserved. See the top-level LICENSE and NOTICE files for details.
3*a0154adeSJed Brown //
4*a0154adeSJed Brown // SPDX-License-Identifier: BSD-2-Clause
5*a0154adeSJed Brown //
6*a0154adeSJed Brown // This file is part of CEED:  http://github.com/ceed
7*a0154adeSJed Brown 
8*a0154adeSJed Brown #include <ceed/ceed.h>
9*a0154adeSJed Brown 
10*a0154adeSJed Brown template <int SIZE>
11*a0154adeSJed Brown //------------------------------------------------------------------------------
12*a0154adeSJed Brown // Read from quadrature points
13*a0154adeSJed Brown //------------------------------------------------------------------------------
14*a0154adeSJed Brown inline __device__ void readQuads(const CeedInt quad, const CeedInt num_qpts,
15*a0154adeSJed Brown                                  const CeedScalar* d_u, CeedScalar* r_u) {
16*a0154adeSJed Brown   for(CeedInt comp = 0; comp < SIZE; comp++) {
17*a0154adeSJed Brown     r_u[comp] = d_u[quad + num_qpts * comp];
18*a0154adeSJed Brown   }
19*a0154adeSJed Brown }
20*a0154adeSJed Brown 
21*a0154adeSJed Brown //------------------------------------------------------------------------------
22*a0154adeSJed Brown // Write at quadrature points
23*a0154adeSJed Brown //------------------------------------------------------------------------------
24*a0154adeSJed Brown template <int SIZE>
25*a0154adeSJed Brown inline __device__ void writeQuads(const CeedInt quad, const CeedInt num_qpts,
26*a0154adeSJed Brown                                   const CeedScalar* r_v, CeedScalar* d_v) {
27*a0154adeSJed Brown   for(CeedInt comp = 0; comp < SIZE; comp++) {
28*a0154adeSJed Brown     d_v[quad + num_qpts * comp] = r_v[comp];
29*a0154adeSJed Brown   }
30*a0154adeSJed Brown }
31*a0154adeSJed Brown 
32*a0154adeSJed Brown //------------------------------------------------------------------------------
33