13d8e8822SJeremy L Thompson // Copyright (c) 2017-2022, Lawrence Livermore National Security, LLC and other CEED contributors. 23d8e8822SJeremy L Thompson // All Rights Reserved. See the top-level LICENSE and NOTICE files for details. 321617c04Sjeremylt // 43d8e8822SJeremy L Thompson // SPDX-License-Identifier: BSD-2-Clause 521617c04Sjeremylt // 63d8e8822SJeremy L Thompson // This file is part of CEED: http://github.com/ceed 721617c04Sjeremylt 849aac155SJeremy L Thompson #include <ceed.h> 9ec3da8bcSJed Brown #include <ceed/backend.h> 103d576824SJeremy L Thompson #include <stdbool.h> 113d576824SJeremy L Thompson #include <stddef.h> 123d576824SJeremy L Thompson #include <stdint.h> 132b730f8bSJeremy L Thompson 1421617c04Sjeremylt #include "ceed-ref.h" 1521617c04Sjeremylt 16f10650afSjeremylt //------------------------------------------------------------------------------ 17f10650afSjeremylt // Setup Input/Output Fields 18f10650afSjeremylt //------------------------------------------------------------------------------ 192b730f8bSJeremy L Thompson static int CeedOperatorSetupFields_Ref(CeedQFunction qf, CeedOperator op, bool is_input, CeedVector *e_vecs_full, CeedVector *e_vecs, 202b730f8bSJeremy L Thompson CeedVector *q_vecs, CeedInt start_e, CeedInt num_fields, CeedInt Q) { 21aedaa0e5Sjeremylt Ceed ceed; 22ad70ee2cSJeremy L Thompson CeedSize e_size, q_size; 23ad70ee2cSJeremy L Thompson CeedInt num_comp, size, P; 24d1d35e2fSjeremylt CeedQFunctionField *qf_fields; 25ad70ee2cSJeremy L Thompson CeedOperatorField *op_fields; 26ad70ee2cSJeremy L Thompson 27e910d748SJeremy L Thompson { 28e910d748SJeremy L Thompson Ceed ceed_parent; 29e910d748SJeremy L Thompson 30ad70ee2cSJeremy L Thompson CeedCallBackend(CeedOperatorGetCeed(op, &ceed)); 31e910d748SJeremy L Thompson CeedCallBackend(CeedGetParent(ceed, &ceed_parent)); 32e910d748SJeremy L Thompson if (ceed_parent) ceed = ceed_parent; 33e910d748SJeremy L Thompson } 344fc1f125SJeremy L Thompson if (is_input) { 352b730f8bSJeremy L Thompson CeedCallBackend(CeedOperatorGetFields(op, NULL, &op_fields, NULL, NULL)); 362b730f8bSJeremy L Thompson CeedCallBackend(CeedQFunctionGetFields(qf, NULL, &qf_fields, NULL, NULL)); 374fc1f125SJeremy L Thompson } else { 382b730f8bSJeremy L Thompson CeedCallBackend(CeedOperatorGetFields(op, NULL, NULL, NULL, &op_fields)); 392b730f8bSJeremy L Thompson CeedCallBackend(CeedQFunctionGetFields(qf, NULL, NULL, NULL, &qf_fields)); 40fe2413ffSjeremylt } 4121617c04Sjeremylt 42885ac19cSjeremylt // Loop over fields 43d1d35e2fSjeremylt for (CeedInt i = 0; i < num_fields; i++) { 44d1d35e2fSjeremylt CeedEvalMode eval_mode; 45edb2538eSJeremy L Thompson CeedElemRestriction elem_rstr; 46ad70ee2cSJeremy L Thompson CeedBasis basis; 47d1d35e2fSjeremylt 48ad70ee2cSJeremy L Thompson CeedCallBackend(CeedQFunctionFieldGetEvalMode(qf_fields[i], &eval_mode)); 49d1d35e2fSjeremylt if (eval_mode != CEED_EVAL_WEIGHT) { 50edb2538eSJeremy L Thompson CeedCallBackend(CeedOperatorFieldGetElemRestriction(op_fields[i], &elem_rstr)); 51edb2538eSJeremy L Thompson CeedCallBackend(CeedElemRestrictionCreateVector(elem_rstr, NULL, &e_vecs_full[i + start_e])); 52135a076eSjeremylt } 53135a076eSjeremylt 54d1d35e2fSjeremylt switch (eval_mode) { 55885ac19cSjeremylt case CEED_EVAL_NONE: 562b730f8bSJeremy L Thompson CeedCallBackend(CeedQFunctionFieldGetSize(qf_fields[i], &size)); 57d2643443SJeremy L Thompson q_size = (CeedSize)Q * size; 582b730f8bSJeremy L Thompson CeedCallBackend(CeedVectorCreate(ceed, q_size, &q_vecs[i])); 59aedaa0e5Sjeremylt break; 60aedaa0e5Sjeremylt case CEED_EVAL_INTERP: 61885ac19cSjeremylt case CEED_EVAL_GRAD: 62a915a514Srezgarshakeri case CEED_EVAL_DIV: 63c4e3f59bSSebastian Grimberg case CEED_EVAL_CURL: 642b730f8bSJeremy L Thompson CeedCallBackend(CeedOperatorFieldGetBasis(op_fields[i], &basis)); 652b730f8bSJeremy L Thompson CeedCallBackend(CeedQFunctionFieldGetSize(qf_fields[i], &size)); 662b730f8bSJeremy L Thompson CeedCallBackend(CeedBasisGetNumNodes(basis, &P)); 672b730f8bSJeremy L Thompson CeedCallBackend(CeedBasisGetNumComponents(basis, &num_comp)); 68d2643443SJeremy L Thompson e_size = (CeedSize)P * num_comp; 692b730f8bSJeremy L Thompson CeedCallBackend(CeedVectorCreate(ceed, e_size, &e_vecs[i])); 70d2643443SJeremy L Thompson q_size = (CeedSize)Q * size; 712b730f8bSJeremy L Thompson CeedCallBackend(CeedVectorCreate(ceed, q_size, &q_vecs[i])); 72885ac19cSjeremylt break; 73885ac19cSjeremylt case CEED_EVAL_WEIGHT: // Only on input fields 742b730f8bSJeremy L Thompson CeedCallBackend(CeedOperatorFieldGetBasis(op_fields[i], &basis)); 75d2643443SJeremy L Thompson q_size = (CeedSize)Q; 762b730f8bSJeremy L Thompson CeedCallBackend(CeedVectorCreate(ceed, q_size, &q_vecs[i])); 772b730f8bSJeremy L Thompson CeedCallBackend(CeedBasisApply(basis, 1, CEED_NOTRANSPOSE, CEED_EVAL_WEIGHT, CEED_VECTOR_NONE, q_vecs[i])); 78885ac19cSjeremylt break; 7921617c04Sjeremylt } 80885ac19cSjeremylt } 81e15f9bd0SJeremy L Thompson return CEED_ERROR_SUCCESS; 8221617c04Sjeremylt } 8321617c04Sjeremylt 84f10650afSjeremylt //------------------------------------------------------------------------------ 85f10650afSjeremylt // Setup Operator 86f10650afSjeremylt //------------------------------------------------------------------------------/* 87885ac19cSjeremylt static int CeedOperatorSetup_Ref(CeedOperator op) { 888c1105f8SJeremy L Thompson bool is_setup_done; 89ad70ee2cSJeremy L Thompson Ceed ceed; 90ad70ee2cSJeremy L Thompson CeedInt Q, num_input_fields, num_output_fields; 91ad70ee2cSJeremy L Thompson CeedQFunctionField *qf_input_fields, *qf_output_fields; 92ad70ee2cSJeremy L Thompson CeedQFunction qf; 93ad70ee2cSJeremy L Thompson CeedOperatorField *op_input_fields, *op_output_fields; 94ad70ee2cSJeremy L Thompson CeedOperator_Ref *impl; 95ad70ee2cSJeremy L Thompson 962b730f8bSJeremy L Thompson CeedCallBackend(CeedOperatorIsSetupDone(op, &is_setup_done)); 978c1105f8SJeremy L Thompson if (is_setup_done) return CEED_ERROR_SUCCESS; 98ad70ee2cSJeremy L Thompson 992b730f8bSJeremy L Thompson CeedCallBackend(CeedOperatorGetCeed(op, &ceed)); 1002b730f8bSJeremy L Thompson CeedCallBackend(CeedOperatorGetData(op, &impl)); 1012b730f8bSJeremy L Thompson CeedCallBackend(CeedOperatorGetQFunction(op, &qf)); 1022b730f8bSJeremy L Thompson CeedCallBackend(CeedOperatorGetNumQuadraturePoints(op, &Q)); 1032b730f8bSJeremy L Thompson CeedCallBackend(CeedQFunctionIsIdentity(qf, &impl->is_identity_qf)); 1042b730f8bSJeremy L Thompson CeedCallBackend(CeedOperatorGetFields(op, &num_input_fields, &op_input_fields, &num_output_fields, &op_output_fields)); 1052b730f8bSJeremy L Thompson CeedCallBackend(CeedQFunctionGetFields(qf, NULL, &qf_input_fields, NULL, &qf_output_fields)); 106885ac19cSjeremylt 107885ac19cSjeremylt // Allocate 1082b730f8bSJeremy L Thompson CeedCallBackend(CeedCalloc(num_input_fields + num_output_fields, &impl->e_vecs_full)); 109885ac19cSjeremylt 1102b730f8bSJeremy L Thompson CeedCallBackend(CeedCalloc(CEED_FIELD_MAX, &impl->input_states)); 1112b730f8bSJeremy L Thompson CeedCallBackend(CeedCalloc(CEED_FIELD_MAX, &impl->e_vecs_in)); 1122b730f8bSJeremy L Thompson CeedCallBackend(CeedCalloc(CEED_FIELD_MAX, &impl->e_vecs_out)); 1132b730f8bSJeremy L Thompson CeedCallBackend(CeedCalloc(CEED_FIELD_MAX, &impl->q_vecs_in)); 1142b730f8bSJeremy L Thompson CeedCallBackend(CeedCalloc(CEED_FIELD_MAX, &impl->q_vecs_out)); 115885ac19cSjeremylt 1164fc1f125SJeremy L Thompson impl->num_inputs = num_input_fields; 1174fc1f125SJeremy L Thompson impl->num_outputs = num_output_fields; 118885ac19cSjeremylt 119d1d35e2fSjeremylt // Set up infield and outfield e_vecs and q_vecs 120885ac19cSjeremylt // Infields 1212b730f8bSJeremy L Thompson CeedCallBackend(CeedOperatorSetupFields_Ref(qf, op, true, impl->e_vecs_full, impl->e_vecs_in, impl->q_vecs_in, 0, num_input_fields, Q)); 122885ac19cSjeremylt // Outfields 1232b730f8bSJeremy L Thompson CeedCallBackend( 1242b730f8bSJeremy L Thompson CeedOperatorSetupFields_Ref(qf, op, false, impl->e_vecs_full, impl->e_vecs_out, impl->q_vecs_out, num_input_fields, num_output_fields, Q)); 125885ac19cSjeremylt 12616911fdaSjeremylt // Identity QFunctions 1270b454692Sjeremylt if (impl->is_identity_qf) { 128d1d35e2fSjeremylt CeedEvalMode in_mode, out_mode; 129d1d35e2fSjeremylt CeedQFunctionField *in_fields, *out_fields; 130ad70ee2cSJeremy L Thompson 1312b730f8bSJeremy L Thompson CeedCallBackend(CeedQFunctionGetFields(qf, NULL, &in_fields, NULL, &out_fields)); 1322b730f8bSJeremy L Thompson CeedCallBackend(CeedQFunctionFieldGetEvalMode(in_fields[0], &in_mode)); 1332b730f8bSJeremy L Thompson CeedCallBackend(CeedQFunctionFieldGetEvalMode(out_fields[0], &out_mode)); 134d1d35e2fSjeremylt 1350b454692Sjeremylt if (in_mode == CEED_EVAL_NONE && out_mode == CEED_EVAL_NONE) { 136edb2538eSJeremy L Thompson impl->is_identity_rstr_op = true; 1370b454692Sjeremylt } else { 138db002c03SJeremy L Thompson CeedCallBackend(CeedVectorReferenceCopy(impl->q_vecs_in[0], &impl->q_vecs_out[0])); 13916911fdaSjeremylt } 14016911fdaSjeremylt } 14116911fdaSjeremylt 1422b730f8bSJeremy L Thompson CeedCallBackend(CeedOperatorSetSetupDone(op)); 143e15f9bd0SJeremy L Thompson return CEED_ERROR_SUCCESS; 144885ac19cSjeremylt } 145885ac19cSjeremylt 146f10650afSjeremylt //------------------------------------------------------------------------------ 147f10650afSjeremylt // Setup Operator Inputs 148f10650afSjeremylt //------------------------------------------------------------------------------ 1492b730f8bSJeremy L Thompson static inline int CeedOperatorSetupInputs_Ref(CeedInt num_input_fields, CeedQFunctionField *qf_input_fields, CeedOperatorField *op_input_fields, 1502b730f8bSJeremy L Thompson CeedVector in_vec, const bool skip_active, CeedScalar *e_data_full[2 * CEED_FIELD_MAX], 151a0162de9SJeremy L Thompson CeedOperator_Ref *impl, CeedRequest *request) { 152ad70ee2cSJeremy L Thompson for (CeedInt i = 0; i < num_input_fields; i++) { 153ad70ee2cSJeremy L Thompson uint64_t state; 154d1d35e2fSjeremylt CeedEvalMode eval_mode; 155d1bcdac9Sjeremylt CeedVector vec; 156edb2538eSJeremy L Thompson CeedElemRestriction elem_rstr; 157885ac19cSjeremylt 158d1bcdac9Sjeremylt // Get input vector 1592b730f8bSJeremy L Thompson CeedCallBackend(CeedOperatorFieldGetVector(op_input_fields[i], &vec)); 1601d102b48SJeremy L Thompson if (vec == CEED_VECTOR_ACTIVE) { 1612b730f8bSJeremy L Thompson if (skip_active) continue; 1622b730f8bSJeremy L Thompson else vec = in_vec; 1631d102b48SJeremy L Thompson } 1641d102b48SJeremy L Thompson 1652b730f8bSJeremy L Thompson CeedCallBackend(CeedQFunctionFieldGetEvalMode(qf_input_fields[i], &eval_mode)); 1661d102b48SJeremy L Thompson // Restrict and Evec 167d1d35e2fSjeremylt if (eval_mode == CEED_EVAL_WEIGHT) { // Skip 1681d102b48SJeremy L Thompson } else { 169668048e2SJed Brown // Restrict 1702b730f8bSJeremy L Thompson CeedCallBackend(CeedVectorGetState(vec, &state)); 1718d713cf6Sjeremylt // Skip restriction if input is unchanged 1724fc1f125SJeremy L Thompson if (state != impl->input_states[i] || vec == in_vec) { 173edb2538eSJeremy L Thompson CeedCallBackend(CeedOperatorFieldGetElemRestriction(op_input_fields[i], &elem_rstr)); 174edb2538eSJeremy L Thompson CeedCallBackend(CeedElemRestrictionApply(elem_rstr, CEED_NOTRANSPOSE, vec, impl->e_vecs_full[i], request)); 1754fc1f125SJeremy L Thompson impl->input_states[i] = state; 1768d713cf6Sjeremylt } 177668048e2SJed Brown // Get evec 1782b730f8bSJeremy L Thompson CeedCallBackend(CeedVectorGetArrayRead(impl->e_vecs_full[i], CEED_MEM_HOST, (const CeedScalar **)&e_data_full[i])); 179885ac19cSjeremylt } 180885ac19cSjeremylt } 181e15f9bd0SJeremy L Thompson return CEED_ERROR_SUCCESS; 182885ac19cSjeremylt } 183885ac19cSjeremylt 184f10650afSjeremylt //------------------------------------------------------------------------------ 185f10650afSjeremylt // Input Basis Action 186f10650afSjeremylt //------------------------------------------------------------------------------ 1872b730f8bSJeremy L Thompson static inline int CeedOperatorInputBasis_Ref(CeedInt e, CeedInt Q, CeedQFunctionField *qf_input_fields, CeedOperatorField *op_input_fields, 1882b730f8bSJeremy L Thompson CeedInt num_input_fields, const bool skip_active, CeedScalar *e_data_full[2 * CEED_FIELD_MAX], 1892b730f8bSJeremy L Thompson CeedOperator_Ref *impl) { 190ad70ee2cSJeremy L Thompson for (CeedInt i = 0; i < num_input_fields; i++) { 191a915a514Srezgarshakeri CeedInt elem_size, size, num_comp; 192d1d35e2fSjeremylt CeedEvalMode eval_mode; 193edb2538eSJeremy L Thompson CeedElemRestriction elem_rstr; 1941d102b48SJeremy L Thompson CeedBasis basis; 1951d102b48SJeremy L Thompson 1961d102b48SJeremy L Thompson // Skip active input 197d1d35e2fSjeremylt if (skip_active) { 1981d102b48SJeremy L Thompson CeedVector vec; 199ad70ee2cSJeremy L Thompson 2002b730f8bSJeremy L Thompson CeedCallBackend(CeedOperatorFieldGetVector(op_input_fields[i], &vec)); 2012b730f8bSJeremy L Thompson if (vec == CEED_VECTOR_ACTIVE) continue; 2021d102b48SJeremy L Thompson } 203d1d35e2fSjeremylt // Get elem_size, eval_mode, size 204edb2538eSJeremy L Thompson CeedCallBackend(CeedOperatorFieldGetElemRestriction(op_input_fields[i], &elem_rstr)); 205edb2538eSJeremy L Thompson CeedCallBackend(CeedElemRestrictionGetElementSize(elem_rstr, &elem_size)); 2062b730f8bSJeremy L Thompson CeedCallBackend(CeedQFunctionFieldGetEvalMode(qf_input_fields[i], &eval_mode)); 2072b730f8bSJeremy L Thompson CeedCallBackend(CeedQFunctionFieldGetSize(qf_input_fields[i], &size)); 208885ac19cSjeremylt // Basis action 209d1d35e2fSjeremylt switch (eval_mode) { 210885ac19cSjeremylt case CEED_EVAL_NONE: 2112b730f8bSJeremy L Thompson CeedCallBackend(CeedVectorSetArray(impl->q_vecs_in[i], CEED_MEM_HOST, CEED_USE_POINTER, &e_data_full[i][e * Q * size])); 212885ac19cSjeremylt break; 213885ac19cSjeremylt case CEED_EVAL_INTERP: 214885ac19cSjeremylt case CEED_EVAL_GRAD: 215a915a514Srezgarshakeri case CEED_EVAL_DIV: 216c4e3f59bSSebastian Grimberg case CEED_EVAL_CURL: 217a915a514Srezgarshakeri CeedCallBackend(CeedOperatorFieldGetBasis(op_input_fields[i], &basis)); 218a915a514Srezgarshakeri CeedCallBackend(CeedBasisGetNumComponents(basis, &num_comp)); 219a915a514Srezgarshakeri CeedCallBackend(CeedVectorSetArray(impl->e_vecs_in[i], CEED_MEM_HOST, CEED_USE_POINTER, &e_data_full[i][e * elem_size * num_comp])); 220c4e3f59bSSebastian Grimberg CeedCallBackend(CeedBasisApply(basis, 1, CEED_NOTRANSPOSE, eval_mode, impl->e_vecs_in[i], impl->q_vecs_in[i])); 221a915a514Srezgarshakeri break; 222885ac19cSjeremylt case CEED_EVAL_WEIGHT: 223885ac19cSjeremylt break; // No action 224885ac19cSjeremylt } 225885ac19cSjeremylt } 226e15f9bd0SJeremy L Thompson return CEED_ERROR_SUCCESS; 227885ac19cSjeremylt } 228885ac19cSjeremylt 229f10650afSjeremylt //------------------------------------------------------------------------------ 230f10650afSjeremylt // Output Basis Action 231f10650afSjeremylt //------------------------------------------------------------------------------ 2322b730f8bSJeremy L Thompson static inline int CeedOperatorOutputBasis_Ref(CeedInt e, CeedInt Q, CeedQFunctionField *qf_output_fields, CeedOperatorField *op_output_fields, 233d1d35e2fSjeremylt CeedInt num_input_fields, CeedInt num_output_fields, CeedOperator op, 2344fc1f125SJeremy L Thompson CeedScalar *e_data_full[2 * CEED_FIELD_MAX], CeedOperator_Ref *impl) { 235ad70ee2cSJeremy L Thompson for (CeedInt i = 0; i < num_output_fields; i++) { 236a915a514Srezgarshakeri CeedInt elem_size, num_comp; 237d1d35e2fSjeremylt CeedEvalMode eval_mode; 238edb2538eSJeremy L Thompson CeedElemRestriction elem_rstr; 2391d102b48SJeremy L Thompson CeedBasis basis; 2401d102b48SJeremy L Thompson 241a915a514Srezgarshakeri // Get elem_size, eval_mode 242edb2538eSJeremy L Thompson CeedCallBackend(CeedOperatorFieldGetElemRestriction(op_output_fields[i], &elem_rstr)); 243edb2538eSJeremy L Thompson CeedCallBackend(CeedElemRestrictionGetElementSize(elem_rstr, &elem_size)); 2442b730f8bSJeremy L Thompson CeedCallBackend(CeedQFunctionFieldGetEvalMode(qf_output_fields[i], &eval_mode)); 245885ac19cSjeremylt // Basis action 246d1d35e2fSjeremylt switch (eval_mode) { 247885ac19cSjeremylt case CEED_EVAL_NONE: 248885ac19cSjeremylt break; // No action 249885ac19cSjeremylt case CEED_EVAL_INTERP: 250885ac19cSjeremylt case CEED_EVAL_GRAD: 251a915a514Srezgarshakeri case CEED_EVAL_DIV: 252c4e3f59bSSebastian Grimberg case CEED_EVAL_CURL: 253a915a514Srezgarshakeri CeedCallBackend(CeedOperatorFieldGetBasis(op_output_fields[i], &basis)); 254a915a514Srezgarshakeri CeedCallBackend(CeedBasisGetNumComponents(basis, &num_comp)); 255a915a514Srezgarshakeri CeedCallBackend( 256a915a514Srezgarshakeri CeedVectorSetArray(impl->e_vecs_out[i], CEED_MEM_HOST, CEED_USE_POINTER, &e_data_full[i + num_input_fields][e * elem_size * num_comp])); 257c4e3f59bSSebastian Grimberg CeedCallBackend(CeedBasisApply(basis, 1, CEED_TRANSPOSE, eval_mode, impl->q_vecs_out[i], impl->e_vecs_out[i])); 258a915a514Srezgarshakeri break; 259c042f62fSJeremy L Thompson // LCOV_EXCL_START 260bbfacfcdSjeremylt case CEED_EVAL_WEIGHT: { 2614ce2993fSjeremylt Ceed ceed; 262ad70ee2cSJeremy L Thompson 2632b730f8bSJeremy L Thompson CeedCallBackend(CeedOperatorGetCeed(op, &ceed)); 2646574a04fSJeremy L Thompson return CeedError(ceed, CEED_ERROR_BACKEND, "CEED_EVAL_WEIGHT cannot be an output evaluation mode"); 2651d102b48SJeremy L Thompson // LCOV_EXCL_STOP 266885ac19cSjeremylt } 267885ac19cSjeremylt } 268885ac19cSjeremylt } 269e15f9bd0SJeremy L Thompson return CEED_ERROR_SUCCESS; 2701d102b48SJeremy L Thompson } 2711d102b48SJeremy L Thompson 272f10650afSjeremylt //------------------------------------------------------------------------------ 273f10650afSjeremylt // Restore Input Vectors 274f10650afSjeremylt //------------------------------------------------------------------------------ 2752b730f8bSJeremy L Thompson static inline int CeedOperatorRestoreInputs_Ref(CeedInt num_input_fields, CeedQFunctionField *qf_input_fields, CeedOperatorField *op_input_fields, 2762b730f8bSJeremy L Thompson const bool skip_active, CeedScalar *e_data_full[2 * CEED_FIELD_MAX], CeedOperator_Ref *impl) { 277ad70ee2cSJeremy L Thompson for (CeedInt i = 0; i < num_input_fields; i++) { 278d1d35e2fSjeremylt CeedEvalMode eval_mode; 2791d102b48SJeremy L Thompson 2801d102b48SJeremy L Thompson // Skip active inputs 281d1d35e2fSjeremylt if (skip_active) { 2821d102b48SJeremy L Thompson CeedVector vec; 283ad70ee2cSJeremy L Thompson 2842b730f8bSJeremy L Thompson CeedCallBackend(CeedOperatorFieldGetVector(op_input_fields[i], &vec)); 2852b730f8bSJeremy L Thompson if (vec == CEED_VECTOR_ACTIVE) continue; 2861d102b48SJeremy L Thompson } 2871d102b48SJeremy L Thompson // Restore input 2882b730f8bSJeremy L Thompson CeedCallBackend(CeedQFunctionFieldGetEvalMode(qf_input_fields[i], &eval_mode)); 289d1d35e2fSjeremylt if (eval_mode == CEED_EVAL_WEIGHT) { // Skip 2901d102b48SJeremy L Thompson } else { 2912b730f8bSJeremy L Thompson CeedCallBackend(CeedVectorRestoreArrayRead(impl->e_vecs_full[i], (const CeedScalar **)&e_data_full[i])); 2921d102b48SJeremy L Thompson } 2931d102b48SJeremy L Thompson } 294e15f9bd0SJeremy L Thompson return CEED_ERROR_SUCCESS; 2951d102b48SJeremy L Thompson } 2961d102b48SJeremy L Thompson 297f10650afSjeremylt //------------------------------------------------------------------------------ 298f10650afSjeremylt // Operator Apply 299f10650afSjeremylt //------------------------------------------------------------------------------ 3002b730f8bSJeremy L Thompson static int CeedOperatorApplyAdd_Ref(CeedOperator op, CeedVector in_vec, CeedVector out_vec, CeedRequest *request) { 301d1d35e2fSjeremylt CeedInt Q, num_elem, num_input_fields, num_output_fields, size; 302ad70ee2cSJeremy L Thompson CeedEvalMode eval_mode; 303ad70ee2cSJeremy L Thompson CeedScalar *e_data_full[2 * CEED_FIELD_MAX] = {NULL}; 304ad70ee2cSJeremy L Thompson CeedQFunctionField *qf_input_fields, *qf_output_fields; 305ad70ee2cSJeremy L Thompson CeedQFunction qf; 306ad70ee2cSJeremy L Thompson CeedOperatorField *op_input_fields, *op_output_fields; 307ad70ee2cSJeremy L Thompson CeedOperator_Ref *impl; 308ad70ee2cSJeremy L Thompson 309ad70ee2cSJeremy L Thompson CeedCallBackend(CeedOperatorGetData(op, &impl)); 310ad70ee2cSJeremy L Thompson CeedCallBackend(CeedOperatorGetQFunction(op, &qf)); 3112b730f8bSJeremy L Thompson CeedCallBackend(CeedOperatorGetNumQuadraturePoints(op, &Q)); 3122b730f8bSJeremy L Thompson CeedCallBackend(CeedOperatorGetNumElements(op, &num_elem)); 3132b730f8bSJeremy L Thompson CeedCallBackend(CeedOperatorGetFields(op, &num_input_fields, &op_input_fields, &num_output_fields, &op_output_fields)); 3142b730f8bSJeremy L Thompson CeedCallBackend(CeedQFunctionGetFields(qf, NULL, &qf_input_fields, NULL, &qf_output_fields)); 3151d102b48SJeremy L Thompson 3161d102b48SJeremy L Thompson // Setup 3172b730f8bSJeremy L Thompson CeedCallBackend(CeedOperatorSetup_Ref(op)); 3181d102b48SJeremy L Thompson 3190b454692Sjeremylt // Restriction only operator 320edb2538eSJeremy L Thompson if (impl->is_identity_rstr_op) { 321edb2538eSJeremy L Thompson CeedElemRestriction elem_rstr; 322ad70ee2cSJeremy L Thompson 323edb2538eSJeremy L Thompson CeedCallBackend(CeedOperatorFieldGetElemRestriction(op_input_fields[0], &elem_rstr)); 324edb2538eSJeremy L Thompson CeedCallBackend(CeedElemRestrictionApply(elem_rstr, CEED_NOTRANSPOSE, in_vec, impl->e_vecs_full[0], request)); 325edb2538eSJeremy L Thompson CeedCallBackend(CeedOperatorFieldGetElemRestriction(op_output_fields[0], &elem_rstr)); 326edb2538eSJeremy L Thompson CeedCallBackend(CeedElemRestrictionApply(elem_rstr, CEED_TRANSPOSE, impl->e_vecs_full[0], out_vec, request)); 3270b454692Sjeremylt return CEED_ERROR_SUCCESS; 3280b454692Sjeremylt } 3290b454692Sjeremylt 3301d102b48SJeremy L Thompson // Input Evecs and Restriction 3312b730f8bSJeremy L Thompson CeedCallBackend(CeedOperatorSetupInputs_Ref(num_input_fields, qf_input_fields, op_input_fields, in_vec, false, e_data_full, impl, request)); 3321d102b48SJeremy L Thompson 3331d102b48SJeremy L Thompson // Output Evecs 334d1d35e2fSjeremylt for (CeedInt i = 0; i < num_output_fields; i++) { 3352b730f8bSJeremy L Thompson CeedCallBackend(CeedVectorGetArrayWrite(impl->e_vecs_full[i + impl->num_inputs], CEED_MEM_HOST, &e_data_full[i + num_input_fields])); 3361d102b48SJeremy L Thompson } 3371d102b48SJeremy L Thompson 3381d102b48SJeremy L Thompson // Loop through elements 339d1d35e2fSjeremylt for (CeedInt e = 0; e < num_elem; e++) { 3401d102b48SJeremy L Thompson // Output pointers 341d1d35e2fSjeremylt for (CeedInt i = 0; i < num_output_fields; i++) { 3422b730f8bSJeremy L Thompson CeedCallBackend(CeedQFunctionFieldGetEvalMode(qf_output_fields[i], &eval_mode)); 343d1d35e2fSjeremylt if (eval_mode == CEED_EVAL_NONE) { 3442b730f8bSJeremy L Thompson CeedCallBackend(CeedQFunctionFieldGetSize(qf_output_fields[i], &size)); 3452b730f8bSJeremy L Thompson CeedCallBackend(CeedVectorSetArray(impl->q_vecs_out[i], CEED_MEM_HOST, CEED_USE_POINTER, &e_data_full[i + num_input_fields][e * Q * size])); 3461d102b48SJeremy L Thompson } 3471d102b48SJeremy L Thompson } 3481d102b48SJeremy L Thompson 34916911fdaSjeremylt // Input basis apply 3502b730f8bSJeremy L Thompson CeedCallBackend(CeedOperatorInputBasis_Ref(e, Q, qf_input_fields, op_input_fields, num_input_fields, false, e_data_full, impl)); 35116911fdaSjeremylt 3521d102b48SJeremy L Thompson // Q function 3530b454692Sjeremylt if (!impl->is_identity_qf) { 3542b730f8bSJeremy L Thompson CeedCallBackend(CeedQFunctionApply(qf, Q, impl->q_vecs_in, impl->q_vecs_out)); 35516911fdaSjeremylt } 3561d102b48SJeremy L Thompson 3571d102b48SJeremy L Thompson // Output basis apply 3582b730f8bSJeremy L Thompson CeedCallBackend( 3592b730f8bSJeremy L Thompson CeedOperatorOutputBasis_Ref(e, Q, qf_output_fields, op_output_fields, num_input_fields, num_output_fields, op, e_data_full, impl)); 3601d102b48SJeremy L Thompson } 361885ac19cSjeremylt 362885ac19cSjeremylt // Output restriction 363d1d35e2fSjeremylt for (CeedInt i = 0; i < num_output_fields; i++) { 364ad70ee2cSJeremy L Thompson CeedVector vec; 365edb2538eSJeremy L Thompson CeedElemRestriction elem_rstr; 366ad70ee2cSJeremy L Thompson 367d1d35e2fSjeremylt // Restore Evec 3682b730f8bSJeremy L Thompson CeedCallBackend(CeedVectorRestoreArray(impl->e_vecs_full[i + impl->num_inputs], &e_data_full[i + num_input_fields])); 369d1bcdac9Sjeremylt // Get output vector 3702b730f8bSJeremy L Thompson CeedCallBackend(CeedOperatorFieldGetVector(op_output_fields[i], &vec)); 371668048e2SJed Brown // Active 3722b730f8bSJeremy L Thompson if (vec == CEED_VECTOR_ACTIVE) vec = out_vec; 3737ca8db16Sjeremylt // Restrict 374edb2538eSJeremy L Thompson CeedCallBackend(CeedOperatorFieldGetElemRestriction(op_output_fields[i], &elem_rstr)); 375edb2538eSJeremy L Thompson CeedCallBackend(CeedElemRestrictionApply(elem_rstr, CEED_TRANSPOSE, impl->e_vecs_full[i + impl->num_inputs], vec, request)); 376885ac19cSjeremylt } 377885ac19cSjeremylt 3787ca8db16Sjeremylt // Restore input arrays 3792b730f8bSJeremy L Thompson CeedCallBackend(CeedOperatorRestoreInputs_Ref(num_input_fields, qf_input_fields, op_input_fields, false, e_data_full, impl)); 380e15f9bd0SJeremy L Thompson return CEED_ERROR_SUCCESS; 38121617c04Sjeremylt } 38221617c04Sjeremylt 383f10650afSjeremylt //------------------------------------------------------------------------------ 38470a7ffb3SJeremy L Thompson // Core code for assembling linear QFunction 385f10650afSjeremylt //------------------------------------------------------------------------------ 3862b730f8bSJeremy L Thompson static inline int CeedOperatorLinearAssembleQFunctionCore_Ref(CeedOperator op, bool build_objects, CeedVector *assembled, CeedElemRestriction *rstr, 38770a7ffb3SJeremy L Thompson CeedRequest *request) { 388e910d748SJeremy L Thompson Ceed ceed, ceed_parent; 389e984cf9aSJeremy L Thompson CeedSize q_size; 390ad70ee2cSJeremy L Thompson CeedInt num_active_in, num_active_out, Q, num_elem, num_input_fields, num_output_fields, size; 391ad70ee2cSJeremy L Thompson CeedScalar *assembled_array, *e_data_full[2 * CEED_FIELD_MAX] = {NULL}; 392ad70ee2cSJeremy L Thompson CeedVector *active_in; 393ad70ee2cSJeremy L Thompson CeedQFunctionField *qf_input_fields, *qf_output_fields; 394ad70ee2cSJeremy L Thompson CeedQFunction qf; 395ad70ee2cSJeremy L Thompson CeedOperatorField *op_input_fields, *op_output_fields; 396ad70ee2cSJeremy L Thompson CeedOperator_Ref *impl; 397ad70ee2cSJeremy L Thompson 3982b730f8bSJeremy L Thompson CeedCallBackend(CeedOperatorGetCeed(op, &ceed)); 399e910d748SJeremy L Thompson CeedCallBackend(CeedOperatorGetFallbackParentCeed(op, &ceed_parent)); 400e984cf9aSJeremy L Thompson CeedCallBackend(CeedOperatorGetData(op, &impl)); 401e984cf9aSJeremy L Thompson active_in = impl->qf_active_in; 402e984cf9aSJeremy L Thompson num_active_in = impl->num_active_in, num_active_out = impl->num_active_out; 403e984cf9aSJeremy L Thompson CeedCallBackend(CeedOperatorGetQFunction(op, &qf)); 404e984cf9aSJeremy L Thompson CeedCallBackend(CeedOperatorGetNumQuadraturePoints(op, &Q)); 405e984cf9aSJeremy L Thompson CeedCallBackend(CeedOperatorGetNumElements(op, &num_elem)); 406e984cf9aSJeremy L Thompson CeedCallBackend(CeedQFunctionGetFields(qf, NULL, &qf_input_fields, NULL, &qf_output_fields)); 407e984cf9aSJeremy L Thompson CeedCallBackend(CeedOperatorGetFields(op, &num_input_fields, &op_input_fields, &num_output_fields, &op_output_fields)); 4081d102b48SJeremy L Thompson 4091d102b48SJeremy L Thompson // Setup 4102b730f8bSJeremy L Thompson CeedCallBackend(CeedOperatorSetup_Ref(op)); 4111d102b48SJeremy L Thompson 412506b1a0cSSebastian Grimberg // Check for restriction only operator 413506b1a0cSSebastian Grimberg CeedCheck(!impl->is_identity_rstr_op, ceed, CEED_ERROR_BACKEND, "Assembling restriction only operators is not supported"); 41416911fdaSjeremylt 4151d102b48SJeremy L Thompson // Input Evecs and Restriction 4162b730f8bSJeremy L Thompson CeedCallBackend(CeedOperatorSetupInputs_Ref(num_input_fields, qf_input_fields, op_input_fields, NULL, true, e_data_full, impl, request)); 4171d102b48SJeremy L Thompson 4181d102b48SJeremy L Thompson // Count number of active input fields 419bb219a0fSJeremy L Thompson if (!num_active_in) { 420d1d35e2fSjeremylt for (CeedInt i = 0; i < num_input_fields; i++) { 421ad70ee2cSJeremy L Thompson CeedScalar *q_vec_array; 422ad70ee2cSJeremy L Thompson CeedVector vec; 423ad70ee2cSJeremy L Thompson 4241d102b48SJeremy L Thompson // Get input vector 4252b730f8bSJeremy L Thompson CeedCallBackend(CeedOperatorFieldGetVector(op_input_fields[i], &vec)); 4261d102b48SJeremy L Thompson // Check if active input 4271d102b48SJeremy L Thompson if (vec == CEED_VECTOR_ACTIVE) { 4282b730f8bSJeremy L Thompson CeedCallBackend(CeedQFunctionFieldGetSize(qf_input_fields[i], &size)); 4292b730f8bSJeremy L Thompson CeedCallBackend(CeedVectorSetValue(impl->q_vecs_in[i], 0.0)); 430ad70ee2cSJeremy L Thompson CeedCallBackend(CeedVectorGetArray(impl->q_vecs_in[i], CEED_MEM_HOST, &q_vec_array)); 4312b730f8bSJeremy L Thompson CeedCallBackend(CeedRealloc(num_active_in + size, &active_in)); 4321d102b48SJeremy L Thompson for (CeedInt field = 0; field < size; field++) { 433d2643443SJeremy L Thompson q_size = (CeedSize)Q; 434e910d748SJeremy L Thompson CeedCallBackend(CeedVectorCreate(ceed_parent, q_size, &active_in[num_active_in + field])); 435ad70ee2cSJeremy L Thompson CeedCallBackend(CeedVectorSetArray(active_in[num_active_in + field], CEED_MEM_HOST, CEED_USE_POINTER, &q_vec_array[field * Q])); 4361d102b48SJeremy L Thompson } 437d1d35e2fSjeremylt num_active_in += size; 438ad70ee2cSJeremy L Thompson CeedCallBackend(CeedVectorRestoreArray(impl->q_vecs_in[i], &q_vec_array)); 4391d102b48SJeremy L Thompson } 4401d102b48SJeremy L Thompson } 4414fc1f125SJeremy L Thompson impl->num_active_in = num_active_in; 442bb219a0fSJeremy L Thompson impl->qf_active_in = active_in; 443bb219a0fSJeremy L Thompson } 4441d102b48SJeremy L Thompson 4451d102b48SJeremy L Thompson // Count number of active output fields 446bb219a0fSJeremy L Thompson if (!num_active_out) { 447d1d35e2fSjeremylt for (CeedInt i = 0; i < num_output_fields; i++) { 448ad70ee2cSJeremy L Thompson CeedVector vec; 449ad70ee2cSJeremy L Thompson 4501d102b48SJeremy L Thompson // Get output vector 4512b730f8bSJeremy L Thompson CeedCallBackend(CeedOperatorFieldGetVector(op_output_fields[i], &vec)); 4521d102b48SJeremy L Thompson // Check if active output 4531d102b48SJeremy L Thompson if (vec == CEED_VECTOR_ACTIVE) { 4542b730f8bSJeremy L Thompson CeedCallBackend(CeedQFunctionFieldGetSize(qf_output_fields[i], &size)); 455d1d35e2fSjeremylt num_active_out += size; 4561d102b48SJeremy L Thompson } 4571d102b48SJeremy L Thompson } 4584fc1f125SJeremy L Thompson impl->num_active_out = num_active_out; 459bb219a0fSJeremy L Thompson } 4601d102b48SJeremy L Thompson 4611d102b48SJeremy L Thompson // Check sizes 4626574a04fSJeremy L Thompson CeedCheck(num_active_in > 0 && num_active_out > 0, ceed, CEED_ERROR_BACKEND, "Cannot assemble QFunction without active inputs and outputs"); 4631d102b48SJeremy L Thompson 46470a7ffb3SJeremy L Thompson // Build objects if needed 46570a7ffb3SJeremy L Thompson if (build_objects) { 466ad70ee2cSJeremy L Thompson const CeedSize l_size = (CeedSize)num_elem * Q * num_active_in * num_active_out; 467d1d35e2fSjeremylt CeedInt strides[3] = {1, Q, num_active_in * num_active_out * Q}; /* *NOPAD* */ 468ad70ee2cSJeremy L Thompson 469ad70ee2cSJeremy L Thompson // Create output restriction 470e910d748SJeremy L Thompson CeedCallBackend(CeedElemRestrictionCreateStrided(ceed_parent, num_elem, Q, num_active_in * num_active_out, 4712b730f8bSJeremy L Thompson num_active_in * num_active_out * num_elem * Q, strides, rstr)); 4721d102b48SJeremy L Thompson // Create assembled vector 473e910d748SJeremy L Thompson CeedCallBackend(CeedVectorCreate(ceed_parent, l_size, assembled)); 47470a7ffb3SJeremy L Thompson } 47570a7ffb3SJeremy L Thompson // Clear output vector 4762b730f8bSJeremy L Thompson CeedCallBackend(CeedVectorSetValue(*assembled, 0.0)); 477ad70ee2cSJeremy L Thompson CeedCallBackend(CeedVectorGetArray(*assembled, CEED_MEM_HOST, &assembled_array)); 4781d102b48SJeremy L Thompson 4791d102b48SJeremy L Thompson // Loop through elements 480d1d35e2fSjeremylt for (CeedInt e = 0; e < num_elem; e++) { 4811d102b48SJeremy L Thompson // Input basis apply 4822b730f8bSJeremy L Thompson CeedCallBackend(CeedOperatorInputBasis_Ref(e, Q, qf_input_fields, op_input_fields, num_input_fields, true, e_data_full, impl)); 4831d102b48SJeremy L Thompson 4841d102b48SJeremy L Thompson // Assemble QFunction 485d1d35e2fSjeremylt for (CeedInt in = 0; in < num_active_in; in++) { 4861d102b48SJeremy L Thompson // Set Inputs 4872b730f8bSJeremy L Thompson CeedCallBackend(CeedVectorSetValue(active_in[in], 1.0)); 488d1d35e2fSjeremylt if (num_active_in > 1) { 4892b730f8bSJeremy L Thompson CeedCallBackend(CeedVectorSetValue(active_in[(in + num_active_in - 1) % num_active_in], 0.0)); 49042ea3801Sjeremylt } 491506b1a0cSSebastian Grimberg if (!impl->is_identity_qf) { 4921d102b48SJeremy L Thompson // Set Outputs 493d1d35e2fSjeremylt for (CeedInt out = 0; out < num_output_fields; out++) { 494ad70ee2cSJeremy L Thompson CeedVector vec; 495ad70ee2cSJeremy L Thompson 4961d102b48SJeremy L Thompson // Get output vector 4972b730f8bSJeremy L Thompson CeedCallBackend(CeedOperatorFieldGetVector(op_output_fields[out], &vec)); 4981d102b48SJeremy L Thompson // Check if active output 4991d102b48SJeremy L Thompson if (vec == CEED_VECTOR_ACTIVE) { 500ad70ee2cSJeremy L Thompson CeedCallBackend(CeedVectorSetArray(impl->q_vecs_out[out], CEED_MEM_HOST, CEED_USE_POINTER, assembled_array)); 5012b730f8bSJeremy L Thompson CeedCallBackend(CeedQFunctionFieldGetSize(qf_output_fields[out], &size)); 502ad70ee2cSJeremy L Thompson assembled_array += size * Q; // Advance the pointer by the size of the output 5031d102b48SJeremy L Thompson } 5041d102b48SJeremy L Thompson } 5051d102b48SJeremy L Thompson // Apply QFunction 5062b730f8bSJeremy L Thompson CeedCallBackend(CeedQFunctionApply(qf, Q, impl->q_vecs_in, impl->q_vecs_out)); 507506b1a0cSSebastian Grimberg } else { 508506b1a0cSSebastian Grimberg const CeedScalar *q_vec_array; 509506b1a0cSSebastian Grimberg 510506b1a0cSSebastian Grimberg // Copy Identity Outputs 511506b1a0cSSebastian Grimberg CeedCallBackend(CeedQFunctionFieldGetSize(qf_output_fields[0], &size)); 512506b1a0cSSebastian Grimberg CeedCallBackend(CeedVectorGetArrayRead(impl->q_vecs_out[0], CEED_MEM_HOST, &q_vec_array)); 513506b1a0cSSebastian Grimberg for (CeedInt i = 0; i < size * Q; i++) assembled_array[i] = q_vec_array[i]; 514506b1a0cSSebastian Grimberg CeedCallBackend(CeedVectorRestoreArrayRead(impl->q_vecs_out[0], &q_vec_array)); 515506b1a0cSSebastian Grimberg assembled_array += size * Q; 516506b1a0cSSebastian Grimberg } 5171d102b48SJeremy L Thompson } 5181d102b48SJeremy L Thompson } 5191d102b48SJeremy L Thompson 5201d102b48SJeremy L Thompson // Un-set output Qvecs to prevent accidental overwrite of Assembled 521506b1a0cSSebastian Grimberg if (!impl->is_identity_qf) { 522d1d35e2fSjeremylt for (CeedInt out = 0; out < num_output_fields; out++) { 523ad70ee2cSJeremy L Thompson CeedVector vec; 524ad70ee2cSJeremy L Thompson 5251d102b48SJeremy L Thompson // Get output vector 5262b730f8bSJeremy L Thompson CeedCallBackend(CeedOperatorFieldGetVector(op_output_fields[out], &vec)); 5271d102b48SJeremy L Thompson // Check if active output 528056ea4bdSJeremy L Thompson if (vec == CEED_VECTOR_ACTIVE && num_elem > 0) { 5292b730f8bSJeremy L Thompson CeedCallBackend(CeedVectorTakeArray(impl->q_vecs_out[out], CEED_MEM_HOST, NULL)); 5301d102b48SJeremy L Thompson } 5311d102b48SJeremy L Thompson } 532506b1a0cSSebastian Grimberg } 5331d102b48SJeremy L Thompson 5341d102b48SJeremy L Thompson // Restore input arrays 5352b730f8bSJeremy L Thompson CeedCallBackend(CeedOperatorRestoreInputs_Ref(num_input_fields, qf_input_fields, op_input_fields, true, e_data_full, impl)); 5361d102b48SJeremy L Thompson 5371d102b48SJeremy L Thompson // Restore output 538ad70ee2cSJeremy L Thompson CeedCallBackend(CeedVectorRestoreArray(*assembled, &assembled_array)); 539e15f9bd0SJeremy L Thompson return CEED_ERROR_SUCCESS; 5401d102b48SJeremy L Thompson } 5411d102b48SJeremy L Thompson 542f10650afSjeremylt //------------------------------------------------------------------------------ 54370a7ffb3SJeremy L Thompson // Assemble Linear QFunction 54470a7ffb3SJeremy L Thompson //------------------------------------------------------------------------------ 5452b730f8bSJeremy L Thompson static int CeedOperatorLinearAssembleQFunction_Ref(CeedOperator op, CeedVector *assembled, CeedElemRestriction *rstr, CeedRequest *request) { 5462b730f8bSJeremy L Thompson return CeedOperatorLinearAssembleQFunctionCore_Ref(op, true, assembled, rstr, request); 54770a7ffb3SJeremy L Thompson } 54870a7ffb3SJeremy L Thompson 54970a7ffb3SJeremy L Thompson //------------------------------------------------------------------------------ 55070a7ffb3SJeremy L Thompson // Update Assembled Linear QFunction 55170a7ffb3SJeremy L Thompson //------------------------------------------------------------------------------ 5522b730f8bSJeremy L Thompson static int CeedOperatorLinearAssembleQFunctionUpdate_Ref(CeedOperator op, CeedVector assembled, CeedElemRestriction rstr, CeedRequest *request) { 5532b730f8bSJeremy L Thompson return CeedOperatorLinearAssembleQFunctionCore_Ref(op, false, &assembled, &rstr, request); 55470a7ffb3SJeremy L Thompson } 55570a7ffb3SJeremy L Thompson 55670a7ffb3SJeremy L Thompson //------------------------------------------------------------------------------ 55748acf710SJeremy L Thompson // Setup Input/Output Fields 55848acf710SJeremy L Thompson //------------------------------------------------------------------------------ 55948acf710SJeremy L Thompson static int CeedOperatorSetupFieldsAtPoints_Ref(CeedQFunction qf, CeedOperator op, bool is_input, CeedVector *e_vecs_full, CeedVector *e_vecs, 56048acf710SJeremy L Thompson CeedVector *q_vecs, CeedInt start_e, CeedInt num_fields, CeedInt Q) { 56148acf710SJeremy L Thompson Ceed ceed; 56248acf710SJeremy L Thompson CeedSize e_size, q_size; 56348acf710SJeremy L Thompson CeedInt max_num_points, num_comp, size, P; 56448acf710SJeremy L Thompson CeedQFunctionField *qf_fields; 56548acf710SJeremy L Thompson CeedOperatorField *op_fields; 56648acf710SJeremy L Thompson 567e910d748SJeremy L Thompson { 568e910d748SJeremy L Thompson Ceed ceed_parent; 569e910d748SJeremy L Thompson 57048acf710SJeremy L Thompson CeedCallBackend(CeedOperatorGetCeed(op, &ceed)); 571e910d748SJeremy L Thompson CeedCallBackend(CeedGetParent(ceed, &ceed_parent)); 572e910d748SJeremy L Thompson if (ceed_parent) ceed = ceed_parent; 573e910d748SJeremy L Thompson } 57448acf710SJeremy L Thompson if (is_input) { 57548acf710SJeremy L Thompson CeedCallBackend(CeedOperatorGetFields(op, NULL, &op_fields, NULL, NULL)); 57648acf710SJeremy L Thompson CeedCallBackend(CeedQFunctionGetFields(qf, NULL, &qf_fields, NULL, NULL)); 57748acf710SJeremy L Thompson } else { 57848acf710SJeremy L Thompson CeedCallBackend(CeedOperatorGetFields(op, NULL, NULL, NULL, &op_fields)); 57948acf710SJeremy L Thompson CeedCallBackend(CeedQFunctionGetFields(qf, NULL, NULL, NULL, &qf_fields)); 58048acf710SJeremy L Thompson } 58148acf710SJeremy L Thompson 58248acf710SJeremy L Thompson // Get max number of points 58348acf710SJeremy L Thompson { 58448acf710SJeremy L Thompson CeedInt dim; 58548acf710SJeremy L Thompson CeedElemRestriction rstr_points = NULL; 58648acf710SJeremy L Thompson CeedOperator_Ref *impl; 58748acf710SJeremy L Thompson 58848acf710SJeremy L Thompson CeedCallBackend(CeedOperatorAtPointsGetPoints(op, &rstr_points, NULL)); 58948acf710SJeremy L Thompson CeedCallBackend(CeedElemRestrictionGetMaxPointsInElement(rstr_points, &max_num_points)); 59048acf710SJeremy L Thompson CeedCallBackend(CeedElemRestrictionGetNumComponents(rstr_points, &dim)); 59148acf710SJeremy L Thompson CeedCallBackend(CeedElemRestrictionDestroy(&rstr_points)); 59248acf710SJeremy L Thompson CeedCallBackend(CeedOperatorGetData(op, &impl)); 59348acf710SJeremy L Thompson if (is_input) CeedCallBackend(CeedVectorCreate(ceed, dim * max_num_points, &impl->point_coords_elem)); 59448acf710SJeremy L Thompson } 59548acf710SJeremy L Thompson 59648acf710SJeremy L Thompson // Loop over fields 59748acf710SJeremy L Thompson for (CeedInt i = 0; i < num_fields; i++) { 59848acf710SJeremy L Thompson CeedEvalMode eval_mode; 59948acf710SJeremy L Thompson CeedBasis basis; 60048acf710SJeremy L Thompson 60148acf710SJeremy L Thompson CeedCallBackend(CeedQFunctionFieldGetEvalMode(qf_fields[i], &eval_mode)); 60248acf710SJeremy L Thompson if (eval_mode != CEED_EVAL_WEIGHT) { 60348acf710SJeremy L Thompson CeedRestrictionType rstr_type; 60448acf710SJeremy L Thompson CeedElemRestriction elem_rstr; 60548acf710SJeremy L Thompson 60648acf710SJeremy L Thompson CeedCallBackend(CeedOperatorFieldGetElemRestriction(op_fields[i], &elem_rstr)); 60748acf710SJeremy L Thompson CeedCallBackend(CeedElemRestrictionGetNumComponents(elem_rstr, &num_comp)); 60848acf710SJeremy L Thompson CeedCallBackend(CeedElemRestrictionGetType(elem_rstr, &rstr_type)); 60948acf710SJeremy L Thompson CeedCallBackend(CeedElemRestrictionCreateVector(elem_rstr, NULL, &e_vecs_full[i + start_e])); 61048acf710SJeremy L Thompson } 61148acf710SJeremy L Thompson 61248acf710SJeremy L Thompson switch (eval_mode) { 61338e83183SJeremy L Thompson case CEED_EVAL_NONE: { 61438e83183SJeremy L Thompson CeedVector vec; 61538e83183SJeremy L Thompson 61648acf710SJeremy L Thompson CeedCallBackend(CeedQFunctionFieldGetSize(qf_fields[i], &size)); 61748acf710SJeremy L Thompson e_size = (CeedSize)max_num_points * size; 61848acf710SJeremy L Thompson CeedCallBackend(CeedVectorCreate(ceed, e_size, &e_vecs[i])); 61938e83183SJeremy L Thompson CeedCallBackend(CeedOperatorFieldGetVector(op_fields[i], &vec)); 62038e83183SJeremy L Thompson if (vec == CEED_VECTOR_ACTIVE || !is_input) { 62138e83183SJeremy L Thompson CeedCallBackend(CeedVectorReferenceCopy(e_vecs[i], &q_vecs[i])); 62238e83183SJeremy L Thompson } else { 62348acf710SJeremy L Thompson q_size = (CeedSize)max_num_points * size; 62448acf710SJeremy L Thompson CeedCallBackend(CeedVectorCreate(ceed, q_size, &q_vecs[i])); 62538e83183SJeremy L Thompson } 62648acf710SJeremy L Thompson break; 62738e83183SJeremy L Thompson } 62848acf710SJeremy L Thompson case CEED_EVAL_INTERP: 62948acf710SJeremy L Thompson case CEED_EVAL_GRAD: 63048acf710SJeremy L Thompson case CEED_EVAL_DIV: 63148acf710SJeremy L Thompson case CEED_EVAL_CURL: 63248acf710SJeremy L Thompson CeedCallBackend(CeedOperatorFieldGetBasis(op_fields[i], &basis)); 63348acf710SJeremy L Thompson CeedCallBackend(CeedQFunctionFieldGetSize(qf_fields[i], &size)); 63448acf710SJeremy L Thompson CeedCallBackend(CeedBasisGetNumNodes(basis, &P)); 63548acf710SJeremy L Thompson CeedCallBackend(CeedBasisGetNumComponents(basis, &num_comp)); 63648acf710SJeremy L Thompson e_size = (CeedSize)P * num_comp; 63748acf710SJeremy L Thompson CeedCallBackend(CeedVectorCreate(ceed, e_size, &e_vecs[i])); 63848acf710SJeremy L Thompson q_size = (CeedSize)max_num_points * size; 63948acf710SJeremy L Thompson CeedCallBackend(CeedVectorCreate(ceed, q_size, &q_vecs[i])); 64048acf710SJeremy L Thompson break; 64148acf710SJeremy L Thompson case CEED_EVAL_WEIGHT: // Only on input fields 64248acf710SJeremy L Thompson CeedCallBackend(CeedOperatorFieldGetBasis(op_fields[i], &basis)); 64348acf710SJeremy L Thompson q_size = (CeedSize)max_num_points; 64448acf710SJeremy L Thompson CeedCallBackend(CeedVectorCreate(ceed, q_size, &q_vecs[i])); 64548acf710SJeremy L Thompson CeedCallBackend( 64648acf710SJeremy L Thompson CeedBasisApplyAtPoints(basis, max_num_points, CEED_NOTRANSPOSE, CEED_EVAL_WEIGHT, CEED_VECTOR_NONE, CEED_VECTOR_NONE, q_vecs[i])); 64748acf710SJeremy L Thompson break; 64848acf710SJeremy L Thompson } 64948acf710SJeremy L Thompson if (is_input && e_vecs[i]) { 65048acf710SJeremy L Thompson CeedCallBackend(CeedVectorSetArray(e_vecs[i], CEED_MEM_HOST, CEED_COPY_VALUES, NULL)); 65148acf710SJeremy L Thompson } 65248acf710SJeremy L Thompson } 65348acf710SJeremy L Thompson return CEED_ERROR_SUCCESS; 65448acf710SJeremy L Thompson } 65548acf710SJeremy L Thompson 65648acf710SJeremy L Thompson //------------------------------------------------------------------------------ 65748acf710SJeremy L Thompson // Setup Operator 65848acf710SJeremy L Thompson //------------------------------------------------------------------------------ 65948acf710SJeremy L Thompson static int CeedOperatorSetupAtPoints_Ref(CeedOperator op) { 66048acf710SJeremy L Thompson bool is_setup_done; 66148acf710SJeremy L Thompson Ceed ceed; 66248acf710SJeremy L Thompson CeedInt Q, num_input_fields, num_output_fields; 66348acf710SJeremy L Thompson CeedQFunctionField *qf_input_fields, *qf_output_fields; 66448acf710SJeremy L Thompson CeedQFunction qf; 66548acf710SJeremy L Thompson CeedOperatorField *op_input_fields, *op_output_fields; 66648acf710SJeremy L Thompson CeedOperator_Ref *impl; 66748acf710SJeremy L Thompson 66848acf710SJeremy L Thompson CeedCallBackend(CeedOperatorIsSetupDone(op, &is_setup_done)); 66948acf710SJeremy L Thompson if (is_setup_done) return CEED_ERROR_SUCCESS; 67048acf710SJeremy L Thompson 67148acf710SJeremy L Thompson CeedCallBackend(CeedOperatorGetCeed(op, &ceed)); 67248acf710SJeremy L Thompson CeedCallBackend(CeedOperatorGetData(op, &impl)); 67348acf710SJeremy L Thompson CeedCallBackend(CeedOperatorGetQFunction(op, &qf)); 67448acf710SJeremy L Thompson CeedCallBackend(CeedOperatorGetNumQuadraturePoints(op, &Q)); 67548acf710SJeremy L Thompson CeedCallBackend(CeedQFunctionIsIdentity(qf, &impl->is_identity_qf)); 67648acf710SJeremy L Thompson CeedCallBackend(CeedOperatorGetFields(op, &num_input_fields, &op_input_fields, &num_output_fields, &op_output_fields)); 67748acf710SJeremy L Thompson CeedCallBackend(CeedQFunctionGetFields(qf, NULL, &qf_input_fields, NULL, &qf_output_fields)); 67848acf710SJeremy L Thompson 67948acf710SJeremy L Thompson // Allocate 68048acf710SJeremy L Thompson CeedCallBackend(CeedCalloc(num_input_fields + num_output_fields, &impl->e_vecs_full)); 68148acf710SJeremy L Thompson 68248acf710SJeremy L Thompson CeedCallBackend(CeedCalloc(CEED_FIELD_MAX, &impl->input_states)); 68348acf710SJeremy L Thompson CeedCallBackend(CeedCalloc(CEED_FIELD_MAX, &impl->e_vecs_in)); 68448acf710SJeremy L Thompson CeedCallBackend(CeedCalloc(CEED_FIELD_MAX, &impl->e_vecs_out)); 68548acf710SJeremy L Thompson CeedCallBackend(CeedCalloc(CEED_FIELD_MAX, &impl->q_vecs_in)); 68648acf710SJeremy L Thompson CeedCallBackend(CeedCalloc(CEED_FIELD_MAX, &impl->q_vecs_out)); 68748acf710SJeremy L Thompson 68848acf710SJeremy L Thompson impl->num_inputs = num_input_fields; 68948acf710SJeremy L Thompson impl->num_outputs = num_output_fields; 69048acf710SJeremy L Thompson 69148acf710SJeremy L Thompson // Set up infield and outfield pointer arrays 69248acf710SJeremy L Thompson // Infields 69348acf710SJeremy L Thompson CeedCallBackend(CeedOperatorSetupFieldsAtPoints_Ref(qf, op, true, impl->e_vecs_full, impl->e_vecs_in, impl->q_vecs_in, 0, num_input_fields, Q)); 69448acf710SJeremy L Thompson // Outfields 69548acf710SJeremy L Thompson CeedCallBackend(CeedOperatorSetupFieldsAtPoints_Ref(qf, op, false, impl->e_vecs_full, impl->e_vecs_out, impl->q_vecs_out, num_input_fields, 69648acf710SJeremy L Thompson num_output_fields, Q)); 69748acf710SJeremy L Thompson 69848acf710SJeremy L Thompson // Identity QFunctions 69948acf710SJeremy L Thompson if (impl->is_identity_qf) { 70048acf710SJeremy L Thompson CeedCallBackend(CeedVectorReferenceCopy(impl->q_vecs_in[0], &impl->q_vecs_out[0])); 70138e83183SJeremy L Thompson CeedCallBackend(CeedVectorReferenceCopy(impl->q_vecs_in[0], &impl->e_vecs_out[0])); 70248acf710SJeremy L Thompson } 70348acf710SJeremy L Thompson 70448acf710SJeremy L Thompson CeedCallBackend(CeedOperatorSetSetupDone(op)); 70548acf710SJeremy L Thompson return CEED_ERROR_SUCCESS; 70648acf710SJeremy L Thompson } 70748acf710SJeremy L Thompson 70848acf710SJeremy L Thompson //------------------------------------------------------------------------------ 70948acf710SJeremy L Thompson // Setup Input Fields 71048acf710SJeremy L Thompson //------------------------------------------------------------------------------ 71148acf710SJeremy L Thompson static inline int CeedOperatorSetupInputsAtPoints_Ref(CeedInt num_input_fields, CeedQFunctionField *qf_input_fields, 71248acf710SJeremy L Thompson CeedOperatorField *op_input_fields, CeedVector in_vec, CeedScalar *e_data[2 * CEED_FIELD_MAX], 71348acf710SJeremy L Thompson CeedOperator_Ref *impl, CeedRequest *request) { 71448acf710SJeremy L Thompson for (CeedInt i = 0; i < num_input_fields; i++) { 71548acf710SJeremy L Thompson uint64_t state; 71648acf710SJeremy L Thompson CeedEvalMode eval_mode; 71748acf710SJeremy L Thompson CeedVector vec; 71848acf710SJeremy L Thompson CeedElemRestriction elem_rstr; 71948acf710SJeremy L Thompson 72048acf710SJeremy L Thompson CeedCallBackend(CeedQFunctionFieldGetEvalMode(qf_input_fields[i], &eval_mode)); 72148acf710SJeremy L Thompson if (eval_mode == CEED_EVAL_WEIGHT) { // Skip 72248acf710SJeremy L Thompson } else { 72348acf710SJeremy L Thompson // Get input vector 72448acf710SJeremy L Thompson CeedCallBackend(CeedOperatorFieldGetVector(op_input_fields[i], &vec)); 72548acf710SJeremy L Thompson CeedCallBackend(CeedOperatorFieldGetElemRestriction(op_input_fields[i], &elem_rstr)); 72648acf710SJeremy L Thompson if (vec != CEED_VECTOR_ACTIVE) { 72748acf710SJeremy L Thompson // Restrict 72848acf710SJeremy L Thompson CeedCallBackend(CeedVectorGetState(vec, &state)); 72948acf710SJeremy L Thompson if (state != impl->input_states[i]) { 73048acf710SJeremy L Thompson CeedCallBackend(CeedElemRestrictionApply(elem_rstr, CEED_NOTRANSPOSE, vec, impl->e_vecs_full[i], request)); 73148acf710SJeremy L Thompson impl->input_states[i] = state; 73248acf710SJeremy L Thompson } 73348acf710SJeremy L Thompson // Get evec 73448acf710SJeremy L Thompson CeedCallBackend(CeedVectorGetArrayRead(impl->e_vecs_full[i], CEED_MEM_HOST, (const CeedScalar **)&e_data[i])); 73548acf710SJeremy L Thompson } 73648acf710SJeremy L Thompson } 73748acf710SJeremy L Thompson } 73848acf710SJeremy L Thompson return CEED_ERROR_SUCCESS; 73948acf710SJeremy L Thompson } 74048acf710SJeremy L Thompson 74148acf710SJeremy L Thompson //------------------------------------------------------------------------------ 74248acf710SJeremy L Thompson // Input Basis Action 74348acf710SJeremy L Thompson //------------------------------------------------------------------------------ 74448acf710SJeremy L Thompson static inline int CeedOperatorInputBasisAtPoints_Ref(CeedInt e, CeedInt num_points_offset, CeedInt num_points, CeedQFunctionField *qf_input_fields, 74548acf710SJeremy L Thompson CeedOperatorField *op_input_fields, CeedInt num_input_fields, CeedVector in_vec, 74648acf710SJeremy L Thompson CeedVector point_coords_elem, bool skip_active, CeedScalar *e_data[2 * CEED_FIELD_MAX], 74748acf710SJeremy L Thompson CeedOperator_Ref *impl, CeedRequest *request) { 74848acf710SJeremy L Thompson for (CeedInt i = 0; i < num_input_fields; i++) { 74948acf710SJeremy L Thompson bool is_active_input = false; 75048acf710SJeremy L Thompson CeedInt elem_size, size, num_comp; 75148acf710SJeremy L Thompson CeedRestrictionType rstr_type; 75248acf710SJeremy L Thompson CeedEvalMode eval_mode; 75348acf710SJeremy L Thompson CeedVector vec; 75448acf710SJeremy L Thompson CeedElemRestriction elem_rstr; 75548acf710SJeremy L Thompson CeedBasis basis; 75648acf710SJeremy L Thompson 75748acf710SJeremy L Thompson CeedCallBackend(CeedOperatorFieldGetVector(op_input_fields[i], &vec)); 75848acf710SJeremy L Thompson // Skip active input 75948acf710SJeremy L Thompson is_active_input = vec == CEED_VECTOR_ACTIVE; 76048acf710SJeremy L Thompson if (skip_active && is_active_input) continue; 76148acf710SJeremy L Thompson 76248acf710SJeremy L Thompson // Get elem_size, eval_mode, size 76348acf710SJeremy L Thompson CeedCallBackend(CeedOperatorFieldGetElemRestriction(op_input_fields[i], &elem_rstr)); 76448acf710SJeremy L Thompson CeedCallBackend(CeedElemRestrictionGetType(elem_rstr, &rstr_type)); 76548acf710SJeremy L Thompson CeedCallBackend(CeedQFunctionFieldGetEvalMode(qf_input_fields[i], &eval_mode)); 76648acf710SJeremy L Thompson CeedCallBackend(CeedQFunctionFieldGetSize(qf_input_fields[i], &size)); 76748acf710SJeremy L Thompson // Restrict block active input 76848acf710SJeremy L Thompson if (is_active_input) { 76948acf710SJeremy L Thompson if (rstr_type == CEED_RESTRICTION_POINTS) { 77048acf710SJeremy L Thompson CeedCallBackend(CeedElemRestrictionApplyAtPointsInElement(elem_rstr, e, CEED_NOTRANSPOSE, in_vec, impl->e_vecs_in[i], request)); 77148acf710SJeremy L Thompson } else { 77248acf710SJeremy L Thompson CeedCallBackend(CeedElemRestrictionApplyBlock(elem_rstr, e, CEED_NOTRANSPOSE, in_vec, impl->e_vecs_in[i], request)); 77348acf710SJeremy L Thompson } 77448acf710SJeremy L Thompson } 77548acf710SJeremy L Thompson // Basis action 77648acf710SJeremy L Thompson switch (eval_mode) { 77748acf710SJeremy L Thompson case CEED_EVAL_NONE: 77848acf710SJeremy L Thompson if (!is_active_input) { 77948acf710SJeremy L Thompson CeedCallBackend(CeedVectorSetArray(impl->q_vecs_in[i], CEED_MEM_HOST, CEED_USE_POINTER, &e_data[i][num_points_offset * size])); 78048acf710SJeremy L Thompson } 78148acf710SJeremy L Thompson break; 78248acf710SJeremy L Thompson // Note - these basis eval modes require FEM fields 78348acf710SJeremy L Thompson case CEED_EVAL_INTERP: 78448acf710SJeremy L Thompson case CEED_EVAL_GRAD: 78548acf710SJeremy L Thompson case CEED_EVAL_DIV: 78648acf710SJeremy L Thompson case CEED_EVAL_CURL: 78748acf710SJeremy L Thompson CeedCallBackend(CeedOperatorFieldGetBasis(op_input_fields[i], &basis)); 78848acf710SJeremy L Thompson if (!is_active_input) { 78948acf710SJeremy L Thompson CeedCallBackend(CeedBasisGetNumComponents(basis, &num_comp)); 79048acf710SJeremy L Thompson CeedCallBackend(CeedElemRestrictionGetElementSize(elem_rstr, &elem_size)); 79148acf710SJeremy L Thompson CeedCallBackend(CeedVectorSetArray(impl->e_vecs_in[i], CEED_MEM_HOST, CEED_USE_POINTER, &e_data[i][e * elem_size * num_comp])); 79248acf710SJeremy L Thompson } 79348acf710SJeremy L Thompson CeedCallBackend( 79448acf710SJeremy L Thompson CeedBasisApplyAtPoints(basis, num_points, CEED_NOTRANSPOSE, eval_mode, point_coords_elem, impl->e_vecs_in[i], impl->q_vecs_in[i])); 79548acf710SJeremy L Thompson break; 79648acf710SJeremy L Thompson case CEED_EVAL_WEIGHT: 79748acf710SJeremy L Thompson break; // No action 79848acf710SJeremy L Thompson } 79948acf710SJeremy L Thompson } 80048acf710SJeremy L Thompson return CEED_ERROR_SUCCESS; 80148acf710SJeremy L Thompson } 80248acf710SJeremy L Thompson 80348acf710SJeremy L Thompson //------------------------------------------------------------------------------ 80448acf710SJeremy L Thompson // Output Basis Action 80548acf710SJeremy L Thompson //------------------------------------------------------------------------------ 80648acf710SJeremy L Thompson static inline int CeedOperatorOutputBasisAtPoints_Ref(CeedInt e, CeedInt num_points_offset, CeedInt num_points, CeedQFunctionField *qf_output_fields, 80748acf710SJeremy L Thompson CeedOperatorField *op_output_fields, CeedInt num_input_fields, CeedInt num_output_fields, 80848acf710SJeremy L Thompson CeedOperator op, CeedVector out_vec, CeedVector point_coords_elem, CeedOperator_Ref *impl, 80948acf710SJeremy L Thompson CeedRequest *request) { 81048acf710SJeremy L Thompson for (CeedInt i = 0; i < num_output_fields; i++) { 81148acf710SJeremy L Thompson CeedRestrictionType rstr_type; 81248acf710SJeremy L Thompson CeedEvalMode eval_mode; 81348acf710SJeremy L Thompson CeedVector vec; 81448acf710SJeremy L Thompson CeedElemRestriction elem_rstr; 81548acf710SJeremy L Thompson CeedBasis basis; 81648acf710SJeremy L Thompson 81748acf710SJeremy L Thompson // Get elem_size, eval_mode, size 81848acf710SJeremy L Thompson CeedCallBackend(CeedOperatorFieldGetElemRestriction(op_output_fields[i], &elem_rstr)); 81948acf710SJeremy L Thompson CeedCallBackend(CeedQFunctionFieldGetEvalMode(qf_output_fields[i], &eval_mode)); 82048acf710SJeremy L Thompson // Basis action 82148acf710SJeremy L Thompson switch (eval_mode) { 82248acf710SJeremy L Thompson case CEED_EVAL_NONE: 82348acf710SJeremy L Thompson break; // No action 82448acf710SJeremy L Thompson case CEED_EVAL_INTERP: 82548acf710SJeremy L Thompson case CEED_EVAL_GRAD: 82648acf710SJeremy L Thompson case CEED_EVAL_DIV: 82748acf710SJeremy L Thompson case CEED_EVAL_CURL: 82848acf710SJeremy L Thompson CeedCallBackend(CeedOperatorFieldGetBasis(op_output_fields[i], &basis)); 82948acf710SJeremy L Thompson CeedCallBackend( 83048acf710SJeremy L Thompson CeedBasisApplyAtPoints(basis, num_points, CEED_TRANSPOSE, eval_mode, point_coords_elem, impl->q_vecs_out[i], impl->e_vecs_out[i])); 83148acf710SJeremy L Thompson break; 83248acf710SJeremy L Thompson // LCOV_EXCL_START 83348acf710SJeremy L Thompson case CEED_EVAL_WEIGHT: { 83448acf710SJeremy L Thompson Ceed ceed; 835*bcbe1c99SJeremy L Thompson 83648acf710SJeremy L Thompson CeedCallBackend(CeedOperatorGetCeed(op, &ceed)); 83748acf710SJeremy L Thompson return CeedError(ceed, CEED_ERROR_BACKEND, "CEED_EVAL_WEIGHT cannot be an output evaluation mode"); 83848acf710SJeremy L Thompson // LCOV_EXCL_STOP 83948acf710SJeremy L Thompson } 84048acf710SJeremy L Thompson } 84148acf710SJeremy L Thompson // Restrict output block 84248acf710SJeremy L Thompson // Get output vector 84348acf710SJeremy L Thompson CeedCallBackend(CeedElemRestrictionGetType(elem_rstr, &rstr_type)); 84448acf710SJeremy L Thompson CeedCallBackend(CeedOperatorFieldGetVector(op_output_fields[i], &vec)); 84548acf710SJeremy L Thompson if (vec == CEED_VECTOR_ACTIVE) vec = out_vec; 84648acf710SJeremy L Thompson // Restrict 84748acf710SJeremy L Thompson if (rstr_type == CEED_RESTRICTION_POINTS) { 84848acf710SJeremy L Thompson CeedCallBackend(CeedElemRestrictionApplyAtPointsInElement(elem_rstr, e, CEED_TRANSPOSE, impl->e_vecs_out[i], vec, request)); 84948acf710SJeremy L Thompson } else { 85048acf710SJeremy L Thompson CeedCallBackend(CeedElemRestrictionApplyBlock(elem_rstr, e, CEED_TRANSPOSE, impl->e_vecs_out[i], vec, request)); 85148acf710SJeremy L Thompson } 85248acf710SJeremy L Thompson } 85348acf710SJeremy L Thompson return CEED_ERROR_SUCCESS; 85448acf710SJeremy L Thompson } 85548acf710SJeremy L Thompson 85648acf710SJeremy L Thompson //------------------------------------------------------------------------------ 85748acf710SJeremy L Thompson // Restore Input Vectors 85848acf710SJeremy L Thompson //------------------------------------------------------------------------------ 85948acf710SJeremy L Thompson static inline int CeedOperatorRestoreInputsAtPoints_Ref(CeedInt num_input_fields, CeedQFunctionField *qf_input_fields, 86048acf710SJeremy L Thompson CeedOperatorField *op_input_fields, CeedScalar *e_data[2 * CEED_FIELD_MAX], 86148acf710SJeremy L Thompson CeedOperator_Ref *impl) { 86248acf710SJeremy L Thompson for (CeedInt i = 0; i < num_input_fields; i++) { 86348acf710SJeremy L Thompson CeedEvalMode eval_mode; 86448acf710SJeremy L Thompson CeedVector vec; 86548acf710SJeremy L Thompson 86648acf710SJeremy L Thompson CeedCallBackend(CeedQFunctionFieldGetEvalMode(qf_input_fields[i], &eval_mode)); 86748acf710SJeremy L Thompson CeedCallBackend(CeedOperatorFieldGetVector(op_input_fields[i], &vec)); 86848acf710SJeremy L Thompson if (eval_mode != CEED_EVAL_WEIGHT && vec != CEED_VECTOR_ACTIVE) { 86948acf710SJeremy L Thompson CeedCallBackend(CeedVectorRestoreArrayRead(impl->e_vecs_full[i], (const CeedScalar **)&e_data[i])); 87048acf710SJeremy L Thompson } 87148acf710SJeremy L Thompson } 87248acf710SJeremy L Thompson return CEED_ERROR_SUCCESS; 87348acf710SJeremy L Thompson } 87448acf710SJeremy L Thompson 87548acf710SJeremy L Thompson //------------------------------------------------------------------------------ 87648acf710SJeremy L Thompson // Operator Apply 87748acf710SJeremy L Thompson //------------------------------------------------------------------------------ 87848acf710SJeremy L Thompson static int CeedOperatorApplyAddAtPoints_Ref(CeedOperator op, CeedVector in_vec, CeedVector out_vec, CeedRequest *request) { 87948acf710SJeremy L Thompson Ceed ceed; 88048acf710SJeremy L Thompson CeedInt num_points_offset = 0, num_input_fields, num_output_fields, num_elem; 88148acf710SJeremy L Thompson CeedScalar *e_data[2 * CEED_FIELD_MAX] = {0}; 88248acf710SJeremy L Thompson CeedVector point_coords = NULL; 88348acf710SJeremy L Thompson CeedElemRestriction rstr_points = NULL; 88448acf710SJeremy L Thompson CeedQFunctionField *qf_input_fields, *qf_output_fields; 88548acf710SJeremy L Thompson CeedQFunction qf; 88648acf710SJeremy L Thompson CeedOperatorField *op_input_fields, *op_output_fields; 88748acf710SJeremy L Thompson CeedOperator_Ref *impl; 88848acf710SJeremy L Thompson 88948acf710SJeremy L Thompson CeedCallBackend(CeedOperatorGetCeed(op, &ceed)); 89048acf710SJeremy L Thompson CeedCallBackend(CeedOperatorGetData(op, &impl)); 89148acf710SJeremy L Thompson CeedCallBackend(CeedOperatorGetNumElements(op, &num_elem)); 89248acf710SJeremy L Thompson CeedCallBackend(CeedOperatorGetQFunction(op, &qf)); 89348acf710SJeremy L Thompson CeedCallBackend(CeedOperatorGetFields(op, &num_input_fields, &op_input_fields, &num_output_fields, &op_output_fields)); 89448acf710SJeremy L Thompson CeedCallBackend(CeedQFunctionGetFields(qf, NULL, &qf_input_fields, NULL, &qf_output_fields)); 89548acf710SJeremy L Thompson 89648acf710SJeremy L Thompson // Setup 89748acf710SJeremy L Thompson CeedCallBackend(CeedOperatorSetupAtPoints_Ref(op)); 89848acf710SJeremy L Thompson 89948acf710SJeremy L Thompson // Point coordinates 90048acf710SJeremy L Thompson CeedCallBackend(CeedOperatorAtPointsGetPoints(op, &rstr_points, &point_coords)); 90148acf710SJeremy L Thompson 90248acf710SJeremy L Thompson // Input Evecs and Restriction 90348acf710SJeremy L Thompson CeedCallBackend(CeedOperatorSetupInputsAtPoints_Ref(num_input_fields, qf_input_fields, op_input_fields, in_vec, e_data, impl, request)); 90448acf710SJeremy L Thompson 90548acf710SJeremy L Thompson // Loop through elements 90648acf710SJeremy L Thompson for (CeedInt e = 0; e < num_elem; e++) { 90748acf710SJeremy L Thompson CeedInt num_points; 90848acf710SJeremy L Thompson 90948acf710SJeremy L Thompson // Setup points for element 91048acf710SJeremy L Thompson CeedCallBackend(CeedElemRestrictionApplyAtPointsInElement(rstr_points, e, CEED_NOTRANSPOSE, point_coords, impl->point_coords_elem, request)); 91148acf710SJeremy L Thompson CeedCallBackend(CeedElemRestrictionGetNumPointsInElement(rstr_points, e, &num_points)); 91248acf710SJeremy L Thompson 91348acf710SJeremy L Thompson // Input basis apply 91448acf710SJeremy L Thompson CeedCallBackend(CeedOperatorInputBasisAtPoints_Ref(e, num_points_offset, num_points, qf_input_fields, op_input_fields, num_input_fields, in_vec, 91548acf710SJeremy L Thompson impl->point_coords_elem, false, e_data, impl, request)); 91648acf710SJeremy L Thompson 91748acf710SJeremy L Thompson // Q function 91848acf710SJeremy L Thompson if (!impl->is_identity_qf) { 91948acf710SJeremy L Thompson CeedCallBackend(CeedQFunctionApply(qf, num_points, impl->q_vecs_in, impl->q_vecs_out)); 92048acf710SJeremy L Thompson } 92148acf710SJeremy L Thompson 92248acf710SJeremy L Thompson // Output basis apply and restriction 92348acf710SJeremy L Thompson CeedCallBackend(CeedOperatorOutputBasisAtPoints_Ref(e, num_points_offset, num_points, qf_output_fields, op_output_fields, num_input_fields, 92448acf710SJeremy L Thompson num_output_fields, op, out_vec, impl->point_coords_elem, impl, request)); 92548acf710SJeremy L Thompson 92648acf710SJeremy L Thompson num_points_offset += num_points; 92748acf710SJeremy L Thompson } 92848acf710SJeremy L Thompson 92948acf710SJeremy L Thompson // Restore input arrays 93048acf710SJeremy L Thompson CeedCallBackend(CeedOperatorRestoreInputsAtPoints_Ref(num_input_fields, qf_input_fields, op_input_fields, e_data, impl)); 93148acf710SJeremy L Thompson 93248acf710SJeremy L Thompson // Cleanup point coordinates 93348acf710SJeremy L Thompson CeedCallBackend(CeedVectorDestroy(&point_coords)); 93448acf710SJeremy L Thompson CeedCallBackend(CeedElemRestrictionDestroy(&rstr_points)); 93548acf710SJeremy L Thompson return CEED_ERROR_SUCCESS; 93648acf710SJeremy L Thompson } 93748acf710SJeremy L Thompson 93848acf710SJeremy L Thompson //------------------------------------------------------------------------------ 939f10650afSjeremylt // Operator Destroy 940f10650afSjeremylt //------------------------------------------------------------------------------ 941f10650afSjeremylt static int CeedOperatorDestroy_Ref(CeedOperator op) { 942f10650afSjeremylt CeedOperator_Ref *impl; 943f10650afSjeremylt 944ad70ee2cSJeremy L Thompson CeedCallBackend(CeedOperatorGetData(op, &impl)); 9454fc1f125SJeremy L Thompson for (CeedInt i = 0; i < impl->num_inputs + impl->num_outputs; i++) { 9462b730f8bSJeremy L Thompson CeedCallBackend(CeedVectorDestroy(&impl->e_vecs_full[i])); 947f10650afSjeremylt } 9482b730f8bSJeremy L Thompson CeedCallBackend(CeedFree(&impl->e_vecs_full)); 9492b730f8bSJeremy L Thompson CeedCallBackend(CeedFree(&impl->input_states)); 950f10650afSjeremylt 9514fc1f125SJeremy L Thompson for (CeedInt i = 0; i < impl->num_inputs; i++) { 9522b730f8bSJeremy L Thompson CeedCallBackend(CeedVectorDestroy(&impl->e_vecs_in[i])); 9532b730f8bSJeremy L Thompson CeedCallBackend(CeedVectorDestroy(&impl->q_vecs_in[i])); 954f10650afSjeremylt } 9552b730f8bSJeremy L Thompson CeedCallBackend(CeedFree(&impl->e_vecs_in)); 9562b730f8bSJeremy L Thompson CeedCallBackend(CeedFree(&impl->q_vecs_in)); 957f10650afSjeremylt 9584fc1f125SJeremy L Thompson for (CeedInt i = 0; i < impl->num_outputs; i++) { 9592b730f8bSJeremy L Thompson CeedCallBackend(CeedVectorDestroy(&impl->e_vecs_out[i])); 9602b730f8bSJeremy L Thompson CeedCallBackend(CeedVectorDestroy(&impl->q_vecs_out[i])); 961f10650afSjeremylt } 9622b730f8bSJeremy L Thompson CeedCallBackend(CeedFree(&impl->e_vecs_out)); 9632b730f8bSJeremy L Thompson CeedCallBackend(CeedFree(&impl->q_vecs_out)); 96448acf710SJeremy L Thompson CeedCallBackend(CeedVectorDestroy(&impl->point_coords_elem)); 965f10650afSjeremylt 966bb219a0fSJeremy L Thompson // QFunction assembly 9674fc1f125SJeremy L Thompson for (CeedInt i = 0; i < impl->num_active_in; i++) { 9682b730f8bSJeremy L Thompson CeedCallBackend(CeedVectorDestroy(&impl->qf_active_in[i])); 969bb219a0fSJeremy L Thompson } 9702b730f8bSJeremy L Thompson CeedCallBackend(CeedFree(&impl->qf_active_in)); 971bb219a0fSJeremy L Thompson 9722b730f8bSJeremy L Thompson CeedCallBackend(CeedFree(&impl)); 973e15f9bd0SJeremy L Thompson return CEED_ERROR_SUCCESS; 974f10650afSjeremylt } 975f10650afSjeremylt 976f10650afSjeremylt //------------------------------------------------------------------------------ 977713f43c3Sjeremylt // Operator Create 978f10650afSjeremylt //------------------------------------------------------------------------------ 97921617c04Sjeremylt int CeedOperatorCreate_Ref(CeedOperator op) { 980fe2413ffSjeremylt Ceed ceed; 9814ce2993fSjeremylt CeedOperator_Ref *impl; 98221617c04Sjeremylt 983ad70ee2cSJeremy L Thompson CeedCallBackend(CeedOperatorGetCeed(op, &ceed)); 9842b730f8bSJeremy L Thompson CeedCallBackend(CeedCalloc(1, &impl)); 9852b730f8bSJeremy L Thompson CeedCallBackend(CeedOperatorSetData(op, impl)); 9862b730f8bSJeremy L Thompson CeedCallBackend(CeedSetBackendFunction(ceed, "Operator", op, "LinearAssembleQFunction", CeedOperatorLinearAssembleQFunction_Ref)); 9872b730f8bSJeremy L Thompson CeedCallBackend(CeedSetBackendFunction(ceed, "Operator", op, "LinearAssembleQFunctionUpdate", CeedOperatorLinearAssembleQFunctionUpdate_Ref)); 9882b730f8bSJeremy L Thompson CeedCallBackend(CeedSetBackendFunction(ceed, "Operator", op, "ApplyAdd", CeedOperatorApplyAdd_Ref)); 9892b730f8bSJeremy L Thompson CeedCallBackend(CeedSetBackendFunction(ceed, "Operator", op, "Destroy", CeedOperatorDestroy_Ref)); 990e15f9bd0SJeremy L Thompson return CEED_ERROR_SUCCESS; 99121617c04Sjeremylt } 9922a86cc9dSSebastian Grimberg 9932a86cc9dSSebastian Grimberg //------------------------------------------------------------------------------ 99448acf710SJeremy L Thompson // Operator Create At Points 99548acf710SJeremy L Thompson //------------------------------------------------------------------------------ 99648acf710SJeremy L Thompson int CeedOperatorCreateAtPoints_Ref(CeedOperator op) { 99748acf710SJeremy L Thompson Ceed ceed; 99848acf710SJeremy L Thompson CeedOperator_Ref *impl; 99948acf710SJeremy L Thompson 100048acf710SJeremy L Thompson CeedCallBackend(CeedOperatorGetCeed(op, &ceed)); 100148acf710SJeremy L Thompson CeedCallBackend(CeedCalloc(1, &impl)); 100248acf710SJeremy L Thompson CeedCallBackend(CeedOperatorSetData(op, impl)); 100348acf710SJeremy L Thompson CeedCallBackend(CeedSetBackendFunction(ceed, "Operator", op, "ApplyAdd", CeedOperatorApplyAddAtPoints_Ref)); 100448acf710SJeremy L Thompson CeedCallBackend(CeedSetBackendFunction(ceed, "Operator", op, "Destroy", CeedOperatorDestroy_Ref)); 100548acf710SJeremy L Thompson return CEED_ERROR_SUCCESS; 100648acf710SJeremy L Thompson } 100748acf710SJeremy L Thompson 100848acf710SJeremy L Thompson //------------------------------------------------------------------------------ 1009