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
8 // Note: intentionally testing strange spacing in include's
9 // clang-format off
10 // Note - ceed/types.h should be used over ceed.h
11 #include <ceed.h>
12 // Note - system headers like math.h and std*.h should be guarded
13 #ifndef CEED_RUNNING_JIT_PASS
14 # include <math.h>
15 #endif
16
17 #include "t406-qfunction-helper.h"
18 // Test duplicate includes of guarded files
19 // Also test include path with "/../"
20 #include "../tests/t406-qfunction-helper.h"
21 // Also test include path with "/../../"
22 #include "output/../../tests/t406-qfunction-helper.h"
23 # include "t406-qfunction-scales.h"
24 // clang-format on
25
26 // Extra define set via CeedAddJitDefine() during JiT
27 #ifndef CEED_RUNNING_JIT_PASS
28 #define COMPILER_DEFINED_SCALE 42
29 #endif
30
setup(void * ctx,const CeedInt Q,const CeedScalar * const * in,CeedScalar * const * out)31 CEED_QFUNCTION(setup)(void *ctx, const CeedInt Q, const CeedScalar *const *in, CeedScalar *const *out) {
32 const CeedScalar *w = in[0];
33 CeedScalar *q_data = out[0];
34 for (CeedInt i = 0; i < Q; i++) {
35 q_data[i] = w[i];
36 }
37 return 0;
38 }
39
mass(void * ctx,const CeedInt Q,const CeedScalar * const * in,CeedScalar * const * out)40 CEED_QFUNCTION(mass)(void *ctx, const CeedInt Q, const CeedScalar *const *in, CeedScalar *const *out) {
41 const CeedScalar *q_data = in[0], *u = in[1];
42 CeedScalar *v = out[0];
43 for (CeedInt i = 0; i < Q; i++) {
44 v[i] = q_data[i] * COMPILER_DEFINED_SCALE * (times_two(u[i]) + times_three(u[i])) * sqrt(1.0 * SCALE_TWO);
45 }
46 return 0;
47 }
48