Line data    Source code

       1              : // Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
       2              : // SPDX-License-Identifier: Apache-2.0
       3              : 
       4              : #ifndef PALACE_LIBCEED_L2_3_QF_H
       5              : #define PALACE_LIBCEED_L2_3_QF_H
       6              : 
       7              : #include "../coeff/coeff_3_qf.h"
       8              : 
       9            0 : CEED_QFUNCTION(f_apply_l2_3)(void *__restrict__ ctx, CeedInt Q, const CeedScalar *const *in,
      10              :                              CeedScalar *const *out)
      11              : {
      12            0 :   const CeedScalar *attr = in[0], *wdetJ = in[0] + Q, *qw = in[1], *u = in[2];
      13            0 :   CeedScalar *v = out[0];
      14              : 
      15            0 :   CeedPragmaSIMD for (CeedInt i = 0; i < Q; i++)
      16              :   {
      17              :     CeedScalar coeff[9];
      18            0 :     CoeffUnpack3((const CeedIntScalar *)ctx, (CeedInt)attr[i], coeff);
      19            0 :     const CeedScalar w = qw[i] * qw[i] / wdetJ[i];
      20              : 
      21            0 :     const CeedScalar u0 = u[i + Q * 0];
      22            0 :     const CeedScalar u1 = u[i + Q * 1];
      23            0 :     const CeedScalar u2 = u[i + Q * 2];
      24            0 :     v[i + Q * 0] = w * (coeff[0] * u0 + coeff[3] * u1 + coeff[6] * u2);
      25            0 :     v[i + Q * 1] = w * (coeff[1] * u0 + coeff[4] * u1 + coeff[7] * u2);
      26            0 :     v[i + Q * 2] = w * (coeff[2] * u0 + coeff[5] * u1 + coeff[8] * u2);
      27              :   }
      28            0 :   return 0;
      29              : }
      30              : 
      31              : #endif  // PALACE_LIBCEED_L2_3_QF_H