diff --git a/examples/fluids/shallow-water/qfunctions/setup_geo.h b/examples/fluids/shallow-water/qfunctions/setup_geo.h index 144791987e..708515497a 100644 --- a/examples/fluids/shallow-water/qfunctions/setup_geo.h +++ b/examples/fluids/shallow-water/qfunctions/setup_geo.h @@ -20,6 +20,8 @@ #ifndef setupgeo_h #define setupgeo_h +#include "../sw_headers.h" + #ifndef __CUDACC__ # include #endif @@ -67,15 +69,23 @@ CEED_QFUNCTION(SetupGeo)(void *ctx, CeedInt Q, // Outputs CeedScalar (*qdata)[CEED_Q_VLA] = (CeedScalar(*)[CEED_Q_VLA])out[0]; // *INDENT-ON* + // Context + const PhysicsContext context = (PhysicsContext)ctx; + const CeedScalar R = context->R; CeedPragmaSIMD // Quadrature Point Loop for (CeedInt i=0; i