xref: /libCEED/rust/libceed-sys/c-src/backends/sycl-ref/ceed-sycl-ref-operator.sycl.cpp (revision 681d0ea73ee05192cf73f31e6e4a886b41175395)
15aed82e4SJeremy L Thompson // Copyright (c) 2017-2024, Lawrence Livermore National Security, LLC and other
2bd882c8aSJames Wright // CEED contributors. All Rights Reserved. See the top-level LICENSE and NOTICE
3bd882c8aSJames Wright // files for details.
4bd882c8aSJames Wright //
5bd882c8aSJames Wright // SPDX-License-Identifier: BSD-2-Clause
6bd882c8aSJames Wright //
7bd882c8aSJames Wright // This file is part of CEED:  http://github.com/ceed
8bd882c8aSJames Wright 
9bd882c8aSJames Wright #include <ceed/backend.h>
10bd882c8aSJames Wright #include <ceed/ceed.h>
11bd882c8aSJames Wright 
12bd882c8aSJames Wright #include <cassert>
13bd882c8aSJames Wright #include <string>
14bd882c8aSJames Wright #include <sycl/sycl.hpp>
15bd882c8aSJames Wright 
16bd882c8aSJames Wright #include "../sycl/ceed-sycl-compile.hpp"
17bd882c8aSJames Wright #include "ceed-sycl-ref.hpp"
18bd882c8aSJames Wright 
19bd882c8aSJames Wright class CeedOperatorSyclLinearDiagonal;
20bd882c8aSJames Wright class CeedOperatorSyclLinearAssemble;
21bd882c8aSJames Wright class CeedOperatorSyclLinearAssembleFallback;
22bd882c8aSJames Wright 
23bd882c8aSJames Wright //------------------------------------------------------------------------------
24bd882c8aSJames Wright //  Get Basis Emode Pointer
25bd882c8aSJames Wright //------------------------------------------------------------------------------
26*681d0ea7SJeremy L Thompson void CeedOperatorGetBasisPointer_Sycl(const CeedScalar **basis_ptr, CeedEvalMode eval_mode, const CeedScalar *identity, const CeedScalar *interp,
27bd882c8aSJames Wright                                       const CeedScalar *grad) {
28*681d0ea7SJeremy L Thompson   switch (eval_mode) {
29bd882c8aSJames Wright     case CEED_EVAL_NONE:
30dd64fc84SJeremy L Thompson       *basis_ptr = identity;
31bd882c8aSJames Wright       break;
32bd882c8aSJames Wright     case CEED_EVAL_INTERP:
33dd64fc84SJeremy L Thompson       *basis_ptr = interp;
34bd882c8aSJames Wright       break;
35bd882c8aSJames Wright     case CEED_EVAL_GRAD:
36dd64fc84SJeremy L Thompson       *basis_ptr = grad;
37bd882c8aSJames Wright       break;
38bd882c8aSJames Wright     case CEED_EVAL_WEIGHT:
39bd882c8aSJames Wright     case CEED_EVAL_DIV:
40bd882c8aSJames Wright     case CEED_EVAL_CURL:
41bd882c8aSJames Wright       break;  // Caught by QF Assembly
42bd882c8aSJames Wright   }
43bd882c8aSJames Wright }
44bd882c8aSJames Wright 
45bd882c8aSJames Wright //------------------------------------------------------------------------------
46bd882c8aSJames Wright // Destroy operator
47bd882c8aSJames Wright //------------------------------------------------------------------------------
48bd882c8aSJames Wright static int CeedOperatorDestroy_Sycl(CeedOperator op) {
49bd882c8aSJames Wright   Ceed               ceed;
50bd882c8aSJames Wright   Ceed_Sycl         *sycl_data;
51dd64fc84SJeremy L Thompson   CeedOperator_Sycl *impl;
52dd64fc84SJeremy L Thompson 
53dd64fc84SJeremy L Thompson   CeedCallBackend(CeedOperatorGetData(op, &impl));
54dd64fc84SJeremy L Thompson   CeedCallBackend(CeedOperatorGetCeed(op, &ceed));
55bd882c8aSJames Wright   CeedCallBackend(CeedGetData(ceed, &sycl_data));
56bd882c8aSJames Wright 
57bd882c8aSJames Wright   // Apply data
58dd64fc84SJeremy L Thompson   for (CeedInt i = 0; i < impl->num_e_in + impl->num_e_out; i++) {
59dd64fc84SJeremy L Thompson     CeedCallBackend(CeedVectorDestroy(&impl->e_vecs[i]));
60bd882c8aSJames Wright   }
61dd64fc84SJeremy L Thompson   CeedCallBackend(CeedFree(&impl->e_vecs));
62bd882c8aSJames Wright 
63dd64fc84SJeremy L Thompson   for (CeedInt i = 0; i < impl->num_e_in; i++) {
64dd64fc84SJeremy L Thompson     CeedCallBackend(CeedVectorDestroy(&impl->q_vecs_in[i]));
65bd882c8aSJames Wright   }
66dd64fc84SJeremy L Thompson   CeedCallBackend(CeedFree(&impl->q_vecs_in));
67bd882c8aSJames Wright 
68dd64fc84SJeremy L Thompson   for (CeedInt i = 0; i < impl->num_e_out; i++) {
69dd64fc84SJeremy L Thompson     CeedCallBackend(CeedVectorDestroy(&impl->q_vecs_out[i]));
70bd882c8aSJames Wright   }
71dd64fc84SJeremy L Thompson   CeedCallBackend(CeedFree(&impl->q_vecs_out));
72bd882c8aSJames Wright 
73*681d0ea7SJeremy L Thompson   // QFunction assembly dataf
74dd64fc84SJeremy L Thompson   for (CeedInt i = 0; i < impl->num_active_in; i++) {
75dd64fc84SJeremy L Thompson     CeedCallBackend(CeedVectorDestroy(&impl->qf_active_in[i]));
76bd882c8aSJames Wright   }
77dd64fc84SJeremy L Thompson   CeedCallBackend(CeedFree(&impl->qf_active_in));
78bd882c8aSJames Wright 
79bd882c8aSJames Wright   // Diag data
80bd882c8aSJames Wright   if (impl->diag) {
81*681d0ea7SJeremy L Thompson     CeedCallBackend(CeedFree(&impl->diag->h_eval_mode_in));
82*681d0ea7SJeremy L Thompson     CeedCallBackend(CeedFree(&impl->diag->h_eval_mode_out));
83bd882c8aSJames Wright 
84bd882c8aSJames Wright     CeedCallSycl(ceed, sycl_data->sycl_queue.wait_and_throw());
85*681d0ea7SJeremy L Thompson     CeedCallSycl(ceed, sycl::free(impl->diag->d_eval_mode_in, sycl_data->sycl_context));
86*681d0ea7SJeremy L Thompson     CeedCallSycl(ceed, sycl::free(impl->diag->d_eval_mode_out, sycl_data->sycl_context));
87bd882c8aSJames Wright     CeedCallSycl(ceed, sycl::free(impl->diag->d_identity, sycl_data->sycl_context));
88dd64fc84SJeremy L Thompson     CeedCallSycl(ceed, sycl::free(impl->diag->d_interp_in, sycl_data->sycl_context));
89dd64fc84SJeremy L Thompson     CeedCallSycl(ceed, sycl::free(impl->diag->d_interp_out, sycl_data->sycl_context));
90dd64fc84SJeremy L Thompson     CeedCallSycl(ceed, sycl::free(impl->diag->d_grad_in, sycl_data->sycl_context));
91dd64fc84SJeremy L Thompson     CeedCallSycl(ceed, sycl::free(impl->diag->d_grad_out, sycl_data->sycl_context));
92dd64fc84SJeremy L Thompson     CeedCallBackend(CeedElemRestrictionDestroy(&impl->diag->point_block_diag_rstr));
93bd882c8aSJames Wright 
94dd64fc84SJeremy L Thompson     CeedCallBackend(CeedVectorDestroy(&impl->diag->elem_diag));
95dd64fc84SJeremy L Thompson     CeedCallBackend(CeedVectorDestroy(&impl->diag->point_block_elem_diag));
96bd882c8aSJames Wright   }
97bd882c8aSJames Wright   CeedCallBackend(CeedFree(&impl->diag));
98bd882c8aSJames Wright 
99bd882c8aSJames Wright   if (impl->asmb) {
100bd882c8aSJames Wright     CeedCallSycl(ceed, sycl_data->sycl_queue.wait_and_throw());
101bd882c8aSJames Wright     CeedCallSycl(ceed, sycl::free(impl->asmb->d_B_in, sycl_data->sycl_context));
102bd882c8aSJames Wright     CeedCallSycl(ceed, sycl::free(impl->asmb->d_B_out, sycl_data->sycl_context));
103bd882c8aSJames Wright   }
104bd882c8aSJames Wright   CeedCallBackend(CeedFree(&impl->asmb));
105bd882c8aSJames Wright 
106bd882c8aSJames Wright   CeedCallBackend(CeedFree(&impl));
107bd882c8aSJames Wright   return CEED_ERROR_SUCCESS;
108bd882c8aSJames Wright }
109bd882c8aSJames Wright 
110bd882c8aSJames Wright //------------------------------------------------------------------------------
111bd882c8aSJames Wright // Setup infields or outfields
112bd882c8aSJames Wright //------------------------------------------------------------------------------
113dd64fc84SJeremy L Thompson static int CeedOperatorSetupFields_Sycl(CeedQFunction qf, CeedOperator op, bool is_input, CeedVector *e_vecs, CeedVector *q_vecs, CeedInt start_e,
114dd64fc84SJeremy L Thompson                                         CeedInt num_fields, CeedInt Q, CeedInt num_elem) {
115bd882c8aSJames Wright   Ceed                ceed;
116dd64fc84SJeremy L Thompson   CeedSize            q_size;
117dd64fc84SJeremy L Thompson   bool                is_strided, skip_restriction;
118dd64fc84SJeremy L Thompson   CeedInt             dim, size;
119dd64fc84SJeremy L Thompson   CeedOperatorField  *op_fields;
120dd64fc84SJeremy L Thompson   CeedQFunctionField *qf_fields;
121bd882c8aSJames Wright 
122dd64fc84SJeremy L Thompson   CeedCallBackend(CeedOperatorGetCeed(op, &ceed));
123dd64fc84SJeremy L Thompson   if (is_input) {
124dd64fc84SJeremy L Thompson     CeedCallBackend(CeedOperatorGetFields(op, NULL, &op_fields, NULL, NULL));
125dd64fc84SJeremy L Thompson     CeedCallBackend(CeedQFunctionGetFields(qf, NULL, &qf_fields, NULL, NULL));
126bd882c8aSJames Wright   } else {
127dd64fc84SJeremy L Thompson     CeedCallBackend(CeedOperatorGetFields(op, NULL, NULL, NULL, &op_fields));
128dd64fc84SJeremy L Thompson     CeedCallBackend(CeedQFunctionGetFields(qf, NULL, NULL, NULL, &qf_fields));
129bd882c8aSJames Wright   }
130bd882c8aSJames Wright 
131bd882c8aSJames Wright   // Loop over fields
132dd64fc84SJeremy L Thompson   for (CeedInt i = 0; i < num_fields; i++) {
133*681d0ea7SJeremy L Thompson     CeedEvalMode        eval_mode;
134dd64fc84SJeremy L Thompson     CeedVector          vec;
135dd64fc84SJeremy L Thompson     CeedElemRestriction rstr;
136dd64fc84SJeremy L Thompson     CeedBasis           basis;
137bd882c8aSJames Wright 
138*681d0ea7SJeremy L Thompson     CeedCallBackend(CeedQFunctionFieldGetEvalMode(qf_fields[i], &eval_mode));
139dd64fc84SJeremy L Thompson 
140dd64fc84SJeremy L Thompson     is_strided       = false;
141dd64fc84SJeremy L Thompson     skip_restriction = false;
142*681d0ea7SJeremy L Thompson     if (eval_mode != CEED_EVAL_WEIGHT) {
143dd64fc84SJeremy L Thompson       CeedCallBackend(CeedOperatorFieldGetElemRestriction(op_fields[i], &rstr));
144bd882c8aSJames Wright 
145bd882c8aSJames Wright       // Check whether this field can skip the element restriction:
146*681d0ea7SJeremy L Thompson       // must be passive input, with  eval_mode NONE, and have a strided restriction with CEED_STRIDES_BACKEND.
147bd882c8aSJames Wright 
148bd882c8aSJames Wright       // First, check whether the field is input or output:
149dd64fc84SJeremy L Thompson       if (is_input) {
150bd882c8aSJames Wright         // Check for passive input:
151dd64fc84SJeremy L Thompson         CeedCallBackend(CeedOperatorFieldGetVector(op_fields[i], &vec));
152dd64fc84SJeremy L Thompson         if (vec != CEED_VECTOR_ACTIVE) {
153*681d0ea7SJeremy L Thompson           // Check  eval_mode
154*681d0ea7SJeremy L Thompson           if (eval_mode == CEED_EVAL_NONE) {
155dd64fc84SJeremy L Thompson             // Check for  is_strided restriction
156dd64fc84SJeremy L Thompson             CeedCallBackend(CeedElemRestrictionIsStrided(rstr, &is_strided));
157dd64fc84SJeremy L Thompson             if (is_strided) {
158bd882c8aSJames Wright               // Check if vector is already in preferred backend ordering
159dd64fc84SJeremy L Thompson               CeedCallBackend(CeedElemRestrictionHasBackendStrides(rstr, &skip_restriction));
160bd882c8aSJames Wright             }
161bd882c8aSJames Wright           }
162bd882c8aSJames Wright         }
163*681d0ea7SJeremy L Thompson         CeedCallBackend(CeedVectorDestroy(&vec));
164bd882c8aSJames Wright       }
165dd64fc84SJeremy L Thompson       if (skip_restriction) {
166bd882c8aSJames Wright         // We do not need an E-Vector, but will use the input field vector's data directly in the operator application
167dd64fc84SJeremy L Thompson         e_vecs[i + start_e] = NULL;
168bd882c8aSJames Wright       } else {
169dd64fc84SJeremy L Thompson         CeedCallBackend(CeedElemRestrictionCreateVector(rstr, NULL, &e_vecs[i + start_e]));
170bd882c8aSJames Wright       }
171*681d0ea7SJeremy L Thompson       CeedCallBackend(CeedElemRestrictionDestroy(&rstr));
172bd882c8aSJames Wright     }
173bd882c8aSJames Wright 
174*681d0ea7SJeremy L Thompson     switch (eval_mode) {
175bd882c8aSJames Wright       case CEED_EVAL_NONE:
176dd64fc84SJeremy L Thompson         CeedCallBackend(CeedQFunctionFieldGetSize(qf_fields[i], &size));
177dd64fc84SJeremy L Thompson         q_size = (CeedSize)num_elem * Q * size;
178dd64fc84SJeremy L Thompson         CeedCallBackend(CeedVectorCreate(ceed, q_size, &q_vecs[i]));
179bd882c8aSJames Wright         break;
180bd882c8aSJames Wright       case CEED_EVAL_INTERP:
181dd64fc84SJeremy L Thompson         CeedCallBackend(CeedQFunctionFieldGetSize(qf_fields[i], &size));
182dd64fc84SJeremy L Thompson         q_size = (CeedSize)num_elem * Q * size;
183dd64fc84SJeremy L Thompson         CeedCallBackend(CeedVectorCreate(ceed, q_size, &q_vecs[i]));
184bd882c8aSJames Wright         break;
185bd882c8aSJames Wright       case CEED_EVAL_GRAD:
186dd64fc84SJeremy L Thompson         CeedCallBackend(CeedOperatorFieldGetBasis(op_fields[i], &basis));
187dd64fc84SJeremy L Thompson         CeedCallBackend(CeedQFunctionFieldGetSize(qf_fields[i], &size));
188bd882c8aSJames Wright         CeedCallBackend(CeedBasisGetDimension(basis, &dim));
189*681d0ea7SJeremy L Thompson         CeedCallBackend(CeedBasisDestroy(&basis));
190dd64fc84SJeremy L Thompson         q_size = (CeedSize)num_elem * Q * size;
191dd64fc84SJeremy L Thompson         CeedCallBackend(CeedVectorCreate(ceed, q_size, &q_vecs[i]));
192bd882c8aSJames Wright         break;
193bd882c8aSJames Wright       case CEED_EVAL_WEIGHT:  // Only on input fields
194dd64fc84SJeremy L Thompson         CeedCallBackend(CeedOperatorFieldGetBasis(op_fields[i], &basis));
195dd64fc84SJeremy L Thompson         q_size = (CeedSize)num_elem * Q;
196dd64fc84SJeremy L Thompson         CeedCallBackend(CeedVectorCreate(ceed, q_size, &q_vecs[i]));
1976078361eSJames Wright         CeedCallBackend(CeedBasisApply(basis, num_elem, CEED_NOTRANSPOSE, CEED_EVAL_WEIGHT, CEED_VECTOR_NONE, q_vecs[i]));
198*681d0ea7SJeremy L Thompson         CeedCallBackend(CeedBasisDestroy(&basis));
199bd882c8aSJames Wright         break;
200bd882c8aSJames Wright       case CEED_EVAL_DIV:
201bd882c8aSJames Wright         break;  // TODO: Not implemented
202bd882c8aSJames Wright       case CEED_EVAL_CURL:
203bd882c8aSJames Wright         break;  // TODO: Not implemented
204bd882c8aSJames Wright     }
205bd882c8aSJames Wright   }
206bd882c8aSJames Wright   return CEED_ERROR_SUCCESS;
207bd882c8aSJames Wright }
208bd882c8aSJames Wright 
209bd882c8aSJames Wright //------------------------------------------------------------------------------
210bd882c8aSJames Wright // CeedOperator needs to connect all the named fields (be they active or
211bd882c8aSJames Wright // passive) to the named inputs and outputs of its CeedQFunction.
212bd882c8aSJames Wright //------------------------------------------------------------------------------
213bd882c8aSJames Wright static int CeedOperatorSetup_Sycl(CeedOperator op) {
214bd882c8aSJames Wright   Ceed                ceed;
215dd64fc84SJeremy L Thompson   bool                is_setup_done;
216dd64fc84SJeremy L Thompson   CeedInt             Q, num_elem, num_input_fields, num_output_fields;
217dd64fc84SJeremy L Thompson   CeedQFunctionField *qf_input_fields, *qf_output_fields;
218bd882c8aSJames Wright   CeedQFunction       qf;
219dd64fc84SJeremy L Thompson   CeedOperatorField  *op_input_fields, *op_output_fields;
220dd64fc84SJeremy L Thompson   CeedOperator_Sycl  *impl;
221dd64fc84SJeremy L Thompson 
222dd64fc84SJeremy L Thompson   CeedCallBackend(CeedOperatorIsSetupDone(op, &is_setup_done));
223dd64fc84SJeremy L Thompson   if (is_setup_done) return CEED_ERROR_SUCCESS;
224dd64fc84SJeremy L Thompson 
225dd64fc84SJeremy L Thompson   CeedCallBackend(CeedOperatorGetCeed(op, &ceed));
226dd64fc84SJeremy L Thompson   CeedCallBackend(CeedOperatorGetData(op, &impl));
227bd882c8aSJames Wright   CeedCallBackend(CeedOperatorGetQFunction(op, &qf));
228bd882c8aSJames Wright   CeedCallBackend(CeedOperatorGetNumQuadraturePoints(op, &Q));
229dd64fc84SJeremy L Thompson   CeedCallBackend(CeedOperatorGetNumElements(op, &num_elem));
230dd64fc84SJeremy L Thompson   CeedCallBackend(CeedOperatorGetFields(op, &num_input_fields, &op_input_fields, &num_output_fields, &op_output_fields));
231dd64fc84SJeremy L Thompson   CeedCallBackend(CeedQFunctionGetFields(qf, NULL, &qf_input_fields, NULL, &qf_output_fields));
232bd882c8aSJames Wright 
233bd882c8aSJames Wright   // Allocate
234dd64fc84SJeremy L Thompson   CeedCallBackend(CeedCalloc(num_input_fields + num_output_fields, &impl->e_vecs));
235bd882c8aSJames Wright 
236dd64fc84SJeremy L Thompson   CeedCallBackend(CeedCalloc(CEED_FIELD_MAX, &impl->q_vecs_in));
237dd64fc84SJeremy L Thompson   CeedCallBackend(CeedCalloc(CEED_FIELD_MAX, &impl->q_vecs_out));
238bd882c8aSJames Wright 
239dd64fc84SJeremy L Thompson   impl->num_e_in  = num_input_fields;
240dd64fc84SJeremy L Thompson   impl->num_e_out = num_output_fields;
241bd882c8aSJames Wright 
242dd64fc84SJeremy L Thompson   // Set up infield and outfield  e_vecs and  q_vecs
243bd882c8aSJames Wright   // Infields
244dd64fc84SJeremy L Thompson   CeedCallBackend(CeedOperatorSetupFields_Sycl(qf, op, true, impl->e_vecs, impl->q_vecs_in, 0, num_input_fields, Q, num_elem));
245bd882c8aSJames Wright   // Outfields
246dd64fc84SJeremy L Thompson   CeedCallBackend(CeedOperatorSetupFields_Sycl(qf, op, false, impl->e_vecs, impl->q_vecs_out, num_input_fields, num_output_fields, Q, num_elem));
247bd882c8aSJames Wright 
248bd882c8aSJames Wright   CeedCallBackend(CeedOperatorSetSetupDone(op));
249bd882c8aSJames Wright   return CEED_ERROR_SUCCESS;
250bd882c8aSJames Wright }
251bd882c8aSJames Wright 
252bd882c8aSJames Wright //------------------------------------------------------------------------------
253bd882c8aSJames Wright // Setup Operator Inputs
254bd882c8aSJames Wright //------------------------------------------------------------------------------
255dd64fc84SJeremy L Thompson static inline int CeedOperatorSetupInputs_Sycl(CeedInt num_input_fields, CeedQFunctionField *qf_input_fields, CeedOperatorField *op_input_fields,
256dd64fc84SJeremy L Thompson                                                CeedVector in_vec, const bool skip_active, CeedScalar *e_data[2 * CEED_FIELD_MAX],
257bd882c8aSJames Wright                                                CeedOperator_Sycl *impl, CeedRequest *request) {
258dd64fc84SJeremy L Thompson   for (CeedInt i = 0; i < num_input_fields; i++) {
259*681d0ea7SJeremy L Thompson     bool         is_active;
260*681d0ea7SJeremy L Thompson     CeedEvalMode eval_mode;
261bd882c8aSJames Wright     CeedVector   vec;
262bd882c8aSJames Wright 
263bd882c8aSJames Wright     // Get input vector
264dd64fc84SJeremy L Thompson     CeedCallBackend(CeedOperatorFieldGetVector(op_input_fields[i], &vec));
265*681d0ea7SJeremy L Thompson     is_active = vec == CEED_VECTOR_ACTIVE;
266*681d0ea7SJeremy L Thompson     if (is_active) {
267dd64fc84SJeremy L Thompson       if (skip_active) continue;
268dd64fc84SJeremy L Thompson       else vec = in_vec;
269bd882c8aSJames Wright     }
270bd882c8aSJames Wright 
271*681d0ea7SJeremy L Thompson     CeedCallBackend(CeedQFunctionFieldGetEvalMode(qf_input_fields[i], &eval_mode));
272*681d0ea7SJeremy L Thompson     if (eval_mode == CEED_EVAL_WEIGHT) {  // Skip
273bd882c8aSJames Wright     } else {
274bd882c8aSJames Wright       // Restrict, if necessary
275dd64fc84SJeremy L Thompson       if (!impl->e_vecs[i]) {
276bd882c8aSJames Wright         // No restriction for this field; read data directly from vec.
277dd64fc84SJeremy L Thompson         CeedCallBackend(CeedVectorGetArrayRead(vec, CEED_MEM_DEVICE, (const CeedScalar **)&e_data[i]));
278bd882c8aSJames Wright       } else {
279*681d0ea7SJeremy L Thompson         CeedElemRestriction rstr;
280*681d0ea7SJeremy L Thompson 
281*681d0ea7SJeremy L Thompson         CeedCallBackend(CeedOperatorFieldGetElemRestriction(op_input_fields[i], &rstr));
282dd64fc84SJeremy L Thompson         CeedCallBackend(CeedElemRestrictionApply(rstr, CEED_NOTRANSPOSE, vec, impl->e_vecs[i], request));
283*681d0ea7SJeremy L Thompson         CeedCallBackend(CeedElemRestrictionDestroy(&rstr));
284dd64fc84SJeremy L Thompson         CeedCallBackend(CeedVectorGetArrayRead(impl->e_vecs[i], CEED_MEM_DEVICE, (const CeedScalar **)&e_data[i]));
285bd882c8aSJames Wright       }
286bd882c8aSJames Wright     }
287*681d0ea7SJeremy L Thompson     if (!is_active) CeedCallBackend(CeedVectorDestroy(&vec));
288bd882c8aSJames Wright   }
289bd882c8aSJames Wright   return CEED_ERROR_SUCCESS;
290bd882c8aSJames Wright }
291bd882c8aSJames Wright 
292bd882c8aSJames Wright //------------------------------------------------------------------------------
293bd882c8aSJames Wright // Input Basis Action
294bd882c8aSJames Wright //------------------------------------------------------------------------------
295dd64fc84SJeremy L Thompson static inline int CeedOperatorInputBasis_Sycl(CeedInt num_elem, CeedQFunctionField *qf_input_fields, CeedOperatorField *op_input_fields,
296dd64fc84SJeremy L Thompson                                               CeedInt num_input_fields, const bool skip_active, CeedScalar *e_data[2 * CEED_FIELD_MAX],
297bd882c8aSJames Wright                                               CeedOperator_Sycl *impl) {
298dd64fc84SJeremy L Thompson   for (CeedInt i = 0; i < num_input_fields; i++) {
299*681d0ea7SJeremy L Thompson     CeedEvalMode eval_mode;
300bd882c8aSJames Wright     CeedBasis    basis;
301bd882c8aSJames Wright 
302bd882c8aSJames Wright     // Skip active input
303dd64fc84SJeremy L Thompson     if (skip_active) {
304*681d0ea7SJeremy L Thompson       bool       is_active;
305bd882c8aSJames Wright       CeedVector vec;
306dd64fc84SJeremy L Thompson 
307dd64fc84SJeremy L Thompson       CeedCallBackend(CeedOperatorFieldGetVector(op_input_fields[i], &vec));
308*681d0ea7SJeremy L Thompson       is_active = vec == CEED_VECTOR_ACTIVE;
309*681d0ea7SJeremy L Thompson       CeedCallBackend(CeedVectorDestroy(&vec));
310*681d0ea7SJeremy L Thompson       if (is_active) continue;
311bd882c8aSJames Wright     }
312bd882c8aSJames Wright     // Basis action
313*681d0ea7SJeremy L Thompson     CeedCallBackend(CeedQFunctionFieldGetEvalMode(qf_input_fields[i], &eval_mode));
314*681d0ea7SJeremy L Thompson     switch (eval_mode) {
315bd882c8aSJames Wright       case CEED_EVAL_NONE:
316dd64fc84SJeremy L Thompson         CeedCallBackend(CeedVectorSetArray(impl->q_vecs_in[i], CEED_MEM_DEVICE, CEED_USE_POINTER, e_data[i]));
317bd882c8aSJames Wright         break;
318bd882c8aSJames Wright       case CEED_EVAL_INTERP:
319bd882c8aSJames Wright       case CEED_EVAL_GRAD:
320dd64fc84SJeremy L Thompson         CeedCallBackend(CeedOperatorFieldGetBasis(op_input_fields[i], &basis));
321*681d0ea7SJeremy L Thompson         CeedCallBackend(CeedBasisApply(basis, num_elem, CEED_NOTRANSPOSE, eval_mode, impl->e_vecs[i], impl->q_vecs_in[i]));
322*681d0ea7SJeremy L Thompson         CeedCallBackend(CeedBasisDestroy(&basis));
323bd882c8aSJames Wright         break;
324bd882c8aSJames Wright       case CEED_EVAL_WEIGHT:
325bd882c8aSJames Wright         break;  // No action
326bd882c8aSJames Wright       case CEED_EVAL_DIV:
327bd882c8aSJames Wright         break;  // TODO: Not implemented
328bd882c8aSJames Wright       case CEED_EVAL_CURL:
329bd882c8aSJames Wright         break;  // TODO: Not implemented
330bd882c8aSJames Wright     }
331bd882c8aSJames Wright   }
332bd882c8aSJames Wright   return CEED_ERROR_SUCCESS;
333bd882c8aSJames Wright }
334bd882c8aSJames Wright 
335bd882c8aSJames Wright //------------------------------------------------------------------------------
336bd882c8aSJames Wright // Restore Input Vectors
337bd882c8aSJames Wright //------------------------------------------------------------------------------
338dd64fc84SJeremy L Thompson static inline int CeedOperatorRestoreInputs_Sycl(CeedInt num_input_fields, CeedQFunctionField *qf_input_fields, CeedOperatorField *op_input_fields,
339dd64fc84SJeremy L Thompson                                                  const bool skip_active, CeedScalar *e_data[2 * CEED_FIELD_MAX], CeedOperator_Sycl *impl) {
340dd64fc84SJeremy L Thompson   for (CeedInt i = 0; i < num_input_fields; i++) {
341*681d0ea7SJeremy L Thompson     bool         is_active;
342*681d0ea7SJeremy L Thompson     CeedEvalMode eval_mode;
343bd882c8aSJames Wright     CeedVector   vec;
344bd882c8aSJames Wright 
345*681d0ea7SJeremy L Thompson     CeedCallBackend(CeedOperatorFieldGetVector(op_input_fields[i], &vec));
346*681d0ea7SJeremy L Thompson     is_active = vec == CEED_VECTOR_ACTIVE;
347bd882c8aSJames Wright     // Skip active input
348dd64fc84SJeremy L Thompson     if (skip_active) {
349*681d0ea7SJeremy L Thompson       if (is_active) continue;
350bd882c8aSJames Wright     }
351*681d0ea7SJeremy L Thompson     CeedCallBackend(CeedQFunctionFieldGetEvalMode(qf_input_fields[i], &eval_mode));
352*681d0ea7SJeremy L Thompson     if (eval_mode == CEED_EVAL_WEIGHT) {  // Skip
353bd882c8aSJames Wright     } else {
354dd64fc84SJeremy L Thompson       if (!impl->e_vecs[i]) {  // This was a  skip_restriction case
355dd64fc84SJeremy L Thompson         CeedCallBackend(CeedVectorRestoreArrayRead(vec, (const CeedScalar **)&e_data[i]));
356bd882c8aSJames Wright       } else {
357dd64fc84SJeremy L Thompson         CeedCallBackend(CeedVectorRestoreArrayRead(impl->e_vecs[i], (const CeedScalar **)&e_data[i]));
358bd882c8aSJames Wright       }
359bd882c8aSJames Wright     }
360*681d0ea7SJeremy L Thompson     if (!is_active) CeedCallBackend(CeedVectorDestroy(&vec));
361bd882c8aSJames Wright   }
362bd882c8aSJames Wright   return CEED_ERROR_SUCCESS;
363bd882c8aSJames Wright }
364bd882c8aSJames Wright 
365bd882c8aSJames Wright //------------------------------------------------------------------------------
366bd882c8aSJames Wright // Apply and add to output
367bd882c8aSJames Wright //------------------------------------------------------------------------------
368dd64fc84SJeremy L Thompson static int CeedOperatorApplyAdd_Sycl(CeedOperator op, CeedVector in_vec, CeedVector out_vec, CeedRequest *request) {
369dd64fc84SJeremy L Thompson   CeedInt             Q, num_elem, elem_size, num_input_fields, num_output_fields, size;
3709d1bceceSJames Wright   CeedEvalMode        eval_mode;
371dd64fc84SJeremy L Thompson   CeedScalar         *e_data[2 * CEED_FIELD_MAX] = {0};
372dd64fc84SJeremy L Thompson   CeedQFunctionField *qf_input_fields, *qf_output_fields;
373bd882c8aSJames Wright   CeedQFunction       qf;
374dd64fc84SJeremy L Thompson   CeedOperatorField  *op_input_fields, *op_output_fields;
375dd64fc84SJeremy L Thompson   CeedOperator_Sycl  *impl;
376dd64fc84SJeremy L Thompson 
377dd64fc84SJeremy L Thompson   CeedCallBackend(CeedOperatorGetData(op, &impl));
378bd882c8aSJames Wright   CeedCallBackend(CeedOperatorGetQFunction(op, &qf));
379bd882c8aSJames Wright   CeedCallBackend(CeedOperatorGetNumQuadraturePoints(op, &Q));
380dd64fc84SJeremy L Thompson   CeedCallBackend(CeedOperatorGetNumElements(op, &num_elem));
381dd64fc84SJeremy L Thompson   CeedCallBackend(CeedOperatorGetFields(op, &num_input_fields, &op_input_fields, &num_output_fields, &op_output_fields));
382dd64fc84SJeremy L Thompson   CeedCallBackend(CeedQFunctionGetFields(qf, NULL, &qf_input_fields, NULL, &qf_output_fields));
383bd882c8aSJames Wright 
384bd882c8aSJames Wright   // Setup
385bd882c8aSJames Wright   CeedCallBackend(CeedOperatorSetup_Sycl(op));
386bd882c8aSJames Wright 
387bd882c8aSJames Wright   // Input Evecs and Restriction
388dd64fc84SJeremy L Thompson   CeedCallBackend(CeedOperatorSetupInputs_Sycl(num_input_fields, qf_input_fields, op_input_fields, in_vec, false, e_data, impl, request));
389bd882c8aSJames Wright 
390bd882c8aSJames Wright   // Input basis apply if needed
391dd64fc84SJeremy L Thompson   CeedCallBackend(CeedOperatorInputBasis_Sycl(num_elem, qf_input_fields, op_input_fields, num_input_fields, false, e_data, impl));
392bd882c8aSJames Wright 
393bd882c8aSJames Wright   // Output pointers, as necessary
394dd64fc84SJeremy L Thompson   for (CeedInt i = 0; i < num_output_fields; i++) {
3959d1bceceSJames Wright     CeedCallBackend(CeedQFunctionFieldGetEvalMode(qf_output_fields[i], &eval_mode));
3969d1bceceSJames Wright     if (eval_mode == CEED_EVAL_NONE) {
397bd882c8aSJames Wright       // Set the output Q-Vector to use the E-Vector data directly
398dd64fc84SJeremy L Thompson       CeedCallBackend(CeedVectorGetArrayWrite(impl->e_vecs[i + impl->num_e_in], CEED_MEM_DEVICE, &e_data[i + num_input_fields]));
399dd64fc84SJeremy L Thompson       CeedCallBackend(CeedVectorSetArray(impl->q_vecs_out[i], CEED_MEM_DEVICE, CEED_USE_POINTER, e_data[i + num_input_fields]));
400bd882c8aSJames Wright     }
401bd882c8aSJames Wright   }
402bd882c8aSJames Wright 
403bd882c8aSJames Wright   // Q function
404dd64fc84SJeremy L Thompson   CeedCallBackend(CeedQFunctionApply(qf, num_elem * Q, impl->q_vecs_in, impl->q_vecs_out));
405bd882c8aSJames Wright 
406bd882c8aSJames Wright   // Output basis apply if needed
407dd64fc84SJeremy L Thompson   for (CeedInt i = 0; i < num_output_fields; i++) {
408dd64fc84SJeremy L Thompson     CeedElemRestriction rstr;
409dd64fc84SJeremy L Thompson     CeedBasis           basis;
410dd64fc84SJeremy L Thompson 
4119d1bceceSJames Wright     // Get elem_size, eval_mode, size
412dd64fc84SJeremy L Thompson     CeedCallBackend(CeedOperatorFieldGetElemRestriction(op_output_fields[i], &rstr));
413dd64fc84SJeremy L Thompson     CeedCallBackend(CeedElemRestrictionGetElementSize(rstr, &elem_size));
414*681d0ea7SJeremy L Thompson     CeedCallBackend(CeedElemRestrictionDestroy(&rstr));
4159d1bceceSJames Wright     CeedCallBackend(CeedQFunctionFieldGetEvalMode(qf_output_fields[i], &eval_mode));
416dd64fc84SJeremy L Thompson     CeedCallBackend(CeedQFunctionFieldGetSize(qf_output_fields[i], &size));
417bd882c8aSJames Wright     // Basis action
4189d1bceceSJames Wright     switch (eval_mode) {
419bd882c8aSJames Wright       case CEED_EVAL_NONE:
420bd882c8aSJames Wright         break;
421bd882c8aSJames Wright       case CEED_EVAL_INTERP:
422bd882c8aSJames Wright       case CEED_EVAL_GRAD:
423dd64fc84SJeremy L Thompson         CeedCallBackend(CeedOperatorFieldGetBasis(op_output_fields[i], &basis));
424*681d0ea7SJeremy L Thompson         CeedCallBackend(CeedBasisApply(basis, num_elem, CEED_TRANSPOSE, eval_mode, impl->q_vecs_out[i], impl->e_vecs[i + impl->num_e_in]));
425*681d0ea7SJeremy L Thompson         CeedCallBackend(CeedBasisDestroy(&basis));
426bd882c8aSJames Wright         break;
427bd882c8aSJames Wright       // LCOV_EXCL_START
428bd882c8aSJames Wright       case CEED_EVAL_WEIGHT:
429bd882c8aSJames Wright         Ceed ceed;
4304e3038a5SJeremy L Thompson 
431bd882c8aSJames Wright         CeedCallBackend(CeedOperatorGetCeed(op, &ceed));
432bd882c8aSJames Wright         return CeedError(ceed, CEED_ERROR_BACKEND, "CEED_EVAL_WEIGHT cannot be an output evaluation mode");
433bd882c8aSJames Wright         break;  // Should not occur
434bd882c8aSJames Wright       case CEED_EVAL_DIV:
4354e3038a5SJeremy L Thompson       case CEED_EVAL_CURL: {
4364e3038a5SJeremy L Thompson         Ceed ceed;
4374e3038a5SJeremy L Thompson 
4384e3038a5SJeremy L Thompson         CeedCallBackend(CeedOperatorGetCeed(op, &ceed));
4394e3038a5SJeremy L Thompson         return CeedError(ceed, CEED_ERROR_BACKEND, "%s not supported", CeedEvalModes[eval_mode]);
4404e3038a5SJeremy L Thompson         break;  // Should not occur
4414e3038a5SJeremy L Thompson       }
442bd882c8aSJames Wright         // LCOV_EXCL_STOP
443bd882c8aSJames Wright     }
444bd882c8aSJames Wright   }
445bd882c8aSJames Wright 
446bd882c8aSJames Wright   // Output restriction
447dd64fc84SJeremy L Thompson   for (CeedInt i = 0; i < num_output_fields; i++) {
448*681d0ea7SJeremy L Thompson     bool                is_active;
449*681d0ea7SJeremy L Thompson     CeedEvalMode        eval_mode;
450dd64fc84SJeremy L Thompson     CeedVector          vec;
451dd64fc84SJeremy L Thompson     CeedElemRestriction rstr;
452dd64fc84SJeremy L Thompson 
453bd882c8aSJames Wright     // Restore evec
4549d1bceceSJames Wright     CeedCallBackend(CeedQFunctionFieldGetEvalMode(qf_output_fields[i], &eval_mode));
4559d1bceceSJames Wright     if (eval_mode == CEED_EVAL_NONE) {
456dd64fc84SJeremy L Thompson       CeedCallBackend(CeedVectorRestoreArray(impl->e_vecs[i + impl->num_e_in], &e_data[i + num_input_fields]));
457bd882c8aSJames Wright     }
458bd882c8aSJames Wright     // Restrict
459*681d0ea7SJeremy L Thompson     CeedCallBackend(CeedOperatorFieldGetVector(op_output_fields[i], &vec));
460*681d0ea7SJeremy L Thompson     is_active = vec == CEED_VECTOR_ACTIVE;
461dd64fc84SJeremy L Thompson     CeedCallBackend(CeedOperatorFieldGetElemRestriction(op_output_fields[i], &rstr));
462*681d0ea7SJeremy L Thompson     if (is_active) vec = out_vec;
463*681d0ea7SJeremy L Thompson     CeedCallBackend(CeedElemRestrictionApply(rstr, CEED_TRANSPOSE, impl->e_vecs[i + impl->num_inputs], vec, request));
464*681d0ea7SJeremy L Thompson     if (!is_active) CeedCallBackend(CeedVectorDestroy(&vec));
465*681d0ea7SJeremy L Thompson     CeedCallBackend(CeedElemRestrictionDestroy(&rstr));
466bd882c8aSJames Wright   }
467bd882c8aSJames Wright 
468bd882c8aSJames Wright   // Restore input arrays
469dd64fc84SJeremy L Thompson   CeedCallBackend(CeedOperatorRestoreInputs_Sycl(num_input_fields, qf_input_fields, op_input_fields, false, e_data, impl));
470bd882c8aSJames Wright   return CEED_ERROR_SUCCESS;
471bd882c8aSJames Wright }
472bd882c8aSJames Wright 
473bd882c8aSJames Wright //------------------------------------------------------------------------------
474bd882c8aSJames Wright // Core code for assembling linear QFunction
475bd882c8aSJames Wright //------------------------------------------------------------------------------
476bd882c8aSJames Wright static inline int CeedOperatorLinearAssembleQFunctionCore_Sycl(CeedOperator op, bool build_objects, CeedVector *assembled, CeedElemRestriction *rstr,
477bd882c8aSJames Wright                                                                CeedRequest *request) {
478dd64fc84SJeremy L Thompson   Ceed                ceed, ceed_parent;
479e984cf9aSJeremy L Thompson   CeedSize            q_size;
480dd64fc84SJeremy L Thompson   CeedInt             num_active_in, num_active_out, Q, num_elem, num_input_fields, num_output_fields, size;
481dd64fc84SJeremy L Thompson   CeedScalar         *assembled_array, *e_data[2 * CEED_FIELD_MAX] = {NULL};
482dd64fc84SJeremy L Thompson   CeedVector         *active_in;
483dd64fc84SJeremy L Thompson   CeedQFunctionField *qf_input_fields, *qf_output_fields;
484dd64fc84SJeremy L Thompson   CeedQFunction       qf;
485dd64fc84SJeremy L Thompson   CeedOperatorField  *op_input_fields, *op_output_fields;
486dd64fc84SJeremy L Thompson   CeedOperator_Sycl  *impl;
487dd64fc84SJeremy L Thompson 
488bd882c8aSJames Wright   CeedCallBackend(CeedOperatorGetCeed(op, &ceed));
489dd64fc84SJeremy L Thompson   CeedCallBackend(CeedOperatorGetFallbackParentCeed(op, &ceed_parent));
490e984cf9aSJeremy L Thompson   CeedCallBackend(CeedOperatorGetData(op, &impl));
491e984cf9aSJeremy L Thompson   CeedCallBackend(CeedOperatorGetQFunction(op, &qf));
492e984cf9aSJeremy L Thompson   CeedCallBackend(CeedOperatorGetNumQuadraturePoints(op, &Q));
493dd64fc84SJeremy L Thompson   CeedCallBackend(CeedOperatorGetNumElements(op, &num_elem));
494dd64fc84SJeremy L Thompson   CeedCallBackend(CeedQFunctionGetFields(qf, NULL, &qf_input_fields, NULL, &qf_output_fields));
495dd64fc84SJeremy L Thompson   CeedCallBackend(CeedOperatorGetFields(op, &num_input_fields, &op_input_fields, &num_output_fields, &op_output_fields));
496dd64fc84SJeremy L Thompson   active_in     = impl->qf_active_in;
497dd64fc84SJeremy L Thompson   num_active_in = impl->num_active_in, num_active_out = impl->num_active_out;
498bd882c8aSJames Wright 
499bd882c8aSJames Wright   // Setup
500bd882c8aSJames Wright   CeedCallBackend(CeedOperatorSetup_Sycl(op));
501bd882c8aSJames Wright 
502bd882c8aSJames Wright   // Input Evecs and Restriction
503dd64fc84SJeremy L Thompson   CeedCallBackend(CeedOperatorSetupInputs_Sycl(num_input_fields, qf_input_fields, op_input_fields, NULL, true, e_data, impl, request));
504bd882c8aSJames Wright 
505bd882c8aSJames Wright   // Count number of active input fields
506dd64fc84SJeremy L Thompson   if (!num_active_in) {
507dd64fc84SJeremy L Thompson     for (CeedInt i = 0; i < num_input_fields; i++) {
508dd64fc84SJeremy L Thompson       CeedScalar *q_vec_array;
509dd64fc84SJeremy L Thompson       CeedVector  vec;
510dd64fc84SJeremy L Thompson 
511bd882c8aSJames Wright       // Check if active input
512*681d0ea7SJeremy L Thompson       CeedCallBackend(CeedOperatorFieldGetVector(op_input_fields[i], &vec));
513bd882c8aSJames Wright       if (vec == CEED_VECTOR_ACTIVE) {
514dd64fc84SJeremy L Thompson         CeedCallBackend(CeedQFunctionFieldGetSize(qf_input_fields[i], &size));
515dd64fc84SJeremy L Thompson         CeedCallBackend(CeedVectorSetValue(impl->q_vecs_in[i], 0.0));
516dd64fc84SJeremy L Thompson         CeedCallBackend(CeedVectorGetArray(impl->q_vecs_in[i], CEED_MEM_DEVICE, &q_vec_array));
517dd64fc84SJeremy L Thompson         CeedCallBackend(CeedRealloc(num_active_in + size, &active_in));
518bd882c8aSJames Wright         for (CeedInt field = 0; field < size; field++) {
519dd64fc84SJeremy L Thompson           q_size = (CeedSize)Q * num_elem;
520dd64fc84SJeremy L Thompson           CeedCallBackend(CeedVectorCreate(ceed, q_size, &active_in[num_active_in + field]));
521dd64fc84SJeremy L Thompson           CeedCallBackend(
522dd64fc84SJeremy L Thompson               CeedVectorSetArray(active_in[num_active_in + field], CEED_MEM_DEVICE, CEED_USE_POINTER, &q_vec_array[field * Q * num_elem]));
523bd882c8aSJames Wright         }
524dd64fc84SJeremy L Thompson         num_active_in += size;
525dd64fc84SJeremy L Thompson         CeedCallBackend(CeedVectorRestoreArray(impl->q_vecs_in[i], &q_vec_array));
526bd882c8aSJames Wright       }
527*681d0ea7SJeremy L Thompson       CeedCallBackend(CeedVectorDestroy(&vec));
528bd882c8aSJames Wright     }
529dd64fc84SJeremy L Thompson     impl->num_active_in = num_active_in;
530dd64fc84SJeremy L Thompson     impl->qf_active_in  = active_in;
531bd882c8aSJames Wright   }
532bd882c8aSJames Wright 
533bd882c8aSJames Wright   // Count number of active output fields
534dd64fc84SJeremy L Thompson   if (!num_active_out) {
535dd64fc84SJeremy L Thompson     for (CeedInt i = 0; i < num_output_fields; i++) {
536dd64fc84SJeremy L Thompson       CeedVector vec;
537dd64fc84SJeremy L Thompson 
538bd882c8aSJames Wright       // Check if active output
539*681d0ea7SJeremy L Thompson       CeedCallBackend(CeedOperatorFieldGetVector(op_output_fields[i], &vec));
540bd882c8aSJames Wright       if (vec == CEED_VECTOR_ACTIVE) {
541dd64fc84SJeremy L Thompson         CeedCallBackend(CeedQFunctionFieldGetSize(qf_output_fields[i], &size));
542dd64fc84SJeremy L Thompson         num_active_out += size;
543bd882c8aSJames Wright       }
544*681d0ea7SJeremy L Thompson       CeedCallBackend(CeedVectorDestroy(&vec));
545bd882c8aSJames Wright     }
546dd64fc84SJeremy L Thompson     impl->num_active_out = num_active_out;
547bd882c8aSJames Wright   }
548bd882c8aSJames Wright 
549bd882c8aSJames Wright   // Check sizes
5504e3038a5SJeremy L Thompson   CeedCheck(num_active_in > 0 && num_active_out > 0, ceed, CEED_ERROR_BACKEND, "Cannot assemble QFunction without active inputs and outputs");
551bd882c8aSJames Wright 
552bd882c8aSJames Wright   // Build objects if needed
553bd882c8aSJames Wright   if (build_objects) {
554dd64fc84SJeremy L Thompson     CeedSize l_size     = (CeedSize)num_elem * Q * num_active_in * num_active_out;
555dd64fc84SJeremy L Thompson     CeedInt  strides[3] = {1, num_elem * Q, Q}; /* *NOPAD* */
556dd64fc84SJeremy L Thompson 
557bd882c8aSJames Wright     // Create output restriction
55822ab0487SJames Wright     CeedCallBackend(CeedElemRestrictionCreateStrided(ceed_parent, num_elem, Q, num_active_in * num_active_out, l_size, strides, rstr));
559bd882c8aSJames Wright     // Create assembled vector
560dd64fc84SJeremy L Thompson     CeedCallBackend(CeedVectorCreate(ceed_parent, l_size, assembled));
561bd882c8aSJames Wright   }
562bd882c8aSJames Wright   CeedCallBackend(CeedVectorSetValue(*assembled, 0.0));
563dd64fc84SJeremy L Thompson   CeedCallBackend(CeedVectorGetArray(*assembled, CEED_MEM_DEVICE, &assembled_array));
564bd882c8aSJames Wright 
565bd882c8aSJames Wright   // Input basis apply
566dd64fc84SJeremy L Thompson   CeedCallBackend(CeedOperatorInputBasis_Sycl(num_elem, qf_input_fields, op_input_fields, num_input_fields, true, e_data, impl));
567bd882c8aSJames Wright 
568bd882c8aSJames Wright   // Assemble QFunction
569dd64fc84SJeremy L Thompson   for (CeedInt in = 0; in < num_active_in; in++) {
570bd882c8aSJames Wright     // Set Inputs
571dd64fc84SJeremy L Thompson     CeedCallBackend(CeedVectorSetValue(active_in[in], 1.0));
572dd64fc84SJeremy L Thompson     if (num_active_in > 1) {
573dd64fc84SJeremy L Thompson       CeedCallBackend(CeedVectorSetValue(active_in[(in + num_active_in - 1) % num_active_in], 0.0));
574bd882c8aSJames Wright     }
575bd882c8aSJames Wright     // Set Outputs
576dd64fc84SJeremy L Thompson     for (CeedInt out = 0; out < num_output_fields; out++) {
577dd64fc84SJeremy L Thompson       CeedVector vec;
578dd64fc84SJeremy L Thompson 
579bd882c8aSJames Wright       // Check if active output
580*681d0ea7SJeremy L Thompson       CeedCallBackend(CeedOperatorFieldGetVector(op_output_fields[out], &vec));
581bd882c8aSJames Wright       if (vec == CEED_VECTOR_ACTIVE) {
582dd64fc84SJeremy L Thompson         CeedCallBackend(CeedVectorSetArray(impl->q_vecs_out[out], CEED_MEM_DEVICE, CEED_USE_POINTER, assembled_array));
583dd64fc84SJeremy L Thompson         CeedCallBackend(CeedQFunctionFieldGetSize(qf_output_fields[out], &size));
584dd64fc84SJeremy L Thompson         assembled_array += size * Q * num_elem;  // Advance the pointer by the size of the output
585bd882c8aSJames Wright       }
586*681d0ea7SJeremy L Thompson       CeedCallBackend(CeedVectorDestroy(&vec));
587bd882c8aSJames Wright     }
588bd882c8aSJames Wright     // Apply QFunction
589dd64fc84SJeremy L Thompson     CeedCallBackend(CeedQFunctionApply(qf, Q * num_elem, impl->q_vecs_in, impl->q_vecs_out));
590bd882c8aSJames Wright   }
591bd882c8aSJames Wright 
592bd882c8aSJames Wright   // Un-set output Qvecs to prevent accidental overwrite of Assembled
593dd64fc84SJeremy L Thompson   for (CeedInt out = 0; out < num_output_fields; out++) {
594dd64fc84SJeremy L Thompson     CeedVector vec;
595dd64fc84SJeremy L Thompson 
596bd882c8aSJames Wright     // Check if active output
597*681d0ea7SJeremy L Thompson     CeedCallBackend(CeedOperatorFieldGetVector(op_output_fields[out], &vec));
598bd882c8aSJames Wright     if (vec == CEED_VECTOR_ACTIVE) {
599dd64fc84SJeremy L Thompson       CeedCallBackend(CeedVectorTakeArray(impl->q_vecs_out[out], CEED_MEM_DEVICE, NULL));
600bd882c8aSJames Wright     }
601*681d0ea7SJeremy L Thompson     CeedCallBackend(CeedVectorDestroy(&vec));
602bd882c8aSJames Wright   }
603bd882c8aSJames Wright 
604bd882c8aSJames Wright   // Restore input arrays
605dd64fc84SJeremy L Thompson   CeedCallBackend(CeedOperatorRestoreInputs_Sycl(num_input_fields, qf_input_fields, op_input_fields, true, e_data, impl));
606bd882c8aSJames Wright 
607bd882c8aSJames Wright   // Restore output
608dd64fc84SJeremy L Thompson   CeedCallBackend(CeedVectorRestoreArray(*assembled, &assembled_array));
609bd882c8aSJames Wright   return CEED_ERROR_SUCCESS;
610bd882c8aSJames Wright }
611bd882c8aSJames Wright 
612bd882c8aSJames Wright //------------------------------------------------------------------------------
613bd882c8aSJames Wright // Assemble Linear QFunction
614bd882c8aSJames Wright //------------------------------------------------------------------------------
615bd882c8aSJames Wright static int CeedOperatorLinearAssembleQFunction_Sycl(CeedOperator op, CeedVector *assembled, CeedElemRestriction *rstr, CeedRequest *request) {
616bd882c8aSJames Wright   return CeedOperatorLinearAssembleQFunctionCore_Sycl(op, true, assembled, rstr, request);
617bd882c8aSJames Wright }
618bd882c8aSJames Wright 
619bd882c8aSJames Wright //------------------------------------------------------------------------------
620bd882c8aSJames Wright // Update Assembled Linear QFunction
621bd882c8aSJames Wright //------------------------------------------------------------------------------
622bd882c8aSJames Wright static int CeedOperatorLinearAssembleQFunctionUpdate_Sycl(CeedOperator op, CeedVector assembled, CeedElemRestriction rstr, CeedRequest *request) {
623bd882c8aSJames Wright   return CeedOperatorLinearAssembleQFunctionCore_Sycl(op, false, &assembled, &rstr, request);
624bd882c8aSJames Wright }
625bd882c8aSJames Wright 
626bd882c8aSJames Wright //------------------------------------------------------------------------------
627bd882c8aSJames Wright // Assemble diagonal setup
628bd882c8aSJames Wright //------------------------------------------------------------------------------
629506b1a0cSSebastian Grimberg static inline int CeedOperatorAssembleDiagonalSetup_Sycl(CeedOperator op) {
630bd882c8aSJames Wright   Ceed                ceed;
631dd64fc84SJeremy L Thompson   Ceed_Sycl          *sycl_data;
632*681d0ea7SJeremy L Thompson   CeedInt             num_input_fields, num_output_fields, num_eval_mode_in = 0, num_comp = 0, dim = 1, num_eval_mode_out = 0;
633*681d0ea7SJeremy L Thompson   CeedEvalMode       *eval_mode_in = NULL, *eval_mode_out = NULL;
634dd64fc84SJeremy L Thompson   CeedBasis           basis_in = NULL, basis_out = NULL;
635dd64fc84SJeremy L Thompson   CeedQFunctionField *qf_fields;
636bd882c8aSJames Wright   CeedQFunction       qf;
637dd64fc84SJeremy L Thompson   CeedOperatorField  *op_fields;
638dd64fc84SJeremy L Thompson   CeedOperator_Sycl  *impl;
639dd64fc84SJeremy L Thompson 
640dd64fc84SJeremy L Thompson   CeedCallBackend(CeedOperatorGetCeed(op, &ceed));
641bd882c8aSJames Wright   CeedCallBackend(CeedOperatorGetQFunction(op, &qf));
642dd64fc84SJeremy L Thompson   CeedCallBackend(CeedQFunctionGetNumArgs(qf, &num_input_fields, &num_output_fields));
643bd882c8aSJames Wright 
644bd882c8aSJames Wright   // Determine active input basis
645dd64fc84SJeremy L Thompson   CeedCallBackend(CeedOperatorGetFields(op, NULL, &op_fields, NULL, NULL));
646dd64fc84SJeremy L Thompson   CeedCallBackend(CeedQFunctionGetFields(qf, NULL, &qf_fields, NULL, NULL));
647dd64fc84SJeremy L Thompson   for (CeedInt i = 0; i < num_input_fields; i++) {
648bd882c8aSJames Wright     CeedVector vec;
649dd64fc84SJeremy L Thompson 
650dd64fc84SJeremy L Thompson     CeedCallBackend(CeedOperatorFieldGetVector(op_fields[i], &vec));
651bd882c8aSJames Wright     if (vec == CEED_VECTOR_ACTIVE) {
652*681d0ea7SJeremy L Thompson       CeedEvalMode eval_mode;
653*681d0ea7SJeremy L Thompson       CeedBasis    basis;
654dd64fc84SJeremy L Thompson 
655*681d0ea7SJeremy L Thompson       CeedCallBackend(CeedOperatorFieldGetBasis(op_fields[i], &basis));
656*681d0ea7SJeremy L Thompson       CeedCheck(!basis_in || basis_in == basis, ceed, CEED_ERROR_BACKEND,
657*681d0ea7SJeremy L Thompson                 "Backend does not implement operator diagonal assembly with multiple active bases");
658*681d0ea7SJeremy L Thompson       if (!basis_in) CeedCallBackend(CeedBasisReferenceCopy(basis, &basis_in));
659*681d0ea7SJeremy L Thompson       CeedCallBackend(CeedBasisDestroy(&basis));
660*681d0ea7SJeremy L Thompson       CeedCallBackend(CeedQFunctionFieldGetEvalMode(qf_fields[i], &eval_mode));
661*681d0ea7SJeremy L Thompson       switch (eval_mode) {
662bd882c8aSJames Wright         case CEED_EVAL_NONE:
663bd882c8aSJames Wright         case CEED_EVAL_INTERP:
664*681d0ea7SJeremy L Thompson           CeedCallBackend(CeedRealloc(num_eval_mode_in + 1, &eval_mode_in));
665*681d0ea7SJeremy L Thompson           eval_mode_in[num_eval_mode_in] = eval_mode;
666*681d0ea7SJeremy L Thompson           num_eval_mode_in += 1;
667bd882c8aSJames Wright           break;
668bd882c8aSJames Wright         case CEED_EVAL_GRAD:
669*681d0ea7SJeremy L Thompson           CeedCallBackend(CeedRealloc(num_eval_mode_in + dim, &eval_mode_in));
670*681d0ea7SJeremy L Thompson           for (CeedInt d = 0; d < dim; d++) eval_mode_in[num_eval_mode_in + d] = eval_mode;
671*681d0ea7SJeremy L Thompson           num_eval_mode_in += dim;
672bd882c8aSJames Wright           break;
673bd882c8aSJames Wright         case CEED_EVAL_WEIGHT:
674bd882c8aSJames Wright         case CEED_EVAL_DIV:
675bd882c8aSJames Wright         case CEED_EVAL_CURL:
676bd882c8aSJames Wright           break;  // Caught by QF Assembly
677bd882c8aSJames Wright       }
678bd882c8aSJames Wright     }
679*681d0ea7SJeremy L Thompson     CeedCallBackend(CeedVectorDestroy(&vec));
680bd882c8aSJames Wright   }
681bd882c8aSJames Wright 
682bd882c8aSJames Wright   // Determine active output basis
683dd64fc84SJeremy L Thompson   CeedCallBackend(CeedOperatorGetFields(op, NULL, NULL, NULL, &op_fields));
684dd64fc84SJeremy L Thompson   CeedCallBackend(CeedQFunctionGetFields(qf, NULL, NULL, NULL, &qf_fields));
685dd64fc84SJeremy L Thompson   for (CeedInt i = 0; i < num_output_fields; i++) {
686bd882c8aSJames Wright     CeedVector vec;
687dd64fc84SJeremy L Thompson 
688dd64fc84SJeremy L Thompson     CeedCallBackend(CeedOperatorFieldGetVector(op_fields[i], &vec));
689bd882c8aSJames Wright     if (vec == CEED_VECTOR_ACTIVE) {
690*681d0ea7SJeremy L Thompson       CeedEvalMode eval_mode;
691*681d0ea7SJeremy L Thompson       CeedBasis    basis;
692dd64fc84SJeremy L Thompson 
693*681d0ea7SJeremy L Thompson       CeedCallBackend(CeedOperatorFieldGetBasis(op_fields[i], &basis));
694*681d0ea7SJeremy L Thompson       CeedCheck(!basis_out || basis_out == basis, ceed, CEED_ERROR_BACKEND,
695*681d0ea7SJeremy L Thompson                 "Backend does not implement operator diagonal assembly with multiple active bases");
696*681d0ea7SJeremy L Thompson       if (!basis_out) CeedCallBackend(CeedBasisReferenceCopy(basis, &basis_out));
697*681d0ea7SJeremy L Thompson       CeedCallBackend(CeedBasisDestroy(&basis));
698*681d0ea7SJeremy L Thompson       CeedCallBackend(CeedQFunctionFieldGetEvalMode(qf_fields[i], &eval_mode));
699*681d0ea7SJeremy L Thompson       switch (eval_mode) {
700bd882c8aSJames Wright         case CEED_EVAL_NONE:
701bd882c8aSJames Wright         case CEED_EVAL_INTERP:
702*681d0ea7SJeremy L Thompson           CeedCallBackend(CeedRealloc(num_eval_mode_in + 1, &eval_mode_in));
703*681d0ea7SJeremy L Thompson           eval_mode_in[num_eval_mode_in] = eval_mode;
704*681d0ea7SJeremy L Thompson           num_eval_mode_in += 1;
705bd882c8aSJames Wright           break;
706bd882c8aSJames Wright         case CEED_EVAL_GRAD:
707*681d0ea7SJeremy L Thompson           CeedCallBackend(CeedRealloc(num_eval_mode_in + dim, &eval_mode_in));
708*681d0ea7SJeremy L Thompson           for (CeedInt d = 0; d < dim; d++) eval_mode_in[num_eval_mode_in + d] = eval_mode;
709*681d0ea7SJeremy L Thompson           num_eval_mode_in += dim;
710bd882c8aSJames Wright           break;
711bd882c8aSJames Wright         case CEED_EVAL_WEIGHT:
712bd882c8aSJames Wright         case CEED_EVAL_DIV:
713bd882c8aSJames Wright         case CEED_EVAL_CURL:
714bd882c8aSJames Wright           break;  // Caught by QF Assembly
715bd882c8aSJames Wright       }
716bd882c8aSJames Wright     }
717*681d0ea7SJeremy L Thompson     CeedCallBackend(CeedVectorDestroy(&vec));
718bd882c8aSJames Wright   }
719bd882c8aSJames Wright 
720bd882c8aSJames Wright   // Operator data struct
721bd882c8aSJames Wright   CeedCallBackend(CeedOperatorGetData(op, &impl));
722bd882c8aSJames Wright   CeedCallBackend(CeedGetData(ceed, &sycl_data));
723bd882c8aSJames Wright   CeedCallBackend(CeedCalloc(1, &impl->diag));
724bd882c8aSJames Wright   CeedOperatorDiag_Sycl *diag = impl->diag;
725dd64fc84SJeremy L Thompson 
726dd64fc84SJeremy L Thompson   diag->basis_in          = basis_in;
727dd64fc84SJeremy L Thompson   diag->basis_out         = basis_out;
728*681d0ea7SJeremy L Thompson   diag->h_eval_mode_in    = eval_mode_in;
729*681d0ea7SJeremy L Thompson   diag->h_eval_mode_out   = eval_mode_out;
730*681d0ea7SJeremy L Thompson   diag->num_eval_mode_in  = num_eval_mode_in;
731*681d0ea7SJeremy L Thompson   diag->num_eval_mode_out = num_eval_mode_out;
732bd882c8aSJames Wright 
733bd882c8aSJames Wright   // Kernel parameters
734dd64fc84SJeremy L Thompson   CeedInt num_nodes, num_qpts;
735dd64fc84SJeremy L Thompson   CeedCallBackend(CeedBasisGetNumNodes(basis_in, &num_nodes));
736dd64fc84SJeremy L Thompson   CeedCallBackend(CeedBasisGetNumQuadraturePoints(basis_in, &num_qpts));
737dd64fc84SJeremy L Thompson   diag->num_nodes = num_nodes;
738dd64fc84SJeremy L Thompson   diag->num_qpts  = num_qpts;
739dd64fc84SJeremy L Thompson   diag->num_comp  = num_comp;
740bd882c8aSJames Wright 
741bd882c8aSJames Wright   // Basis matrices
742dd64fc84SJeremy L Thompson   const CeedInt     i_len = num_qpts * num_nodes;
743dd64fc84SJeremy L Thompson   const CeedInt     g_len = num_qpts * num_nodes * dim;
744dd64fc84SJeremy L Thompson   const CeedScalar *interp_in, *interp_out, *grad_in, *grad_out;
745bd882c8aSJames Wright 
746bd882c8aSJames Wright   // CEED_EVAL_NONE
747bd882c8aSJames Wright   CeedScalar *identity      = NULL;
748dd64fc84SJeremy L Thompson   bool        has_eval_none = false;
749*681d0ea7SJeremy L Thompson   for (CeedInt i = 0; i < num_eval_mode_in; i++) has_eval_none = has_eval_none || (eval_mode_in[i] == CEED_EVAL_NONE);
750*681d0ea7SJeremy L Thompson   for (CeedInt i = 0; i < num_eval_mode_out; i++) has_eval_none = has_eval_none || (eval_mode_out[i] == CEED_EVAL_NONE);
751bd882c8aSJames Wright 
7521f4b1b45SUmesh Unnikrishnan   std::vector<sycl::event> e;
7531f4b1b45SUmesh Unnikrishnan 
7541f4b1b45SUmesh Unnikrishnan   if (!sycl_data->sycl_queue.is_in_order()) e = {sycl_data->sycl_queue.ext_oneapi_submit_barrier()};
755bd882c8aSJames Wright 
756bd882c8aSJames Wright   std::vector<sycl::event> copy_events;
7571f4b1b45SUmesh Unnikrishnan 
758dd64fc84SJeremy L Thompson   if (has_eval_none) {
759dd64fc84SJeremy L Thompson     CeedCallBackend(CeedCalloc(num_qpts * num_nodes, &identity));
760dd64fc84SJeremy L Thompson     for (CeedSize i = 0; i < (num_nodes < num_qpts ? num_nodes : num_qpts); i++) identity[i * num_nodes + i] = 1.0;
761dd64fc84SJeremy L Thompson     CeedCallSycl(ceed, diag->d_identity = sycl::malloc_device<CeedScalar>(i_len, sycl_data->sycl_device, sycl_data->sycl_context));
7621f4b1b45SUmesh Unnikrishnan     sycl::event identity_copy = sycl_data->sycl_queue.copy<CeedScalar>(identity, diag->d_identity, i_len, e);
763bd882c8aSJames Wright     copy_events.push_back(identity_copy);
764bd882c8aSJames Wright   }
765bd882c8aSJames Wright 
766bd882c8aSJames Wright   // CEED_EVAL_INTERP
767dd64fc84SJeremy L Thompson   CeedCallBackend(CeedBasisGetInterp(basis_in, &interp_in));
768dd64fc84SJeremy L Thompson   CeedCallSycl(ceed, diag->d_interp_in = sycl::malloc_device<CeedScalar>(i_len, sycl_data->sycl_device, sycl_data->sycl_context));
7691f4b1b45SUmesh Unnikrishnan   sycl::event interp_in_copy = sycl_data->sycl_queue.copy<CeedScalar>(interp_in, diag->d_interp_in, i_len, e);
770dd64fc84SJeremy L Thompson   copy_events.push_back(interp_in_copy);
771bd882c8aSJames Wright 
772dd64fc84SJeremy L Thompson   CeedCallBackend(CeedBasisGetInterp(basis_out, &interp_out));
773dd64fc84SJeremy L Thompson   CeedCallSycl(ceed, diag->d_interp_out = sycl::malloc_device<CeedScalar>(i_len, sycl_data->sycl_device, sycl_data->sycl_context));
7741f4b1b45SUmesh Unnikrishnan   sycl::event interp_out_copy = sycl_data->sycl_queue.copy<CeedScalar>(interp_out, diag->d_interp_out, i_len, e);
775dd64fc84SJeremy L Thompson   copy_events.push_back(interp_out_copy);
776bd882c8aSJames Wright 
777bd882c8aSJames Wright   // CEED_EVAL_GRAD
778dd64fc84SJeremy L Thompson   CeedCallBackend(CeedBasisGetGrad(basis_in, &grad_in));
779dd64fc84SJeremy L Thompson   CeedCallSycl(ceed, diag->d_grad_in = sycl::malloc_device<CeedScalar>(g_len, sycl_data->sycl_device, sycl_data->sycl_context));
7801f4b1b45SUmesh Unnikrishnan   sycl::event grad_in_copy = sycl_data->sycl_queue.copy<CeedScalar>(grad_in, diag->d_grad_in, g_len, e);
781dd64fc84SJeremy L Thompson   copy_events.push_back(grad_in_copy);
782bd882c8aSJames Wright 
783dd64fc84SJeremy L Thompson   CeedCallBackend(CeedBasisGetGrad(basis_out, &grad_out));
784dd64fc84SJeremy L Thompson   CeedCallSycl(ceed, diag->d_grad_out = sycl::malloc_device<CeedScalar>(g_len, sycl_data->sycl_device, sycl_data->sycl_context));
7851f4b1b45SUmesh Unnikrishnan   sycl::event grad_out_copy = sycl_data->sycl_queue.copy<CeedScalar>(grad_out, diag->d_grad_out, g_len, e);
786dd64fc84SJeremy L Thompson   copy_events.push_back(grad_out_copy);
787bd882c8aSJames Wright 
788*681d0ea7SJeremy L Thompson   // Arrays of  eval_modes
789*681d0ea7SJeremy L Thompson   CeedCallSycl(ceed, diag->d_eval_mode_in = sycl::malloc_device<CeedEvalMode>(num_eval_mode_in, sycl_data->sycl_device, sycl_data->sycl_context));
790*681d0ea7SJeremy L Thompson   sycl::event eval_mode_in_copy = sycl_data->sycl_queue.copy<CeedEvalMode>(eval_mode_in, diag->d_eval_mode_in, num_eval_mode_in, e);
791*681d0ea7SJeremy L Thompson   copy_events.push_back(eval_mode_in_copy);
792bd882c8aSJames Wright 
793*681d0ea7SJeremy L Thompson   CeedCallSycl(ceed, diag->d_eval_mode_out = sycl::malloc_device<CeedEvalMode>(num_eval_mode_out, sycl_data->sycl_device, sycl_data->sycl_context));
794*681d0ea7SJeremy L Thompson   sycl::event eval_mode_out_copy = sycl_data->sycl_queue.copy<CeedEvalMode>(eval_mode_out, diag->d_eval_mode_out, num_eval_mode_out, e);
795*681d0ea7SJeremy L Thompson   copy_events.push_back(eval_mode_out_copy);
796bd882c8aSJames Wright 
797bd882c8aSJames Wright   // Restriction
798*681d0ea7SJeremy L Thompson   {
799*681d0ea7SJeremy L Thompson     CeedElemRestriction rstr_out;
800*681d0ea7SJeremy L Thompson 
801*681d0ea7SJeremy L Thompson     CeedCallBackend(CeedOperatorGetActiveElemRestrictions(op, NULL, &rstr_out));
802dd64fc84SJeremy L Thompson     diag->diag_rstr = rstr_out;
803*681d0ea7SJeremy L Thompson     CeedCallBackend(CeedElemRestrictionDestroy(&rstr_out));
804*681d0ea7SJeremy L Thompson   }
805bd882c8aSJames Wright 
806bd882c8aSJames Wright   // Wait for all copies to complete and handle exceptions
807bd882c8aSJames Wright   CeedCallSycl(ceed, sycl::event::wait_and_throw(copy_events));
808bd882c8aSJames Wright   return CEED_ERROR_SUCCESS;
809bd882c8aSJames Wright }
810bd882c8aSJames Wright 
811bd882c8aSJames Wright //------------------------------------------------------------------------------
812bd882c8aSJames Wright //  Kernel for diagonal assembly
813bd882c8aSJames Wright //------------------------------------------------------------------------------
814dd64fc84SJeremy L Thompson static int CeedOperatorLinearDiagonal_Sycl(sycl::queue &sycl_queue, const bool is_point_block, const CeedInt num_elem,
815dd64fc84SJeremy L Thompson                                            const CeedOperatorDiag_Sycl *diag, const CeedScalar *assembled_qf_array, CeedScalar *elem_diag_array) {
816dd64fc84SJeremy L Thompson   const CeedSize      num_nodes         = diag->num_nodes;
817dd64fc84SJeremy L Thompson   const CeedSize      num_qpts          = diag->num_qpts;
818dd64fc84SJeremy L Thompson   const CeedSize      num_comp          = diag->num_comp;
819*681d0ea7SJeremy L Thompson   const CeedSize      num_eval_mode_in  = diag->num_eval_mode_in;
820*681d0ea7SJeremy L Thompson   const CeedSize      num_eval_mode_out = diag->num_eval_mode_out;
821bd882c8aSJames Wright   const CeedScalar   *identity          = diag->d_identity;
822dd64fc84SJeremy L Thompson   const CeedScalar   *interp_in         = diag->d_interp_in;
823dd64fc84SJeremy L Thompson   const CeedScalar   *grad_in           = diag->d_grad_in;
824dd64fc84SJeremy L Thompson   const CeedScalar   *interp_out        = diag->d_interp_out;
825dd64fc84SJeremy L Thompson   const CeedScalar   *grad_out          = diag->d_grad_out;
826*681d0ea7SJeremy L Thompson   const CeedEvalMode *eval_mode_in      = diag->d_eval_mode_in;
827*681d0ea7SJeremy L Thompson   const CeedEvalMode *eval_mode_out     = diag->d_eval_mode_out;
828bd882c8aSJames Wright 
829dd64fc84SJeremy L Thompson   sycl::range<1> kernel_range(num_elem * num_nodes);
830bd882c8aSJames Wright 
8311f4b1b45SUmesh Unnikrishnan   std::vector<sycl::event> e;
8321f4b1b45SUmesh Unnikrishnan 
8331f4b1b45SUmesh Unnikrishnan   if (!sycl_queue.is_in_order()) e = {sycl_queue.ext_oneapi_submit_barrier()};
8341f4b1b45SUmesh Unnikrishnan 
8351f4b1b45SUmesh Unnikrishnan   sycl_queue.parallel_for<CeedOperatorSyclLinearDiagonal>(kernel_range, e, [=](sycl::id<1> idx) {
836dd64fc84SJeremy L Thompson     const CeedInt tid = idx % num_nodes;
837dd64fc84SJeremy L Thompson     const CeedInt e   = idx / num_nodes;
838bd882c8aSJames Wright 
839bd882c8aSJames Wright     // Compute the diagonal of B^T D B
840bd882c8aSJames Wright     // Each element
841dd64fc84SJeremy L Thompson     CeedInt d_out = -1;
842bd882c8aSJames Wright     // Each basis eval mode pair
843*681d0ea7SJeremy L Thompson     for (CeedSize e_out = 0; e_out < num_eval_mode_out; e_out++) {
844bd882c8aSJames Wright       const CeedScalar *bt = NULL;
845dd64fc84SJeremy L Thompson 
846*681d0ea7SJeremy L Thompson       if (eval_mode_out[e_out] == CEED_EVAL_GRAD) ++d_out;
847*681d0ea7SJeremy L Thompson       CeedOperatorGetBasisPointer_Sycl(&bt, eval_mode_out[e_out], identity, interp_out, &grad_out[d_out * num_qpts * num_nodes]);
848dd64fc84SJeremy L Thompson       CeedInt d_in = -1;
849dd64fc84SJeremy L Thompson 
850*681d0ea7SJeremy L Thompson       for (CeedSize e_in = 0; e_in < num_eval_mode_in; e_in++) {
851bd882c8aSJames Wright         const CeedScalar *b = NULL;
852dd64fc84SJeremy L Thompson 
853*681d0ea7SJeremy L Thompson         if (eval_mode_in[e_in] == CEED_EVAL_GRAD) ++d_in;
854*681d0ea7SJeremy L Thompson         CeedOperatorGetBasisPointer_Sycl(&b, eval_mode_in[e_in], identity, interp_in, &grad_in[d_in * num_qpts * num_nodes]);
855bd882c8aSJames Wright         // Each component
856dd64fc84SJeremy L Thompson         for (CeedSize comp_out = 0; comp_out < num_comp; comp_out++) {
857bd882c8aSJames Wright           // Each qpoint/node pair
858dd64fc84SJeremy L Thompson           if (is_point_block) {
859bd882c8aSJames Wright             // Point Block Diagonal
860dd64fc84SJeremy L Thompson             for (CeedInt comp_in = 0; comp_in < num_comp; comp_in++) {
861dd64fc84SJeremy L Thompson               CeedScalar e_value = 0.0;
862dd64fc84SJeremy L Thompson 
863dd64fc84SJeremy L Thompson               for (CeedSize q = 0; q < num_qpts; q++) {
864dd64fc84SJeremy L Thompson                 const CeedScalar qf_value =
865*681d0ea7SJeremy L Thompson                     assembled_qf_array[((((e_in * num_comp + comp_in) * num_eval_mode_out + e_out) * num_comp + comp_out) * num_elem + e) * num_qpts +
866dd64fc84SJeremy L Thompson                                        q];
867dd64fc84SJeremy L Thompson 
868dd64fc84SJeremy L Thompson                 e_value += bt[q * num_nodes + tid] * qf_value * b[q * num_nodes + tid];
869bd882c8aSJames Wright               }
870dd64fc84SJeremy L Thompson               elem_diag_array[((comp_out * num_comp + comp_in) * num_elem + e) * num_nodes + tid] += e_value;
871bd882c8aSJames Wright             }
872bd882c8aSJames Wright           } else {
873bd882c8aSJames Wright             // Diagonal Only
874dd64fc84SJeremy L Thompson             CeedScalar e_value = 0.0;
875dd64fc84SJeremy L Thompson 
876dd64fc84SJeremy L Thompson             for (CeedSize q = 0; q < num_qpts; q++) {
877dd64fc84SJeremy L Thompson               const CeedScalar qf_value =
878*681d0ea7SJeremy L Thompson                   assembled_qf_array[((((e_in * num_comp + comp_out) * num_eval_mode_out + e_out) * num_comp + comp_out) * num_elem + e) * num_qpts +
879*681d0ea7SJeremy L Thompson                                      q];
880dd64fc84SJeremy L Thompson               e_value += bt[q * num_nodes + tid] * qf_value * b[q * num_nodes + tid];
881bd882c8aSJames Wright             }
882dd64fc84SJeremy L Thompson             elem_diag_array[(comp_out * num_elem + e) * num_nodes + tid] += e_value;
883bd882c8aSJames Wright           }
884bd882c8aSJames Wright         }
885bd882c8aSJames Wright       }
886bd882c8aSJames Wright     }
887bd882c8aSJames Wright   });
888bd882c8aSJames Wright   return CEED_ERROR_SUCCESS;
889bd882c8aSJames Wright }
890bd882c8aSJames Wright 
891bd882c8aSJames Wright //------------------------------------------------------------------------------
892bd882c8aSJames Wright // Assemble diagonal common code
893bd882c8aSJames Wright //------------------------------------------------------------------------------
894dd64fc84SJeremy L Thompson static inline int CeedOperatorAssembleDiagonalCore_Sycl(CeedOperator op, CeedVector assembled, CeedRequest *request, const bool is_point_block) {
895bd882c8aSJames Wright   Ceed                ceed;
896bd882c8aSJames Wright   Ceed_Sycl          *sycl_data;
897dd64fc84SJeremy L Thompson   CeedInt             num_elem;
898dd64fc84SJeremy L Thompson   CeedScalar         *elem_diag_array;
899dd64fc84SJeremy L Thompson   const CeedScalar   *assembled_qf_array;
900dd64fc84SJeremy L Thompson   CeedVector          assembled_qf = NULL;
901dd64fc84SJeremy L Thompson   CeedElemRestriction rstr         = NULL;
902dd64fc84SJeremy L Thompson   CeedOperator_Sycl  *impl;
903dd64fc84SJeremy L Thompson 
904dd64fc84SJeremy L Thompson   CeedCallBackend(CeedOperatorGetCeed(op, &ceed));
905dd64fc84SJeremy L Thompson   CeedCallBackend(CeedOperatorGetData(op, &impl));
906bd882c8aSJames Wright   CeedCallBackend(CeedGetData(ceed, &sycl_data));
907bd882c8aSJames Wright 
908bd882c8aSJames Wright   // Assemble QFunction
909dd64fc84SJeremy L Thompson   CeedCallBackend(CeedOperatorLinearAssembleQFunctionBuildOrUpdate(op, &assembled_qf, &rstr, request));
910bd882c8aSJames Wright   CeedCallBackend(CeedElemRestrictionDestroy(&rstr));
911bd882c8aSJames Wright 
912bd882c8aSJames Wright   // Setup
913bd882c8aSJames Wright   if (!impl->diag) {
914506b1a0cSSebastian Grimberg     CeedCallBackend(CeedOperatorAssembleDiagonalSetup_Sycl(op));
915bd882c8aSJames Wright   }
916bd882c8aSJames Wright   CeedOperatorDiag_Sycl *diag = impl->diag;
917dd64fc84SJeremy L Thompson 
918bd882c8aSJames Wright   assert(diag != NULL);
919bd882c8aSJames Wright 
920bd882c8aSJames Wright   // Restriction
921dd64fc84SJeremy L Thompson   if (is_point_block && !diag->point_block_diag_rstr) {
922506b1a0cSSebastian Grimberg     CeedCallBackend(CeedOperatorCreateActivePointBlockRestriction(diag->diag_rstr, &diag->point_block_diag_rstr));
923bd882c8aSJames Wright   }
924dd64fc84SJeremy L Thompson   CeedElemRestriction diag_rstr = is_point_block ? diag->point_block_diag_rstr : diag->diag_rstr;
925bd882c8aSJames Wright 
926bd882c8aSJames Wright   // Create diagonal vector
927dd64fc84SJeremy L Thompson   CeedVector elem_diag = is_point_block ? diag->point_block_elem_diag : diag->elem_diag;
928dd64fc84SJeremy L Thompson 
929dd64fc84SJeremy L Thompson   if (!elem_diag) {
930dd64fc84SJeremy L Thompson     CeedCallBackend(CeedElemRestrictionCreateVector(diag_rstr, NULL, &elem_diag));
931dd64fc84SJeremy L Thompson     if (is_point_block) diag->point_block_elem_diag = elem_diag;
932dd64fc84SJeremy L Thompson     else diag->elem_diag = elem_diag;
933bd882c8aSJames Wright   }
934dd64fc84SJeremy L Thompson   CeedCallBackend(CeedVectorSetValue(elem_diag, 0.0));
935bd882c8aSJames Wright 
936bd882c8aSJames Wright   // Assemble element operator diagonals
937dd64fc84SJeremy L Thompson   CeedCallBackend(CeedVectorGetArray(elem_diag, CEED_MEM_DEVICE, &elem_diag_array));
938dd64fc84SJeremy L Thompson   CeedCallBackend(CeedVectorGetArrayRead(assembled_qf, CEED_MEM_DEVICE, &assembled_qf_array));
939dd64fc84SJeremy L Thompson   CeedCallBackend(CeedElemRestrictionGetNumElements(diag_rstr, &num_elem));
940bd882c8aSJames Wright 
941bd882c8aSJames Wright   // Compute the diagonal of B^T D B
942dd64fc84SJeremy L Thompson   CeedCallBackend(CeedOperatorLinearDiagonal_Sycl(sycl_data->sycl_queue, is_point_block, num_elem, diag, assembled_qf_array, elem_diag_array));
943bd882c8aSJames Wright 
944bd882c8aSJames Wright   // Wait for queue to complete and handle exceptions
945bd882c8aSJames Wright   sycl_data->sycl_queue.wait_and_throw();
946bd882c8aSJames Wright 
947bd882c8aSJames Wright   // Restore arrays
948dd64fc84SJeremy L Thompson   CeedCallBackend(CeedVectorRestoreArray(elem_diag, &elem_diag_array));
949dd64fc84SJeremy L Thompson   CeedCallBackend(CeedVectorRestoreArrayRead(assembled_qf, &assembled_qf_array));
950bd882c8aSJames Wright 
951bd882c8aSJames Wright   // Assemble local operator diagonal
952dd64fc84SJeremy L Thompson   CeedCallBackend(CeedElemRestrictionApply(diag_rstr, CEED_TRANSPOSE, elem_diag, assembled, request));
953bd882c8aSJames Wright 
954bd882c8aSJames Wright   // Cleanup
955dd64fc84SJeremy L Thompson   CeedCallBackend(CeedVectorDestroy(&assembled_qf));
956bd882c8aSJames Wright   return CEED_ERROR_SUCCESS;
957bd882c8aSJames Wright }
958bd882c8aSJames Wright 
959bd882c8aSJames Wright //------------------------------------------------------------------------------
960bd882c8aSJames Wright // Assemble Linear Diagonal
961bd882c8aSJames Wright //------------------------------------------------------------------------------
962bd882c8aSJames Wright static int CeedOperatorLinearAssembleAddDiagonal_Sycl(CeedOperator op, CeedVector assembled, CeedRequest *request) {
963bd882c8aSJames Wright   CeedCallBackend(CeedOperatorAssembleDiagonalCore_Sycl(op, assembled, request, false));
964bd882c8aSJames Wright   return CEED_ERROR_SUCCESS;
965bd882c8aSJames Wright }
966bd882c8aSJames Wright 
967bd882c8aSJames Wright //------------------------------------------------------------------------------
968bd882c8aSJames Wright // Assemble Linear Point Block Diagonal
969bd882c8aSJames Wright //------------------------------------------------------------------------------
970bd882c8aSJames Wright static int CeedOperatorLinearAssembleAddPointBlockDiagonal_Sycl(CeedOperator op, CeedVector assembled, CeedRequest *request) {
971bd882c8aSJames Wright   CeedCallBackend(CeedOperatorAssembleDiagonalCore_Sycl(op, assembled, request, true));
972bd882c8aSJames Wright   return CEED_ERROR_SUCCESS;
973bd882c8aSJames Wright }
974bd882c8aSJames Wright 
975bd882c8aSJames Wright //------------------------------------------------------------------------------
976bd882c8aSJames Wright // Single operator assembly setup
977bd882c8aSJames Wright //------------------------------------------------------------------------------
978bd882c8aSJames Wright static int CeedSingleOperatorAssembleSetup_Sycl(CeedOperator op) {
979bd882c8aSJames Wright   Ceed    ceed;
980*681d0ea7SJeremy L Thompson   CeedInt num_input_fields, num_output_fields, num_eval_mode_in = 0, dim = 1, num_B_in_mats_to_load = 0, size_B_in = 0, num_eval_mode_out = 0,
981dd64fc84SJeremy L Thompson                                                num_B_out_mats_to_load = 0, size_B_out = 0, num_qpts = 0, elem_size = 0, num_elem, num_comp,
982dd64fc84SJeremy L Thompson                                                mat_start = 0;
983dd64fc84SJeremy L Thompson   CeedEvalMode       *eval_mode_in = NULL, *eval_mode_out = NULL;
984dd64fc84SJeremy L Thompson   const CeedScalar   *interp_in, *grad_in;
985dd64fc84SJeremy L Thompson   CeedElemRestriction rstr_in = NULL, rstr_out = NULL;
986dd64fc84SJeremy L Thompson   CeedBasis           basis_in = NULL, basis_out = NULL;
987dd64fc84SJeremy L Thompson   CeedQFunctionField *qf_fields;
988dd64fc84SJeremy L Thompson   CeedQFunction       qf;
989dd64fc84SJeremy L Thompson   CeedOperatorField  *input_fields, *output_fields;
990bd882c8aSJames Wright   CeedOperator_Sycl  *impl;
991dd64fc84SJeremy L Thompson 
992dd64fc84SJeremy L Thompson   CeedCallBackend(CeedOperatorGetCeed(op, &ceed));
993bd882c8aSJames Wright   CeedCallBackend(CeedOperatorGetData(op, &impl));
994bd882c8aSJames Wright 
995bd882c8aSJames Wright   // Get input and output fields
996bd882c8aSJames Wright   CeedCallBackend(CeedOperatorGetFields(op, &num_input_fields, &input_fields, &num_output_fields, &output_fields));
997bd882c8aSJames Wright 
998bd882c8aSJames Wright   // Determine active input basis eval mode
999bd882c8aSJames Wright   CeedCallBackend(CeedOperatorGetQFunction(op, &qf));
1000bd882c8aSJames Wright   CeedCallBackend(CeedQFunctionGetFields(qf, NULL, &qf_fields, NULL, NULL));
1001bd882c8aSJames Wright   // Note that the kernel will treat each dimension of a gradient action separately;
1002*681d0ea7SJeremy L Thompson   // i.e., when an active input has a CEED_EVAL_GRAD mode, num_ eval_mode_in will increment by dim.
1003*681d0ea7SJeremy L Thompson   // However, for the purposes of loading the B matrices, it will be treated as one mode, and we will load/copy the entire gradient matrix at once,
1004dd64fc84SJeremy L Thompson   // so num_B_in_mats_to_load will be incremented by 1.
1005bd882c8aSJames Wright   for (CeedInt i = 0; i < num_input_fields; i++) {
1006bd882c8aSJames Wright     CeedVector vec;
1007dd64fc84SJeremy L Thompson 
1008bd882c8aSJames Wright     CeedCallBackend(CeedOperatorFieldGetVector(input_fields[i], &vec));
1009bd882c8aSJames Wright     if (vec == CEED_VECTOR_ACTIVE) {
1010*681d0ea7SJeremy L Thompson       CeedEvalMode        eval_mode;
1011*681d0ea7SJeremy L Thompson       CeedElemRestriction elem_rstr;
1012*681d0ea7SJeremy L Thompson       CeedBasis           basis;
1013*681d0ea7SJeremy L Thompson 
1014*681d0ea7SJeremy L Thompson       CeedCallBackend(CeedOperatorFieldGetBasis(input_fields[i], &basis));
1015*681d0ea7SJeremy L Thompson       CeedCheck(!basis_in || basis_in == basis, ceed, CEED_ERROR_BACKEND, "Backend does not implement operator assembly with multiple active bases");
1016*681d0ea7SJeremy L Thompson       if (!basis_in) CeedCallBackend(CeedBasisReferenceCopy(basis, &basis_in));
1017*681d0ea7SJeremy L Thompson       CeedCallBackend(CeedBasisDestroy(&basis));
1018bd882c8aSJames Wright       CeedCallBackend(CeedBasisGetDimension(basis_in, &dim));
1019dd64fc84SJeremy L Thompson       CeedCallBackend(CeedBasisGetNumQuadraturePoints(basis_in, &num_qpts));
1020*681d0ea7SJeremy L Thompson       CeedCallBackend(CeedOperatorFieldGetElemRestriction(input_fields[i], &elem_rstr));
1021*681d0ea7SJeremy L Thompson       if (!rstr_in) CeedCallBackend(CeedElemRestrictionReferenceCopy(elem_rstr, &rstr_in));
1022*681d0ea7SJeremy L Thompson       CeedCallBackend(CeedElemRestrictionDestroy(&elem_rstr));
1023dd64fc84SJeremy L Thompson       CeedCallBackend(CeedElemRestrictionGetElementSize(rstr_in, &elem_size));
1024bd882c8aSJames Wright       CeedCallBackend(CeedQFunctionFieldGetEvalMode(qf_fields[i], &eval_mode));
1025bd882c8aSJames Wright       if (eval_mode != CEED_EVAL_NONE) {
1026bd882c8aSJames Wright         CeedCallBackend(CeedRealloc(num_B_in_mats_to_load + 1, &eval_mode_in));
1027bd882c8aSJames Wright         eval_mode_in[num_B_in_mats_to_load] = eval_mode;
1028bd882c8aSJames Wright         num_B_in_mats_to_load += 1;
1029bd882c8aSJames Wright         if (eval_mode == CEED_EVAL_GRAD) {
1030*681d0ea7SJeremy L Thompson           num_eval_mode_in += dim;
1031dd64fc84SJeremy L Thompson           size_B_in += dim * elem_size * num_qpts;
1032bd882c8aSJames Wright         } else {
1033*681d0ea7SJeremy L Thompson           num_eval_mode_in += 1;
1034dd64fc84SJeremy L Thompson           size_B_in += elem_size * num_qpts;
1035bd882c8aSJames Wright         }
1036bd882c8aSJames Wright       }
1037bd882c8aSJames Wright     }
1038*681d0ea7SJeremy L Thompson     CeedCallBackend(CeedVectorDestroy(&vec));
1039bd882c8aSJames Wright   }
1040bd882c8aSJames Wright 
1041bd882c8aSJames Wright   // Determine active output basis; basis_out and rstr_out only used if same as input, TODO
1042bd882c8aSJames Wright   CeedCallBackend(CeedQFunctionGetFields(qf, NULL, NULL, NULL, &qf_fields));
1043bd882c8aSJames Wright   for (CeedInt i = 0; i < num_output_fields; i++) {
1044bd882c8aSJames Wright     CeedVector vec;
1045dd64fc84SJeremy L Thompson 
1046bd882c8aSJames Wright     CeedCallBackend(CeedOperatorFieldGetVector(output_fields[i], &vec));
1047bd882c8aSJames Wright     if (vec == CEED_VECTOR_ACTIVE) {
1048*681d0ea7SJeremy L Thompson       CeedEvalMode        eval_mode;
1049*681d0ea7SJeremy L Thompson       CeedElemRestriction elem_rstr;
1050*681d0ea7SJeremy L Thompson       CeedBasis           basis;
1051*681d0ea7SJeremy L Thompson 
1052*681d0ea7SJeremy L Thompson       CeedCallBackend(CeedOperatorFieldGetBasis(output_fields[i], &basis));
1053*681d0ea7SJeremy L Thompson       CeedCheck(!basis_out || basis_out == basis, ceed, CEED_ERROR_BACKEND,
1054*681d0ea7SJeremy L Thompson                 "Backend does not implement operator assembly with multiple active bases");
1055*681d0ea7SJeremy L Thompson       if (!basis_out) CeedCallBackend(CeedBasisReferenceCopy(basis, &basis_out));
1056*681d0ea7SJeremy L Thompson       CeedCallBackend(CeedBasisDestroy(&basis));
1057*681d0ea7SJeremy L Thompson       CeedCallBackend(CeedOperatorFieldGetElemRestriction(output_fields[i], &elem_rstr));
1058*681d0ea7SJeremy L Thompson       if (!rstr_out) CeedCallBackend(CeedElemRestrictionReferenceCopy(elem_rstr, &rstr_out));
1059*681d0ea7SJeremy L Thompson       CeedCallBackend(CeedElemRestrictionDestroy(&elem_rstr));
1060bd882c8aSJames Wright       CeedCallBackend(CeedQFunctionFieldGetEvalMode(qf_fields[i], &eval_mode));
1061bd882c8aSJames Wright       if (eval_mode != CEED_EVAL_NONE) {
1062bd882c8aSJames Wright         CeedCallBackend(CeedRealloc(num_B_out_mats_to_load + 1, &eval_mode_out));
1063bd882c8aSJames Wright         eval_mode_out[num_B_out_mats_to_load] = eval_mode;
1064bd882c8aSJames Wright         num_B_out_mats_to_load += 1;
1065bd882c8aSJames Wright         if (eval_mode == CEED_EVAL_GRAD) {
1066*681d0ea7SJeremy L Thompson           num_eval_mode_out += dim;
1067dd64fc84SJeremy L Thompson           size_B_out += dim * elem_size * num_qpts;
1068bd882c8aSJames Wright         } else {
1069*681d0ea7SJeremy L Thompson           num_eval_mode_out += 1;
1070dd64fc84SJeremy L Thompson           size_B_out += elem_size * num_qpts;
1071bd882c8aSJames Wright         }
1072bd882c8aSJames Wright       }
1073bd882c8aSJames Wright     }
1074*681d0ea7SJeremy L Thompson     CeedCallBackend(CeedVectorDestroy(&vec));
1075bd882c8aSJames Wright   }
1076*681d0ea7SJeremy L Thompson   CeedCheck(num_eval_mode_in > 0 && num_eval_mode_out > 0, ceed, CEED_ERROR_UNSUPPORTED, "Cannot assemble operator without inputs/outputs");
1077bd882c8aSJames Wright 
1078dd64fc84SJeremy L Thompson   CeedCallBackend(CeedElemRestrictionGetNumElements(rstr_in, &num_elem));
1079dd64fc84SJeremy L Thompson   CeedCallBackend(CeedElemRestrictionGetNumComponents(rstr_in, &num_comp));
1080bd882c8aSJames Wright 
1081bd882c8aSJames Wright   CeedCallBackend(CeedCalloc(1, &impl->asmb));
1082bd882c8aSJames Wright   CeedOperatorAssemble_Sycl *asmb = impl->asmb;
1083dd64fc84SJeremy L Thompson   asmb->num_elem                  = num_elem;
1084bd882c8aSJames Wright 
1085bd882c8aSJames Wright   Ceed_Sycl *sycl_data;
1086bd882c8aSJames Wright   CeedCallBackend(CeedGetData(ceed, &sycl_data));
1087bd882c8aSJames Wright 
1088bd882c8aSJames Wright   // Kernel setup
1089004e4986SSebastian Grimberg   int elems_per_block     = 1;
1090004e4986SSebastian Grimberg   asmb->elems_per_block   = elems_per_block;
1091dd64fc84SJeremy L Thompson   asmb->block_size_x      = elem_size;
1092dd64fc84SJeremy L Thompson   asmb->block_size_y      = elem_size;
1093*681d0ea7SJeremy L Thompson   asmb->num_eval_mode_in  = num_eval_mode_in;
1094*681d0ea7SJeremy L Thompson   asmb->num_eval_mode_out = num_eval_mode_out;
1095dd64fc84SJeremy L Thompson   asmb->num_qpts          = num_qpts;
1096dd64fc84SJeremy L Thompson   asmb->num_nodes         = elem_size;
10975700671aSSebastian Grimberg   asmb->block_size        = elem_size * elem_size * elems_per_block;
1098dd64fc84SJeremy L Thompson   asmb->num_comp          = num_comp;
1099bd882c8aSJames Wright 
1100bd882c8aSJames Wright   // Build 'full' B matrices (not 1D arrays used for tensor-product matrices
1101bd882c8aSJames Wright   CeedCallBackend(CeedBasisGetInterp(basis_in, &interp_in));
1102bd882c8aSJames Wright   CeedCallBackend(CeedBasisGetGrad(basis_in, &grad_in));
1103bd882c8aSJames Wright 
1104bd882c8aSJames Wright   // Load into B_in, in order that they will be used in eval_mode
1105bd882c8aSJames Wright   CeedCallSycl(ceed, asmb->d_B_in = sycl::malloc_device<CeedScalar>(size_B_in, sycl_data->sycl_device, sycl_data->sycl_context));
1106bd882c8aSJames Wright   for (int i = 0; i < num_B_in_mats_to_load; i++) {
1107bd882c8aSJames Wright     CeedEvalMode eval_mode = eval_mode_in[i];
1108dd64fc84SJeremy L Thompson 
1109bd882c8aSJames Wright     if (eval_mode == CEED_EVAL_INTERP) {
11101f4b1b45SUmesh Unnikrishnan       std::vector<sycl::event> e;
11111f4b1b45SUmesh Unnikrishnan 
11121f4b1b45SUmesh Unnikrishnan       if (!sycl_data->sycl_queue.is_in_order()) e = {sycl_data->sycl_queue.ext_oneapi_submit_barrier()};
11131f4b1b45SUmesh Unnikrishnan       sycl_data->sycl_queue.copy<CeedScalar>(interp_in, &asmb->d_B_in[mat_start], elem_size * num_qpts, e);
1114dd64fc84SJeremy L Thompson       mat_start += elem_size * num_qpts;
1115bd882c8aSJames Wright     } else if (eval_mode == CEED_EVAL_GRAD) {
11161f4b1b45SUmesh Unnikrishnan       std::vector<sycl::event> e;
11171f4b1b45SUmesh Unnikrishnan 
11181f4b1b45SUmesh Unnikrishnan       if (!sycl_data->sycl_queue.is_in_order()) e = {sycl_data->sycl_queue.ext_oneapi_submit_barrier()};
11191f4b1b45SUmesh Unnikrishnan       sycl_data->sycl_queue.copy<CeedScalar>(grad_in, &asmb->d_B_in[mat_start], dim * elem_size * num_qpts, e);
1120dd64fc84SJeremy L Thompson       mat_start += dim * elem_size * num_qpts;
1121bd882c8aSJames Wright     }
1122bd882c8aSJames Wright   }
1123bd882c8aSJames Wright 
1124bd882c8aSJames Wright   const CeedScalar *interp_out, *grad_out;
1125bd882c8aSJames Wright   // Note that this function currently assumes 1 basis, so this should always be true
1126bd882c8aSJames Wright   // for now
1127bd882c8aSJames Wright   if (basis_out == basis_in) {
1128bd882c8aSJames Wright     interp_out = interp_in;
1129bd882c8aSJames Wright     grad_out   = grad_in;
1130bd882c8aSJames Wright   } else {
1131bd882c8aSJames Wright     CeedCallBackend(CeedBasisGetInterp(basis_out, &interp_out));
1132bd882c8aSJames Wright     CeedCallBackend(CeedBasisGetGrad(basis_out, &grad_out));
1133bd882c8aSJames Wright   }
1134bd882c8aSJames Wright 
1135bd882c8aSJames Wright   // Load into B_out, in order that they will be used in eval_mode
1136bd882c8aSJames Wright   mat_start = 0;
1137bd882c8aSJames Wright   CeedCallSycl(ceed, asmb->d_B_out = sycl::malloc_device<CeedScalar>(size_B_out, sycl_data->sycl_device, sycl_data->sycl_context));
1138bd882c8aSJames Wright   for (int i = 0; i < num_B_out_mats_to_load; i++) {
1139bd882c8aSJames Wright     CeedEvalMode eval_mode = eval_mode_out[i];
1140dd64fc84SJeremy L Thompson 
1141bd882c8aSJames Wright     if (eval_mode == CEED_EVAL_INTERP) {
11421f4b1b45SUmesh Unnikrishnan       std::vector<sycl::event> e;
11431f4b1b45SUmesh Unnikrishnan 
11441f4b1b45SUmesh Unnikrishnan       if (!sycl_data->sycl_queue.is_in_order()) e = {sycl_data->sycl_queue.ext_oneapi_submit_barrier()};
11451f4b1b45SUmesh Unnikrishnan       sycl_data->sycl_queue.copy<CeedScalar>(interp_out, &asmb->d_B_out[mat_start], elem_size * num_qpts, e);
1146dd64fc84SJeremy L Thompson       mat_start += elem_size * num_qpts;
1147bd882c8aSJames Wright     } else if (eval_mode == CEED_EVAL_GRAD) {
11481f4b1b45SUmesh Unnikrishnan       std::vector<sycl::event> e;
11491f4b1b45SUmesh Unnikrishnan 
11501f4b1b45SUmesh Unnikrishnan       if (!sycl_data->sycl_queue.is_in_order()) e = {sycl_data->sycl_queue.ext_oneapi_submit_barrier()};
11511f4b1b45SUmesh Unnikrishnan       sycl_data->sycl_queue.copy<CeedScalar>(grad_out, &asmb->d_B_out[mat_start], dim * elem_size * num_qpts, e);
1152dd64fc84SJeremy L Thompson       mat_start += dim * elem_size * num_qpts;
1153bd882c8aSJames Wright     }
1154bd882c8aSJames Wright   }
1155*681d0ea7SJeremy L Thompson   CeedCallBackend(CeedElemRestrictionDestroy(&rstr_in));
1156*681d0ea7SJeremy L Thompson   CeedCallBackend(CeedElemRestrictionDestroy(&rstr_out));
1157*681d0ea7SJeremy L Thompson   CeedCallBackend(CeedBasisDestroy(&basis_in));
1158*681d0ea7SJeremy L Thompson   CeedCallBackend(CeedBasisDestroy(&basis_out));
1159bd882c8aSJames Wright   return CEED_ERROR_SUCCESS;
1160bd882c8aSJames Wright }
1161bd882c8aSJames Wright 
1162bd882c8aSJames Wright //------------------------------------------------------------------------------
1163bd882c8aSJames Wright // Matrix assembly kernel for low-order elements (3D thread block)
1164bd882c8aSJames Wright //------------------------------------------------------------------------------
1165bd882c8aSJames Wright static int CeedOperatorLinearAssemble_Sycl(sycl::queue &sycl_queue, const CeedOperator_Sycl *impl, const CeedScalar *qf_array,
1166bd882c8aSJames Wright                                            CeedScalar *values_array) {
1167bd882c8aSJames Wright   // This kernels assumes B_in and B_out have the same number of quadrature points and basis points.
1168bd882c8aSJames Wright   // TODO: expand to more general cases
1169bd882c8aSJames Wright   CeedOperatorAssemble_Sycl *asmb              = impl->asmb;
1170dd64fc84SJeremy L Thompson   const CeedInt              num_elem          = asmb->num_elem;
1171dd64fc84SJeremy L Thompson   const CeedSize             num_nodes         = asmb->num_nodes;
1172dd64fc84SJeremy L Thompson   const CeedSize             num_comp          = asmb->num_comp;
1173dd64fc84SJeremy L Thompson   const CeedSize             num_qpts          = asmb->num_qpts;
1174*681d0ea7SJeremy L Thompson   const CeedSize             num_eval_mode_in  = asmb->num_eval_mode_in;
1175*681d0ea7SJeremy L Thompson   const CeedSize             num_eval_mode_out = asmb->num_eval_mode_out;
1176bd882c8aSJames Wright 
1177bd882c8aSJames Wright   // Strides for final output ordering, determined by the reference (inference) implementation of the symbolic assembly, slowest --> fastest: element,
1178bd882c8aSJames Wright   // comp_in, comp_out, node_row, node_col
1179dd64fc84SJeremy L Thompson   const CeedSize comp_out_stride = num_nodes * num_nodes;
1180dd64fc84SJeremy L Thompson   const CeedSize comp_in_stride  = comp_out_stride * num_comp;
1181dd64fc84SJeremy L Thompson   const CeedSize e_stride        = comp_in_stride * num_comp;
1182*681d0ea7SJeremy L Thompson   // Strides for QF array, slowest --> fastest:  eval_mode_in, comp_in,  eval_mode_out, comp_out, elem, qpt
1183dd64fc84SJeremy L Thompson   const CeedSize q_e_stride             = num_qpts;
1184dd64fc84SJeremy L Thompson   const CeedSize q_comp_out_stride      = num_elem * q_e_stride;
1185*681d0ea7SJeremy L Thompson   const CeedSize q_eval_mode_out_stride = q_comp_out_stride * num_comp;
1186*681d0ea7SJeremy L Thompson   const CeedSize q_comp_in_stride       = q_eval_mode_out_stride * num_eval_mode_out;
1187*681d0ea7SJeremy L Thompson   const CeedSize q_eval_mode_in_stride  = q_comp_in_stride * num_comp;
1188bd882c8aSJames Wright 
1189bd882c8aSJames Wright   CeedScalar *B_in, *B_out;
1190bd882c8aSJames Wright   B_in                       = asmb->d_B_in;
1191bd882c8aSJames Wright   B_out                      = asmb->d_B_out;
1192bd882c8aSJames Wright   const CeedInt block_size_x = asmb->block_size_x;
1193bd882c8aSJames Wright   const CeedInt block_size_y = asmb->block_size_y;
1194bd882c8aSJames Wright 
1195dd64fc84SJeremy L Thompson   sycl::range<3> kernel_range(num_elem, block_size_y, block_size_x);
1196bd882c8aSJames Wright 
11971f4b1b45SUmesh Unnikrishnan   std::vector<sycl::event> e;
11981f4b1b45SUmesh Unnikrishnan 
11991f4b1b45SUmesh Unnikrishnan   if (!sycl_queue.is_in_order()) e = {sycl_queue.ext_oneapi_submit_barrier()};
12001f4b1b45SUmesh Unnikrishnan   sycl_queue.parallel_for<CeedOperatorSyclLinearAssemble>(kernel_range, e, [=](sycl::id<3> idx) {
1201bd882c8aSJames Wright     const int e = idx.get(0);  // Element index
1202bd882c8aSJames Wright     const int l = idx.get(1);  // The output column index of each B^TDB operation
1203bd882c8aSJames Wright     const int i = idx.get(2);  // The output row index of each B^TDB operation
1204bd882c8aSJames Wright                                // such that we have (Bout^T)_ij D_jk Bin_kl = C_il
1205dd64fc84SJeremy L Thompson     for (CeedSize comp_in = 0; comp_in < num_comp; comp_in++) {
1206dd64fc84SJeremy L Thompson       for (CeedSize comp_out = 0; comp_out < num_comp; comp_out++) {
1207bd882c8aSJames Wright         CeedScalar result        = 0.0;
1208dd64fc84SJeremy L Thompson         CeedSize   qf_index_comp = q_comp_in_stride * comp_in + q_comp_out_stride * comp_out + q_e_stride * e;
1209dd64fc84SJeremy L Thompson 
1210*681d0ea7SJeremy L Thompson         for (CeedSize eval_mode_in = 0; eval_mode_in < num_eval_mode_in; eval_mode_in++) {
1211*681d0ea7SJeremy L Thompson           CeedSize b_in_index = eval_mode_in * num_qpts * num_nodes;
1212dd64fc84SJeremy L Thompson 
1213*681d0ea7SJeremy L Thompson           for (CeedSize eval_mode_out = 0; eval_mode_out < num_eval_mode_out; eval_mode_out++) {
1214*681d0ea7SJeremy L Thompson             CeedSize b_out_index = eval_mode_out * num_qpts * num_nodes;
1215*681d0ea7SJeremy L Thompson             CeedSize qf_index    = qf_index_comp + q_eval_mode_out_stride * eval_mode_out + q_eval_mode_in_stride * eval_mode_in;
1216dd64fc84SJeremy L Thompson 
1217bd882c8aSJames Wright             // Perform the B^T D B operation for this 'chunk' of D (the qf_array)
1218dd64fc84SJeremy L Thompson             for (CeedSize j = 0; j < num_qpts; j++) {
1219dd64fc84SJeremy L Thompson               result += B_out[b_out_index + j * num_nodes + i] * qf_array[qf_index + j] * B_in[b_in_index + j * num_nodes + l];
1220bd882c8aSJames Wright             }
1221*681d0ea7SJeremy L Thompson           }  // end of  eval_mode_out
1222*681d0ea7SJeremy L Thompson         }  // end of  eval_mode_in
1223dd64fc84SJeremy L Thompson         CeedSize val_index = comp_in_stride * comp_in + comp_out_stride * comp_out + e_stride * e + num_nodes * i + l;
1224dd64fc84SJeremy L Thompson 
1225bd882c8aSJames Wright         values_array[val_index] = result;
1226bd882c8aSJames Wright       }  // end of out component
1227bd882c8aSJames Wright     }  // end of in component
1228bd882c8aSJames Wright   });
1229bd882c8aSJames Wright   return CEED_ERROR_SUCCESS;
1230bd882c8aSJames Wright }
1231bd882c8aSJames Wright 
1232bd882c8aSJames Wright //------------------------------------------------------------------------------
1233bd882c8aSJames Wright // Fallback kernel for larger orders (1D thread block)
1234bd882c8aSJames Wright //------------------------------------------------------------------------------
1235bd882c8aSJames Wright /*
1236bd882c8aSJames Wright static int CeedOperatorLinearAssembleFallback_Sycl(sycl::queue &sycl_queue, const CeedOperator_Sycl *impl, const CeedScalar *qf_array,
1237bd882c8aSJames Wright                                                    CeedScalar *values_array) {
1238bd882c8aSJames Wright   // This kernel assumes B_in and B_out have the same number of quadrature points and basis points.
1239bd882c8aSJames Wright   // TODO: expand to more general cases
1240bd882c8aSJames Wright   CeedOperatorAssemble_Sycl *asmb        = impl->asmb;
1241dd64fc84SJeremy L Thompson   const CeedInt              num_elem       = asmb->num_elem;
1242dd64fc84SJeremy L Thompson   const CeedInt              num_nodes      = asmb->num_nodes;
1243dd64fc84SJeremy L Thompson   const CeedInt              num_comp       = asmb->num_comp;
1244dd64fc84SJeremy L Thompson   const CeedInt              num_qpts       = asmb->num_qpts;
1245*681d0ea7SJeremy L Thompson   const CeedInt              num_eval_mode_in  = asmb->num_eval_mode_in;
1246*681d0ea7SJeremy L Thompson   const CeedInt              num_eval_mode_out = asmb->num_eval_mode_out;
1247bd882c8aSJames Wright 
1248bd882c8aSJames Wright   // Strides for final output ordering, determined by the reference (interface) implementation of the symbolic assembly, slowest --> fastest: elememt,
1249bd882c8aSJames Wright   // comp_in, comp_out, node_row, node_col
1250dd64fc84SJeremy L Thompson   const CeedInt comp_out_stride = num_nodes * num_nodes;
1251dd64fc84SJeremy L Thompson   const CeedInt comp_in_stride  = comp_out_stride * num_comp;
1252dd64fc84SJeremy L Thompson   const CeedInt e_stride        = comp_in_stride * num_comp;
1253*681d0ea7SJeremy L Thompson   // Strides for QF array, slowest --> fastest:  eval_mode_in, comp_in,  eval_mode_out, comp_out, elem, qpt
1254dd64fc84SJeremy L Thompson   const CeedInt q_e_stride         = num_qpts;
1255dd64fc84SJeremy L Thompson   const CeedInt q_comp_out_stride  = num_elem * q_e_stride;
1256*681d0ea7SJeremy L Thompson   const CeedInt q_eval_mode_out_stride = q_comp_out_stride * num_comp;
1257*681d0ea7SJeremy L Thompson   const CeedInt q_comp_in_stride   = q_eval_mode_out_stride * num_eval_mode_out;
1258*681d0ea7SJeremy L Thompson   const CeedInt q_eval_mode_in_stride  = q_comp_in_stride * num_comp;
1259bd882c8aSJames Wright 
1260bd882c8aSJames Wright   CeedScalar *B_in, *B_out;
1261bd882c8aSJames Wright   B_in                        = asmb->d_B_in;
1262bd882c8aSJames Wright   B_out                       = asmb->d_B_out;
1263004e4986SSebastian Grimberg   const CeedInt elems_per_block = asmb->elems_per_block;
1264bd882c8aSJames Wright   const CeedInt block_size_x  = asmb->block_size_x;
1265bd882c8aSJames Wright   const CeedInt block_size_y  = asmb->block_size_y;  // This will be 1 for the fallback kernel
1266bd882c8aSJames Wright 
1267004e4986SSebastian Grimberg   const CeedInt     grid = num_elem / elems_per_block + ((num_elem / elems_per_block * elems_per_block < num_elem) ? 1 : 0);
1268004e4986SSebastian Grimberg   sycl::range<3>    local_range(block_size_x, block_size_y, elems_per_block);
1269004e4986SSebastian Grimberg   sycl::range<3>    global_range(grid * block_size_x, block_size_y, elems_per_block);
1270bd882c8aSJames Wright   sycl::nd_range<3> kernel_range(global_range, local_range);
1271bd882c8aSJames Wright 
1272bd882c8aSJames Wright   sycl_queue.parallel_for<CeedOperatorSyclLinearAssembleFallback>(kernel_range, [=](sycl::nd_item<3> work_item) {
1273bd882c8aSJames Wright     const CeedInt blockIdx  = work_item.get_group(0);
1274bd882c8aSJames Wright     const CeedInt gridDimx  = work_item.get_group_range(0);
1275bd882c8aSJames Wright     const CeedInt threadIdx = work_item.get_local_id(0);
1276bd882c8aSJames Wright     const CeedInt threadIdz = work_item.get_local_id(2);
1277bd882c8aSJames Wright     const CeedInt blockDimz = work_item.get_local_range(2);
1278bd882c8aSJames Wright 
1279bd882c8aSJames Wright     const int l = threadIdx;  // The output column index of each B^TDB operation
1280bd882c8aSJames Wright                               // such that we have (Bout^T)_ij D_jk Bin_kl = C_il
1281dd64fc84SJeremy L Thompson     for (CeedInt e = blockIdx * blockDimz + threadIdz; e < num_elem; e += gridDimx * blockDimz) {
1282dd64fc84SJeremy L Thompson       for (CeedInt comp_in = 0; comp_in < num_comp; comp_in++) {
1283dd64fc84SJeremy L Thompson         for (CeedInt comp_out = 0; comp_out < num_comp; comp_out++) {
1284dd64fc84SJeremy L Thompson           for (CeedInt i = 0; i < num_nodes; i++) {
1285bd882c8aSJames Wright             CeedScalar result        = 0.0;
1286dd64fc84SJeremy L Thompson             CeedInt    qf_index_comp = q_comp_in_stride * comp_in + q_comp_out_stride * comp_out + q_e_stride * e;
1287*681d0ea7SJeremy L Thompson             for (CeedInt  eval_mode_in = 0;  eval_mode_in < num_eval_mode_in;  eval_mode_in++) {
1288*681d0ea7SJeremy L Thompson               CeedInt b_in_index =  eval_mode_in * num_qpts * num_nodes;
1289*681d0ea7SJeremy L Thompson               for (CeedInt  eval_mode_out = 0;  eval_mode_out < num_eval_mode_out;  eval_mode_out++) {
1290*681d0ea7SJeremy L Thompson                 CeedInt b_out_index =  eval_mode_out * num_qpts * num_nodes;
1291*681d0ea7SJeremy L Thompson                 CeedInt qf_index    = qf_index_comp + q_eval_mode_out_stride *  eval_mode_out + q_eval_mode_in_stride *  eval_mode_in;
1292bd882c8aSJames Wright                 // Perform the B^T D B operation for this 'chunk' of D (the qf_array)
1293dd64fc84SJeremy L Thompson                 for (CeedInt j = 0; j < num_qpts; j++) {
1294dd64fc84SJeremy L Thompson                   result += B_out[b_out_index + j * num_nodes + i] * qf_array[qf_index + j] * B_in[b_in_index + j * num_nodes + l];
1295bd882c8aSJames Wright                 }
1296*681d0ea7SJeremy L Thompson               }  // end of  eval_mode_out
1297*681d0ea7SJeremy L Thompson             }    // end of  eval_mode_in
1298dd64fc84SJeremy L Thompson             CeedInt val_index       = comp_in_stride * comp_in + comp_out_stride * comp_out + e_stride * e + num_nodes * i + l;
1299bd882c8aSJames Wright             values_array[val_index] = result;
1300bd882c8aSJames Wright           }  // end of loop over element node index, i
1301bd882c8aSJames Wright         }    // end of out component
1302bd882c8aSJames Wright       }      // end of in component
1303bd882c8aSJames Wright     }        // end of element loop
1304bd882c8aSJames Wright   });
1305bd882c8aSJames Wright   return CEED_ERROR_SUCCESS;
1306bd882c8aSJames Wright }*/
1307bd882c8aSJames Wright 
1308bd882c8aSJames Wright //------------------------------------------------------------------------------
1309bd882c8aSJames Wright // Assemble matrix data for COO matrix of assembled operator.
1310bd882c8aSJames Wright // The sparsity pattern is set by CeedOperatorLinearAssembleSymbolic.
1311bd882c8aSJames Wright //
1312bd882c8aSJames Wright // Note that this (and other assembly routines) currently assume only one active
1313bd882c8aSJames Wright // input restriction/basis per operator (could have multiple basis eval modes).
1314bd882c8aSJames Wright // TODO: allow multiple active input restrictions/basis objects
1315bd882c8aSJames Wright //------------------------------------------------------------------------------
1316bd882c8aSJames Wright static int CeedSingleOperatorAssemble_Sycl(CeedOperator op, CeedInt offset, CeedVector values) {
1317bd882c8aSJames Wright   Ceed                ceed;
1318bd882c8aSJames Wright   Ceed_Sycl          *sycl_data;
1319dd64fc84SJeremy L Thompson   CeedScalar         *values_array;
1320dd64fc84SJeremy L Thompson   const CeedScalar   *qf_array;
1321dd64fc84SJeremy L Thompson   CeedVector          assembled_qf = NULL;
1322dd64fc84SJeremy L Thompson   CeedElemRestriction rstr_q       = NULL;
1323dd64fc84SJeremy L Thompson   CeedOperator_Sycl  *impl;
1324dd64fc84SJeremy L Thompson 
1325dd64fc84SJeremy L Thompson   CeedCallBackend(CeedOperatorGetCeed(op, &ceed));
1326dd64fc84SJeremy L Thompson   CeedCallBackend(CeedOperatorGetData(op, &impl));
1327bd882c8aSJames Wright   CeedCallBackend(CeedGetData(ceed, &sycl_data));
1328bd882c8aSJames Wright 
1329bd882c8aSJames Wright   // Setup
1330bd882c8aSJames Wright   if (!impl->asmb) {
1331bd882c8aSJames Wright     CeedCallBackend(CeedSingleOperatorAssembleSetup_Sycl(op));
1332bd882c8aSJames Wright     assert(impl->asmb != NULL);
1333bd882c8aSJames Wright   }
1334bd882c8aSJames Wright 
1335bd882c8aSJames Wright   // Assemble QFunction
1336bd882c8aSJames Wright   CeedCallBackend(CeedOperatorLinearAssembleQFunctionBuildOrUpdate(op, &assembled_qf, &rstr_q, CEED_REQUEST_IMMEDIATE));
1337bd882c8aSJames Wright   CeedCallBackend(CeedElemRestrictionDestroy(&rstr_q));
1338bd882c8aSJames Wright   CeedCallBackend(CeedVectorGetArrayWrite(values, CEED_MEM_DEVICE, &values_array));
1339bd882c8aSJames Wright   values_array += offset;
1340bd882c8aSJames Wright   CeedCallBackend(CeedVectorGetArrayRead(assembled_qf, CEED_MEM_DEVICE, &qf_array));
1341bd882c8aSJames Wright 
1342bd882c8aSJames Wright   // Compute B^T D B
1343bd882c8aSJames Wright   CeedCallBackend(CeedOperatorLinearAssemble_Sycl(sycl_data->sycl_queue, impl, qf_array, values_array));
1344bd882c8aSJames Wright 
1345bd882c8aSJames Wright   // Wait for kernels to be completed
1346bd882c8aSJames Wright   // Kris: Review if this is necessary -- enqueing an async barrier may be sufficient
1347bd882c8aSJames Wright   sycl_data->sycl_queue.wait_and_throw();
1348bd882c8aSJames Wright 
1349bd882c8aSJames Wright   // Restore arrays
1350bd882c8aSJames Wright   CeedCallBackend(CeedVectorRestoreArray(values, &values_array));
1351bd882c8aSJames Wright   CeedCallBackend(CeedVectorRestoreArrayRead(assembled_qf, &qf_array));
1352bd882c8aSJames Wright 
1353bd882c8aSJames Wright   // Cleanup
1354bd882c8aSJames Wright   CeedCallBackend(CeedVectorDestroy(&assembled_qf));
1355bd882c8aSJames Wright   return CEED_ERROR_SUCCESS;
1356bd882c8aSJames Wright }
1357bd882c8aSJames Wright 
1358bd882c8aSJames Wright //------------------------------------------------------------------------------
1359bd882c8aSJames Wright // Create operator
1360bd882c8aSJames Wright //------------------------------------------------------------------------------
1361bd882c8aSJames Wright int CeedOperatorCreate_Sycl(CeedOperator op) {
1362bd882c8aSJames Wright   Ceed               ceed;
1363bd882c8aSJames Wright   CeedOperator_Sycl *impl;
1364bd882c8aSJames Wright 
1365dd64fc84SJeremy L Thompson   CeedCallBackend(CeedOperatorGetCeed(op, &ceed));
1366dd64fc84SJeremy L Thompson 
1367bd882c8aSJames Wright   CeedCallBackend(CeedCalloc(1, &impl));
1368bd882c8aSJames Wright   CeedCallBackend(CeedOperatorSetData(op, impl));
1369bd882c8aSJames Wright 
1370bd882c8aSJames Wright   CeedCallBackend(CeedSetBackendFunctionCpp(ceed, "Operator", op, "LinearAssembleQFunction", CeedOperatorLinearAssembleQFunction_Sycl));
1371bd882c8aSJames Wright   CeedCallBackend(CeedSetBackendFunctionCpp(ceed, "Operator", op, "LinearAssembleQFunctionUpdate", CeedOperatorLinearAssembleQFunctionUpdate_Sycl));
1372bd882c8aSJames Wright   CeedCallBackend(CeedSetBackendFunctionCpp(ceed, "Operator", op, "LinearAssembleAddDiagonal", CeedOperatorLinearAssembleAddDiagonal_Sycl));
1373bd882c8aSJames Wright   CeedCallBackend(
1374bd882c8aSJames Wright       CeedSetBackendFunctionCpp(ceed, "Operator", op, "LinearAssembleAddPointBlockDiagonal", CeedOperatorLinearAssembleAddPointBlockDiagonal_Sycl));
1375bd882c8aSJames Wright   CeedCallBackend(CeedSetBackendFunctionCpp(ceed, "Operator", op, "LinearAssembleSingle", CeedSingleOperatorAssemble_Sycl));
1376bd882c8aSJames Wright   CeedCallBackend(CeedSetBackendFunctionCpp(ceed, "Operator", op, "ApplyAdd", CeedOperatorApplyAdd_Sycl));
1377bd882c8aSJames Wright   CeedCallBackend(CeedSetBackendFunctionCpp(ceed, "Operator", op, "Destroy", CeedOperatorDestroy_Sycl));
1378bd882c8aSJames Wright   return CEED_ERROR_SUCCESS;
1379bd882c8aSJames Wright }
1380bd882c8aSJames Wright 
1381bd882c8aSJames Wright //------------------------------------------------------------------------------
1382