From 9fddab1ac911df326d9ff1d7bcc1dbf29d84aef5 Mon Sep 17 00:00:00 2001
From: Jeremy L Thompson <jeremy@jeremylt.org>
Date: Tue, 17 Sep 2024 15:57:44 -0600
Subject: [PATCH] sycl - more

---
 backends/sycl-ref/ceed-sycl-ref-operator.sycl.cpp | 12 +++---------
 1 file changed, 3 insertions(+), 9 deletions(-)

diff --git a/backends/sycl-ref/ceed-sycl-ref-operator.sycl.cpp b/backends/sycl-ref/ceed-sycl-ref-operator.sycl.cpp
index 72ae1bc02b..8debf68db5 100644
--- a/backends/sycl-ref/ceed-sycl-ref-operator.sycl.cpp
+++ b/backends/sycl-ref/ceed-sycl-ref-operator.sycl.cpp
@@ -673,7 +673,6 @@ static inline int CeedOperatorAssembleDiagonalSetup_Sycl(CeedOperator op) {
                 "Backend does not implement operator diagonal assembly with multiple active bases");
       if (!basis_in) CeedCallBackend(CeedBasisReferenceCopy(basis, &basis_in));
       CeedCallBackend(CeedBasisDestroy(&basis));
-      CeedCallBackend(CeedBasisGetNumComponents(basis_in, &num_comp));
       CeedCallBackend(CeedBasisGetDimension(basis_in, &dim));
       CeedCallBackend(CeedQFunctionFieldGetEvalMode(qf_fields[i], &eval_mode));
       switch (eval_mode) {
@@ -741,7 +740,6 @@ static inline int CeedOperatorAssembleDiagonalSetup_Sycl(CeedOperator op) {
     }
     CeedCallBackend(CeedVectorDestroy(&vec));
   }
-  CeedCallBackend(CeedElemRestrictionDestroy(&rstr_out));
 
   // Operator data struct
   CeedCallBackend(CeedOperatorGetData(op, &impl));
@@ -760,6 +758,7 @@ static inline int CeedOperatorAssembleDiagonalSetup_Sycl(CeedOperator op) {
   CeedInt num_nodes, num_qpts;
   CeedCallBackend(CeedBasisGetNumNodes(basis_in, &num_nodes));
   CeedCallBackend(CeedBasisGetNumQuadraturePoints(basis_in, &num_qpts));
+  CeedCallBackend(CeedBasisGetNumComponents(basis_in, &num_comp));
   diag->num_nodes = num_nodes;
   diag->num_qpts  = num_qpts;
   diag->num_comp  = num_comp;
@@ -821,13 +820,8 @@ static inline int CeedOperatorAssembleDiagonalSetup_Sycl(CeedOperator op) {
   copy_events.push_back(eval_mode_out_copy);
 
   // Restriction
-  {
-    CeedElemRestriction rstr_out;
-
-    CeedCallBackend(CeedOperatorGetActiveElemRestrictions(op, NULL, &rstr_out));
-    CeedCallBackend(CeedElemRestrictionReferenceCopy(rstr_out, &diag->diag_rstr));
-    CeedCallBackend(CeedElemRestrictionDestroy(&rstr_out));
-  }
+  CeedCallBackend(CeedElemRestrictionReferenceCopy(rstr_out, &diag->diag_rstr));
+  CeedCallBackend(CeedElemRestrictionDestroy(&rstr_out));
 
   // Cleanup
   CeedCallBackend(CeedBasisDestroy(&basis_in));