xref: /libCEED/include/ceed/backend.h (revision 53ef2869b49ed7dd91aa8ced0c36f3ed2fcf2517)
1 // Copyright (c) 2017-2022, Lawrence Livermore National Security, LLC and other CEED contributors.
2 // All Rights Reserved. See the top-level LICENSE and NOTICE files for details.
3 //
4 // SPDX-License-Identifier: BSD-2-Clause
5 //
6 // This file is part of CEED:  http://github.com/ceed
7 
8 /// @file
9 /// Public header for backend components of libCEED
10 #ifndef CEED_BACKEND_H
11 #define CEED_BACKEND_H
12 
13 #include <ceed.h>
14 #include <limits.h>
15 #include <stdbool.h>
16 
17 #if defined(__clang_analyzer__)
18 #define CEED_INTERN
19 #elif defined(__cplusplus)
20 #define CEED_INTERN extern "C" CEED_VISIBILITY(hidden)
21 #else
22 #define CEED_INTERN extern CEED_VISIBILITY(hidden)
23 #endif
24 
25 #define CEED_UNUSED __attribute__((unused))
26 
27 #define CEED_MAX_RESOURCE_LEN 1024
28 #define CEED_MAX_BACKEND_PRIORITY UINT_MAX
29 #define CEED_ALIGN 64
30 #define CEED_COMPOSITE_MAX 16
31 #define CEED_FIELD_MAX 16
32 
33 #ifndef CeedPragmaOptimizeOff
34 #if defined(__clang__)
35 /// This macro provides the ability to disable optimization flags for functions that are sensitive to floating point optimizations.
36 /// @ingroup Ceed
37 #define CeedPragmaOptimizeOff _Pragma("clang optimize off")
38 #elif defined(__GNUC__)
39 #define CeedPragmaOptimizeOff _Pragma("GCC push_options") _Pragma("GCC optimize 0")
40 #elif defined(__INTEL_COMPILER) || defined(__INTEL_LLVM_COMPILER)
41 #define CeedPragmaOptimizeOff _Pragma("optimize('', off)")
42 #else
43 #define CeedPragmaOptimizeOff
44 #endif
45 #endif
46 
47 #ifndef CeedPragmaOptimizeOn
48 #if defined(__clang__)
49 /// This macro restores previously set optimization flags after CeedPragmaOptimizeOff.
50 /// @ingroup Ceed
51 #define CeedPragmaOptimizeOn _Pragma("clang optimize on")
52 #elif defined(__GNUC__)
53 #define CeedPragmaOptimizeOn _Pragma("GCC pop_options")
54 #elif defined(__INTEL_COMPILER) || defined(__INTEL_LLVM_COMPILER)
55 #define CeedPragmaOptimizeOff _Pragma("optimize('', on)")
56 #else
57 #define CeedPragmaOptimizeOn
58 #endif
59 #endif
60 
61 /**
62   This enum supples common colors for CeedDebug256 debugging output.
63   Set the environment variable `CEED_DEBUG = 1` to activate debugging output.
64 
65   @ingroup Ceed
66   @ref     Backend
67 **/
68 typedef enum {
69   /// Success color
70   CEED_DEBUG_COLOR_SUCCESS = 108,
71   /// Warning color
72   CEED_DEBUG_COLOR_WARNING = 208,
73   /// Error color
74   CEED_DEBUG_COLOR_ERROR = 196,
75   /// Use native terminal coloring
76   CEED_DEBUG_COLOR_NONE = 255,
77 } CeedDebugColor;
78 
79 CEED_EXTERN void CeedDebugImpl256(const unsigned char, const char *, ...);
80 CEED_EXTERN bool CeedDebugFlag(const Ceed ceed);
81 CEED_EXTERN bool CeedDebugFlagEnv(void);
82 /**
83   Print debugging information in color
84 
85   @param[in] ceed  Ceed
86   @param[in] color Color to print with
87 
88   @ingroup Ceed
89   @ref     Backend
90 **/
91 #define CeedDebug256(ceed, color, ...)                               \
92   {                                                                  \
93     if (CeedDebugFlag(ceed)) CeedDebugImpl256(color, ##__VA_ARGS__); \
94   }
95 /**
96   Print debugging information to terminal
97 
98   @param[in] ceed  Ceed
99 
100   @ingroup Ceed
101   @ref     Backend
102 **/
103 #define CeedDebug(ceed, ...) CeedDebug256(ceed, (unsigned char)CEED_DEBUG_COLOR_NONE, ##__VA_ARGS__)
104 /**
105   Print debugging information in color without Ceed to reference
106 
107   @param[in] color Color to print with
108 
109   @ingroup Ceed
110   @ref     Backend
111 **/
112 #define CeedDebugEnv256(color, ...)                                 \
113   {                                                                 \
114     if (CeedDebugFlagEnv()) CeedDebugImpl256(color, ##__VA_ARGS__); \
115   }
116 /**
117   Print debugging information to terminal without Ceed to reference
118 
119   @ingroup Ceed
120   @ref     Backend
121 **/
122 #define CeedDebugEnv(...) CeedDebugEnv256((unsigned char)CEED_DEBUG_COLOR_NONE, ##__VA_ARGS__)
123 /**
124   Print warning information in color
125 
126   @ingroup Ceed
127   @ref     Backend
128 **/
129 #define CeedWarn(...) \
130   { CeedDebugImpl256(CEED_DEBUG_COLOR_WARNING, ##__VA_ARGS__); }
131 
132 /**
133   Swap the values of two CeedScalars
134 
135   @param[in,out] a  First CeedScalar
136   @param[in,out] b  Second CeedScalar
137 
138   @ingroup Ceed
139   @ref     Backend
140 **/
141 #define CeedScalarSwap(a, b)        \
142   {                                 \
143     const CeedScalar temp_ = a;     \
144     a                      = b;     \
145     b                      = temp_; \
146   }
147 
148 /// Handle for object handling TensorContraction
149 /// @ingroup CeedBasis
150 typedef struct CeedTensorContract_private *CeedTensorContract;
151 
152 /// Handle for object handling assembled QFunction data
153 /// @ingroup CeedOperator
154 typedef struct CeedQFunctionAssemblyData_private *CeedQFunctionAssemblyData;
155 
156 /// Handle for object handling assembled Operator data
157 /// @ingroup CeedOperator
158 typedef struct CeedOperatorAssemblyData_private *CeedOperatorAssemblyData;
159 
160 /* In the next 3 functions, p has to be the address of a pointer type, i.e. p has to be a pointer to a pointer. */
161 CEED_INTERN int CeedMallocArray(size_t n, size_t unit, void *p);
162 CEED_INTERN int CeedCallocArray(size_t n, size_t unit, void *p);
163 CEED_INTERN int CeedReallocArray(size_t n, size_t unit, void *p);
164 CEED_INTERN int CeedStringAllocCopy(const char *source, char **copy);
165 CEED_INTERN int CeedFree(void *p);
166 
167 /**
168   @brief Calls a libCEED function and then checks the resulting error code.
169   If the error code is non-zero, then the error handler is called and the call from the current function with the error code.
170 
171   @ref Developer
172 **/
173 #define CeedCall(...)        \
174   do {                       \
175     int ierr_ = __VA_ARGS__; \
176     if (ierr_) return ierr_; \
177   } while (0)
178 
179 /**
180   @brief Calls a libCEED function and then checks the resulting error code.
181   If the error code is non-zero, then the error handler is called and the call from the current function with the error code.
182   All interface level error codes are upgraded to `CEED_ERROR_BACKEND`.
183 
184   @ref Developer
185 **/
186 #define CeedCallBackend(...)                                                     \
187   do {                                                                           \
188     int ierr_ = __VA_ARGS__;                                                     \
189     if (ierr_) return (ierr_ > CEED_ERROR_SUCCESS) ? CEED_ERROR_BACKEND : ierr_; \
190   } while (0)
191 
192 /**
193   @brief Check that a particular condition is true and returns a `CeedError` if not.
194 
195   @ref Developer
196 **/
197 #define CeedCheck(cond, ceed, ecode, ...)                    \
198   do {                                                       \
199     if (!(cond)) return CeedError(ceed, ecode, __VA_ARGS__); \
200   } while (0)
201 
202 /* Note - these are legacy macros that should be removed eventually */
203 #define CeedChk(...) CeedCall(__VA_ARGS__)
204 #define CeedChkBackend(...) CeedCallBackend(__VA_ARGS__)
205 
206 /* Note that CeedMalloc and CeedCalloc will, generally, return pointers with different memory alignments:
207    CeedMalloc returns pointers aligned at CEED_ALIGN bytes, while CeedCalloc uses the alignment of calloc. */
208 #define CeedMalloc(n, p) CeedMallocArray((n), sizeof(**(p)), p)
209 #define CeedCalloc(n, p) CeedCallocArray((n), sizeof(**(p)), p)
210 #define CeedRealloc(n, p) CeedReallocArray((n), sizeof(**(p)), p)
211 
212 CEED_EXTERN int CeedRegister(const char *prefix, int (*init)(const char *, Ceed), unsigned int priority);
213 CEED_EXTERN int CeedRegisterImpl(const char *prefix, int (*init)(const char *, Ceed), unsigned int priority);
214 
215 CEED_EXTERN int CeedIsDebug(Ceed ceed, bool *is_debug);
216 CEED_EXTERN int CeedGetResourceRoot(Ceed ceed, const char *resource, const char *delineator, char **resource_root);
217 CEED_EXTERN int CeedGetParent(Ceed ceed, Ceed *parent);
218 CEED_EXTERN int CeedGetDelegate(Ceed ceed, Ceed *delegate);
219 CEED_EXTERN int CeedSetDelegate(Ceed ceed, Ceed delegate);
220 CEED_EXTERN int CeedGetObjectDelegate(Ceed ceed, Ceed *delegate, const char *obj_name);
221 CEED_EXTERN int CeedSetObjectDelegate(Ceed ceed, Ceed delegate, const char *obj_name);
222 CEED_EXTERN int CeedGetOperatorFallbackResource(Ceed ceed, const char **resource);
223 CEED_EXTERN int CeedGetOperatorFallbackCeed(Ceed ceed, Ceed *fallback_ceed);
224 CEED_EXTERN int CeedSetOperatorFallbackResource(Ceed ceed, const char *resource);
225 CEED_EXTERN int CeedSetDeterministic(Ceed ceed, bool is_deterministic);
226 CEED_EXTERN int CeedSetBackendFunction(Ceed ceed, const char *type, void *object, const char *func_name, int (*f)());
227 CEED_EXTERN int CeedGetData(Ceed ceed, void *data);
228 CEED_EXTERN int CeedSetData(Ceed ceed, void *data);
229 CEED_EXTERN int CeedReference(Ceed ceed);
230 
231 CEED_EXTERN int CeedVectorHasValidArray(CeedVector vec, bool *has_valid_array);
232 CEED_EXTERN int CeedVectorHasBorrowedArrayOfType(CeedVector vec, CeedMemType mem_type, bool *has_borrowed_array_of_type);
233 CEED_EXTERN int CeedVectorHasValidArray(CeedVector vec, bool *has_valid_array);
234 CEED_EXTERN int CeedVectorGetState(CeedVector vec, uint64_t *state);
235 CEED_EXTERN int CeedVectorGetData(CeedVector vec, void *data);
236 CEED_EXTERN int CeedVectorSetData(CeedVector vec, void *data);
237 CEED_EXTERN int CeedVectorReference(CeedVector vec);
238 
239 /// Type of element restriction;
240 /// @ingroup CeedElemRestriction
241 typedef enum {
242   /// Standard element restriction with offsets
243   CEED_RESTRICTION_STANDARD = 1,
244   /// Oriented element restriction
245   CEED_RESTRICTION_ORIENTED = 2,
246   /// Curl-oriented element restriction
247   CEED_RESTRICTION_CURL_ORIENTED = 3,
248   /// Strided element restriction
249   CEED_RESTRICTION_STRIDED = 4,
250 } CeedRestrictionType;
251 
252 CEED_EXTERN int CeedElemRestrictionGetType(CeedElemRestriction rstr, CeedRestrictionType *rstr_type);
253 CEED_EXTERN int CeedElemRestrictionIsStrided(CeedElemRestriction rstr, bool *is_strided);
254 CEED_EXTERN int CeedElemRestrictionGetStrides(CeedElemRestriction rstr, CeedInt (*strides)[3]);
255 CEED_EXTERN int CeedElemRestrictionHasBackendStrides(CeedElemRestriction rstr, bool *has_backend_strides);
256 CEED_EXTERN int CeedElemRestrictionGetOffsets(CeedElemRestriction rstr, CeedMemType mem_type, const CeedInt **offsets);
257 CEED_EXTERN int CeedElemRestrictionRestoreOffsets(CeedElemRestriction rstr, const CeedInt **offsets);
258 CEED_EXTERN int CeedElemRestrictionGetOrientations(CeedElemRestriction rstr, CeedMemType mem_type, const bool **orients);
259 CEED_EXTERN int CeedElemRestrictionRestoreOrientations(CeedElemRestriction rstr, const bool **orients);
260 CEED_EXTERN int CeedElemRestrictionGetCurlOrientations(CeedElemRestriction rstr, CeedMemType mem_type, const CeedInt8 **curl_orients);
261 CEED_EXTERN int CeedElemRestrictionRestoreCurlOrientations(CeedElemRestriction rstr, const CeedInt8 **curl_orients);
262 CEED_EXTERN int CeedElemRestrictionGetELayout(CeedElemRestriction rstr, CeedInt (*layout)[3]);
263 CEED_EXTERN int CeedElemRestrictionSetELayout(CeedElemRestriction rstr, CeedInt layout[3]);
264 CEED_EXTERN int CeedElemRestrictionGetData(CeedElemRestriction rstr, void *data);
265 CEED_EXTERN int CeedElemRestrictionSetData(CeedElemRestriction rstr, void *data);
266 CEED_EXTERN int CeedElemRestrictionReference(CeedElemRestriction rstr);
267 CEED_EXTERN int CeedElemRestrictionGetFlopsEstimate(CeedElemRestriction rstr, CeedTransposeMode t_mode, CeedSize *flops);
268 
269 /// Type of FE space;
270 /// @ingroup CeedBasis
271 typedef enum {
272   /// H^1 FE space
273   CEED_FE_SPACE_H1 = 1,
274   /// H(div) FE space
275   CEED_FE_SPACE_HDIV = 2,
276   /// H(curl) FE space
277   CEED_FE_SPACE_HCURL = 3,
278 } CeedFESpace;
279 CEED_EXTERN const char *const CeedFESpaces[];
280 
281 CEED_EXTERN int CeedBasisGetCollocatedGrad(CeedBasis basis, CeedScalar *colo_grad_1d);
282 CEED_EXTERN int CeedBasisIsTensor(CeedBasis basis, bool *is_tensor);
283 CEED_EXTERN int CeedBasisGetData(CeedBasis basis, void *data);
284 CEED_EXTERN int CeedBasisSetData(CeedBasis basis, void *data);
285 CEED_EXTERN int CeedBasisReference(CeedBasis basis);
286 CEED_EXTERN int CeedBasisGetNumQuadratureComponents(CeedBasis basis, CeedEvalMode eval_mode, CeedInt *q_comp);
287 CEED_EXTERN int CeedBasisGetFlopsEstimate(CeedBasis basis, CeedTransposeMode t_mode, CeedEvalMode eval_mode, CeedSize *flops);
288 CEED_EXTERN int CeedBasisGetFESpace(CeedBasis basis, CeedFESpace *fe_space);
289 CEED_EXTERN int CeedBasisGetTopologyDimension(CeedElemTopology topo, CeedInt *dim);
290 CEED_EXTERN int CeedBasisGetTensorContract(CeedBasis basis, CeedTensorContract *contract);
291 CEED_EXTERN int CeedBasisSetTensorContract(CeedBasis basis, CeedTensorContract contract);
292 
293 CEED_EXTERN int CeedTensorContractCreate(Ceed ceed, CeedBasis basis, CeedTensorContract *contract);
294 CEED_EXTERN int CeedTensorContractApply(CeedTensorContract contract, CeedInt A, CeedInt B, CeedInt C, CeedInt J, const CeedScalar *__restrict__ t,
295                                         CeedTransposeMode t_mode, const CeedInt Add, const CeedScalar *__restrict__ u, CeedScalar *__restrict__ v);
296 CEED_EXTERN int CeedTensorContractStridedApply(CeedTensorContract contract, CeedInt A, CeedInt B, CeedInt C, CeedInt D, CeedInt J,
297                                                const CeedScalar *__restrict__ t, CeedTransposeMode t_mode, const CeedInt add,
298                                                const CeedScalar *__restrict__ u, CeedScalar *__restrict__ v);
299 CEED_EXTERN int CeedTensorContractGetCeed(CeedTensorContract contract, Ceed *ceed);
300 CEED_EXTERN int CeedTensorContractGetData(CeedTensorContract contract, void *data);
301 CEED_EXTERN int CeedTensorContractSetData(CeedTensorContract contract, void *data);
302 CEED_EXTERN int CeedTensorContractReference(CeedTensorContract contract);
303 CEED_EXTERN int CeedTensorContractReferenceCopy(CeedTensorContract tensor, CeedTensorContract *tensor_copy);
304 CEED_EXTERN int CeedTensorContractDestroy(CeedTensorContract *contract);
305 
306 CEED_EXTERN int CeedQFunctionRegister(const char *name, const char *source, CeedInt vec_length, CeedQFunctionUser f,
307                                       int (*init)(Ceed, const char *, CeedQFunction));
308 CEED_EXTERN int CeedQFunctionSetFortranStatus(CeedQFunction qf, bool status);
309 CEED_EXTERN int CeedQFunctionGetVectorLength(CeedQFunction qf, CeedInt *vec_length);
310 CEED_EXTERN int CeedQFunctionGetNumArgs(CeedQFunction qf, CeedInt *num_input_fields, CeedInt *num_output_fields);
311 CEED_EXTERN int CeedQFunctionGetKernelName(CeedQFunction qf, char **kernel_name);
312 CEED_EXTERN int CeedQFunctionGetSourcePath(CeedQFunction qf, char **source_path);
313 CEED_EXTERN int CeedQFunctionLoadSourceToBuffer(CeedQFunction qf, char **source_buffer);
314 CEED_EXTERN int CeedQFunctionGetUserFunction(CeedQFunction qf, CeedQFunctionUser *f);
315 CEED_EXTERN int CeedQFunctionGetContext(CeedQFunction qf, CeedQFunctionContext *ctx);
316 CEED_EXTERN int CeedQFunctionGetContextData(CeedQFunction qf, CeedMemType mem_type, void *data);
317 CEED_EXTERN int CeedQFunctionRestoreContextData(CeedQFunction qf, void *data);
318 CEED_EXTERN int CeedQFunctionGetInnerContext(CeedQFunction qf, CeedQFunctionContext *ctx);
319 CEED_EXTERN int CeedQFunctionGetInnerContextData(CeedQFunction qf, CeedMemType mem_type, void *data);
320 CEED_EXTERN int CeedQFunctionRestoreInnerContextData(CeedQFunction qf, void *data);
321 CEED_EXTERN int CeedQFunctionIsIdentity(CeedQFunction qf, bool *is_identity);
322 CEED_EXTERN int CeedQFunctionIsContextWritable(CeedQFunction qf, bool *is_writable);
323 CEED_EXTERN int CeedQFunctionGetData(CeedQFunction qf, void *data);
324 CEED_EXTERN int CeedQFunctionSetData(CeedQFunction qf, void *data);
325 CEED_EXTERN int CeedQFunctionReference(CeedQFunction qf);
326 CEED_EXTERN int CeedQFunctionGetFlopsEstimate(CeedQFunction qf, CeedSize *flops);
327 
328 CEED_EXTERN int CeedQFunctionContextGetCeed(CeedQFunctionContext ctx, Ceed *ceed);
329 CEED_EXTERN int CeedQFunctionContextHasValidData(CeedQFunctionContext ctx, bool *has_valid_data);
330 CEED_EXTERN int CeedQFunctionContextHasBorrowedDataOfType(CeedQFunctionContext ctx, CeedMemType mem_type, bool *has_borrowed_data_of_type);
331 CEED_EXTERN int CeedQFunctionContextGetState(CeedQFunctionContext ctx, uint64_t *state);
332 CEED_EXTERN int CeedQFunctionContextGetBackendData(CeedQFunctionContext ctx, void *data);
333 CEED_EXTERN int CeedQFunctionContextSetBackendData(CeedQFunctionContext ctx, void *data);
334 CEED_EXTERN int CeedQFunctionContextGetFieldLabel(CeedQFunctionContext ctx, const char *field_name, CeedContextFieldLabel *field_label);
335 CEED_EXTERN int CeedQFunctionContextSetGeneric(CeedQFunctionContext ctx, CeedContextFieldLabel field_label, CeedContextFieldType field_type,
336                                                void *value);
337 CEED_EXTERN int CeedQFunctionContextGetGenericRead(CeedQFunctionContext ctx, CeedContextFieldLabel field_label, CeedContextFieldType field_type,
338                                                    size_t *num_values, void *value);
339 CEED_EXTERN int CeedQFunctionContextRestoreGenericRead(CeedQFunctionContext ctx, CeedContextFieldLabel field_label, CeedContextFieldType field_type,
340                                                        void *value);
341 CEED_EXTERN int CeedQFunctionContextSetDouble(CeedQFunctionContext ctx, CeedContextFieldLabel field_label, double *values);
342 CEED_EXTERN int CeedQFunctionContextGetDoubleRead(CeedQFunctionContext ctx, CeedContextFieldLabel field_label, size_t *num_values,
343                                                   const double **values);
344 CEED_EXTERN int CeedQFunctionContextRestoreDoubleRead(CeedQFunctionContext ctx, CeedContextFieldLabel field_label, const double **values);
345 CEED_EXTERN int CeedQFunctionContextSetInt32(CeedQFunctionContext ctx, CeedContextFieldLabel field_label, int *values);
346 CEED_EXTERN int CeedQFunctionContextGetInt32Read(CeedQFunctionContext ctx, CeedContextFieldLabel field_label, size_t *num_values, const int **values);
347 CEED_EXTERN int CeedQFunctionContextRestoreInt32Read(CeedQFunctionContext ctx, CeedContextFieldLabel field_label, const int **values);
348 CEED_EXTERN int CeedQFunctionContextGetDataDestroy(CeedQFunctionContext ctx, CeedMemType *f_mem_type, CeedQFunctionContextDataDestroyUser *f);
349 CEED_EXTERN int CeedQFunctionContextReference(CeedQFunctionContext ctx);
350 
351 CEED_EXTERN int CeedQFunctionAssemblyDataCreate(Ceed ceed, CeedQFunctionAssemblyData *data);
352 CEED_EXTERN int CeedQFunctionAssemblyDataReference(CeedQFunctionAssemblyData data);
353 CEED_EXTERN int CeedQFunctionAssemblyDataSetReuse(CeedQFunctionAssemblyData data, bool reuse_assembly_data);
354 CEED_EXTERN int CeedQFunctionAssemblyDataSetUpdateNeeded(CeedQFunctionAssemblyData data, bool needs_data_update);
355 CEED_EXTERN int CeedQFunctionAssemblyDataIsUpdateNeeded(CeedQFunctionAssemblyData data, bool *is_update_needed);
356 CEED_EXTERN int CeedQFunctionAssemblyDataReferenceCopy(CeedQFunctionAssemblyData data, CeedQFunctionAssemblyData *data_copy);
357 CEED_EXTERN int CeedQFunctionAssemblyDataIsSetup(CeedQFunctionAssemblyData data, bool *is_setup);
358 CEED_EXTERN int CeedQFunctionAssemblyDataSetObjects(CeedQFunctionAssemblyData data, CeedVector vec, CeedElemRestriction rstr);
359 CEED_EXTERN int CeedQFunctionAssemblyDataGetObjects(CeedQFunctionAssemblyData data, CeedVector *vec, CeedElemRestriction *rstr);
360 CEED_EXTERN int CeedQFunctionAssemblyDataDestroy(CeedQFunctionAssemblyData *data);
361 
362 CEED_EXTERN int CeedOperatorAssemblyDataCreate(Ceed ceed, CeedOperator op, CeedOperatorAssemblyData *data);
363 CEED_EXTERN int CeedOperatorAssemblyDataGetEvalModes(CeedOperatorAssemblyData data, CeedInt *num_active_bases, CeedInt **num_eval_modes_in,
364                                                      const CeedEvalMode ***eval_modes_in, CeedSize ***eval_mode_offsets_in,
365                                                      CeedInt **num_eval_modes_out, const CeedEvalMode ***eval_modes_out,
366                                                      CeedSize ***eval_mode_offsets_out, CeedSize *num_output_components);
367 CEED_EXTERN int CeedOperatorAssemblyDataGetBases(CeedOperatorAssemblyData data, CeedInt *num_active_bases, CeedBasis **active_bases,
368                                                  const CeedScalar ***assembled_bases_in, const CeedScalar ***assembled_bases_out);
369 CEED_EXTERN int CeedOperatorAssemblyDataGetElemRestrictions(CeedOperatorAssemblyData data, CeedInt *num_active_elem_rstrs,
370                                                             CeedElemRestriction **active_elem_rstrs);
371 CEED_EXTERN int CeedOperatorAssemblyDataDestroy(CeedOperatorAssemblyData *data);
372 
373 CEED_EXTERN int CeedOperatorGetOperatorAssemblyData(CeedOperator op, CeedOperatorAssemblyData *data);
374 CEED_EXTERN int CeedOperatorGetActiveBasis(CeedOperator op, CeedBasis *active_basis);
375 CEED_EXTERN int CeedOperatorGetActiveElemRestriction(CeedOperator op, CeedElemRestriction *active_rstr);
376 CEED_EXTERN int CeedOperatorGetNumArgs(CeedOperator op, CeedInt *num_args);
377 CEED_EXTERN int CeedOperatorIsSetupDone(CeedOperator op, bool *is_setup_done);
378 CEED_EXTERN int CeedOperatorGetQFunction(CeedOperator op, CeedQFunction *qf);
379 CEED_EXTERN int CeedOperatorIsComposite(CeedOperator op, bool *is_composite);
380 CEED_EXTERN int CeedOperatorGetData(CeedOperator op, void *data);
381 CEED_EXTERN int CeedOperatorSetData(CeedOperator op, void *data);
382 CEED_EXTERN int CeedOperatorReference(CeedOperator op);
383 CEED_EXTERN int CeedOperatorGetFallback(CeedOperator op, CeedOperator *op_fallback);
384 CEED_EXTERN int CeedOperatorGetFallbackParent(CeedOperator op, CeedOperator *parent);
385 CEED_EXTERN int CeedOperatorGetFallbackParentCeed(CeedOperator op, Ceed *parent);
386 CEED_EXTERN int CeedOperatorSetSetupDone(CeedOperator op);
387 
388 CEED_INTERN int CeedMatrixMatrixMultiply(Ceed ceed, const CeedScalar *mat_A, const CeedScalar *mat_B, CeedScalar *mat_C, CeedInt m, CeedInt n,
389                                          CeedInt kk);
390 CEED_EXTERN int CeedQRFactorization(Ceed ceed, CeedScalar *mat, CeedScalar *tau, CeedInt m, CeedInt n);
391 CEED_EXTERN int CeedHouseholderApplyQ(CeedScalar *mat_A, const CeedScalar *mat_Q, const CeedScalar *tau, CeedTransposeMode t_mode, CeedInt m,
392                                       CeedInt n, CeedInt k, CeedInt row, CeedInt col);
393 CEED_EXTERN int CeedSymmetricSchurDecomposition(Ceed ceed, CeedScalar *mat, CeedScalar *lambda, CeedInt n);
394 CEED_EXTERN int CeedSimultaneousDiagonalization(Ceed ceed, CeedScalar *mat_A, CeedScalar *mat_B, CeedScalar *x, CeedScalar *lambda, CeedInt n);
395 
396 #endif  // CEED_BACKEND_H
397