xref: /libCEED/rust/libceed-sys/c-src/backends/memcheck/ceed-memcheck.h (revision 3345ba21ea3223800ed4bc1b9a4502bd60b77d9f)
15aed82e4SJeremy L Thompson // Copyright (c) 2017-2024, Lawrence Livermore National Security, LLC and other CEED contributors.
23d8e8822SJeremy L Thompson // All Rights Reserved. See the top-level LICENSE and NOTICE files for details.
3fc7cf9a0Sjeremylt //
43d8e8822SJeremy L Thompson // SPDX-License-Identifier: BSD-2-Clause
5fc7cf9a0Sjeremylt //
63d8e8822SJeremy L Thompson // This file is part of CEED:  http://github.com/ceed
7509d4af6SJeremy L Thompson #pragma once
83d576824SJeremy L Thompson 
949aac155SJeremy L Thompson #include <ceed.h>
10ec3da8bcSJed Brown #include <ceed/backend.h>
11fc7cf9a0Sjeremylt 
12fc7cf9a0Sjeremylt typedef struct {
139937a20eSJeremy L Thompson   // Internal array buffer
149937a20eSJeremy L Thompson   int         allocated_block_id;
159798701eSJeremy L Thompson   CeedScalar *array_allocated;
169937a20eSJeremy L Thompson   // Owned external array
179937a20eSJeremy L Thompson   int         owned_block_id;
189798701eSJeremy L Thompson   CeedScalar *array_owned;
199937a20eSJeremy L Thompson   // Borrowed external array
209937a20eSJeremy L Thompson   int         borrowed_block_id;
219798701eSJeremy L Thompson   CeedScalar *array_borrowed;
229937a20eSJeremy L Thompson   // Externally viewable read-only array
239937a20eSJeremy L Thompson   int         read_only_block_id;
249798701eSJeremy L Thompson   CeedScalar *array_read_only_copy;
259937a20eSJeremy L Thompson   // Externally viewable writable array
269937a20eSJeremy L Thompson   bool        is_write_only_access;
279937a20eSJeremy L Thompson   int         writable_block_id;
281a3e18b3SJeremy L Thompson   CeedScalar *array_writable_copy;
299798701eSJeremy L Thompson } CeedVector_Memcheck;
309798701eSJeremy L Thompson 
319798701eSJeremy L Thompson typedef struct {
329e82028bSJeremy L Thompson   const CeedInt  *offsets;
339e82028bSJeremy L Thompson   CeedInt        *offsets_allocated;
349e82028bSJeremy L Thompson   const bool     *orients; /* Orientation, if it exists, is true when the dof must be flipped */
359e82028bSJeremy L Thompson   bool           *orients_allocated;
369e82028bSJeremy L Thompson   const CeedInt8 *curl_orients; /* Tridiagonal matrix (row-major) for a general transformation during restriction */
379e82028bSJeremy L Thompson   CeedInt8       *curl_orients_allocated;
389e82028bSJeremy L Thompson   int (*Apply)(CeedElemRestriction, CeedInt, CeedInt, CeedInt, CeedInt, CeedInt, CeedTransposeMode, bool, bool, CeedVector, CeedVector,
399e82028bSJeremy L Thompson                CeedRequest *);
409e82028bSJeremy L Thompson } CeedElemRestriction_Memcheck;
419e82028bSJeremy L Thompson 
429e82028bSJeremy L Thompson typedef struct {
43*3345ba21SJeremy L Thompson   bool               setup_done;
44fc7cf9a0Sjeremylt   const CeedScalar **inputs;
45fc7cf9a0Sjeremylt   CeedScalar       **outputs;
46fc7cf9a0Sjeremylt } CeedQFunction_Memcheck;
47fc7cf9a0Sjeremylt 
480f58c348SJeremy L Thompson typedef struct {
490f58c348SJeremy L Thompson   int   mem_block_id;
500f58c348SJeremy L Thompson   void *data;
510f58c348SJeremy L Thompson   void *data_allocated;
520f58c348SJeremy L Thompson   void *data_owned;
538e457467SJeremy L Thompson   void *data_borrowed;
548e457467SJeremy L Thompson   void *data_read_only_copy;
550f58c348SJeremy L Thompson } CeedQFunctionContext_Memcheck;
560f58c348SJeremy L Thompson 
579798701eSJeremy L Thompson CEED_INTERN int CeedVectorCreate_Memcheck(CeedSize n, CeedVector vec);
589798701eSJeremy L Thompson 
599e82028bSJeremy L Thompson CEED_INTERN int CeedElemRestrictionCreate_Memcheck(CeedMemType mem_type, CeedCopyMode copy_mode, const CeedInt *offsets, const bool *orients,
609e82028bSJeremy L Thompson                                                    const CeedInt8 *curl_orients, CeedElemRestriction r);
619e82028bSJeremy L Thompson 
62fc7cf9a0Sjeremylt CEED_INTERN int CeedQFunctionCreate_Memcheck(CeedQFunction qf);
633d576824SJeremy L Thompson 
640f58c348SJeremy L Thompson CEED_INTERN int CeedQFunctionContextCreate_Memcheck(CeedQFunctionContext ctx);
65