xref: /libCEED/rust/libceed-sys/c-src/backends/sycl-ref/ceed-sycl-ref-operator.sycl.cpp (revision ac2269bd6b9128b5262c5b76aac90f6c3e7aeebf)
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 //------------------------------------------------------------------------------
26681d0ea7SJeremy L Thompson void CeedOperatorGetBasisPointer_Sycl(const CeedScalar **basis_ptr, CeedEvalMode eval_mode, const CeedScalar *identity, const CeedScalar *interp,
27bd882c8aSJames Wright                                       const CeedScalar *grad) {
28681d0ea7SJeremy 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*ac2269bdSJeremy L Thompson   // QFunction assembly data
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) {
81681d0ea7SJeremy L Thompson     CeedCallBackend(CeedFree(&impl->diag->h_eval_mode_in));
82681d0ea7SJeremy L Thompson     CeedCallBackend(CeedFree(&impl->diag->h_eval_mode_out));
83bd882c8aSJames Wright 
84bd882c8aSJames Wright     CeedCallSycl(ceed, sycl_data->sycl_queue.wait_and_throw());
85681d0ea7SJeremy L Thompson     CeedCallSycl(ceed, sycl::free(impl->diag->d_eval_mode_in, sycl_data->sycl_context));
86681d0ea7SJeremy 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++) {
133681d0ea7SJeremy L Thompson     CeedEvalMode        eval_mode;
134dd64fc84SJeremy L Thompson     CeedVector          vec;
135*ac2269bdSJeremy L Thompson     CeedElemRestriction elem_rstr;
136dd64fc84SJeremy L Thompson     CeedBasis           basis;
137bd882c8aSJames Wright 
138681d0ea7SJeremy L Thompson     CeedCallBackend(CeedQFunctionFieldGetEvalMode(qf_fields[i], &eval_mode));
139dd64fc84SJeremy L Thompson 
140dd64fc84SJeremy L Thompson     is_strided       = false;
141dd64fc84SJeremy L Thompson     skip_restriction = false;
142681d0ea7SJeremy L Thompson     if (eval_mode != CEED_EVAL_WEIGHT) {
143*ac2269bdSJeremy L Thompson       CeedCallBackend(CeedOperatorFieldGetElemRestriction(op_fields[i], &elem_rstr));
144bd882c8aSJames Wright 
145bd882c8aSJames Wright       // Check whether this field can skip the element restriction:
146681d0ea7SJeremy 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) {
153681d0ea7SJeremy L Thompson           // Check  eval_mode
154681d0ea7SJeremy L Thompson           if (eval_mode == CEED_EVAL_NONE) {
155dd64fc84SJeremy L Thompson             // Check for  is_strided restriction
156*ac2269bdSJeremy L Thompson             CeedCallBackend(CeedElemRestrictionIsStrided(elem_rstr, &is_strided));
157dd64fc84SJeremy L Thompson             if (is_strided) {
158bd882c8aSJames Wright               // Check if vector is already in preferred backend ordering
159*ac2269bdSJeremy L Thompson               CeedCallBackend(CeedElemRestrictionHasBackendStrides(elem_rstr, &skip_restriction));
160bd882c8aSJames Wright             }
161bd882c8aSJames Wright           }
162bd882c8aSJames Wright         }
163681d0ea7SJeremy 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 {
169*ac2269bdSJeremy L Thompson         CeedCallBackend(CeedElemRestrictionCreateVector(elem_rstr, NULL, &e_vecs[i + start_e]));
170bd882c8aSJames Wright       }
171*ac2269bdSJeremy L Thompson       CeedCallBackend(CeedElemRestrictionDestroy(&elem_rstr));
172bd882c8aSJames Wright     }
173bd882c8aSJames Wright 
174681d0ea7SJeremy 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));
189681d0ea7SJeremy 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]));
198681d0ea7SJeremy 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++) {
259681d0ea7SJeremy L Thompson     bool         is_active;
260681d0ea7SJeremy 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));
265681d0ea7SJeremy L Thompson     is_active = vec == CEED_VECTOR_ACTIVE;
266681d0ea7SJeremy L Thompson     if (is_active) {
267dd64fc84SJeremy L Thompson       if (skip_active) continue;
268dd64fc84SJeremy L Thompson       else vec = in_vec;
269bd882c8aSJames Wright     }
270bd882c8aSJames Wright 
271681d0ea7SJeremy L Thompson     CeedCallBackend(CeedQFunctionFieldGetEvalMode(qf_input_fields[i], &eval_mode));
272681d0ea7SJeremy 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*ac2269bdSJeremy L Thompson         CeedElemRestriction elem_rstr;
280681d0ea7SJeremy L Thompson 
281*ac2269bdSJeremy L Thompson         CeedCallBackend(CeedOperatorFieldGetElemRestriction(op_input_fields[i], &elem_rstr));
282*ac2269bdSJeremy L Thompson         CeedCallBackend(CeedElemRestrictionApply(elem_rstr, CEED_NOTRANSPOSE, vec, impl->e_vecs[i], request));
283*ac2269bdSJeremy L Thompson         CeedCallBackend(CeedElemRestrictionDestroy(&elem_rstr));
284dd64fc84SJeremy L Thompson         CeedCallBackend(CeedVectorGetArrayRead(impl->e_vecs[i], CEED_MEM_DEVICE, (const CeedScalar **)&e_data[i]));
285bd882c8aSJames Wright       }
286bd882c8aSJames Wright     }
287681d0ea7SJeremy 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++) {
299681d0ea7SJeremy L Thompson     CeedEvalMode eval_mode;
300bd882c8aSJames Wright 
301bd882c8aSJames Wright     // Skip active input
302dd64fc84SJeremy L Thompson     if (skip_active) {
303681d0ea7SJeremy L Thompson       bool       is_active;
304bd882c8aSJames Wright       CeedVector vec;
305dd64fc84SJeremy L Thompson 
306dd64fc84SJeremy L Thompson       CeedCallBackend(CeedOperatorFieldGetVector(op_input_fields[i], &vec));
307681d0ea7SJeremy L Thompson       is_active = vec == CEED_VECTOR_ACTIVE;
308681d0ea7SJeremy L Thompson       CeedCallBackend(CeedVectorDestroy(&vec));
309681d0ea7SJeremy L Thompson       if (is_active) continue;
310bd882c8aSJames Wright     }
311bd882c8aSJames Wright     // Basis action
312681d0ea7SJeremy L Thompson     CeedCallBackend(CeedQFunctionFieldGetEvalMode(qf_input_fields[i], &eval_mode));
313681d0ea7SJeremy L Thompson     switch (eval_mode) {
314bd882c8aSJames Wright       case CEED_EVAL_NONE:
315dd64fc84SJeremy L Thompson         CeedCallBackend(CeedVectorSetArray(impl->q_vecs_in[i], CEED_MEM_DEVICE, CEED_USE_POINTER, e_data[i]));
316bd882c8aSJames Wright         break;
317bd882c8aSJames Wright       case CEED_EVAL_INTERP:
318*ac2269bdSJeremy L Thompson       case CEED_EVAL_GRAD: {
319*ac2269bdSJeremy L Thompson         CeedBasis basis;
320*ac2269bdSJeremy L Thompson 
321dd64fc84SJeremy L Thompson         CeedCallBackend(CeedOperatorFieldGetBasis(op_input_fields[i], &basis));
322681d0ea7SJeremy L Thompson         CeedCallBackend(CeedBasisApply(basis, num_elem, CEED_NOTRANSPOSE, eval_mode, impl->e_vecs[i], impl->q_vecs_in[i]));
323681d0ea7SJeremy L Thompson         CeedCallBackend(CeedBasisDestroy(&basis));
324bd882c8aSJames Wright         break;
325*ac2269bdSJeremy L Thompson       }
326bd882c8aSJames Wright       case CEED_EVAL_WEIGHT:
327bd882c8aSJames Wright         break;  // No action
328bd882c8aSJames Wright       case CEED_EVAL_DIV:
329bd882c8aSJames Wright         break;  // TODO: Not implemented
330bd882c8aSJames Wright       case CEED_EVAL_CURL:
331bd882c8aSJames Wright         break;  // TODO: Not implemented
332bd882c8aSJames Wright     }
333bd882c8aSJames Wright   }
334bd882c8aSJames Wright   return CEED_ERROR_SUCCESS;
335bd882c8aSJames Wright }
336bd882c8aSJames Wright 
337bd882c8aSJames Wright //------------------------------------------------------------------------------
338bd882c8aSJames Wright // Restore Input Vectors
339bd882c8aSJames Wright //------------------------------------------------------------------------------
340dd64fc84SJeremy L Thompson static inline int CeedOperatorRestoreInputs_Sycl(CeedInt num_input_fields, CeedQFunctionField *qf_input_fields, CeedOperatorField *op_input_fields,
341dd64fc84SJeremy L Thompson                                                  const bool skip_active, CeedScalar *e_data[2 * CEED_FIELD_MAX], CeedOperator_Sycl *impl) {
342dd64fc84SJeremy L Thompson   for (CeedInt i = 0; i < num_input_fields; i++) {
343681d0ea7SJeremy L Thompson     bool         is_active;
344681d0ea7SJeremy L Thompson     CeedEvalMode eval_mode;
345bd882c8aSJames Wright     CeedVector   vec;
346bd882c8aSJames Wright 
347681d0ea7SJeremy L Thompson     CeedCallBackend(CeedOperatorFieldGetVector(op_input_fields[i], &vec));
348681d0ea7SJeremy L Thompson     is_active = vec == CEED_VECTOR_ACTIVE;
349bd882c8aSJames Wright     // Skip active input
350dd64fc84SJeremy L Thompson     if (skip_active) {
351681d0ea7SJeremy L Thompson       if (is_active) continue;
352bd882c8aSJames Wright     }
353681d0ea7SJeremy L Thompson     CeedCallBackend(CeedQFunctionFieldGetEvalMode(qf_input_fields[i], &eval_mode));
354681d0ea7SJeremy L Thompson     if (eval_mode == CEED_EVAL_WEIGHT) {  // Skip
355bd882c8aSJames Wright     } else {
356dd64fc84SJeremy L Thompson       if (!impl->e_vecs[i]) {  // This was a  skip_restriction case
357dd64fc84SJeremy L Thompson         CeedCallBackend(CeedVectorRestoreArrayRead(vec, (const CeedScalar **)&e_data[i]));
358bd882c8aSJames Wright       } else {
359dd64fc84SJeremy L Thompson         CeedCallBackend(CeedVectorRestoreArrayRead(impl->e_vecs[i], (const CeedScalar **)&e_data[i]));
360bd882c8aSJames Wright       }
361bd882c8aSJames Wright     }
362681d0ea7SJeremy L Thompson     if (!is_active) CeedCallBackend(CeedVectorDestroy(&vec));
363bd882c8aSJames Wright   }
364bd882c8aSJames Wright   return CEED_ERROR_SUCCESS;
365bd882c8aSJames Wright }
366bd882c8aSJames Wright 
367bd882c8aSJames Wright //------------------------------------------------------------------------------
368bd882c8aSJames Wright // Apply and add to output
369bd882c8aSJames Wright //------------------------------------------------------------------------------
370dd64fc84SJeremy L Thompson static int CeedOperatorApplyAdd_Sycl(CeedOperator op, CeedVector in_vec, CeedVector out_vec, CeedRequest *request) {
371dd64fc84SJeremy L Thompson   CeedInt             Q, num_elem, elem_size, num_input_fields, num_output_fields, size;
3729d1bceceSJames Wright   CeedEvalMode        eval_mode;
373dd64fc84SJeremy L Thompson   CeedScalar         *e_data[2 * CEED_FIELD_MAX] = {0};
374dd64fc84SJeremy L Thompson   CeedQFunctionField *qf_input_fields, *qf_output_fields;
375bd882c8aSJames Wright   CeedQFunction       qf;
376dd64fc84SJeremy L Thompson   CeedOperatorField  *op_input_fields, *op_output_fields;
377dd64fc84SJeremy L Thompson   CeedOperator_Sycl  *impl;
378dd64fc84SJeremy L Thompson 
379dd64fc84SJeremy L Thompson   CeedCallBackend(CeedOperatorGetData(op, &impl));
380bd882c8aSJames Wright   CeedCallBackend(CeedOperatorGetQFunction(op, &qf));
381bd882c8aSJames Wright   CeedCallBackend(CeedOperatorGetNumQuadraturePoints(op, &Q));
382dd64fc84SJeremy L Thompson   CeedCallBackend(CeedOperatorGetNumElements(op, &num_elem));
383dd64fc84SJeremy L Thompson   CeedCallBackend(CeedOperatorGetFields(op, &num_input_fields, &op_input_fields, &num_output_fields, &op_output_fields));
384dd64fc84SJeremy L Thompson   CeedCallBackend(CeedQFunctionGetFields(qf, NULL, &qf_input_fields, NULL, &qf_output_fields));
385bd882c8aSJames Wright 
386bd882c8aSJames Wright   // Setup
387bd882c8aSJames Wright   CeedCallBackend(CeedOperatorSetup_Sycl(op));
388bd882c8aSJames Wright 
389bd882c8aSJames Wright   // Input Evecs and Restriction
390dd64fc84SJeremy L Thompson   CeedCallBackend(CeedOperatorSetupInputs_Sycl(num_input_fields, qf_input_fields, op_input_fields, in_vec, false, e_data, impl, request));
391bd882c8aSJames Wright 
392bd882c8aSJames Wright   // Input basis apply if needed
393dd64fc84SJeremy L Thompson   CeedCallBackend(CeedOperatorInputBasis_Sycl(num_elem, qf_input_fields, op_input_fields, num_input_fields, false, e_data, impl));
394bd882c8aSJames Wright 
395bd882c8aSJames Wright   // Output pointers, as necessary
396dd64fc84SJeremy L Thompson   for (CeedInt i = 0; i < num_output_fields; i++) {
3979d1bceceSJames Wright     CeedCallBackend(CeedQFunctionFieldGetEvalMode(qf_output_fields[i], &eval_mode));
3989d1bceceSJames Wright     if (eval_mode == CEED_EVAL_NONE) {
399bd882c8aSJames Wright       // Set the output Q-Vector to use the E-Vector data directly
400dd64fc84SJeremy L Thompson       CeedCallBackend(CeedVectorGetArrayWrite(impl->e_vecs[i + impl->num_e_in], CEED_MEM_DEVICE, &e_data[i + num_input_fields]));
401dd64fc84SJeremy L Thompson       CeedCallBackend(CeedVectorSetArray(impl->q_vecs_out[i], CEED_MEM_DEVICE, CEED_USE_POINTER, e_data[i + num_input_fields]));
402bd882c8aSJames Wright     }
403bd882c8aSJames Wright   }
404bd882c8aSJames Wright 
405bd882c8aSJames Wright   // Q function
406dd64fc84SJeremy L Thompson   CeedCallBackend(CeedQFunctionApply(qf, num_elem * Q, impl->q_vecs_in, impl->q_vecs_out));
407bd882c8aSJames Wright 
408bd882c8aSJames Wright   // Output basis apply if needed
409dd64fc84SJeremy L Thompson   for (CeedInt i = 0; i < num_output_fields; i++) {
410*ac2269bdSJeremy L Thompson     CeedElemRestriction elem_rstr;
411dd64fc84SJeremy L Thompson 
4129d1bceceSJames Wright     // Get elem_size, eval_mode, size
413*ac2269bdSJeremy L Thompson     CeedCallBackend(CeedOperatorFieldGetElemRestriction(op_output_fields[i], &elem_rstr));
414*ac2269bdSJeremy L Thompson     CeedCallBackend(CeedElemRestrictionGetElementSize(elem_rstr, &elem_size));
415*ac2269bdSJeremy L Thompson     CeedCallBackend(CeedElemRestrictionDestroy(&elem_rstr));
4169d1bceceSJames Wright     CeedCallBackend(CeedQFunctionFieldGetEvalMode(qf_output_fields[i], &eval_mode));
417dd64fc84SJeremy L Thompson     CeedCallBackend(CeedQFunctionFieldGetSize(qf_output_fields[i], &size));
418bd882c8aSJames Wright     // Basis action
4199d1bceceSJames Wright     switch (eval_mode) {
420bd882c8aSJames Wright       case CEED_EVAL_NONE:
421bd882c8aSJames Wright         break;
422bd882c8aSJames Wright       case CEED_EVAL_INTERP:
423*ac2269bdSJeremy L Thompson       case CEED_EVAL_GRAD: {
424*ac2269bdSJeremy L Thompson         CeedBasis basis;
425*ac2269bdSJeremy L Thompson 
426dd64fc84SJeremy L Thompson         CeedCallBackend(CeedOperatorFieldGetBasis(op_output_fields[i], &basis));
427681d0ea7SJeremy L Thompson         CeedCallBackend(CeedBasisApply(basis, num_elem, CEED_TRANSPOSE, eval_mode, impl->q_vecs_out[i], impl->e_vecs[i + impl->num_e_in]));
428681d0ea7SJeremy L Thompson         CeedCallBackend(CeedBasisDestroy(&basis));
429bd882c8aSJames Wright         break;
430*ac2269bdSJeremy L Thompson       }
431bd882c8aSJames Wright       // LCOV_EXCL_START
432*ac2269bdSJeremy L Thompson       case CEED_EVAL_WEIGHT: {
433bd882c8aSJames Wright         Ceed ceed;
4344e3038a5SJeremy L Thompson 
435bd882c8aSJames Wright         CeedCallBackend(CeedOperatorGetCeed(op, &ceed));
436bd882c8aSJames Wright         return CeedError(ceed, CEED_ERROR_BACKEND, "CEED_EVAL_WEIGHT cannot be an output evaluation mode");
437bd882c8aSJames Wright         break;  // Should not occur
438*ac2269bdSJeremy L Thompson       }
439bd882c8aSJames Wright       case CEED_EVAL_DIV:
4404e3038a5SJeremy L Thompson       case CEED_EVAL_CURL: {
4414e3038a5SJeremy L Thompson         Ceed ceed;
4424e3038a5SJeremy L Thompson 
4434e3038a5SJeremy L Thompson         CeedCallBackend(CeedOperatorGetCeed(op, &ceed));
4444e3038a5SJeremy L Thompson         return CeedError(ceed, CEED_ERROR_BACKEND, "%s not supported", CeedEvalModes[eval_mode]);
4454e3038a5SJeremy L Thompson         break;  // Should not occur
4464e3038a5SJeremy L Thompson       }
447bd882c8aSJames Wright         // LCOV_EXCL_STOP
448bd882c8aSJames Wright     }
449bd882c8aSJames Wright   }
450bd882c8aSJames Wright 
451bd882c8aSJames Wright   // Output restriction
452dd64fc84SJeremy L Thompson   for (CeedInt i = 0; i < num_output_fields; i++) {
453681d0ea7SJeremy L Thompson     bool                is_active;
454681d0ea7SJeremy L Thompson     CeedEvalMode        eval_mode;
455dd64fc84SJeremy L Thompson     CeedVector          vec;
456*ac2269bdSJeremy L Thompson     CeedElemRestriction elem_rstr;
457dd64fc84SJeremy L Thompson 
458bd882c8aSJames Wright     // Restore evec
4599d1bceceSJames Wright     CeedCallBackend(CeedQFunctionFieldGetEvalMode(qf_output_fields[i], &eval_mode));
4609d1bceceSJames Wright     if (eval_mode == CEED_EVAL_NONE) {
461dd64fc84SJeremy L Thompson       CeedCallBackend(CeedVectorRestoreArray(impl->e_vecs[i + impl->num_e_in], &e_data[i + num_input_fields]));
462bd882c8aSJames Wright     }
463bd882c8aSJames Wright     // Restrict
464681d0ea7SJeremy L Thompson     CeedCallBackend(CeedOperatorFieldGetVector(op_output_fields[i], &vec));
465681d0ea7SJeremy L Thompson     is_active = vec == CEED_VECTOR_ACTIVE;
466*ac2269bdSJeremy L Thompson     CeedCallBackend(CeedOperatorFieldGetElemRestriction(op_output_fields[i], &elem_rstr));
467681d0ea7SJeremy L Thompson     if (is_active) vec = out_vec;
468*ac2269bdSJeremy L Thompson     CeedCallBackend(CeedElemRestrictionApply(elem_rstr, CEED_TRANSPOSE, impl->e_vecs[i + impl->num_e_in], vec, request));
469681d0ea7SJeremy L Thompson     if (!is_active) CeedCallBackend(CeedVectorDestroy(&vec));
470*ac2269bdSJeremy L Thompson     CeedCallBackend(CeedElemRestrictionDestroy(&elem_rstr));
471bd882c8aSJames Wright   }
472bd882c8aSJames Wright 
473bd882c8aSJames Wright   // Restore input arrays
474dd64fc84SJeremy L Thompson   CeedCallBackend(CeedOperatorRestoreInputs_Sycl(num_input_fields, qf_input_fields, op_input_fields, false, e_data, impl));
475bd882c8aSJames Wright   return CEED_ERROR_SUCCESS;
476bd882c8aSJames Wright }
477bd882c8aSJames Wright 
478bd882c8aSJames Wright //------------------------------------------------------------------------------
479bd882c8aSJames Wright // Core code for assembling linear QFunction
480bd882c8aSJames Wright //------------------------------------------------------------------------------
481*ac2269bdSJeremy L Thompson static inline int CeedOperatorLinearAssembleQFunctionCore_Sycl(CeedOperator op, bool build_objects, CeedVector *assembled,
482*ac2269bdSJeremy L Thompson                                                                CeedElemRestriction *elem_rstr, CeedRequest *request) {
483dd64fc84SJeremy L Thompson   Ceed                ceed, ceed_parent;
484e984cf9aSJeremy L Thompson   CeedSize            q_size;
485dd64fc84SJeremy L Thompson   CeedInt             num_active_in, num_active_out, Q, num_elem, num_input_fields, num_output_fields, size;
486dd64fc84SJeremy L Thompson   CeedScalar         *assembled_array, *e_data[2 * CEED_FIELD_MAX] = {NULL};
487dd64fc84SJeremy L Thompson   CeedVector         *active_in;
488dd64fc84SJeremy L Thompson   CeedQFunctionField *qf_input_fields, *qf_output_fields;
489dd64fc84SJeremy L Thompson   CeedQFunction       qf;
490dd64fc84SJeremy L Thompson   CeedOperatorField  *op_input_fields, *op_output_fields;
491dd64fc84SJeremy L Thompson   CeedOperator_Sycl  *impl;
492dd64fc84SJeremy L Thompson 
493bd882c8aSJames Wright   CeedCallBackend(CeedOperatorGetCeed(op, &ceed));
494dd64fc84SJeremy L Thompson   CeedCallBackend(CeedOperatorGetFallbackParentCeed(op, &ceed_parent));
495e984cf9aSJeremy L Thompson   CeedCallBackend(CeedOperatorGetData(op, &impl));
496e984cf9aSJeremy L Thompson   CeedCallBackend(CeedOperatorGetQFunction(op, &qf));
497e984cf9aSJeremy L Thompson   CeedCallBackend(CeedOperatorGetNumQuadraturePoints(op, &Q));
498dd64fc84SJeremy L Thompson   CeedCallBackend(CeedOperatorGetNumElements(op, &num_elem));
499dd64fc84SJeremy L Thompson   CeedCallBackend(CeedQFunctionGetFields(qf, NULL, &qf_input_fields, NULL, &qf_output_fields));
500dd64fc84SJeremy L Thompson   CeedCallBackend(CeedOperatorGetFields(op, &num_input_fields, &op_input_fields, &num_output_fields, &op_output_fields));
501dd64fc84SJeremy L Thompson   active_in     = impl->qf_active_in;
502dd64fc84SJeremy L Thompson   num_active_in = impl->num_active_in, num_active_out = impl->num_active_out;
503bd882c8aSJames Wright 
504bd882c8aSJames Wright   // Setup
505bd882c8aSJames Wright   CeedCallBackend(CeedOperatorSetup_Sycl(op));
506bd882c8aSJames Wright 
507bd882c8aSJames Wright   // Input Evecs and Restriction
508dd64fc84SJeremy L Thompson   CeedCallBackend(CeedOperatorSetupInputs_Sycl(num_input_fields, qf_input_fields, op_input_fields, NULL, true, e_data, impl, request));
509bd882c8aSJames Wright 
510bd882c8aSJames Wright   // Count number of active input fields
511dd64fc84SJeremy L Thompson   if (!num_active_in) {
512dd64fc84SJeremy L Thompson     for (CeedInt i = 0; i < num_input_fields; i++) {
513dd64fc84SJeremy L Thompson       CeedScalar *q_vec_array;
514dd64fc84SJeremy L Thompson       CeedVector  vec;
515dd64fc84SJeremy L Thompson 
516bd882c8aSJames Wright       // Check if active input
517681d0ea7SJeremy L Thompson       CeedCallBackend(CeedOperatorFieldGetVector(op_input_fields[i], &vec));
518bd882c8aSJames Wright       if (vec == CEED_VECTOR_ACTIVE) {
519dd64fc84SJeremy L Thompson         CeedCallBackend(CeedQFunctionFieldGetSize(qf_input_fields[i], &size));
520dd64fc84SJeremy L Thompson         CeedCallBackend(CeedVectorSetValue(impl->q_vecs_in[i], 0.0));
521dd64fc84SJeremy L Thompson         CeedCallBackend(CeedVectorGetArray(impl->q_vecs_in[i], CEED_MEM_DEVICE, &q_vec_array));
522dd64fc84SJeremy L Thompson         CeedCallBackend(CeedRealloc(num_active_in + size, &active_in));
523bd882c8aSJames Wright         for (CeedInt field = 0; field < size; field++) {
524dd64fc84SJeremy L Thompson           q_size = (CeedSize)Q * num_elem;
525dd64fc84SJeremy L Thompson           CeedCallBackend(CeedVectorCreate(ceed, q_size, &active_in[num_active_in + field]));
526dd64fc84SJeremy L Thompson           CeedCallBackend(
527dd64fc84SJeremy L Thompson               CeedVectorSetArray(active_in[num_active_in + field], CEED_MEM_DEVICE, CEED_USE_POINTER, &q_vec_array[field * Q * num_elem]));
528bd882c8aSJames Wright         }
529dd64fc84SJeremy L Thompson         num_active_in += size;
530dd64fc84SJeremy L Thompson         CeedCallBackend(CeedVectorRestoreArray(impl->q_vecs_in[i], &q_vec_array));
531bd882c8aSJames Wright       }
532681d0ea7SJeremy L Thompson       CeedCallBackend(CeedVectorDestroy(&vec));
533bd882c8aSJames Wright     }
534dd64fc84SJeremy L Thompson     impl->num_active_in = num_active_in;
535dd64fc84SJeremy L Thompson     impl->qf_active_in  = active_in;
536bd882c8aSJames Wright   }
537bd882c8aSJames Wright 
538bd882c8aSJames Wright   // Count number of active output fields
539dd64fc84SJeremy L Thompson   if (!num_active_out) {
540dd64fc84SJeremy L Thompson     for (CeedInt i = 0; i < num_output_fields; i++) {
541dd64fc84SJeremy L Thompson       CeedVector vec;
542dd64fc84SJeremy L Thompson 
543bd882c8aSJames Wright       // Check if active output
544681d0ea7SJeremy L Thompson       CeedCallBackend(CeedOperatorFieldGetVector(op_output_fields[i], &vec));
545bd882c8aSJames Wright       if (vec == CEED_VECTOR_ACTIVE) {
546dd64fc84SJeremy L Thompson         CeedCallBackend(CeedQFunctionFieldGetSize(qf_output_fields[i], &size));
547dd64fc84SJeremy L Thompson         num_active_out += size;
548bd882c8aSJames Wright       }
549681d0ea7SJeremy L Thompson       CeedCallBackend(CeedVectorDestroy(&vec));
550bd882c8aSJames Wright     }
551dd64fc84SJeremy L Thompson     impl->num_active_out = num_active_out;
552bd882c8aSJames Wright   }
553bd882c8aSJames Wright 
554bd882c8aSJames Wright   // Check sizes
5554e3038a5SJeremy L Thompson   CeedCheck(num_active_in > 0 && num_active_out > 0, ceed, CEED_ERROR_BACKEND, "Cannot assemble QFunction without active inputs and outputs");
556bd882c8aSJames Wright 
557bd882c8aSJames Wright   // Build objects if needed
558bd882c8aSJames Wright   if (build_objects) {
559dd64fc84SJeremy L Thompson     CeedSize l_size     = (CeedSize)num_elem * Q * num_active_in * num_active_out;
560dd64fc84SJeremy L Thompson     CeedInt  strides[3] = {1, num_elem * Q, Q}; /* *NOPAD* */
561dd64fc84SJeremy L Thompson 
562bd882c8aSJames Wright     // Create output restriction
563*ac2269bdSJeremy L Thompson     CeedCallBackend(CeedElemRestrictionCreateStrided(ceed_parent, num_elem, Q, num_active_in * num_active_out, l_size, strides, elem_rstr));
564bd882c8aSJames Wright     // Create assembled vector
565dd64fc84SJeremy L Thompson     CeedCallBackend(CeedVectorCreate(ceed_parent, l_size, assembled));
566bd882c8aSJames Wright   }
567bd882c8aSJames Wright   CeedCallBackend(CeedVectorSetValue(*assembled, 0.0));
568dd64fc84SJeremy L Thompson   CeedCallBackend(CeedVectorGetArray(*assembled, CEED_MEM_DEVICE, &assembled_array));
569bd882c8aSJames Wright 
570bd882c8aSJames Wright   // Input basis apply
571dd64fc84SJeremy L Thompson   CeedCallBackend(CeedOperatorInputBasis_Sycl(num_elem, qf_input_fields, op_input_fields, num_input_fields, true, e_data, impl));
572bd882c8aSJames Wright 
573bd882c8aSJames Wright   // Assemble QFunction
574dd64fc84SJeremy L Thompson   for (CeedInt in = 0; in < num_active_in; in++) {
575bd882c8aSJames Wright     // Set Inputs
576dd64fc84SJeremy L Thompson     CeedCallBackend(CeedVectorSetValue(active_in[in], 1.0));
577dd64fc84SJeremy L Thompson     if (num_active_in > 1) {
578dd64fc84SJeremy L Thompson       CeedCallBackend(CeedVectorSetValue(active_in[(in + num_active_in - 1) % num_active_in], 0.0));
579bd882c8aSJames Wright     }
580bd882c8aSJames Wright     // Set Outputs
581dd64fc84SJeremy L Thompson     for (CeedInt out = 0; out < num_output_fields; out++) {
582dd64fc84SJeremy L Thompson       CeedVector vec;
583dd64fc84SJeremy L Thompson 
584bd882c8aSJames Wright       // Check if active output
585681d0ea7SJeremy L Thompson       CeedCallBackend(CeedOperatorFieldGetVector(op_output_fields[out], &vec));
586bd882c8aSJames Wright       if (vec == CEED_VECTOR_ACTIVE) {
587dd64fc84SJeremy L Thompson         CeedCallBackend(CeedVectorSetArray(impl->q_vecs_out[out], CEED_MEM_DEVICE, CEED_USE_POINTER, assembled_array));
588dd64fc84SJeremy L Thompson         CeedCallBackend(CeedQFunctionFieldGetSize(qf_output_fields[out], &size));
589dd64fc84SJeremy L Thompson         assembled_array += size * Q * num_elem;  // Advance the pointer by the size of the output
590bd882c8aSJames Wright       }
591681d0ea7SJeremy L Thompson       CeedCallBackend(CeedVectorDestroy(&vec));
592bd882c8aSJames Wright     }
593bd882c8aSJames Wright     // Apply QFunction
594dd64fc84SJeremy L Thompson     CeedCallBackend(CeedQFunctionApply(qf, Q * num_elem, impl->q_vecs_in, impl->q_vecs_out));
595bd882c8aSJames Wright   }
596bd882c8aSJames Wright 
597bd882c8aSJames Wright   // Un-set output Qvecs to prevent accidental overwrite of Assembled
598dd64fc84SJeremy L Thompson   for (CeedInt out = 0; out < num_output_fields; out++) {
599dd64fc84SJeremy L Thompson     CeedVector vec;
600dd64fc84SJeremy L Thompson 
601bd882c8aSJames Wright     // Check if active output
602681d0ea7SJeremy L Thompson     CeedCallBackend(CeedOperatorFieldGetVector(op_output_fields[out], &vec));
603bd882c8aSJames Wright     if (vec == CEED_VECTOR_ACTIVE) {
604dd64fc84SJeremy L Thompson       CeedCallBackend(CeedVectorTakeArray(impl->q_vecs_out[out], CEED_MEM_DEVICE, NULL));
605bd882c8aSJames Wright     }
606681d0ea7SJeremy L Thompson     CeedCallBackend(CeedVectorDestroy(&vec));
607bd882c8aSJames Wright   }
608bd882c8aSJames Wright 
609bd882c8aSJames Wright   // Restore input arrays
610dd64fc84SJeremy L Thompson   CeedCallBackend(CeedOperatorRestoreInputs_Sycl(num_input_fields, qf_input_fields, op_input_fields, true, e_data, impl));
611bd882c8aSJames Wright 
612bd882c8aSJames Wright   // Restore output
613dd64fc84SJeremy L Thompson   CeedCallBackend(CeedVectorRestoreArray(*assembled, &assembled_array));
614bd882c8aSJames Wright   return CEED_ERROR_SUCCESS;
615bd882c8aSJames Wright }
616bd882c8aSJames Wright 
617bd882c8aSJames Wright //------------------------------------------------------------------------------
618bd882c8aSJames Wright // Assemble Linear QFunction
619bd882c8aSJames Wright //------------------------------------------------------------------------------
620*ac2269bdSJeremy L Thompson static int CeedOperatorLinearAssembleQFunction_Sycl(CeedOperator op, CeedVector *assembled, CeedElemRestriction *elem_rstr, CeedRequest *request) {
621*ac2269bdSJeremy L Thompson   return CeedOperatorLinearAssembleQFunctionCore_Sycl(op, true, assembled, elem_rstr, request);
622bd882c8aSJames Wright }
623bd882c8aSJames Wright 
624bd882c8aSJames Wright //------------------------------------------------------------------------------
625bd882c8aSJames Wright // Update Assembled Linear QFunction
626bd882c8aSJames Wright //------------------------------------------------------------------------------
627*ac2269bdSJeremy L Thompson static int CeedOperatorLinearAssembleQFunctionUpdate_Sycl(CeedOperator op, CeedVector assembled, CeedElemRestriction elem_rstr,
628*ac2269bdSJeremy L Thompson                                                           CeedRequest *request) {
629*ac2269bdSJeremy L Thompson   return CeedOperatorLinearAssembleQFunctionCore_Sycl(op, false, &assembled, &elem_rstr, request);
630bd882c8aSJames Wright }
631bd882c8aSJames Wright 
632bd882c8aSJames Wright //------------------------------------------------------------------------------
633bd882c8aSJames Wright // Assemble diagonal setup
634bd882c8aSJames Wright //------------------------------------------------------------------------------
635506b1a0cSSebastian Grimberg static inline int CeedOperatorAssembleDiagonalSetup_Sycl(CeedOperator op) {
636bd882c8aSJames Wright   Ceed                ceed;
637dd64fc84SJeremy L Thompson   Ceed_Sycl          *sycl_data;
638681d0ea7SJeremy L Thompson   CeedInt             num_input_fields, num_output_fields, num_eval_mode_in = 0, num_comp = 0, dim = 1, num_eval_mode_out = 0;
639681d0ea7SJeremy L Thompson   CeedEvalMode       *eval_mode_in = NULL, *eval_mode_out = NULL;
640dd64fc84SJeremy L Thompson   CeedBasis           basis_in = NULL, basis_out = NULL;
641dd64fc84SJeremy L Thompson   CeedQFunctionField *qf_fields;
642bd882c8aSJames Wright   CeedQFunction       qf;
643dd64fc84SJeremy L Thompson   CeedOperatorField  *op_fields;
644dd64fc84SJeremy L Thompson   CeedOperator_Sycl  *impl;
645dd64fc84SJeremy L Thompson 
646dd64fc84SJeremy L Thompson   CeedCallBackend(CeedOperatorGetCeed(op, &ceed));
647bd882c8aSJames Wright   CeedCallBackend(CeedOperatorGetQFunction(op, &qf));
648dd64fc84SJeremy L Thompson   CeedCallBackend(CeedQFunctionGetNumArgs(qf, &num_input_fields, &num_output_fields));
649bd882c8aSJames Wright 
650bd882c8aSJames Wright   // Determine active input basis
651dd64fc84SJeremy L Thompson   CeedCallBackend(CeedOperatorGetFields(op, NULL, &op_fields, NULL, NULL));
652dd64fc84SJeremy L Thompson   CeedCallBackend(CeedQFunctionGetFields(qf, NULL, &qf_fields, NULL, NULL));
653dd64fc84SJeremy L Thompson   for (CeedInt i = 0; i < num_input_fields; i++) {
654bd882c8aSJames Wright     CeedVector vec;
655dd64fc84SJeremy L Thompson 
656dd64fc84SJeremy L Thompson     CeedCallBackend(CeedOperatorFieldGetVector(op_fields[i], &vec));
657bd882c8aSJames Wright     if (vec == CEED_VECTOR_ACTIVE) {
658681d0ea7SJeremy L Thompson       CeedEvalMode eval_mode;
659681d0ea7SJeremy L Thompson       CeedBasis    basis;
660dd64fc84SJeremy L Thompson 
661681d0ea7SJeremy L Thompson       CeedCallBackend(CeedOperatorFieldGetBasis(op_fields[i], &basis));
662681d0ea7SJeremy L Thompson       CeedCheck(!basis_in || basis_in == basis, ceed, CEED_ERROR_BACKEND,
663681d0ea7SJeremy L Thompson                 "Backend does not implement operator diagonal assembly with multiple active bases");
664681d0ea7SJeremy L Thompson       if (!basis_in) CeedCallBackend(CeedBasisReferenceCopy(basis, &basis_in));
665681d0ea7SJeremy L Thompson       CeedCallBackend(CeedBasisDestroy(&basis));
666681d0ea7SJeremy L Thompson       CeedCallBackend(CeedQFunctionFieldGetEvalMode(qf_fields[i], &eval_mode));
667681d0ea7SJeremy L Thompson       switch (eval_mode) {
668bd882c8aSJames Wright         case CEED_EVAL_NONE:
669bd882c8aSJames Wright         case CEED_EVAL_INTERP:
670681d0ea7SJeremy L Thompson           CeedCallBackend(CeedRealloc(num_eval_mode_in + 1, &eval_mode_in));
671681d0ea7SJeremy L Thompson           eval_mode_in[num_eval_mode_in] = eval_mode;
672681d0ea7SJeremy L Thompson           num_eval_mode_in += 1;
673bd882c8aSJames Wright           break;
674bd882c8aSJames Wright         case CEED_EVAL_GRAD:
675681d0ea7SJeremy L Thompson           CeedCallBackend(CeedRealloc(num_eval_mode_in + dim, &eval_mode_in));
676681d0ea7SJeremy L Thompson           for (CeedInt d = 0; d < dim; d++) eval_mode_in[num_eval_mode_in + d] = eval_mode;
677681d0ea7SJeremy L Thompson           num_eval_mode_in += dim;
678bd882c8aSJames Wright           break;
679bd882c8aSJames Wright         case CEED_EVAL_WEIGHT:
680bd882c8aSJames Wright         case CEED_EVAL_DIV:
681bd882c8aSJames Wright         case CEED_EVAL_CURL:
682bd882c8aSJames Wright           break;  // Caught by QF Assembly
683bd882c8aSJames Wright       }
684bd882c8aSJames Wright     }
685681d0ea7SJeremy L Thompson     CeedCallBackend(CeedVectorDestroy(&vec));
686bd882c8aSJames Wright   }
687bd882c8aSJames Wright 
688bd882c8aSJames Wright   // Determine active output basis
689dd64fc84SJeremy L Thompson   CeedCallBackend(CeedOperatorGetFields(op, NULL, NULL, NULL, &op_fields));
690dd64fc84SJeremy L Thompson   CeedCallBackend(CeedQFunctionGetFields(qf, NULL, NULL, NULL, &qf_fields));
691dd64fc84SJeremy L Thompson   for (CeedInt i = 0; i < num_output_fields; i++) {
692bd882c8aSJames Wright     CeedVector vec;
693dd64fc84SJeremy L Thompson 
694dd64fc84SJeremy L Thompson     CeedCallBackend(CeedOperatorFieldGetVector(op_fields[i], &vec));
695bd882c8aSJames Wright     if (vec == CEED_VECTOR_ACTIVE) {
696681d0ea7SJeremy L Thompson       CeedEvalMode eval_mode;
697681d0ea7SJeremy L Thompson       CeedBasis    basis;
698dd64fc84SJeremy L Thompson 
699681d0ea7SJeremy L Thompson       CeedCallBackend(CeedOperatorFieldGetBasis(op_fields[i], &basis));
700681d0ea7SJeremy L Thompson       CeedCheck(!basis_out || basis_out == basis, ceed, CEED_ERROR_BACKEND,
701681d0ea7SJeremy L Thompson                 "Backend does not implement operator diagonal assembly with multiple active bases");
702681d0ea7SJeremy L Thompson       if (!basis_out) CeedCallBackend(CeedBasisReferenceCopy(basis, &basis_out));
703681d0ea7SJeremy L Thompson       CeedCallBackend(CeedBasisDestroy(&basis));
704681d0ea7SJeremy L Thompson       CeedCallBackend(CeedQFunctionFieldGetEvalMode(qf_fields[i], &eval_mode));
705681d0ea7SJeremy L Thompson       switch (eval_mode) {
706bd882c8aSJames Wright         case CEED_EVAL_NONE:
707bd882c8aSJames Wright         case CEED_EVAL_INTERP:
708681d0ea7SJeremy L Thompson           CeedCallBackend(CeedRealloc(num_eval_mode_in + 1, &eval_mode_in));
709681d0ea7SJeremy L Thompson           eval_mode_in[num_eval_mode_in] = eval_mode;
710681d0ea7SJeremy L Thompson           num_eval_mode_in += 1;
711bd882c8aSJames Wright           break;
712bd882c8aSJames Wright         case CEED_EVAL_GRAD:
713681d0ea7SJeremy L Thompson           CeedCallBackend(CeedRealloc(num_eval_mode_in + dim, &eval_mode_in));
714681d0ea7SJeremy L Thompson           for (CeedInt d = 0; d < dim; d++) eval_mode_in[num_eval_mode_in + d] = eval_mode;
715681d0ea7SJeremy L Thompson           num_eval_mode_in += dim;
716bd882c8aSJames Wright           break;
717bd882c8aSJames Wright         case CEED_EVAL_WEIGHT:
718bd882c8aSJames Wright         case CEED_EVAL_DIV:
719bd882c8aSJames Wright         case CEED_EVAL_CURL:
720bd882c8aSJames Wright           break;  // Caught by QF Assembly
721bd882c8aSJames Wright       }
722bd882c8aSJames Wright     }
723681d0ea7SJeremy L Thompson     CeedCallBackend(CeedVectorDestroy(&vec));
724bd882c8aSJames Wright   }
725bd882c8aSJames Wright 
726bd882c8aSJames Wright   // Operator data struct
727bd882c8aSJames Wright   CeedCallBackend(CeedOperatorGetData(op, &impl));
728bd882c8aSJames Wright   CeedCallBackend(CeedGetData(ceed, &sycl_data));
729bd882c8aSJames Wright   CeedCallBackend(CeedCalloc(1, &impl->diag));
730bd882c8aSJames Wright   CeedOperatorDiag_Sycl *diag = impl->diag;
731dd64fc84SJeremy L Thompson 
732dd64fc84SJeremy L Thompson   diag->basis_in          = basis_in;
733dd64fc84SJeremy L Thompson   diag->basis_out         = basis_out;
734681d0ea7SJeremy L Thompson   diag->h_eval_mode_in    = eval_mode_in;
735681d0ea7SJeremy L Thompson   diag->h_eval_mode_out   = eval_mode_out;
736681d0ea7SJeremy L Thompson   diag->num_eval_mode_in  = num_eval_mode_in;
737681d0ea7SJeremy L Thompson   diag->num_eval_mode_out = num_eval_mode_out;
738bd882c8aSJames Wright 
739bd882c8aSJames Wright   // Kernel parameters
740dd64fc84SJeremy L Thompson   CeedInt num_nodes, num_qpts;
741dd64fc84SJeremy L Thompson   CeedCallBackend(CeedBasisGetNumNodes(basis_in, &num_nodes));
742dd64fc84SJeremy L Thompson   CeedCallBackend(CeedBasisGetNumQuadraturePoints(basis_in, &num_qpts));
743dd64fc84SJeremy L Thompson   diag->num_nodes = num_nodes;
744dd64fc84SJeremy L Thompson   diag->num_qpts  = num_qpts;
745dd64fc84SJeremy L Thompson   diag->num_comp  = num_comp;
746bd882c8aSJames Wright 
747bd882c8aSJames Wright   // Basis matrices
748dd64fc84SJeremy L Thompson   const CeedInt     i_len = num_qpts * num_nodes;
749dd64fc84SJeremy L Thompson   const CeedInt     g_len = num_qpts * num_nodes * dim;
750dd64fc84SJeremy L Thompson   const CeedScalar *interp_in, *interp_out, *grad_in, *grad_out;
751bd882c8aSJames Wright 
752bd882c8aSJames Wright   // CEED_EVAL_NONE
753bd882c8aSJames Wright   CeedScalar *identity      = NULL;
754dd64fc84SJeremy L Thompson   bool        has_eval_none = false;
755681d0ea7SJeremy 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);
756681d0ea7SJeremy 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);
757bd882c8aSJames Wright 
7581f4b1b45SUmesh Unnikrishnan   std::vector<sycl::event> e;
7591f4b1b45SUmesh Unnikrishnan 
7601f4b1b45SUmesh Unnikrishnan   if (!sycl_data->sycl_queue.is_in_order()) e = {sycl_data->sycl_queue.ext_oneapi_submit_barrier()};
761bd882c8aSJames Wright 
762bd882c8aSJames Wright   std::vector<sycl::event> copy_events;
7631f4b1b45SUmesh Unnikrishnan 
764dd64fc84SJeremy L Thompson   if (has_eval_none) {
765dd64fc84SJeremy L Thompson     CeedCallBackend(CeedCalloc(num_qpts * num_nodes, &identity));
766dd64fc84SJeremy L Thompson     for (CeedSize i = 0; i < (num_nodes < num_qpts ? num_nodes : num_qpts); i++) identity[i * num_nodes + i] = 1.0;
767dd64fc84SJeremy L Thompson     CeedCallSycl(ceed, diag->d_identity = sycl::malloc_device<CeedScalar>(i_len, sycl_data->sycl_device, sycl_data->sycl_context));
7681f4b1b45SUmesh Unnikrishnan     sycl::event identity_copy = sycl_data->sycl_queue.copy<CeedScalar>(identity, diag->d_identity, i_len, e);
769bd882c8aSJames Wright     copy_events.push_back(identity_copy);
770bd882c8aSJames Wright   }
771bd882c8aSJames Wright 
772bd882c8aSJames Wright   // CEED_EVAL_INTERP
773dd64fc84SJeremy L Thompson   CeedCallBackend(CeedBasisGetInterp(basis_in, &interp_in));
774dd64fc84SJeremy L Thompson   CeedCallSycl(ceed, diag->d_interp_in = sycl::malloc_device<CeedScalar>(i_len, sycl_data->sycl_device, sycl_data->sycl_context));
7751f4b1b45SUmesh Unnikrishnan   sycl::event interp_in_copy = sycl_data->sycl_queue.copy<CeedScalar>(interp_in, diag->d_interp_in, i_len, e);
776dd64fc84SJeremy L Thompson   copy_events.push_back(interp_in_copy);
777bd882c8aSJames Wright 
778dd64fc84SJeremy L Thompson   CeedCallBackend(CeedBasisGetInterp(basis_out, &interp_out));
779dd64fc84SJeremy L Thompson   CeedCallSycl(ceed, diag->d_interp_out = sycl::malloc_device<CeedScalar>(i_len, sycl_data->sycl_device, sycl_data->sycl_context));
7801f4b1b45SUmesh Unnikrishnan   sycl::event interp_out_copy = sycl_data->sycl_queue.copy<CeedScalar>(interp_out, diag->d_interp_out, i_len, e);
781dd64fc84SJeremy L Thompson   copy_events.push_back(interp_out_copy);
782bd882c8aSJames Wright 
783bd882c8aSJames Wright   // CEED_EVAL_GRAD
784dd64fc84SJeremy L Thompson   CeedCallBackend(CeedBasisGetGrad(basis_in, &grad_in));
785dd64fc84SJeremy L Thompson   CeedCallSycl(ceed, diag->d_grad_in = sycl::malloc_device<CeedScalar>(g_len, sycl_data->sycl_device, sycl_data->sycl_context));
7861f4b1b45SUmesh Unnikrishnan   sycl::event grad_in_copy = sycl_data->sycl_queue.copy<CeedScalar>(grad_in, diag->d_grad_in, g_len, e);
787dd64fc84SJeremy L Thompson   copy_events.push_back(grad_in_copy);
788bd882c8aSJames Wright 
789dd64fc84SJeremy L Thompson   CeedCallBackend(CeedBasisGetGrad(basis_out, &grad_out));
790dd64fc84SJeremy L Thompson   CeedCallSycl(ceed, diag->d_grad_out = sycl::malloc_device<CeedScalar>(g_len, sycl_data->sycl_device, sycl_data->sycl_context));
7911f4b1b45SUmesh Unnikrishnan   sycl::event grad_out_copy = sycl_data->sycl_queue.copy<CeedScalar>(grad_out, diag->d_grad_out, g_len, e);
792dd64fc84SJeremy L Thompson   copy_events.push_back(grad_out_copy);
793bd882c8aSJames Wright 
794681d0ea7SJeremy L Thompson   // Arrays of  eval_modes
795681d0ea7SJeremy 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));
796681d0ea7SJeremy 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);
797681d0ea7SJeremy L Thompson   copy_events.push_back(eval_mode_in_copy);
798bd882c8aSJames Wright 
799681d0ea7SJeremy 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));
800681d0ea7SJeremy 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);
801681d0ea7SJeremy L Thompson   copy_events.push_back(eval_mode_out_copy);
802bd882c8aSJames Wright 
803bd882c8aSJames Wright   // Restriction
804681d0ea7SJeremy L Thompson   {
805681d0ea7SJeremy L Thompson     CeedElemRestriction rstr_out;
806681d0ea7SJeremy L Thompson 
807681d0ea7SJeremy L Thompson     CeedCallBackend(CeedOperatorGetActiveElemRestrictions(op, NULL, &rstr_out));
808dd64fc84SJeremy L Thompson     diag->diag_rstr = rstr_out;
809681d0ea7SJeremy L Thompson     CeedCallBackend(CeedElemRestrictionDestroy(&rstr_out));
810681d0ea7SJeremy L Thompson   }
811bd882c8aSJames Wright 
812bd882c8aSJames Wright   // Wait for all copies to complete and handle exceptions
813bd882c8aSJames Wright   CeedCallSycl(ceed, sycl::event::wait_and_throw(copy_events));
814bd882c8aSJames Wright   return CEED_ERROR_SUCCESS;
815bd882c8aSJames Wright }
816bd882c8aSJames Wright 
817bd882c8aSJames Wright //------------------------------------------------------------------------------
818bd882c8aSJames Wright //  Kernel for diagonal assembly
819bd882c8aSJames Wright //------------------------------------------------------------------------------
820dd64fc84SJeremy L Thompson static int CeedOperatorLinearDiagonal_Sycl(sycl::queue &sycl_queue, const bool is_point_block, const CeedInt num_elem,
821dd64fc84SJeremy L Thompson                                            const CeedOperatorDiag_Sycl *diag, const CeedScalar *assembled_qf_array, CeedScalar *elem_diag_array) {
822dd64fc84SJeremy L Thompson   const CeedSize      num_nodes         = diag->num_nodes;
823dd64fc84SJeremy L Thompson   const CeedSize      num_qpts          = diag->num_qpts;
824dd64fc84SJeremy L Thompson   const CeedSize      num_comp          = diag->num_comp;
825681d0ea7SJeremy L Thompson   const CeedSize      num_eval_mode_in  = diag->num_eval_mode_in;
826681d0ea7SJeremy L Thompson   const CeedSize      num_eval_mode_out = diag->num_eval_mode_out;
827bd882c8aSJames Wright   const CeedScalar   *identity          = diag->d_identity;
828dd64fc84SJeremy L Thompson   const CeedScalar   *interp_in         = diag->d_interp_in;
829dd64fc84SJeremy L Thompson   const CeedScalar   *grad_in           = diag->d_grad_in;
830dd64fc84SJeremy L Thompson   const CeedScalar   *interp_out        = diag->d_interp_out;
831dd64fc84SJeremy L Thompson   const CeedScalar   *grad_out          = diag->d_grad_out;
832681d0ea7SJeremy L Thompson   const CeedEvalMode *eval_mode_in      = diag->d_eval_mode_in;
833681d0ea7SJeremy L Thompson   const CeedEvalMode *eval_mode_out     = diag->d_eval_mode_out;
834bd882c8aSJames Wright 
835dd64fc84SJeremy L Thompson   sycl::range<1> kernel_range(num_elem * num_nodes);
836bd882c8aSJames Wright 
8371f4b1b45SUmesh Unnikrishnan   std::vector<sycl::event> e;
8381f4b1b45SUmesh Unnikrishnan 
8391f4b1b45SUmesh Unnikrishnan   if (!sycl_queue.is_in_order()) e = {sycl_queue.ext_oneapi_submit_barrier()};
8401f4b1b45SUmesh Unnikrishnan 
8411f4b1b45SUmesh Unnikrishnan   sycl_queue.parallel_for<CeedOperatorSyclLinearDiagonal>(kernel_range, e, [=](sycl::id<1> idx) {
842dd64fc84SJeremy L Thompson     const CeedInt tid = idx % num_nodes;
843dd64fc84SJeremy L Thompson     const CeedInt e   = idx / num_nodes;
844bd882c8aSJames Wright 
845bd882c8aSJames Wright     // Compute the diagonal of B^T D B
846bd882c8aSJames Wright     // Each element
847dd64fc84SJeremy L Thompson     CeedInt d_out = -1;
848bd882c8aSJames Wright     // Each basis eval mode pair
849681d0ea7SJeremy L Thompson     for (CeedSize e_out = 0; e_out < num_eval_mode_out; e_out++) {
850bd882c8aSJames Wright       const CeedScalar *bt = NULL;
851dd64fc84SJeremy L Thompson 
852681d0ea7SJeremy L Thompson       if (eval_mode_out[e_out] == CEED_EVAL_GRAD) ++d_out;
853681d0ea7SJeremy L Thompson       CeedOperatorGetBasisPointer_Sycl(&bt, eval_mode_out[e_out], identity, interp_out, &grad_out[d_out * num_qpts * num_nodes]);
854dd64fc84SJeremy L Thompson       CeedInt d_in = -1;
855dd64fc84SJeremy L Thompson 
856681d0ea7SJeremy L Thompson       for (CeedSize e_in = 0; e_in < num_eval_mode_in; e_in++) {
857bd882c8aSJames Wright         const CeedScalar *b = NULL;
858dd64fc84SJeremy L Thompson 
859681d0ea7SJeremy L Thompson         if (eval_mode_in[e_in] == CEED_EVAL_GRAD) ++d_in;
860681d0ea7SJeremy L Thompson         CeedOperatorGetBasisPointer_Sycl(&b, eval_mode_in[e_in], identity, interp_in, &grad_in[d_in * num_qpts * num_nodes]);
861bd882c8aSJames Wright         // Each component
862dd64fc84SJeremy L Thompson         for (CeedSize comp_out = 0; comp_out < num_comp; comp_out++) {
863bd882c8aSJames Wright           // Each qpoint/node pair
864dd64fc84SJeremy L Thompson           if (is_point_block) {
865bd882c8aSJames Wright             // Point Block Diagonal
866dd64fc84SJeremy L Thompson             for (CeedInt comp_in = 0; comp_in < num_comp; comp_in++) {
867dd64fc84SJeremy L Thompson               CeedScalar e_value = 0.0;
868dd64fc84SJeremy L Thompson 
869dd64fc84SJeremy L Thompson               for (CeedSize q = 0; q < num_qpts; q++) {
870dd64fc84SJeremy L Thompson                 const CeedScalar qf_value =
871681d0ea7SJeremy 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 +
872dd64fc84SJeremy L Thompson                                        q];
873dd64fc84SJeremy L Thompson 
874dd64fc84SJeremy L Thompson                 e_value += bt[q * num_nodes + tid] * qf_value * b[q * num_nodes + tid];
875bd882c8aSJames Wright               }
876dd64fc84SJeremy L Thompson               elem_diag_array[((comp_out * num_comp + comp_in) * num_elem + e) * num_nodes + tid] += e_value;
877bd882c8aSJames Wright             }
878bd882c8aSJames Wright           } else {
879bd882c8aSJames Wright             // Diagonal Only
880dd64fc84SJeremy L Thompson             CeedScalar e_value = 0.0;
881dd64fc84SJeremy L Thompson 
882dd64fc84SJeremy L Thompson             for (CeedSize q = 0; q < num_qpts; q++) {
883dd64fc84SJeremy L Thompson               const CeedScalar qf_value =
884681d0ea7SJeremy 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 +
885681d0ea7SJeremy L Thompson                                      q];
886dd64fc84SJeremy L Thompson               e_value += bt[q * num_nodes + tid] * qf_value * b[q * num_nodes + tid];
887bd882c8aSJames Wright             }
888dd64fc84SJeremy L Thompson             elem_diag_array[(comp_out * num_elem + e) * num_nodes + tid] += e_value;
889bd882c8aSJames Wright           }
890bd882c8aSJames Wright         }
891bd882c8aSJames Wright       }
892bd882c8aSJames Wright     }
893bd882c8aSJames Wright   });
894bd882c8aSJames Wright   return CEED_ERROR_SUCCESS;
895bd882c8aSJames Wright }
896bd882c8aSJames Wright 
897bd882c8aSJames Wright //------------------------------------------------------------------------------
898bd882c8aSJames Wright // Assemble diagonal common code
899bd882c8aSJames Wright //------------------------------------------------------------------------------
900dd64fc84SJeremy L Thompson static inline int CeedOperatorAssembleDiagonalCore_Sycl(CeedOperator op, CeedVector assembled, CeedRequest *request, const bool is_point_block) {
901bd882c8aSJames Wright   Ceed               ceed;
902bd882c8aSJames Wright   Ceed_Sycl         *sycl_data;
903dd64fc84SJeremy L Thompson   CeedInt            num_elem;
904dd64fc84SJeremy L Thompson   CeedScalar        *elem_diag_array;
905dd64fc84SJeremy L Thompson   const CeedScalar  *assembled_qf_array;
906dd64fc84SJeremy L Thompson   CeedVector         assembled_qf = NULL;
907dd64fc84SJeremy L Thompson   CeedOperator_Sycl *impl;
908dd64fc84SJeremy L Thompson 
909dd64fc84SJeremy L Thompson   CeedCallBackend(CeedOperatorGetCeed(op, &ceed));
910dd64fc84SJeremy L Thompson   CeedCallBackend(CeedOperatorGetData(op, &impl));
911bd882c8aSJames Wright   CeedCallBackend(CeedGetData(ceed, &sycl_data));
912bd882c8aSJames Wright 
913bd882c8aSJames Wright   // Assemble QFunction
914*ac2269bdSJeremy L Thompson   {
915*ac2269bdSJeremy L Thompson     CeedElemRestriction elem_rstr = NULL;
916*ac2269bdSJeremy L Thompson 
917*ac2269bdSJeremy L Thompson     CeedCallBackend(CeedOperatorLinearAssembleQFunctionBuildOrUpdate(op, &assembled_qf, &elem_rstr, request));
918*ac2269bdSJeremy L Thompson     CeedCallBackend(CeedElemRestrictionDestroy(&elem_rstr));
919*ac2269bdSJeremy L Thompson   }
920bd882c8aSJames Wright 
921bd882c8aSJames Wright   // Setup
922bd882c8aSJames Wright   if (!impl->diag) {
923506b1a0cSSebastian Grimberg     CeedCallBackend(CeedOperatorAssembleDiagonalSetup_Sycl(op));
924bd882c8aSJames Wright   }
925bd882c8aSJames Wright   CeedOperatorDiag_Sycl *diag = impl->diag;
926dd64fc84SJeremy L Thompson 
927bd882c8aSJames Wright   assert(diag != NULL);
928bd882c8aSJames Wright 
929bd882c8aSJames Wright   // Restriction
930dd64fc84SJeremy L Thompson   if (is_point_block && !diag->point_block_diag_rstr) {
931506b1a0cSSebastian Grimberg     CeedCallBackend(CeedOperatorCreateActivePointBlockRestriction(diag->diag_rstr, &diag->point_block_diag_rstr));
932bd882c8aSJames Wright   }
933dd64fc84SJeremy L Thompson   CeedElemRestriction diag_rstr = is_point_block ? diag->point_block_diag_rstr : diag->diag_rstr;
934bd882c8aSJames Wright 
935bd882c8aSJames Wright   // Create diagonal vector
936dd64fc84SJeremy L Thompson   CeedVector elem_diag = is_point_block ? diag->point_block_elem_diag : diag->elem_diag;
937dd64fc84SJeremy L Thompson 
938dd64fc84SJeremy L Thompson   if (!elem_diag) {
939dd64fc84SJeremy L Thompson     CeedCallBackend(CeedElemRestrictionCreateVector(diag_rstr, NULL, &elem_diag));
940dd64fc84SJeremy L Thompson     if (is_point_block) diag->point_block_elem_diag = elem_diag;
941dd64fc84SJeremy L Thompson     else diag->elem_diag = elem_diag;
942bd882c8aSJames Wright   }
943dd64fc84SJeremy L Thompson   CeedCallBackend(CeedVectorSetValue(elem_diag, 0.0));
944bd882c8aSJames Wright 
945bd882c8aSJames Wright   // Assemble element operator diagonals
946dd64fc84SJeremy L Thompson   CeedCallBackend(CeedVectorGetArray(elem_diag, CEED_MEM_DEVICE, &elem_diag_array));
947dd64fc84SJeremy L Thompson   CeedCallBackend(CeedVectorGetArrayRead(assembled_qf, CEED_MEM_DEVICE, &assembled_qf_array));
948dd64fc84SJeremy L Thompson   CeedCallBackend(CeedElemRestrictionGetNumElements(diag_rstr, &num_elem));
949bd882c8aSJames Wright 
950bd882c8aSJames Wright   // Compute the diagonal of B^T D B
951dd64fc84SJeremy L Thompson   CeedCallBackend(CeedOperatorLinearDiagonal_Sycl(sycl_data->sycl_queue, is_point_block, num_elem, diag, assembled_qf_array, elem_diag_array));
952bd882c8aSJames Wright 
953bd882c8aSJames Wright   // Wait for queue to complete and handle exceptions
954bd882c8aSJames Wright   sycl_data->sycl_queue.wait_and_throw();
955bd882c8aSJames Wright 
956bd882c8aSJames Wright   // Restore arrays
957dd64fc84SJeremy L Thompson   CeedCallBackend(CeedVectorRestoreArray(elem_diag, &elem_diag_array));
958dd64fc84SJeremy L Thompson   CeedCallBackend(CeedVectorRestoreArrayRead(assembled_qf, &assembled_qf_array));
959bd882c8aSJames Wright 
960bd882c8aSJames Wright   // Assemble local operator diagonal
961dd64fc84SJeremy L Thompson   CeedCallBackend(CeedElemRestrictionApply(diag_rstr, CEED_TRANSPOSE, elem_diag, assembled, request));
962bd882c8aSJames Wright 
963bd882c8aSJames Wright   // Cleanup
964dd64fc84SJeremy L Thompson   CeedCallBackend(CeedVectorDestroy(&assembled_qf));
965bd882c8aSJames Wright   return CEED_ERROR_SUCCESS;
966bd882c8aSJames Wright }
967bd882c8aSJames Wright 
968bd882c8aSJames Wright //------------------------------------------------------------------------------
969bd882c8aSJames Wright // Assemble Linear Diagonal
970bd882c8aSJames Wright //------------------------------------------------------------------------------
971bd882c8aSJames Wright static int CeedOperatorLinearAssembleAddDiagonal_Sycl(CeedOperator op, CeedVector assembled, CeedRequest *request) {
972bd882c8aSJames Wright   CeedCallBackend(CeedOperatorAssembleDiagonalCore_Sycl(op, assembled, request, false));
973bd882c8aSJames Wright   return CEED_ERROR_SUCCESS;
974bd882c8aSJames Wright }
975bd882c8aSJames Wright 
976bd882c8aSJames Wright //------------------------------------------------------------------------------
977bd882c8aSJames Wright // Assemble Linear Point Block Diagonal
978bd882c8aSJames Wright //------------------------------------------------------------------------------
979bd882c8aSJames Wright static int CeedOperatorLinearAssembleAddPointBlockDiagonal_Sycl(CeedOperator op, CeedVector assembled, CeedRequest *request) {
980bd882c8aSJames Wright   CeedCallBackend(CeedOperatorAssembleDiagonalCore_Sycl(op, assembled, request, true));
981bd882c8aSJames Wright   return CEED_ERROR_SUCCESS;
982bd882c8aSJames Wright }
983bd882c8aSJames Wright 
984bd882c8aSJames Wright //------------------------------------------------------------------------------
985bd882c8aSJames Wright // Single operator assembly setup
986bd882c8aSJames Wright //------------------------------------------------------------------------------
987bd882c8aSJames Wright static int CeedSingleOperatorAssembleSetup_Sycl(CeedOperator op) {
988bd882c8aSJames Wright   Ceed    ceed;
989681d0ea7SJeremy 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,
990dd64fc84SJeremy L Thompson                                                num_B_out_mats_to_load = 0, size_B_out = 0, num_qpts = 0, elem_size = 0, num_elem, num_comp,
991dd64fc84SJeremy L Thompson                                                mat_start = 0;
992dd64fc84SJeremy L Thompson   CeedEvalMode       *eval_mode_in = NULL, *eval_mode_out = NULL;
993dd64fc84SJeremy L Thompson   const CeedScalar   *interp_in, *grad_in;
994dd64fc84SJeremy L Thompson   CeedElemRestriction rstr_in = NULL, rstr_out = NULL;
995dd64fc84SJeremy L Thompson   CeedBasis           basis_in = NULL, basis_out = NULL;
996dd64fc84SJeremy L Thompson   CeedQFunctionField *qf_fields;
997dd64fc84SJeremy L Thompson   CeedQFunction       qf;
998dd64fc84SJeremy L Thompson   CeedOperatorField  *input_fields, *output_fields;
999bd882c8aSJames Wright   CeedOperator_Sycl  *impl;
1000dd64fc84SJeremy L Thompson 
1001dd64fc84SJeremy L Thompson   CeedCallBackend(CeedOperatorGetCeed(op, &ceed));
1002bd882c8aSJames Wright   CeedCallBackend(CeedOperatorGetData(op, &impl));
1003bd882c8aSJames Wright 
1004bd882c8aSJames Wright   // Get input and output fields
1005bd882c8aSJames Wright   CeedCallBackend(CeedOperatorGetFields(op, &num_input_fields, &input_fields, &num_output_fields, &output_fields));
1006bd882c8aSJames Wright 
1007bd882c8aSJames Wright   // Determine active input basis eval mode
1008bd882c8aSJames Wright   CeedCallBackend(CeedOperatorGetQFunction(op, &qf));
1009bd882c8aSJames Wright   CeedCallBackend(CeedQFunctionGetFields(qf, NULL, &qf_fields, NULL, NULL));
1010bd882c8aSJames Wright   // Note that the kernel will treat each dimension of a gradient action separately;
1011681d0ea7SJeremy L Thompson   // i.e., when an active input has a CEED_EVAL_GRAD mode, num_ eval_mode_in will increment by dim.
1012681d0ea7SJeremy 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,
1013dd64fc84SJeremy L Thompson   // so num_B_in_mats_to_load will be incremented by 1.
1014bd882c8aSJames Wright   for (CeedInt i = 0; i < num_input_fields; i++) {
1015bd882c8aSJames Wright     CeedVector vec;
1016dd64fc84SJeremy L Thompson 
1017bd882c8aSJames Wright     CeedCallBackend(CeedOperatorFieldGetVector(input_fields[i], &vec));
1018bd882c8aSJames Wright     if (vec == CEED_VECTOR_ACTIVE) {
1019681d0ea7SJeremy L Thompson       CeedEvalMode        eval_mode;
1020681d0ea7SJeremy L Thompson       CeedElemRestriction elem_rstr;
1021681d0ea7SJeremy L Thompson       CeedBasis           basis;
1022681d0ea7SJeremy L Thompson 
1023681d0ea7SJeremy L Thompson       CeedCallBackend(CeedOperatorFieldGetBasis(input_fields[i], &basis));
1024681d0ea7SJeremy L Thompson       CeedCheck(!basis_in || basis_in == basis, ceed, CEED_ERROR_BACKEND, "Backend does not implement operator assembly with multiple active bases");
1025681d0ea7SJeremy L Thompson       if (!basis_in) CeedCallBackend(CeedBasisReferenceCopy(basis, &basis_in));
1026681d0ea7SJeremy L Thompson       CeedCallBackend(CeedBasisDestroy(&basis));
1027bd882c8aSJames Wright       CeedCallBackend(CeedBasisGetDimension(basis_in, &dim));
1028dd64fc84SJeremy L Thompson       CeedCallBackend(CeedBasisGetNumQuadraturePoints(basis_in, &num_qpts));
1029681d0ea7SJeremy L Thompson       CeedCallBackend(CeedOperatorFieldGetElemRestriction(input_fields[i], &elem_rstr));
1030681d0ea7SJeremy L Thompson       if (!rstr_in) CeedCallBackend(CeedElemRestrictionReferenceCopy(elem_rstr, &rstr_in));
1031681d0ea7SJeremy L Thompson       CeedCallBackend(CeedElemRestrictionDestroy(&elem_rstr));
1032dd64fc84SJeremy L Thompson       CeedCallBackend(CeedElemRestrictionGetElementSize(rstr_in, &elem_size));
1033bd882c8aSJames Wright       CeedCallBackend(CeedQFunctionFieldGetEvalMode(qf_fields[i], &eval_mode));
1034bd882c8aSJames Wright       if (eval_mode != CEED_EVAL_NONE) {
1035bd882c8aSJames Wright         CeedCallBackend(CeedRealloc(num_B_in_mats_to_load + 1, &eval_mode_in));
1036bd882c8aSJames Wright         eval_mode_in[num_B_in_mats_to_load] = eval_mode;
1037bd882c8aSJames Wright         num_B_in_mats_to_load += 1;
1038bd882c8aSJames Wright         if (eval_mode == CEED_EVAL_GRAD) {
1039681d0ea7SJeremy L Thompson           num_eval_mode_in += dim;
1040dd64fc84SJeremy L Thompson           size_B_in += dim * elem_size * num_qpts;
1041bd882c8aSJames Wright         } else {
1042681d0ea7SJeremy L Thompson           num_eval_mode_in += 1;
1043dd64fc84SJeremy L Thompson           size_B_in += elem_size * num_qpts;
1044bd882c8aSJames Wright         }
1045bd882c8aSJames Wright       }
1046bd882c8aSJames Wright     }
1047681d0ea7SJeremy L Thompson     CeedCallBackend(CeedVectorDestroy(&vec));
1048bd882c8aSJames Wright   }
1049bd882c8aSJames Wright 
1050bd882c8aSJames Wright   // Determine active output basis; basis_out and rstr_out only used if same as input, TODO
1051bd882c8aSJames Wright   CeedCallBackend(CeedQFunctionGetFields(qf, NULL, NULL, NULL, &qf_fields));
1052bd882c8aSJames Wright   for (CeedInt i = 0; i < num_output_fields; i++) {
1053bd882c8aSJames Wright     CeedVector vec;
1054dd64fc84SJeremy L Thompson 
1055bd882c8aSJames Wright     CeedCallBackend(CeedOperatorFieldGetVector(output_fields[i], &vec));
1056bd882c8aSJames Wright     if (vec == CEED_VECTOR_ACTIVE) {
1057681d0ea7SJeremy L Thompson       CeedEvalMode        eval_mode;
1058681d0ea7SJeremy L Thompson       CeedElemRestriction elem_rstr;
1059681d0ea7SJeremy L Thompson       CeedBasis           basis;
1060681d0ea7SJeremy L Thompson 
1061681d0ea7SJeremy L Thompson       CeedCallBackend(CeedOperatorFieldGetBasis(output_fields[i], &basis));
1062681d0ea7SJeremy L Thompson       CeedCheck(!basis_out || basis_out == basis, ceed, CEED_ERROR_BACKEND,
1063681d0ea7SJeremy L Thompson                 "Backend does not implement operator assembly with multiple active bases");
1064681d0ea7SJeremy L Thompson       if (!basis_out) CeedCallBackend(CeedBasisReferenceCopy(basis, &basis_out));
1065681d0ea7SJeremy L Thompson       CeedCallBackend(CeedBasisDestroy(&basis));
1066681d0ea7SJeremy L Thompson       CeedCallBackend(CeedOperatorFieldGetElemRestriction(output_fields[i], &elem_rstr));
1067681d0ea7SJeremy L Thompson       if (!rstr_out) CeedCallBackend(CeedElemRestrictionReferenceCopy(elem_rstr, &rstr_out));
1068681d0ea7SJeremy L Thompson       CeedCallBackend(CeedElemRestrictionDestroy(&elem_rstr));
1069bd882c8aSJames Wright       CeedCallBackend(CeedQFunctionFieldGetEvalMode(qf_fields[i], &eval_mode));
1070bd882c8aSJames Wright       if (eval_mode != CEED_EVAL_NONE) {
1071bd882c8aSJames Wright         CeedCallBackend(CeedRealloc(num_B_out_mats_to_load + 1, &eval_mode_out));
1072bd882c8aSJames Wright         eval_mode_out[num_B_out_mats_to_load] = eval_mode;
1073bd882c8aSJames Wright         num_B_out_mats_to_load += 1;
1074bd882c8aSJames Wright         if (eval_mode == CEED_EVAL_GRAD) {
1075681d0ea7SJeremy L Thompson           num_eval_mode_out += dim;
1076dd64fc84SJeremy L Thompson           size_B_out += dim * elem_size * num_qpts;
1077bd882c8aSJames Wright         } else {
1078681d0ea7SJeremy L Thompson           num_eval_mode_out += 1;
1079dd64fc84SJeremy L Thompson           size_B_out += elem_size * num_qpts;
1080bd882c8aSJames Wright         }
1081bd882c8aSJames Wright       }
1082bd882c8aSJames Wright     }
1083681d0ea7SJeremy L Thompson     CeedCallBackend(CeedVectorDestroy(&vec));
1084bd882c8aSJames Wright   }
1085681d0ea7SJeremy L Thompson   CeedCheck(num_eval_mode_in > 0 && num_eval_mode_out > 0, ceed, CEED_ERROR_UNSUPPORTED, "Cannot assemble operator without inputs/outputs");
1086bd882c8aSJames Wright 
1087dd64fc84SJeremy L Thompson   CeedCallBackend(CeedElemRestrictionGetNumElements(rstr_in, &num_elem));
1088dd64fc84SJeremy L Thompson   CeedCallBackend(CeedElemRestrictionGetNumComponents(rstr_in, &num_comp));
1089bd882c8aSJames Wright 
1090bd882c8aSJames Wright   CeedCallBackend(CeedCalloc(1, &impl->asmb));
1091bd882c8aSJames Wright   CeedOperatorAssemble_Sycl *asmb = impl->asmb;
1092dd64fc84SJeremy L Thompson   asmb->num_elem                  = num_elem;
1093bd882c8aSJames Wright 
1094bd882c8aSJames Wright   Ceed_Sycl *sycl_data;
1095bd882c8aSJames Wright   CeedCallBackend(CeedGetData(ceed, &sycl_data));
1096bd882c8aSJames Wright 
1097bd882c8aSJames Wright   // Kernel setup
1098004e4986SSebastian Grimberg   int elems_per_block     = 1;
1099004e4986SSebastian Grimberg   asmb->elems_per_block   = elems_per_block;
1100dd64fc84SJeremy L Thompson   asmb->block_size_x      = elem_size;
1101dd64fc84SJeremy L Thompson   asmb->block_size_y      = elem_size;
1102681d0ea7SJeremy L Thompson   asmb->num_eval_mode_in  = num_eval_mode_in;
1103681d0ea7SJeremy L Thompson   asmb->num_eval_mode_out = num_eval_mode_out;
1104dd64fc84SJeremy L Thompson   asmb->num_qpts          = num_qpts;
1105dd64fc84SJeremy L Thompson   asmb->num_nodes         = elem_size;
11065700671aSSebastian Grimberg   asmb->block_size        = elem_size * elem_size * elems_per_block;
1107dd64fc84SJeremy L Thompson   asmb->num_comp          = num_comp;
1108bd882c8aSJames Wright 
1109bd882c8aSJames Wright   // Build 'full' B matrices (not 1D arrays used for tensor-product matrices
1110bd882c8aSJames Wright   CeedCallBackend(CeedBasisGetInterp(basis_in, &interp_in));
1111bd882c8aSJames Wright   CeedCallBackend(CeedBasisGetGrad(basis_in, &grad_in));
1112bd882c8aSJames Wright 
1113bd882c8aSJames Wright   // Load into B_in, in order that they will be used in eval_mode
1114bd882c8aSJames Wright   CeedCallSycl(ceed, asmb->d_B_in = sycl::malloc_device<CeedScalar>(size_B_in, sycl_data->sycl_device, sycl_data->sycl_context));
1115bd882c8aSJames Wright   for (int i = 0; i < num_B_in_mats_to_load; i++) {
1116bd882c8aSJames Wright     CeedEvalMode eval_mode = eval_mode_in[i];
1117dd64fc84SJeremy L Thompson 
1118bd882c8aSJames Wright     if (eval_mode == CEED_EVAL_INTERP) {
11191f4b1b45SUmesh Unnikrishnan       std::vector<sycl::event> e;
11201f4b1b45SUmesh Unnikrishnan 
11211f4b1b45SUmesh Unnikrishnan       if (!sycl_data->sycl_queue.is_in_order()) e = {sycl_data->sycl_queue.ext_oneapi_submit_barrier()};
11221f4b1b45SUmesh Unnikrishnan       sycl_data->sycl_queue.copy<CeedScalar>(interp_in, &asmb->d_B_in[mat_start], elem_size * num_qpts, e);
1123dd64fc84SJeremy L Thompson       mat_start += elem_size * num_qpts;
1124bd882c8aSJames Wright     } else if (eval_mode == CEED_EVAL_GRAD) {
11251f4b1b45SUmesh Unnikrishnan       std::vector<sycl::event> e;
11261f4b1b45SUmesh Unnikrishnan 
11271f4b1b45SUmesh Unnikrishnan       if (!sycl_data->sycl_queue.is_in_order()) e = {sycl_data->sycl_queue.ext_oneapi_submit_barrier()};
11281f4b1b45SUmesh Unnikrishnan       sycl_data->sycl_queue.copy<CeedScalar>(grad_in, &asmb->d_B_in[mat_start], dim * elem_size * num_qpts, e);
1129dd64fc84SJeremy L Thompson       mat_start += dim * elem_size * num_qpts;
1130bd882c8aSJames Wright     }
1131bd882c8aSJames Wright   }
1132bd882c8aSJames Wright 
1133bd882c8aSJames Wright   const CeedScalar *interp_out, *grad_out;
1134bd882c8aSJames Wright   // Note that this function currently assumes 1 basis, so this should always be true
1135bd882c8aSJames Wright   // for now
1136bd882c8aSJames Wright   if (basis_out == basis_in) {
1137bd882c8aSJames Wright     interp_out = interp_in;
1138bd882c8aSJames Wright     grad_out   = grad_in;
1139bd882c8aSJames Wright   } else {
1140bd882c8aSJames Wright     CeedCallBackend(CeedBasisGetInterp(basis_out, &interp_out));
1141bd882c8aSJames Wright     CeedCallBackend(CeedBasisGetGrad(basis_out, &grad_out));
1142bd882c8aSJames Wright   }
1143bd882c8aSJames Wright 
1144bd882c8aSJames Wright   // Load into B_out, in order that they will be used in eval_mode
1145bd882c8aSJames Wright   mat_start = 0;
1146bd882c8aSJames Wright   CeedCallSycl(ceed, asmb->d_B_out = sycl::malloc_device<CeedScalar>(size_B_out, sycl_data->sycl_device, sycl_data->sycl_context));
1147bd882c8aSJames Wright   for (int i = 0; i < num_B_out_mats_to_load; i++) {
1148bd882c8aSJames Wright     CeedEvalMode eval_mode = eval_mode_out[i];
1149dd64fc84SJeremy L Thompson 
1150bd882c8aSJames Wright     if (eval_mode == CEED_EVAL_INTERP) {
11511f4b1b45SUmesh Unnikrishnan       std::vector<sycl::event> e;
11521f4b1b45SUmesh Unnikrishnan 
11531f4b1b45SUmesh Unnikrishnan       if (!sycl_data->sycl_queue.is_in_order()) e = {sycl_data->sycl_queue.ext_oneapi_submit_barrier()};
11541f4b1b45SUmesh Unnikrishnan       sycl_data->sycl_queue.copy<CeedScalar>(interp_out, &asmb->d_B_out[mat_start], elem_size * num_qpts, e);
1155dd64fc84SJeremy L Thompson       mat_start += elem_size * num_qpts;
1156bd882c8aSJames Wright     } else if (eval_mode == CEED_EVAL_GRAD) {
11571f4b1b45SUmesh Unnikrishnan       std::vector<sycl::event> e;
11581f4b1b45SUmesh Unnikrishnan 
11591f4b1b45SUmesh Unnikrishnan       if (!sycl_data->sycl_queue.is_in_order()) e = {sycl_data->sycl_queue.ext_oneapi_submit_barrier()};
11601f4b1b45SUmesh Unnikrishnan       sycl_data->sycl_queue.copy<CeedScalar>(grad_out, &asmb->d_B_out[mat_start], dim * elem_size * num_qpts, e);
1161dd64fc84SJeremy L Thompson       mat_start += dim * elem_size * num_qpts;
1162bd882c8aSJames Wright     }
1163bd882c8aSJames Wright   }
1164681d0ea7SJeremy L Thompson   CeedCallBackend(CeedElemRestrictionDestroy(&rstr_in));
1165681d0ea7SJeremy L Thompson   CeedCallBackend(CeedElemRestrictionDestroy(&rstr_out));
1166681d0ea7SJeremy L Thompson   CeedCallBackend(CeedBasisDestroy(&basis_in));
1167681d0ea7SJeremy L Thompson   CeedCallBackend(CeedBasisDestroy(&basis_out));
1168bd882c8aSJames Wright   return CEED_ERROR_SUCCESS;
1169bd882c8aSJames Wright }
1170bd882c8aSJames Wright 
1171bd882c8aSJames Wright //------------------------------------------------------------------------------
1172bd882c8aSJames Wright // Matrix assembly kernel for low-order elements (3D thread block)
1173bd882c8aSJames Wright //------------------------------------------------------------------------------
1174bd882c8aSJames Wright static int CeedOperatorLinearAssemble_Sycl(sycl::queue &sycl_queue, const CeedOperator_Sycl *impl, const CeedScalar *qf_array,
1175bd882c8aSJames Wright                                            CeedScalar *values_array) {
1176bd882c8aSJames Wright   // This kernels assumes B_in and B_out have the same number of quadrature points and basis points.
1177bd882c8aSJames Wright   // TODO: expand to more general cases
1178bd882c8aSJames Wright   CeedOperatorAssemble_Sycl *asmb              = impl->asmb;
1179dd64fc84SJeremy L Thompson   const CeedInt              num_elem          = asmb->num_elem;
1180dd64fc84SJeremy L Thompson   const CeedSize             num_nodes         = asmb->num_nodes;
1181dd64fc84SJeremy L Thompson   const CeedSize             num_comp          = asmb->num_comp;
1182dd64fc84SJeremy L Thompson   const CeedSize             num_qpts          = asmb->num_qpts;
1183681d0ea7SJeremy L Thompson   const CeedSize             num_eval_mode_in  = asmb->num_eval_mode_in;
1184681d0ea7SJeremy L Thompson   const CeedSize             num_eval_mode_out = asmb->num_eval_mode_out;
1185bd882c8aSJames Wright 
1186bd882c8aSJames Wright   // Strides for final output ordering, determined by the reference (inference) implementation of the symbolic assembly, slowest --> fastest: element,
1187bd882c8aSJames Wright   // comp_in, comp_out, node_row, node_col
1188dd64fc84SJeremy L Thompson   const CeedSize comp_out_stride = num_nodes * num_nodes;
1189dd64fc84SJeremy L Thompson   const CeedSize comp_in_stride  = comp_out_stride * num_comp;
1190dd64fc84SJeremy L Thompson   const CeedSize e_stride        = comp_in_stride * num_comp;
1191681d0ea7SJeremy L Thompson   // Strides for QF array, slowest --> fastest:  eval_mode_in, comp_in,  eval_mode_out, comp_out, elem, qpt
1192dd64fc84SJeremy L Thompson   const CeedSize q_e_stride             = num_qpts;
1193dd64fc84SJeremy L Thompson   const CeedSize q_comp_out_stride      = num_elem * q_e_stride;
1194681d0ea7SJeremy L Thompson   const CeedSize q_eval_mode_out_stride = q_comp_out_stride * num_comp;
1195681d0ea7SJeremy L Thompson   const CeedSize q_comp_in_stride       = q_eval_mode_out_stride * num_eval_mode_out;
1196681d0ea7SJeremy L Thompson   const CeedSize q_eval_mode_in_stride  = q_comp_in_stride * num_comp;
1197bd882c8aSJames Wright 
1198bd882c8aSJames Wright   CeedScalar *B_in, *B_out;
1199bd882c8aSJames Wright   B_in                       = asmb->d_B_in;
1200bd882c8aSJames Wright   B_out                      = asmb->d_B_out;
1201bd882c8aSJames Wright   const CeedInt block_size_x = asmb->block_size_x;
1202bd882c8aSJames Wright   const CeedInt block_size_y = asmb->block_size_y;
1203bd882c8aSJames Wright 
1204dd64fc84SJeremy L Thompson   sycl::range<3> kernel_range(num_elem, block_size_y, block_size_x);
1205bd882c8aSJames Wright 
12061f4b1b45SUmesh Unnikrishnan   std::vector<sycl::event> e;
12071f4b1b45SUmesh Unnikrishnan 
12081f4b1b45SUmesh Unnikrishnan   if (!sycl_queue.is_in_order()) e = {sycl_queue.ext_oneapi_submit_barrier()};
12091f4b1b45SUmesh Unnikrishnan   sycl_queue.parallel_for<CeedOperatorSyclLinearAssemble>(kernel_range, e, [=](sycl::id<3> idx) {
1210bd882c8aSJames Wright     const int e = idx.get(0);  // Element index
1211bd882c8aSJames Wright     const int l = idx.get(1);  // The output column index of each B^TDB operation
1212bd882c8aSJames Wright     const int i = idx.get(2);  // The output row index of each B^TDB operation
1213bd882c8aSJames Wright                                // such that we have (Bout^T)_ij D_jk Bin_kl = C_il
1214dd64fc84SJeremy L Thompson     for (CeedSize comp_in = 0; comp_in < num_comp; comp_in++) {
1215dd64fc84SJeremy L Thompson       for (CeedSize comp_out = 0; comp_out < num_comp; comp_out++) {
1216bd882c8aSJames Wright         CeedScalar result        = 0.0;
1217dd64fc84SJeremy L Thompson         CeedSize   qf_index_comp = q_comp_in_stride * comp_in + q_comp_out_stride * comp_out + q_e_stride * e;
1218dd64fc84SJeremy L Thompson 
1219681d0ea7SJeremy L Thompson         for (CeedSize eval_mode_in = 0; eval_mode_in < num_eval_mode_in; eval_mode_in++) {
1220681d0ea7SJeremy L Thompson           CeedSize b_in_index = eval_mode_in * num_qpts * num_nodes;
1221dd64fc84SJeremy L Thompson 
1222681d0ea7SJeremy L Thompson           for (CeedSize eval_mode_out = 0; eval_mode_out < num_eval_mode_out; eval_mode_out++) {
1223681d0ea7SJeremy L Thompson             CeedSize b_out_index = eval_mode_out * num_qpts * num_nodes;
1224681d0ea7SJeremy L Thompson             CeedSize qf_index    = qf_index_comp + q_eval_mode_out_stride * eval_mode_out + q_eval_mode_in_stride * eval_mode_in;
1225dd64fc84SJeremy L Thompson 
1226bd882c8aSJames Wright             // Perform the B^T D B operation for this 'chunk' of D (the qf_array)
1227dd64fc84SJeremy L Thompson             for (CeedSize j = 0; j < num_qpts; j++) {
1228dd64fc84SJeremy 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];
1229bd882c8aSJames Wright             }
1230681d0ea7SJeremy L Thompson           }  // end of  eval_mode_out
1231681d0ea7SJeremy L Thompson         }  // end of  eval_mode_in
1232dd64fc84SJeremy L Thompson         CeedSize val_index = comp_in_stride * comp_in + comp_out_stride * comp_out + e_stride * e + num_nodes * i + l;
1233dd64fc84SJeremy L Thompson 
1234bd882c8aSJames Wright         values_array[val_index] = result;
1235bd882c8aSJames Wright       }  // end of out component
1236bd882c8aSJames Wright     }  // end of in component
1237bd882c8aSJames Wright   });
1238bd882c8aSJames Wright   return CEED_ERROR_SUCCESS;
1239bd882c8aSJames Wright }
1240bd882c8aSJames Wright 
1241bd882c8aSJames Wright //------------------------------------------------------------------------------
1242bd882c8aSJames Wright // Fallback kernel for larger orders (1D thread block)
1243bd882c8aSJames Wright //------------------------------------------------------------------------------
1244bd882c8aSJames Wright /*
1245bd882c8aSJames Wright static int CeedOperatorLinearAssembleFallback_Sycl(sycl::queue &sycl_queue, const CeedOperator_Sycl *impl, const CeedScalar *qf_array,
1246bd882c8aSJames Wright                                                    CeedScalar *values_array) {
1247bd882c8aSJames Wright   // This kernel assumes B_in and B_out have the same number of quadrature points and basis points.
1248bd882c8aSJames Wright   // TODO: expand to more general cases
1249bd882c8aSJames Wright   CeedOperatorAssemble_Sycl *asmb        = impl->asmb;
1250dd64fc84SJeremy L Thompson   const CeedInt              num_elem       = asmb->num_elem;
1251dd64fc84SJeremy L Thompson   const CeedInt              num_nodes      = asmb->num_nodes;
1252dd64fc84SJeremy L Thompson   const CeedInt              num_comp       = asmb->num_comp;
1253dd64fc84SJeremy L Thompson   const CeedInt              num_qpts       = asmb->num_qpts;
1254681d0ea7SJeremy L Thompson   const CeedInt              num_eval_mode_in  = asmb->num_eval_mode_in;
1255681d0ea7SJeremy L Thompson   const CeedInt              num_eval_mode_out = asmb->num_eval_mode_out;
1256bd882c8aSJames Wright 
1257bd882c8aSJames Wright   // Strides for final output ordering, determined by the reference (interface) implementation of the symbolic assembly, slowest --> fastest: elememt,
1258bd882c8aSJames Wright   // comp_in, comp_out, node_row, node_col
1259dd64fc84SJeremy L Thompson   const CeedInt comp_out_stride = num_nodes * num_nodes;
1260dd64fc84SJeremy L Thompson   const CeedInt comp_in_stride  = comp_out_stride * num_comp;
1261dd64fc84SJeremy L Thompson   const CeedInt e_stride        = comp_in_stride * num_comp;
1262681d0ea7SJeremy L Thompson   // Strides for QF array, slowest --> fastest:  eval_mode_in, comp_in,  eval_mode_out, comp_out, elem, qpt
1263dd64fc84SJeremy L Thompson   const CeedInt q_e_stride         = num_qpts;
1264dd64fc84SJeremy L Thompson   const CeedInt q_comp_out_stride  = num_elem * q_e_stride;
1265681d0ea7SJeremy L Thompson   const CeedInt q_eval_mode_out_stride = q_comp_out_stride * num_comp;
1266681d0ea7SJeremy L Thompson   const CeedInt q_comp_in_stride   = q_eval_mode_out_stride * num_eval_mode_out;
1267681d0ea7SJeremy L Thompson   const CeedInt q_eval_mode_in_stride  = q_comp_in_stride * num_comp;
1268bd882c8aSJames Wright 
1269bd882c8aSJames Wright   CeedScalar *B_in, *B_out;
1270bd882c8aSJames Wright   B_in                        = asmb->d_B_in;
1271bd882c8aSJames Wright   B_out                       = asmb->d_B_out;
1272004e4986SSebastian Grimberg   const CeedInt elems_per_block = asmb->elems_per_block;
1273bd882c8aSJames Wright   const CeedInt block_size_x  = asmb->block_size_x;
1274bd882c8aSJames Wright   const CeedInt block_size_y  = asmb->block_size_y;  // This will be 1 for the fallback kernel
1275bd882c8aSJames Wright 
1276004e4986SSebastian Grimberg   const CeedInt     grid = num_elem / elems_per_block + ((num_elem / elems_per_block * elems_per_block < num_elem) ? 1 : 0);
1277004e4986SSebastian Grimberg   sycl::range<3>    local_range(block_size_x, block_size_y, elems_per_block);
1278004e4986SSebastian Grimberg   sycl::range<3>    global_range(grid * block_size_x, block_size_y, elems_per_block);
1279bd882c8aSJames Wright   sycl::nd_range<3> kernel_range(global_range, local_range);
1280bd882c8aSJames Wright 
1281bd882c8aSJames Wright   sycl_queue.parallel_for<CeedOperatorSyclLinearAssembleFallback>(kernel_range, [=](sycl::nd_item<3> work_item) {
1282bd882c8aSJames Wright     const CeedInt blockIdx  = work_item.get_group(0);
1283bd882c8aSJames Wright     const CeedInt gridDimx  = work_item.get_group_range(0);
1284bd882c8aSJames Wright     const CeedInt threadIdx = work_item.get_local_id(0);
1285bd882c8aSJames Wright     const CeedInt threadIdz = work_item.get_local_id(2);
1286bd882c8aSJames Wright     const CeedInt blockDimz = work_item.get_local_range(2);
1287bd882c8aSJames Wright 
1288bd882c8aSJames Wright     const int l = threadIdx;  // The output column index of each B^TDB operation
1289bd882c8aSJames Wright                               // such that we have (Bout^T)_ij D_jk Bin_kl = C_il
1290dd64fc84SJeremy L Thompson     for (CeedInt e = blockIdx * blockDimz + threadIdz; e < num_elem; e += gridDimx * blockDimz) {
1291dd64fc84SJeremy L Thompson       for (CeedInt comp_in = 0; comp_in < num_comp; comp_in++) {
1292dd64fc84SJeremy L Thompson         for (CeedInt comp_out = 0; comp_out < num_comp; comp_out++) {
1293dd64fc84SJeremy L Thompson           for (CeedInt i = 0; i < num_nodes; i++) {
1294bd882c8aSJames Wright             CeedScalar result        = 0.0;
1295dd64fc84SJeremy L Thompson             CeedInt    qf_index_comp = q_comp_in_stride * comp_in + q_comp_out_stride * comp_out + q_e_stride * e;
1296681d0ea7SJeremy L Thompson             for (CeedInt  eval_mode_in = 0;  eval_mode_in < num_eval_mode_in;  eval_mode_in++) {
1297681d0ea7SJeremy L Thompson               CeedInt b_in_index =  eval_mode_in * num_qpts * num_nodes;
1298681d0ea7SJeremy L Thompson               for (CeedInt  eval_mode_out = 0;  eval_mode_out < num_eval_mode_out;  eval_mode_out++) {
1299681d0ea7SJeremy L Thompson                 CeedInt b_out_index =  eval_mode_out * num_qpts * num_nodes;
1300681d0ea7SJeremy L Thompson                 CeedInt qf_index    = qf_index_comp + q_eval_mode_out_stride *  eval_mode_out + q_eval_mode_in_stride *  eval_mode_in;
1301bd882c8aSJames Wright                 // Perform the B^T D B operation for this 'chunk' of D (the qf_array)
1302dd64fc84SJeremy L Thompson                 for (CeedInt j = 0; j < num_qpts; j++) {
1303dd64fc84SJeremy 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];
1304bd882c8aSJames Wright                 }
1305681d0ea7SJeremy L Thompson               }  // end of  eval_mode_out
1306681d0ea7SJeremy L Thompson             }    // end of  eval_mode_in
1307dd64fc84SJeremy L Thompson             CeedInt val_index       = comp_in_stride * comp_in + comp_out_stride * comp_out + e_stride * e + num_nodes * i + l;
1308bd882c8aSJames Wright             values_array[val_index] = result;
1309bd882c8aSJames Wright           }  // end of loop over element node index, i
1310bd882c8aSJames Wright         }    // end of out component
1311bd882c8aSJames Wright       }      // end of in component
1312bd882c8aSJames Wright     }        // end of element loop
1313bd882c8aSJames Wright   });
1314bd882c8aSJames Wright   return CEED_ERROR_SUCCESS;
1315bd882c8aSJames Wright }*/
1316bd882c8aSJames Wright 
1317bd882c8aSJames Wright //------------------------------------------------------------------------------
1318bd882c8aSJames Wright // Assemble matrix data for COO matrix of assembled operator.
1319bd882c8aSJames Wright // The sparsity pattern is set by CeedOperatorLinearAssembleSymbolic.
1320bd882c8aSJames Wright //
1321bd882c8aSJames Wright // Note that this (and other assembly routines) currently assume only one active
1322bd882c8aSJames Wright // input restriction/basis per operator (could have multiple basis eval modes).
1323bd882c8aSJames Wright // TODO: allow multiple active input restrictions/basis objects
1324bd882c8aSJames Wright //------------------------------------------------------------------------------
1325bd882c8aSJames Wright static int CeedSingleOperatorAssemble_Sycl(CeedOperator op, CeedInt offset, CeedVector values) {
1326bd882c8aSJames Wright   Ceed                ceed;
1327bd882c8aSJames Wright   Ceed_Sycl          *sycl_data;
1328dd64fc84SJeremy L Thompson   CeedScalar         *values_array;
1329dd64fc84SJeremy L Thompson   const CeedScalar   *qf_array;
1330dd64fc84SJeremy L Thompson   CeedVector          assembled_qf = NULL;
1331dd64fc84SJeremy L Thompson   CeedElemRestriction rstr_q       = NULL;
1332dd64fc84SJeremy L Thompson   CeedOperator_Sycl  *impl;
1333dd64fc84SJeremy L Thompson 
1334dd64fc84SJeremy L Thompson   CeedCallBackend(CeedOperatorGetCeed(op, &ceed));
1335dd64fc84SJeremy L Thompson   CeedCallBackend(CeedOperatorGetData(op, &impl));
1336bd882c8aSJames Wright   CeedCallBackend(CeedGetData(ceed, &sycl_data));
1337bd882c8aSJames Wright 
1338bd882c8aSJames Wright   // Setup
1339bd882c8aSJames Wright   if (!impl->asmb) {
1340bd882c8aSJames Wright     CeedCallBackend(CeedSingleOperatorAssembleSetup_Sycl(op));
1341bd882c8aSJames Wright     assert(impl->asmb != NULL);
1342bd882c8aSJames Wright   }
1343bd882c8aSJames Wright 
1344bd882c8aSJames Wright   // Assemble QFunction
1345bd882c8aSJames Wright   CeedCallBackend(CeedOperatorLinearAssembleQFunctionBuildOrUpdate(op, &assembled_qf, &rstr_q, CEED_REQUEST_IMMEDIATE));
1346bd882c8aSJames Wright   CeedCallBackend(CeedElemRestrictionDestroy(&rstr_q));
1347bd882c8aSJames Wright   CeedCallBackend(CeedVectorGetArrayWrite(values, CEED_MEM_DEVICE, &values_array));
1348bd882c8aSJames Wright   values_array += offset;
1349bd882c8aSJames Wright   CeedCallBackend(CeedVectorGetArrayRead(assembled_qf, CEED_MEM_DEVICE, &qf_array));
1350bd882c8aSJames Wright 
1351bd882c8aSJames Wright   // Compute B^T D B
1352bd882c8aSJames Wright   CeedCallBackend(CeedOperatorLinearAssemble_Sycl(sycl_data->sycl_queue, impl, qf_array, values_array));
1353bd882c8aSJames Wright 
1354bd882c8aSJames Wright   // Wait for kernels to be completed
1355bd882c8aSJames Wright   // Kris: Review if this is necessary -- enqueing an async barrier may be sufficient
1356bd882c8aSJames Wright   sycl_data->sycl_queue.wait_and_throw();
1357bd882c8aSJames Wright 
1358bd882c8aSJames Wright   // Restore arrays
1359bd882c8aSJames Wright   CeedCallBackend(CeedVectorRestoreArray(values, &values_array));
1360bd882c8aSJames Wright   CeedCallBackend(CeedVectorRestoreArrayRead(assembled_qf, &qf_array));
1361bd882c8aSJames Wright 
1362bd882c8aSJames Wright   // Cleanup
1363bd882c8aSJames Wright   CeedCallBackend(CeedVectorDestroy(&assembled_qf));
1364bd882c8aSJames Wright   return CEED_ERROR_SUCCESS;
1365bd882c8aSJames Wright }
1366bd882c8aSJames Wright 
1367bd882c8aSJames Wright //------------------------------------------------------------------------------
1368bd882c8aSJames Wright // Create operator
1369bd882c8aSJames Wright //------------------------------------------------------------------------------
1370bd882c8aSJames Wright int CeedOperatorCreate_Sycl(CeedOperator op) {
1371bd882c8aSJames Wright   Ceed               ceed;
1372bd882c8aSJames Wright   CeedOperator_Sycl *impl;
1373bd882c8aSJames Wright 
1374dd64fc84SJeremy L Thompson   CeedCallBackend(CeedOperatorGetCeed(op, &ceed));
1375dd64fc84SJeremy L Thompson 
1376bd882c8aSJames Wright   CeedCallBackend(CeedCalloc(1, &impl));
1377bd882c8aSJames Wright   CeedCallBackend(CeedOperatorSetData(op, impl));
1378bd882c8aSJames Wright 
1379bd882c8aSJames Wright   CeedCallBackend(CeedSetBackendFunctionCpp(ceed, "Operator", op, "LinearAssembleQFunction", CeedOperatorLinearAssembleQFunction_Sycl));
1380bd882c8aSJames Wright   CeedCallBackend(CeedSetBackendFunctionCpp(ceed, "Operator", op, "LinearAssembleQFunctionUpdate", CeedOperatorLinearAssembleQFunctionUpdate_Sycl));
1381bd882c8aSJames Wright   CeedCallBackend(CeedSetBackendFunctionCpp(ceed, "Operator", op, "LinearAssembleAddDiagonal", CeedOperatorLinearAssembleAddDiagonal_Sycl));
1382bd882c8aSJames Wright   CeedCallBackend(
1383bd882c8aSJames Wright       CeedSetBackendFunctionCpp(ceed, "Operator", op, "LinearAssembleAddPointBlockDiagonal", CeedOperatorLinearAssembleAddPointBlockDiagonal_Sycl));
1384bd882c8aSJames Wright   CeedCallBackend(CeedSetBackendFunctionCpp(ceed, "Operator", op, "LinearAssembleSingle", CeedSingleOperatorAssemble_Sycl));
1385bd882c8aSJames Wright   CeedCallBackend(CeedSetBackendFunctionCpp(ceed, "Operator", op, "ApplyAdd", CeedOperatorApplyAdd_Sycl));
1386bd882c8aSJames Wright   CeedCallBackend(CeedSetBackendFunctionCpp(ceed, "Operator", op, "Destroy", CeedOperatorDestroy_Sycl));
1387bd882c8aSJames Wright   return CEED_ERROR_SUCCESS;
1388bd882c8aSJames Wright }
1389bd882c8aSJames Wright 
1390bd882c8aSJames Wright //------------------------------------------------------------------------------
1391