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