1cb32e2e7SValeria Barra // Copyright (c) 2017, Lawrence Livermore National Security, LLC. Produced at 2cb32e2e7SValeria Barra // the Lawrence Livermore National Laboratory. LLNL-CODE-734707. All Rights 3cb32e2e7SValeria Barra // reserved. See files LICENSE and NOTICE for details. 4cb32e2e7SValeria Barra // 5cb32e2e7SValeria Barra // This file is part of CEED, a collection of benchmarks, miniapps, software 6cb32e2e7SValeria Barra // libraries and APIs for efficient high-order finite element and spectral 7cb32e2e7SValeria Barra // element discretizations for exascale applications. For more information and 8cb32e2e7SValeria Barra // source code availability see http://github.com/ceed. 9cb32e2e7SValeria Barra // 10cb32e2e7SValeria Barra // The CEED research is supported by the Exascale Computing Project 17-SC-20-SC, 11cb32e2e7SValeria Barra // a collaborative effort of two U.S. Department of Energy organizations (Office 12cb32e2e7SValeria Barra // of Science and the National Nuclear Security Administration) responsible for 13cb32e2e7SValeria Barra // the planning and preparation of a capable exascale ecosystem, including 14cb32e2e7SValeria Barra // software, applications, hardware, advanced system engineering and early 15cb32e2e7SValeria Barra // testbed platforms, in support of the nation's exascale computing imperative. 16cb32e2e7SValeria Barra 17cb32e2e7SValeria Barra /// @file 18cb32e2e7SValeria Barra /// libCEED QFunctions for mass operator example using PETSc 19cb32e2e7SValeria Barra 20f6b55d2cSvaleriabarra #ifndef bp2_h 21f6b55d2cSvaleriabarra #define bp2_h 22f6b55d2cSvaleriabarra 23cb32e2e7SValeria Barra #ifndef __CUDACC__ 24cb32e2e7SValeria Barra # include <math.h> 25cb32e2e7SValeria Barra #endif 26cb32e2e7SValeria Barra 27e83e87a5Sjeremylt // ----------------------------------------------------------------------------- 28ed264d09SValeria Barra // This QFunction sets up the rhs and true solution for the problem 29cb32e2e7SValeria Barra // ----------------------------------------------------------------------------- 30cb32e2e7SValeria Barra CEED_QFUNCTION(SetupMassRhs3)(void *ctx, const CeedInt Q, 31cb32e2e7SValeria Barra const CeedScalar *const *in, 32cb32e2e7SValeria Barra CeedScalar *const *out) { 33e83e87a5Sjeremylt const CeedScalar *x = in[0], *w = in[1]; 34cb32e2e7SValeria Barra CeedScalar *true_soln = out[0], *rhs = out[1]; 35cb32e2e7SValeria Barra 36cb32e2e7SValeria Barra // Quadrature Point Loop 37cb32e2e7SValeria Barra CeedPragmaSIMD 38cb32e2e7SValeria Barra for (CeedInt i=0; i<Q; i++) { 39cb32e2e7SValeria Barra // Component 1 40cb32e2e7SValeria Barra true_soln[i+0*Q] = sqrt(x[i]*x[i] + x[i+Q]*x[i+Q] + x[i+2*Q]*x[i+2*Q]); 41cb32e2e7SValeria Barra // Component 2 4282311801Sjeremylt true_soln[i+1*Q] = 2 * true_soln[i+0*Q]; 43cb32e2e7SValeria Barra // Component 3 4482311801Sjeremylt true_soln[i+2*Q] = 3 * true_soln[i+0*Q]; 45cb32e2e7SValeria Barra 46cb32e2e7SValeria Barra // Component 1 47e83e87a5Sjeremylt rhs[i+0*Q] = w[i] * true_soln[i+0*Q]; 48cb32e2e7SValeria Barra // Component 2 4982311801Sjeremylt rhs[i+1*Q] = 2 * rhs[i+0*Q]; 50cb32e2e7SValeria Barra // Component 3 5182311801Sjeremylt rhs[i+2*Q] = 3 * rhs[i+0*Q]; 52cb32e2e7SValeria Barra } // End of Quadrature Point Loop 53cb32e2e7SValeria Barra return 0; 54cb32e2e7SValeria Barra } 55cb32e2e7SValeria Barra 56e83e87a5Sjeremylt // ----------------------------------------------------------------------------- 57ed264d09SValeria Barra // This QFunction applies the mass operator for a vector field of 3 components. 58ed264d09SValeria Barra // 59ed264d09SValeria Barra // Inputs: 60ed264d09SValeria Barra // u - Input vector at quadrature points 61*9b072555Sjeremylt // q_data - Geometric factors 62ed264d09SValeria Barra // 63ed264d09SValeria Barra // Output: 64ed264d09SValeria Barra // v - Output vector (test functions) at quadrature points 65ed264d09SValeria Barra // 66cb32e2e7SValeria Barra // ----------------------------------------------------------------------------- 67cb32e2e7SValeria Barra CEED_QFUNCTION(Mass3)(void *ctx, const CeedInt Q, 68cb32e2e7SValeria Barra const CeedScalar *const *in, CeedScalar *const *out) { 69*9b072555Sjeremylt const CeedScalar *u = in[0], *q_data = in[1]; 70cb32e2e7SValeria Barra CeedScalar *v = out[0]; 71cb32e2e7SValeria Barra 72cb32e2e7SValeria Barra // Quadrature Point Loop 73cb32e2e7SValeria Barra CeedPragmaSIMD 74cb32e2e7SValeria Barra for (CeedInt i=0; i<Q; i++) { 75cb32e2e7SValeria Barra // Component 1 76*9b072555Sjeremylt v[i+0*Q] = q_data[i] * u[i+0*Q]; 77cb32e2e7SValeria Barra // Component 2 78*9b072555Sjeremylt v[i+1*Q] = q_data[i] * u[i+1*Q]; 79cb32e2e7SValeria Barra // Component 3 80*9b072555Sjeremylt v[i+2*Q] = q_data[i] * u[i+2*Q]; 81cb32e2e7SValeria Barra } // End of Quadrature Point Loop 82cb32e2e7SValeria Barra return 0; 83cb32e2e7SValeria Barra } 84cb32e2e7SValeria Barra // ----------------------------------------------------------------------------- 85f6b55d2cSvaleriabarra 86f6b55d2cSvaleriabarra #endif // bp2_h 87