1d275d636SJeremy L Thompson // Copyright (c) 2017-2025, 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. 3d7b241e6Sjeremylt // 43d8e8822SJeremy L Thompson // SPDX-License-Identifier: BSD-2-Clause 5d7b241e6Sjeremylt // 63d8e8822SJeremy L Thompson // This file is part of CEED: http://github.com/ceed 7d7b241e6Sjeremylt 8d7b241e6Sjeremylt #define _POSIX_C_SOURCE 200112 93d576824SJeremy L Thompson #include <ceed-impl.h> 1049aac155SJeremy L Thompson #include <ceed.h> 112b730f8bSJeremy L Thompson #include <ceed/backend.h> 12aedaa0e5Sjeremylt #include <limits.h> 13d7b241e6Sjeremylt #include <stdarg.h> 1449aac155SJeremy L Thompson #include <stdbool.h> 156e79d475Sjeremylt #include <stddef.h> 16d7b241e6Sjeremylt #include <stdio.h> 17d7b241e6Sjeremylt #include <stdlib.h> 18d7b241e6Sjeremylt #include <string.h> 19d7b241e6Sjeremylt 20d7b241e6Sjeremylt /// @cond DOXYGEN_SKIP 21d7b241e6Sjeremylt static CeedRequest ceed_request_immediate; 22d7b241e6Sjeremylt static CeedRequest ceed_request_ordered; 23d7b241e6Sjeremylt 24d7b241e6Sjeremylt static struct { 25d7b241e6Sjeremylt char prefix[CEED_MAX_RESOURCE_LEN]; 26d7b241e6Sjeremylt int (*init)(const char *resource, Ceed f); 27d7b241e6Sjeremylt unsigned int priority; 28d7b241e6Sjeremylt } backends[32]; 29d7b241e6Sjeremylt static size_t num_backends; 30fe2413ffSjeremylt 31eaae1aeaSJames Wright #define CEED_FTABLE_ENTRY(class, method) {#class #method, offsetof(struct class##_private, method)} 32d7b241e6Sjeremylt /// @endcond 33d7b241e6Sjeremylt 34d7b241e6Sjeremylt /// @file 35d7b241e6Sjeremylt /// Implementation of core components of Ceed library 367a982d89SJeremy L. Thompson 377a982d89SJeremy L. Thompson /// @addtogroup CeedUser 38d7b241e6Sjeremylt /// @{ 39d7b241e6Sjeremylt 40dfdf5a53Sjeremylt /** 41dfdf5a53Sjeremylt @brief Request immediate completion 42dfdf5a53Sjeremylt 43ca94c3ddSJeremy L Thompson This predefined constant is passed as the @ref CeedRequest argument to interfaces when the caller wishes for the operation to be performed immediately. 44ca94c3ddSJeremy L Thompson The code 45dfdf5a53Sjeremylt 46dfdf5a53Sjeremylt @code 47dfdf5a53Sjeremylt CeedOperatorApply(op, ..., CEED_REQUEST_IMMEDIATE); 48dfdf5a53Sjeremylt @endcode 49dfdf5a53Sjeremylt 50dfdf5a53Sjeremylt is semantically equivalent to 51dfdf5a53Sjeremylt 52dfdf5a53Sjeremylt @code 53dfdf5a53Sjeremylt CeedRequest request; 54dfdf5a53Sjeremylt CeedOperatorApply(op, ..., &request); 55dfdf5a53Sjeremylt CeedRequestWait(&request); 56dfdf5a53Sjeremylt @endcode 57dfdf5a53Sjeremylt 58dfdf5a53Sjeremylt @sa CEED_REQUEST_ORDERED 59dfdf5a53Sjeremylt **/ 60d7b241e6Sjeremylt CeedRequest *const CEED_REQUEST_IMMEDIATE = &ceed_request_immediate; 61d7b241e6Sjeremylt 62d7b241e6Sjeremylt /** 63b11c1e72Sjeremylt @brief Request ordered completion 64d7b241e6Sjeremylt 65ca94c3ddSJeremy L Thompson This predefined constant is passed as the @ref CeedRequest argument to interfaces when the caller wishes for the operation to be completed in the order that it is submitted to the device. 66ca94c3ddSJeremy L Thompson It is typically used in a construct such as: 67d7b241e6Sjeremylt 68d7b241e6Sjeremylt @code 69d7b241e6Sjeremylt CeedRequest request; 70d7b241e6Sjeremylt CeedOperatorApply(op1, ..., CEED_REQUEST_ORDERED); 71d7b241e6Sjeremylt CeedOperatorApply(op2, ..., &request); 72d7b241e6Sjeremylt // other optional work 738b2d6f4aSMatthew Knepley CeedRequestWait(&request); 74d7b241e6Sjeremylt @endcode 75d7b241e6Sjeremylt 76ea61e9acSJeremy L Thompson which allows the sequence to complete asynchronously but does not start `op2` until `op1` has completed. 77d7b241e6Sjeremylt 78ea61e9acSJeremy L Thompson @todo The current implementation is overly strict, offering equivalent semantics to @ref CEED_REQUEST_IMMEDIATE. 79d7b241e6Sjeremylt 80d7b241e6Sjeremylt @sa CEED_REQUEST_IMMEDIATE 81d7b241e6Sjeremylt */ 82d7b241e6Sjeremylt CeedRequest *const CEED_REQUEST_ORDERED = &ceed_request_ordered; 83d7b241e6Sjeremylt 84b11c1e72Sjeremylt /** 85ca94c3ddSJeremy L Thompson @brief Wait for a @ref CeedRequest to complete. 86dfdf5a53Sjeremylt 87ca94c3ddSJeremy L Thompson Calling @ref CeedRequestWait() on a `NULL` request is a no-op. 887a982d89SJeremy L. Thompson 89ca94c3ddSJeremy L Thompson @param[in,out] req Address of @ref CeedRequest to wait for; zeroed on completion. 907a982d89SJeremy L. Thompson 917a982d89SJeremy L. Thompson @return An error code: 0 - success, otherwise - failure 927a982d89SJeremy L. Thompson 937a982d89SJeremy L. Thompson @ref User 94b11c1e72Sjeremylt **/ 957a982d89SJeremy L. Thompson int CeedRequestWait(CeedRequest *req) { 962b730f8bSJeremy L Thompson if (!*req) return CEED_ERROR_SUCCESS; 972b730f8bSJeremy L Thompson return CeedError(NULL, CEED_ERROR_UNSUPPORTED, "CeedRequestWait not implemented"); 98683faae0SJed Brown } 997a982d89SJeremy L. Thompson 1007a982d89SJeremy L. Thompson /// @} 1017a982d89SJeremy L. Thompson 1027a982d89SJeremy L. Thompson /// ---------------------------------------------------------------------------- 1037a982d89SJeremy L. Thompson /// Ceed Library Internal Functions 1047a982d89SJeremy L. Thompson /// ---------------------------------------------------------------------------- 1057a982d89SJeremy L. Thompson /// @addtogroup CeedDeveloper 1067a982d89SJeremy L. Thompson /// @{ 107d7b241e6Sjeremylt 1086a406739SJeremy L Thompson /** 1096a406739SJeremy L Thompson @brief Register a Ceed backend internally. 1104385fb7fSSebastian Grimberg 111ca94c3ddSJeremy L Thompson Note: Backends should call @ref CeedRegister() instead. 1126a406739SJeremy L Thompson 113ea61e9acSJeremy L Thompson @param[in] prefix Prefix of resources for this backend to respond to. 114ea61e9acSJeremy L Thompson For example, the reference backend responds to "/cpu/self". 115ca94c3ddSJeremy L Thompson @param[in] init Initialization function called by @ref CeedInit() when the backend is selected to drive the requested resource 116ea61e9acSJeremy L Thompson @param[in] priority Integer priority. 117ca94c3ddSJeremy L Thompson Lower values are preferred in case the resource requested by @ref CeedInit() has non-unique best prefix match. 1186a406739SJeremy L Thompson 1196a406739SJeremy L Thompson @return An error code: 0 - success, otherwise - failure 1206a406739SJeremy L Thompson 1216a406739SJeremy L Thompson @ref Developer 1226a406739SJeremy L Thompson **/ 1232b730f8bSJeremy L Thompson int CeedRegisterImpl(const char *prefix, int (*init)(const char *, Ceed), unsigned int priority) { 12458c07c4fSSebastian Grimberg int ierr = 0; 1256a406739SJeremy L Thompson 12658c07c4fSSebastian Grimberg CeedPragmaCritical(CeedRegisterImpl) { 12758c07c4fSSebastian Grimberg if (num_backends < sizeof(backends) / sizeof(backends[0])) { 1286a406739SJeremy L Thompson strncpy(backends[num_backends].prefix, prefix, CEED_MAX_RESOURCE_LEN); 1296a406739SJeremy L Thompson backends[num_backends].prefix[CEED_MAX_RESOURCE_LEN - 1] = 0; 1306a406739SJeremy L Thompson backends[num_backends].init = init; 1316a406739SJeremy L Thompson backends[num_backends].priority = priority; 1326a406739SJeremy L Thompson num_backends++; 13358c07c4fSSebastian Grimberg } else { 13458c07c4fSSebastian Grimberg ierr = 1; 13558c07c4fSSebastian Grimberg } 13658c07c4fSSebastian Grimberg } 13758c07c4fSSebastian Grimberg CeedCheck(ierr == 0, NULL, CEED_ERROR_MAJOR, "Too many backends"); 1386a406739SJeremy L Thompson return CEED_ERROR_SUCCESS; 1396a406739SJeremy L Thompson } 1406a406739SJeremy L Thompson 14173501bfeSJeremy L Thompson /** 14273501bfeSJeremy L Thompson @brief Create a work vector space for a `ceed` 14373501bfeSJeremy L Thompson 14473501bfeSJeremy L Thompson @param[in,out] ceed `Ceed` to create work vector space for 14573501bfeSJeremy L Thompson 14673501bfeSJeremy L Thompson @return An error code: 0 - success, otherwise - failure 14773501bfeSJeremy L Thompson 14873501bfeSJeremy L Thompson @ref Developer 14973501bfeSJeremy L Thompson **/ 15073501bfeSJeremy L Thompson static int CeedWorkVectorsCreate(Ceed ceed) { 15173501bfeSJeremy L Thompson CeedCall(CeedCalloc(1, &ceed->work_vectors)); 15273501bfeSJeremy L Thompson return CEED_ERROR_SUCCESS; 15373501bfeSJeremy L Thompson } 15473501bfeSJeremy L Thompson 15573501bfeSJeremy L Thompson /** 15673501bfeSJeremy L Thompson @brief Destroy a work vector space for a `ceed` 15773501bfeSJeremy L Thompson 15873501bfeSJeremy L Thompson @param[in,out] ceed `Ceed` to destroy work vector space for 15973501bfeSJeremy L Thompson 16073501bfeSJeremy L Thompson @return An error code: 0 - success, otherwise - failure 16173501bfeSJeremy L Thompson 16273501bfeSJeremy L Thompson @ref Developer 16373501bfeSJeremy L Thompson **/ 16473501bfeSJeremy L Thompson static int CeedWorkVectorsDestroy(Ceed ceed) { 16573501bfeSJeremy L Thompson if (!ceed->work_vectors) return CEED_ERROR_SUCCESS; 16673501bfeSJeremy L Thompson for (CeedSize i = 0; i < ceed->work_vectors->num_vecs; i++) { 16773501bfeSJeremy L Thompson CeedCheck(!ceed->work_vectors->is_in_use[i], ceed, CEED_ERROR_ACCESS, "Work vector %" CeedSize_FMT " checked out but not returned"); 16873501bfeSJeremy L Thompson ceed->ref_count += 2; // Note: increase ref_count to prevent Ceed destructor from triggering again 16973501bfeSJeremy L Thompson CeedCall(CeedVectorDestroy(&ceed->work_vectors->vecs[i])); 17073501bfeSJeremy L Thompson ceed->ref_count -= 1; // Note: restore ref_count 17173501bfeSJeremy L Thompson } 17273501bfeSJeremy L Thompson CeedCall(CeedFree(&ceed->work_vectors->is_in_use)); 17373501bfeSJeremy L Thompson CeedCall(CeedFree(&ceed->work_vectors->vecs)); 17473501bfeSJeremy L Thompson CeedCall(CeedFree(&ceed->work_vectors)); 17573501bfeSJeremy L Thompson return CEED_ERROR_SUCCESS; 17673501bfeSJeremy L Thompson } 17773501bfeSJeremy L Thompson 1787a982d89SJeremy L. Thompson /// @} 179d7b241e6Sjeremylt 1807a982d89SJeremy L. Thompson /// ---------------------------------------------------------------------------- 1817a982d89SJeremy L. Thompson /// Ceed Backend API 1827a982d89SJeremy L. Thompson /// ---------------------------------------------------------------------------- 1837a982d89SJeremy L. Thompson /// @addtogroup CeedBackend 1847a982d89SJeremy L. Thompson /// @{ 185d7b241e6Sjeremylt 186b11c1e72Sjeremylt /** 187ca94c3ddSJeremy L Thompson @brief Return value of `CEED_DEBUG` environment variable 18860f9e2d6SJeremy L Thompson 189ca94c3ddSJeremy L Thompson @param[in] ceed `Ceed` context 19060f9e2d6SJeremy L Thompson 191ca94c3ddSJeremy L Thompson @return Boolean value: true - debugging mode enabled 1923f21f6b1SJeremy L Thompson false - debugging mode disabled 19360f9e2d6SJeremy L Thompson 19460f9e2d6SJeremy L Thompson @ref Backend 19560f9e2d6SJeremy L Thompson **/ 196fc6bbcedSJeremy L Thompson // LCOV_EXCL_START 1972b730f8bSJeremy L Thompson bool CeedDebugFlag(const Ceed ceed) { return ceed->is_debug; } 198fc6bbcedSJeremy L Thompson // LCOV_EXCL_STOP 19960f9e2d6SJeremy L Thompson 20060f9e2d6SJeremy L Thompson /** 201ca94c3ddSJeremy L Thompson @brief Return value of `CEED_DEBUG` environment variable 20260f9e2d6SJeremy L Thompson 203ca94c3ddSJeremy L Thompson @return Boolean value: true - debugging mode enabled 2043f21f6b1SJeremy L Thompson false - debugging mode disabled 2053f21f6b1SJeremy L Thompson 2063f21f6b1SJeremy L Thompson @ref Backend 2073f21f6b1SJeremy L Thompson **/ 2083f21f6b1SJeremy L Thompson // LCOV_EXCL_START 2091c66c397SJeremy L Thompson bool CeedDebugFlagEnv(void) { return getenv("CEED_DEBUG") || getenv("DEBUG") || getenv("DBG"); } 2103f21f6b1SJeremy L Thompson // LCOV_EXCL_STOP 2113f21f6b1SJeremy L Thompson 2123f21f6b1SJeremy L Thompson /** 2133f21f6b1SJeremy L Thompson @brief Print debugging information in color 2143f21f6b1SJeremy L Thompson 215ca94c3ddSJeremy L Thompson @param[in] color Color to print 216ca94c3ddSJeremy L Thompson @param[in] format Printing format 21760f9e2d6SJeremy L Thompson 21860f9e2d6SJeremy L Thompson @ref Backend 21960f9e2d6SJeremy L Thompson **/ 220fc6bbcedSJeremy L Thompson // LCOV_EXCL_START 2213f21f6b1SJeremy L Thompson void CeedDebugImpl256(const unsigned char color, const char *format, ...) { 22260f9e2d6SJeremy L Thompson va_list args; 22360f9e2d6SJeremy L Thompson va_start(args, format); 22460f9e2d6SJeremy L Thompson fflush(stdout); 2252b730f8bSJeremy L Thompson if (color != CEED_DEBUG_COLOR_NONE) fprintf(stdout, "\033[38;5;%dm", color); 22660f9e2d6SJeremy L Thompson vfprintf(stdout, format, args); 2272b730f8bSJeremy L Thompson if (color != CEED_DEBUG_COLOR_NONE) fprintf(stdout, "\033[m"); 22860f9e2d6SJeremy L Thompson fprintf(stdout, "\n"); 22960f9e2d6SJeremy L Thompson fflush(stdout); 23060f9e2d6SJeremy L Thompson va_end(args); 23160f9e2d6SJeremy L Thompson } 232fc6bbcedSJeremy L Thompson // LCOV_EXCL_STOP 23360f9e2d6SJeremy L Thompson 23460f9e2d6SJeremy L Thompson /** 235ca94c3ddSJeremy L Thompson @brief Allocate an array on the host; use @ref CeedMalloc(). 236b11c1e72Sjeremylt 237ea61e9acSJeremy L Thompson Memory usage can be tracked by the library. 238ea61e9acSJeremy L Thompson This ensures sufficient alignment for vectorization and should be used for large allocations. 239b11c1e72Sjeremylt 240ea61e9acSJeremy L Thompson @param[in] n Number of units to allocate 241ea61e9acSJeremy L Thompson @param[in] unit Size of each unit 242ca94c3ddSJeremy L Thompson @param[out] p Address of pointer to hold the result 243b11c1e72Sjeremylt 244b11c1e72Sjeremylt @return An error code: 0 - success, otherwise - failure 245b11c1e72Sjeremylt 2467a982d89SJeremy L. Thompson @ref Backend 247ca94c3ddSJeremy L Thompson 248ca94c3ddSJeremy L Thompson @sa CeedFree() 249b11c1e72Sjeremylt **/ 250d7b241e6Sjeremylt int CeedMallocArray(size_t n, size_t unit, void *p) { 251d7b241e6Sjeremylt int ierr = posix_memalign((void **)p, CEED_ALIGN, n * unit); 2526574a04fSJeremy L Thompson CeedCheck(ierr == 0, NULL, CEED_ERROR_MAJOR, "posix_memalign failed to allocate %zd members of size %zd\n", n, unit); 253e15f9bd0SJeremy L Thompson return CEED_ERROR_SUCCESS; 254d7b241e6Sjeremylt } 255d7b241e6Sjeremylt 256b11c1e72Sjeremylt /** 257ca94c3ddSJeremy L Thompson @brief Allocate a cleared (zeroed) array on the host; use @ref CeedCalloc(). 258b11c1e72Sjeremylt 259b11c1e72Sjeremylt Memory usage can be tracked by the library. 260b11c1e72Sjeremylt 261ea61e9acSJeremy L Thompson @param[in] n Number of units to allocate 262ea61e9acSJeremy L Thompson @param[in] unit Size of each unit 263ca94c3ddSJeremy L Thompson @param[out] p Address of pointer to hold the result 264b11c1e72Sjeremylt 265b11c1e72Sjeremylt @return An error code: 0 - success, otherwise - failure 266b11c1e72Sjeremylt 2677a982d89SJeremy L. Thompson @ref Backend 268ca94c3ddSJeremy L Thompson 269ca94c3ddSJeremy L Thompson @sa CeedFree() 270b11c1e72Sjeremylt **/ 271d7b241e6Sjeremylt int CeedCallocArray(size_t n, size_t unit, void *p) { 272d7b241e6Sjeremylt *(void **)p = calloc(n, unit); 2736574a04fSJeremy L Thompson CeedCheck(!n || !unit || *(void **)p, NULL, CEED_ERROR_MAJOR, "calloc failed to allocate %zd members of size %zd\n", n, unit); 274e15f9bd0SJeremy L Thompson return CEED_ERROR_SUCCESS; 275d7b241e6Sjeremylt } 276d7b241e6Sjeremylt 277b11c1e72Sjeremylt /** 278ca94c3ddSJeremy L Thompson @brief Reallocate an array on the host; use @ref CeedRealloc(). 279b11c1e72Sjeremylt 280b11c1e72Sjeremylt Memory usage can be tracked by the library. 281b11c1e72Sjeremylt 282ea61e9acSJeremy L Thompson @param[in] n Number of units to allocate 283ea61e9acSJeremy L Thompson @param[in] unit Size of each unit 284ca94c3ddSJeremy L Thompson @param[out] p Address of pointer to hold the result 285b11c1e72Sjeremylt 286b11c1e72Sjeremylt @return An error code: 0 - success, otherwise - failure 287b11c1e72Sjeremylt 2887a982d89SJeremy L. Thompson @ref Backend 289ca94c3ddSJeremy L Thompson 290ca94c3ddSJeremy L Thompson @sa CeedFree() 291b11c1e72Sjeremylt **/ 292d7b241e6Sjeremylt int CeedReallocArray(size_t n, size_t unit, void *p) { 293d7b241e6Sjeremylt *(void **)p = realloc(*(void **)p, n * unit); 2946574a04fSJeremy L Thompson CeedCheck(!n || !unit || *(void **)p, NULL, CEED_ERROR_MAJOR, "realloc failed to allocate %zd members of size %zd\n", n, unit); 295e15f9bd0SJeremy L Thompson return CEED_ERROR_SUCCESS; 296d7b241e6Sjeremylt } 297d7b241e6Sjeremylt 298f7e22acaSJeremy L Thompson /** 299ca94c3ddSJeremy L Thompson @brief Allocate a cleared string buffer on the host. 300f7e22acaSJeremy L Thompson 301f7e22acaSJeremy L Thompson Memory usage can be tracked by the library. 302f7e22acaSJeremy L Thompson 303ea61e9acSJeremy L Thompson @param[in] source Pointer to string to be copied 304ea61e9acSJeremy L Thompson @param[out] copy Pointer to variable to hold newly allocated string copy 305f7e22acaSJeremy L Thompson 306f7e22acaSJeremy L Thompson @return An error code: 0 - success, otherwise - failure 307f7e22acaSJeremy L Thompson 308f7e22acaSJeremy L Thompson @ref Backend 309ca94c3ddSJeremy L Thompson 310ca94c3ddSJeremy L Thompson @sa CeedFree() 311f7e22acaSJeremy L Thompson **/ 312f7e22acaSJeremy L Thompson int CeedStringAllocCopy(const char *source, char **copy) { 313f7e22acaSJeremy L Thompson size_t len = strlen(source); 3142b730f8bSJeremy L Thompson CeedCall(CeedCalloc(len + 1, copy)); 315d602d780SJeremy L Thompson memcpy(*copy, source, len); 316f7e22acaSJeremy L Thompson return CEED_ERROR_SUCCESS; 317f7e22acaSJeremy L Thompson } 318f7e22acaSJeremy L Thompson 319ca94c3ddSJeremy L Thompson /** Free memory allocated using @ref CeedMalloc() or @ref CeedCalloc() 32034138859Sjeremylt 321ca94c3ddSJeremy L Thompson @param[in,out] p Address of pointer to memory. 322ca94c3ddSJeremy L Thompson This argument is of type `void*` to avoid needing a cast, but is the address of the pointer (which is zeroed) rather than the pointer. 323ca94c3ddSJeremy L Thompson 324ca94c3ddSJeremy L Thompson @return An error code: 0 - success, otherwise - failure 325ca94c3ddSJeremy L Thompson 326ca94c3ddSJeremy L Thompson @ref Backend 32734138859Sjeremylt **/ 328d7b241e6Sjeremylt int CeedFree(void *p) { 329d7b241e6Sjeremylt free(*(void **)p); 330d7b241e6Sjeremylt *(void **)p = NULL; 331e15f9bd0SJeremy L Thompson return CEED_ERROR_SUCCESS; 332d7b241e6Sjeremylt } 333d7b241e6Sjeremylt 334f5d1e504SJeremy L Thompson /** Internal helper to manage handoff of user `source_array` to backend with proper @ref CeedCopyMode behavior. 335f5d1e504SJeremy L Thompson 336f5d1e504SJeremy L Thompson @param[in] source_array Source data provided by user 337f5d1e504SJeremy L Thompson @param[in] copy_mode Copy mode for the data 338f5d1e504SJeremy L Thompson @param[in] num_values Number of values to handle 33917afdf5cSJames Wright @param[in] size_unit Size of array element in bytes 340f5d1e504SJeremy L Thompson @param[in,out] target_array_owned Pointer to location to allocated or hold owned data, may be freed if already allocated 341f5d1e504SJeremy L Thompson @param[out] target_array_borrowed Pointer to location to hold borrowed data 342f5d1e504SJeremy L Thompson @param[out] target_array Pointer to location for data 343f5d1e504SJeremy L Thompson 344f5d1e504SJeremy L Thompson @return An error code: 0 - success, otherwise - failure 345f5d1e504SJeremy L Thompson 346f5d1e504SJeremy L Thompson @ref Backend 347f5d1e504SJeremy L Thompson **/ 348f5d1e504SJeremy L Thompson static inline int CeedSetHostGenericArray(const void *source_array, CeedCopyMode copy_mode, size_t size_unit, CeedSize num_values, 349f5d1e504SJeremy L Thompson void *target_array_owned, void *target_array_borrowed, void *target_array) { 350f5d1e504SJeremy L Thompson switch (copy_mode) { 351f5d1e504SJeremy L Thompson case CEED_COPY_VALUES: 352cc3bdf8cSJeremy L Thompson if (!*(void **)target_array) { 353cc3bdf8cSJeremy L Thompson if (*(void **)target_array_borrowed) { 354cc3bdf8cSJeremy L Thompson *(void **)target_array = *(void **)target_array_borrowed; 355cc3bdf8cSJeremy L Thompson } else { 356f5d1e504SJeremy L Thompson if (!*(void **)target_array_owned) CeedCall(CeedCallocArray(num_values, size_unit, target_array_owned)); 357f5d1e504SJeremy L Thompson *(void **)target_array = *(void **)target_array_owned; 358cc3bdf8cSJeremy L Thompson } 359cc3bdf8cSJeremy L Thompson } 360cc3bdf8cSJeremy L Thompson if (source_array) memcpy(*(void **)target_array, source_array, size_unit * num_values); 361f5d1e504SJeremy L Thompson break; 362f5d1e504SJeremy L Thompson case CEED_OWN_POINTER: 363f5d1e504SJeremy L Thompson CeedCall(CeedFree(target_array_owned)); 364f5d1e504SJeremy L Thompson *(void **)target_array_owned = (void *)source_array; 365f5d1e504SJeremy L Thompson *(void **)target_array_borrowed = NULL; 366f5d1e504SJeremy L Thompson *(void **)target_array = *(void **)target_array_owned; 367f5d1e504SJeremy L Thompson break; 368f5d1e504SJeremy L Thompson case CEED_USE_POINTER: 369f5d1e504SJeremy L Thompson CeedCall(CeedFree(target_array_owned)); 370f5d1e504SJeremy L Thompson *(void **)target_array_owned = NULL; 371f5d1e504SJeremy L Thompson *(void **)target_array_borrowed = (void *)source_array; 372f5d1e504SJeremy L Thompson *(void **)target_array = *(void **)target_array_borrowed; 373f5d1e504SJeremy L Thompson } 374f5d1e504SJeremy L Thompson return CEED_ERROR_SUCCESS; 375f5d1e504SJeremy L Thompson } 376f5d1e504SJeremy L Thompson 377f5d1e504SJeremy L Thompson /** Manage handoff of user `bool` `source_array` to backend with proper @ref CeedCopyMode behavior. 378f5d1e504SJeremy L Thompson 379f5d1e504SJeremy L Thompson @param[in] source_array Source data provided by user 380f5d1e504SJeremy L Thompson @param[in] copy_mode Copy mode for the data 381f5d1e504SJeremy L Thompson @param[in] num_values Number of values to handle 382f5d1e504SJeremy L Thompson @param[in,out] target_array_owned Pointer to location to allocated or hold owned data, may be freed if already allocated 383f5d1e504SJeremy L Thompson @param[out] target_array_borrowed Pointer to location to hold borrowed data 384f5d1e504SJeremy L Thompson @param[out] target_array Pointer to location for data 385f5d1e504SJeremy L Thompson 386f5d1e504SJeremy L Thompson @return An error code: 0 - success, otherwise - failure 387f5d1e504SJeremy L Thompson 388f5d1e504SJeremy L Thompson @ref Backend 389f5d1e504SJeremy L Thompson **/ 390f5d1e504SJeremy L Thompson int CeedSetHostBoolArray(const bool *source_array, CeedCopyMode copy_mode, CeedSize num_values, const bool **target_array_owned, 391f5d1e504SJeremy L Thompson const bool **target_array_borrowed, const bool **target_array) { 392f5d1e504SJeremy L Thompson CeedCall(CeedSetHostGenericArray(source_array, copy_mode, sizeof(bool), num_values, target_array_owned, target_array_borrowed, target_array)); 393f5d1e504SJeremy L Thompson return CEED_ERROR_SUCCESS; 394f5d1e504SJeremy L Thompson } 395f5d1e504SJeremy L Thompson 396f5d1e504SJeremy L Thompson /** Manage handoff of user `CeedInt8` `source_array` to backend with proper @ref CeedCopyMode behavior. 397f5d1e504SJeremy L Thompson 398f5d1e504SJeremy L Thompson @param[in] source_array Source data provided by user 399f5d1e504SJeremy L Thompson @param[in] copy_mode Copy mode for the data 400f5d1e504SJeremy L Thompson @param[in] num_values Number of values to handle 401f5d1e504SJeremy L Thompson @param[in,out] target_array_owned Pointer to location to allocated or hold owned data, may be freed if already allocated 402f5d1e504SJeremy L Thompson @param[out] target_array_borrowed Pointer to location to hold borrowed data 403f5d1e504SJeremy L Thompson @param[out] target_array Pointer to location for data 404f5d1e504SJeremy L Thompson 405f5d1e504SJeremy L Thompson @return An error code: 0 - success, otherwise - failure 406f5d1e504SJeremy L Thompson 407f5d1e504SJeremy L Thompson @ref Backend 408f5d1e504SJeremy L Thompson **/ 409f5d1e504SJeremy L Thompson int CeedSetHostCeedInt8Array(const CeedInt8 *source_array, CeedCopyMode copy_mode, CeedSize num_values, const CeedInt8 **target_array_owned, 410f5d1e504SJeremy L Thompson const CeedInt8 **target_array_borrowed, const CeedInt8 **target_array) { 411f5d1e504SJeremy L Thompson CeedCall(CeedSetHostGenericArray(source_array, copy_mode, sizeof(CeedInt8), num_values, target_array_owned, target_array_borrowed, target_array)); 412f5d1e504SJeremy L Thompson return CEED_ERROR_SUCCESS; 413f5d1e504SJeremy L Thompson } 414f5d1e504SJeremy L Thompson 415f5d1e504SJeremy L Thompson /** Manage handoff of user `CeedInt` `source_array` to backend with proper @ref CeedCopyMode behavior. 416f5d1e504SJeremy L Thompson 417f5d1e504SJeremy L Thompson @param[in] source_array Source data provided by user 418f5d1e504SJeremy L Thompson @param[in] copy_mode Copy mode for the data 419f5d1e504SJeremy L Thompson @param[in] num_values Number of values to handle 420f5d1e504SJeremy L Thompson @param[in,out] target_array_owned Pointer to location to allocated or hold owned data, may be freed if already allocated 421f5d1e504SJeremy L Thompson @param[out] target_array_borrowed Pointer to location to hold borrowed data 422f5d1e504SJeremy L Thompson @param[out] target_array Pointer to location for data 423f5d1e504SJeremy L Thompson 424f5d1e504SJeremy L Thompson @return An error code: 0 - success, otherwise - failure 425f5d1e504SJeremy L Thompson 426f5d1e504SJeremy L Thompson @ref Backend 427f5d1e504SJeremy L Thompson **/ 428f5d1e504SJeremy L Thompson int CeedSetHostCeedIntArray(const CeedInt *source_array, CeedCopyMode copy_mode, CeedSize num_values, const CeedInt **target_array_owned, 429f5d1e504SJeremy L Thompson const CeedInt **target_array_borrowed, const CeedInt **target_array) { 430f5d1e504SJeremy L Thompson CeedCall(CeedSetHostGenericArray(source_array, copy_mode, sizeof(CeedInt), num_values, target_array_owned, target_array_borrowed, target_array)); 431f5d1e504SJeremy L Thompson return CEED_ERROR_SUCCESS; 432f5d1e504SJeremy L Thompson } 433f5d1e504SJeremy L Thompson 434f5d1e504SJeremy L Thompson /** Manage handoff of user `CeedScalar` `source_array` to backend with proper @ref CeedCopyMode behavior. 435f5d1e504SJeremy L Thompson 436f5d1e504SJeremy L Thompson @param[in] source_array Source data provided by user 437f5d1e504SJeremy L Thompson @param[in] copy_mode Copy mode for the data 438f5d1e504SJeremy L Thompson @param[in] num_values Number of values to handle 439f5d1e504SJeremy L Thompson @param[in,out] target_array_owned Pointer to location to allocated or hold owned data, may be freed if already allocated 440f5d1e504SJeremy L Thompson @param[out] target_array_borrowed Pointer to location to hold borrowed data 441f5d1e504SJeremy L Thompson @param[out] target_array Pointer to location for data 442f5d1e504SJeremy L Thompson 443f5d1e504SJeremy L Thompson @return An error code: 0 - success, otherwise - failure 444f5d1e504SJeremy L Thompson 445f5d1e504SJeremy L Thompson @ref Backend 446f5d1e504SJeremy L Thompson **/ 447f5d1e504SJeremy L Thompson int CeedSetHostCeedScalarArray(const CeedScalar *source_array, CeedCopyMode copy_mode, CeedSize num_values, const CeedScalar **target_array_owned, 448f5d1e504SJeremy L Thompson const CeedScalar **target_array_borrowed, const CeedScalar **target_array) { 449f5d1e504SJeremy L Thompson CeedCall(CeedSetHostGenericArray(source_array, copy_mode, sizeof(CeedScalar), num_values, target_array_owned, target_array_borrowed, target_array)); 450f5d1e504SJeremy L Thompson return CEED_ERROR_SUCCESS; 451f5d1e504SJeremy L Thompson } 452f5d1e504SJeremy L Thompson 453d7b241e6Sjeremylt /** 454ca94c3ddSJeremy L Thompson @brief Register a `Ceed` backend 455d7b241e6Sjeremylt 456ea61e9acSJeremy L Thompson @param[in] prefix Prefix of resources for this backend to respond to. 457ea61e9acSJeremy L Thompson For example, the reference backend responds to "/cpu/self". 458ca94c3ddSJeremy L Thompson @param[in] init Initialization function called by @ref CeedInit() when the backend is selected to drive the requested resource 459ea61e9acSJeremy L Thompson @param[in] priority Integer priority. 460ca94c3ddSJeremy L Thompson Lower values are preferred in case the resource requested by @ref CeedInit() has non-unique best prefix match. 461b11c1e72Sjeremylt 462b11c1e72Sjeremylt @return An error code: 0 - success, otherwise - failure 463dfdf5a53Sjeremylt 4647a982d89SJeremy L. Thompson @ref Backend 465b11c1e72Sjeremylt **/ 4662b730f8bSJeremy L Thompson int CeedRegister(const char *prefix, int (*init)(const char *, Ceed), unsigned int priority) { 46710243053SJeremy L Thompson CeedDebugEnv("Backend Register: %s", prefix); 4686a406739SJeremy L Thompson CeedRegisterImpl(prefix, init, priority); 469e15f9bd0SJeremy L Thompson return CEED_ERROR_SUCCESS; 470d7b241e6Sjeremylt } 471d7b241e6Sjeremylt 472b11c1e72Sjeremylt /** 47360f9e2d6SJeremy L Thompson @brief Return debugging status flag 47460f9e2d6SJeremy L Thompson 475ca94c3ddSJeremy L Thompson @param[in] ceed `Ceed` context to get debugging flag 476ea61e9acSJeremy L Thompson @param[out] is_debug Variable to store debugging flag 47760f9e2d6SJeremy L Thompson 47860f9e2d6SJeremy L Thompson @return An error code: 0 - success, otherwise - failure 47960f9e2d6SJeremy L Thompson 480d1d35e2fSjeremylt @ref Backend 48160f9e2d6SJeremy L Thompson **/ 482d1d35e2fSjeremylt int CeedIsDebug(Ceed ceed, bool *is_debug) { 4833f21f6b1SJeremy L Thompson *is_debug = ceed->is_debug; 484e15f9bd0SJeremy L Thompson return CEED_ERROR_SUCCESS; 48560f9e2d6SJeremy L Thompson } 48660f9e2d6SJeremy L Thompson 48760f9e2d6SJeremy L Thompson /** 488bf84744cSJeremy L Thompson @brief Get the root of the requested resource. 489bf84744cSJeremy L Thompson 490bf84744cSJeremy L Thompson Note: Caller is responsible for calling @ref CeedFree() on the `resource_root`. 491bc246734SJeremy L Thompson 492ca94c3ddSJeremy L Thompson @param[in] ceed `Ceed` context to get resource name of 493ca94c3ddSJeremy L Thompson @param[in] resource Full user specified resource 494ca94c3ddSJeremy L Thompson @param[in] delineator Delineator to break `resource_root` and `resource_spec` 495bc246734SJeremy L Thompson @param[out] resource_root Variable to store resource root 496bc246734SJeremy L Thompson 497bc246734SJeremy L Thompson @return An error code: 0 - success, otherwise - failure 498bc246734SJeremy L Thompson 499bc246734SJeremy L Thompson @ref Backend 500bc246734SJeremy L Thompson **/ 501bc246734SJeremy L Thompson int CeedGetResourceRoot(Ceed ceed, const char *resource, const char *delineator, char **resource_root) { 502bc246734SJeremy L Thompson char *device_spec = strstr(resource, delineator); 503bc246734SJeremy L Thompson size_t resource_root_len = device_spec ? (size_t)(device_spec - resource) + 1 : strlen(resource) + 1; 5041c66c397SJeremy L Thompson 505bc246734SJeremy L Thompson CeedCall(CeedCalloc(resource_root_len, resource_root)); 506bc246734SJeremy L Thompson memcpy(*resource_root, resource, resource_root_len - 1); 507bc246734SJeremy L Thompson return CEED_ERROR_SUCCESS; 508bc246734SJeremy L Thompson } 509bc246734SJeremy L Thompson 510bc246734SJeremy L Thompson /** 511ca94c3ddSJeremy L Thompson @brief Retrieve a parent `Ceed` context 5127a982d89SJeremy L. Thompson 513ca94c3ddSJeremy L Thompson @param[in] ceed `Ceed` context to retrieve parent of 5147a982d89SJeremy L. Thompson @param[out] parent Address to save the parent to 5157a982d89SJeremy L. Thompson 5167a982d89SJeremy L. Thompson @return An error code: 0 - success, otherwise - failure 5177a982d89SJeremy L. Thompson 5187a982d89SJeremy L. Thompson @ref Backend 5197a982d89SJeremy L. Thompson **/ 5207a982d89SJeremy L. Thompson int CeedGetParent(Ceed ceed, Ceed *parent) { 5217a982d89SJeremy L. Thompson if (ceed->parent) { 5222b730f8bSJeremy L Thompson CeedCall(CeedGetParent(ceed->parent, parent)); 523e15f9bd0SJeremy L Thompson return CEED_ERROR_SUCCESS; 5247a982d89SJeremy L. Thompson } 5259bc66399SJeremy L Thompson *parent = NULL; 5269bc66399SJeremy L Thompson CeedCall(CeedReferenceCopy(ceed, parent)); 527e15f9bd0SJeremy L Thompson return CEED_ERROR_SUCCESS; 5287a982d89SJeremy L. Thompson } 5297a982d89SJeremy L. Thompson 5307a982d89SJeremy L. Thompson /** 531ca94c3ddSJeremy L Thompson @brief Retrieve a delegate `Ceed` context 5327a982d89SJeremy L. Thompson 533ca94c3ddSJeremy L Thompson @param[in] ceed `Ceed` context to retrieve delegate of 5347a982d89SJeremy L. Thompson @param[out] delegate Address to save the delegate to 5357a982d89SJeremy L. Thompson 5367a982d89SJeremy L. Thompson @return An error code: 0 - success, otherwise - failure 5377a982d89SJeremy L. Thompson 5387a982d89SJeremy L. Thompson @ref Backend 5397a982d89SJeremy L. Thompson **/ 5407a982d89SJeremy L. Thompson int CeedGetDelegate(Ceed ceed, Ceed *delegate) { 5419bc66399SJeremy L Thompson *delegate = NULL; 5429bc66399SJeremy L Thompson if (ceed->delegate) CeedCall(CeedReferenceCopy(ceed->delegate, delegate)); 543e15f9bd0SJeremy L Thompson return CEED_ERROR_SUCCESS; 5447a982d89SJeremy L. Thompson } 5457a982d89SJeremy L. Thompson 5467a982d89SJeremy L. Thompson /** 547ca94c3ddSJeremy L Thompson @brief Set a delegate `Ceed` context 5487a982d89SJeremy L. Thompson 549ca94c3ddSJeremy L Thompson This function allows a `Ceed` context to set a delegate `Ceed` context. 550ca94c3ddSJeremy L Thompson All backend implementations default to the delegate `Ceed` context, unless overridden. 5517a982d89SJeremy L. Thompson 552ca94c3ddSJeremy L Thompson @param[in] ceed `Ceed` context to set delegate of 5537a982d89SJeremy L. Thompson @param[out] delegate Address to set the delegate to 5547a982d89SJeremy L. Thompson 5557a982d89SJeremy L. Thompson @return An error code: 0 - success, otherwise - failure 5567a982d89SJeremy L. Thompson 5577a982d89SJeremy L. Thompson @ref Backend 5587a982d89SJeremy L. Thompson **/ 5597a982d89SJeremy L. Thompson int CeedSetDelegate(Ceed ceed, Ceed delegate) { 5609bc66399SJeremy L Thompson CeedCall(CeedReferenceCopy(delegate, &ceed->delegate)); 5617a982d89SJeremy L. Thompson delegate->parent = ceed; 562e15f9bd0SJeremy L Thompson return CEED_ERROR_SUCCESS; 5637a982d89SJeremy L. Thompson } 5647a982d89SJeremy L. Thompson 5657a982d89SJeremy L. Thompson /** 566ca94c3ddSJeremy L Thompson @brief Retrieve a delegate `Ceed` context for a specific object type 5677a982d89SJeremy L. Thompson 568ca94c3ddSJeremy L Thompson @param[in] ceed `Ceed` context to retrieve delegate of 5697a982d89SJeremy L. Thompson @param[out] delegate Address to save the delegate to 570d1d35e2fSjeremylt @param[in] obj_name Name of the object type to retrieve delegate for 5717a982d89SJeremy L. Thompson 5727a982d89SJeremy L. Thompson @return An error code: 0 - success, otherwise - failure 5737a982d89SJeremy L. Thompson 5747a982d89SJeremy L. Thompson @ref Backend 5757a982d89SJeremy L. Thompson **/ 576d1d35e2fSjeremylt int CeedGetObjectDelegate(Ceed ceed, Ceed *delegate, const char *obj_name) { 5777a982d89SJeremy L. Thompson // Check for object delegate 5782b730f8bSJeremy L Thompson for (CeedInt i = 0; i < ceed->obj_delegate_count; i++) { 579d1d35e2fSjeremylt if (!strcmp(obj_name, ceed->obj_delegates->obj_name)) { 5809bc66399SJeremy L Thompson *delegate = NULL; 5819bc66399SJeremy L Thompson CeedCall(CeedReferenceCopy(ceed->obj_delegates->delegate, delegate)); 582e15f9bd0SJeremy L Thompson return CEED_ERROR_SUCCESS; 5837a982d89SJeremy L. Thompson } 5842b730f8bSJeremy L Thompson } 5857a982d89SJeremy L. Thompson 5867a982d89SJeremy L. Thompson // Use default delegate if no object delegate 5872b730f8bSJeremy L Thompson CeedCall(CeedGetDelegate(ceed, delegate)); 588e15f9bd0SJeremy L Thompson return CEED_ERROR_SUCCESS; 5897a982d89SJeremy L. Thompson } 5907a982d89SJeremy L. Thompson 5917a982d89SJeremy L. Thompson /** 592ca94c3ddSJeremy L Thompson @brief Set a delegate `Ceed` context for a specific object type 5937a982d89SJeremy L. Thompson 594ca94c3ddSJeremy L Thompson This function allows a `Ceed` context to set a delegate `Ceed` context for a given type of `Ceed` object. 595ca94c3ddSJeremy L Thompson All backend implementations default to the delegate `Ceed` context for this object. 596ca94c3ddSJeremy L Thompson For example, `CeedSetObjectDelegate(ceed, delegate, "Basis")` uses delegate implementations for all `CeedBasis` backend functions. 5977a982d89SJeremy L. Thompson 598ca94c3ddSJeremy L Thompson @param[in,out] ceed `Ceed` context to set delegate of 599ca94c3ddSJeremy L Thompson @param[in] delegate `Ceed` context to use for delegation 600d1d35e2fSjeremylt @param[in] obj_name Name of the object type to set delegate for 6017a982d89SJeremy L. Thompson 6027a982d89SJeremy L. Thompson @return An error code: 0 - success, otherwise - failure 6037a982d89SJeremy L. Thompson 6047a982d89SJeremy L. Thompson @ref Backend 6057a982d89SJeremy L. Thompson **/ 606d1d35e2fSjeremylt int CeedSetObjectDelegate(Ceed ceed, Ceed delegate, const char *obj_name) { 607d1d35e2fSjeremylt CeedInt count = ceed->obj_delegate_count; 6087a982d89SJeremy L. Thompson 6097a982d89SJeremy L. Thompson // Malloc or Realloc 6107a982d89SJeremy L. Thompson if (count) { 6112b730f8bSJeremy L Thompson CeedCall(CeedRealloc(count + 1, &ceed->obj_delegates)); 6127a982d89SJeremy L. Thompson } else { 6132b730f8bSJeremy L Thompson CeedCall(CeedCalloc(1, &ceed->obj_delegates)); 6147a982d89SJeremy L. Thompson } 615d1d35e2fSjeremylt ceed->obj_delegate_count++; 6167a982d89SJeremy L. Thompson 6177a982d89SJeremy L. Thompson // Set object delegate 6189bc66399SJeremy L Thompson CeedCall(CeedReferenceCopy(delegate, &ceed->obj_delegates[count].delegate)); 6192b730f8bSJeremy L Thompson CeedCall(CeedStringAllocCopy(obj_name, &ceed->obj_delegates[count].obj_name)); 6207a982d89SJeremy L. Thompson 6217a982d89SJeremy L. Thompson // Set delegate parent 6227a982d89SJeremy L. Thompson delegate->parent = ceed; 623e15f9bd0SJeremy L Thompson return CEED_ERROR_SUCCESS; 6247a982d89SJeremy L. Thompson } 6257a982d89SJeremy L. Thompson 6267a982d89SJeremy L. Thompson /** 627ca94c3ddSJeremy L Thompson @brief Get the fallback `Ceed` for `CeedOperator` 6288687e1d4SJeremy L Thompson 629ca94c3ddSJeremy L Thompson @param[in] ceed `Ceed` context 630ca94c3ddSJeremy L Thompson @param[out] fallback_ceed Variable to store fallback `Ceed` 6318687e1d4SJeremy L Thompson 6328687e1d4SJeremy L Thompson @return An error code: 0 - success, otherwise - failure 6338687e1d4SJeremy L Thompson 6348687e1d4SJeremy L Thompson @ref Backend 6358687e1d4SJeremy L Thompson **/ 6368687e1d4SJeremy L Thompson int CeedGetOperatorFallbackCeed(Ceed ceed, Ceed *fallback_ceed) { 637*46b50f9eSZach Atkins if (ceed->op_fallback_ceed) { 638ca38d01dSJeremy L Thompson CeedDebug256(ceed, CEED_DEBUG_COLOR_SUCCESS, "---------- Ceed Fallback ----------\n"); 639*46b50f9eSZach Atkins CeedDebug(ceed, "Falling back from Ceed with backend %s at address %p to Ceed with backend %s at address %p", ceed->resource, ceed, 640*46b50f9eSZach Atkins ceed->op_fallback_ceed->resource, ceed->op_fallback_ceed); 641d04bbc78SJeremy L Thompson } 6428687e1d4SJeremy L Thompson 6439bc66399SJeremy L Thompson *fallback_ceed = NULL; 6449bc66399SJeremy L Thompson if (ceed->op_fallback_ceed) CeedCall(CeedReferenceCopy(ceed->op_fallback_ceed, fallback_ceed)); 6458687e1d4SJeremy L Thompson return CEED_ERROR_SUCCESS; 6468687e1d4SJeremy L Thompson } 6478687e1d4SJeremy L Thompson 6488687e1d4SJeremy L Thompson /** 649ca94c3ddSJeremy L Thompson @brief Set the fallback resource for `CeedOperator`. 6504385fb7fSSebastian Grimberg 651*46b50f9eSZach Atkins The current fallback, if any, is freed by calling this function. 6527a982d89SJeremy L. Thompson 653ca94c3ddSJeremy L Thompson @param[in,out] ceed `Ceed` context 654*46b50f9eSZach Atkins @param[in] fallback_ceed `Ceed` context to create fallback operators 6557a982d89SJeremy L. Thompson 6567a982d89SJeremy L. Thompson @return An error code: 0 - success, otherwise - failure 6577a982d89SJeremy L. Thompson 6587a982d89SJeremy L. Thompson @ref Backend 6597a982d89SJeremy L. Thompson **/ 660*46b50f9eSZach Atkins int CeedSetOperatorFallbackCeed(Ceed ceed, Ceed fallback_ceed) { 661*46b50f9eSZach Atkins CeedCall(CeedReferenceCopy(fallback_ceed, &ceed->op_fallback_ceed)); 662*46b50f9eSZach Atkins fallback_ceed->parent = ceed; 663e15f9bd0SJeremy L Thompson return CEED_ERROR_SUCCESS; 6647a982d89SJeremy L. Thompson } 6657a982d89SJeremy L. Thompson 6667a982d89SJeremy L. Thompson /** 667ca94c3ddSJeremy L Thompson @brief Flag `Ceed` context as deterministic 6689525855cSJeremy L Thompson 669ca94c3ddSJeremy L Thompson @param[in] ceed `Ceed` to flag as deterministic 67096b902e2Sjeremylt @param[out] is_deterministic Deterministic status to set 6719525855cSJeremy L Thompson 6729525855cSJeremy L Thompson @return An error code: 0 - success, otherwise - failure 6739525855cSJeremy L Thompson 6749525855cSJeremy L Thompson @ref Backend 6759525855cSJeremy L Thompson **/ 676d1d35e2fSjeremylt int CeedSetDeterministic(Ceed ceed, bool is_deterministic) { 677d1d35e2fSjeremylt ceed->is_deterministic = is_deterministic; 678e15f9bd0SJeremy L Thompson return CEED_ERROR_SUCCESS; 6799525855cSJeremy L Thompson } 6809525855cSJeremy L Thompson 6819525855cSJeremy L Thompson /** 682ca94c3ddSJeremy L Thompson @brief Set a backend function. 6837a982d89SJeremy L. Thompson 684ea61e9acSJeremy L Thompson This function is used for a backend to set the function associated with the Ceed objects. 685ca94c3ddSJeremy L Thompson For example, `CeedSetBackendFunction(ceed, "Ceed", ceed, "VectorCreate", BackendVectorCreate)` sets the backend implementation of @ref CeedVectorCreate() and `CeedSetBackendFunction(ceed, "Basis", basis, "Apply", BackendBasisApply)` sets the backend implementation of @ref CeedBasisApply(). 6869fd66db6SSebastian Grimberg Note, the prefix 'Ceed' is not required for the object type ("Basis" vs "CeedBasis"). 6877a982d89SJeremy L. Thompson 688ca94c3ddSJeremy L Thompson @param[in] ceed `Ceed` context for error handling 689ea61e9acSJeremy L Thompson @param[in] type Type of Ceed object to set function for 6907a982d89SJeremy L. Thompson @param[out] object Ceed object to set function for 691ea61e9acSJeremy L Thompson @param[in] func_name Name of function to set 692ea61e9acSJeremy L Thompson @param[in] f Function to set 6937a982d89SJeremy L. Thompson 6947a982d89SJeremy L. Thompson @return An error code: 0 - success, otherwise - failure 6957a982d89SJeremy L. Thompson 6967a982d89SJeremy L. Thompson @ref Backend 6977a982d89SJeremy L. Thompson **/ 698897d4737SSebastian Grimberg int CeedSetBackendFunctionImpl(Ceed ceed, const char *type, void *object, const char *func_name, void (*f)(void)) { 699d1d35e2fSjeremylt char lookup_name[CEED_MAX_RESOURCE_LEN + 1] = ""; 7007a982d89SJeremy L. Thompson 7017a982d89SJeremy L. Thompson // Build lookup name 7022b730f8bSJeremy L Thompson if (strcmp(type, "Ceed")) strncat(lookup_name, "Ceed", CEED_MAX_RESOURCE_LEN); 703d1d35e2fSjeremylt strncat(lookup_name, type, CEED_MAX_RESOURCE_LEN); 704d1d35e2fSjeremylt strncat(lookup_name, func_name, CEED_MAX_RESOURCE_LEN); 7057a982d89SJeremy L. Thompson 7067a982d89SJeremy L. Thompson // Find and use offset 7072b730f8bSJeremy L Thompson for (CeedInt i = 0; ceed->f_offsets[i].func_name; i++) { 708d1d35e2fSjeremylt if (!strcmp(ceed->f_offsets[i].func_name, lookup_name)) { 709d1d35e2fSjeremylt size_t offset = ceed->f_offsets[i].offset; 7107a982d89SJeremy L. Thompson int (**fpointer)(void) = (int (**)(void))((char *)object + offset); // *NOPAD* 7111c66c397SJeremy L Thompson 712897d4737SSebastian Grimberg *fpointer = (int (*)(void))f; 713e15f9bd0SJeremy L Thompson return CEED_ERROR_SUCCESS; 7147a982d89SJeremy L. Thompson } 7152b730f8bSJeremy L Thompson } 7167a982d89SJeremy L. Thompson 7177a982d89SJeremy L. Thompson // LCOV_EXCL_START 7182b730f8bSJeremy L Thompson return CeedError(ceed, CEED_ERROR_UNSUPPORTED, "Requested function '%s' was not found for CEED object '%s'", func_name, type); 7197a982d89SJeremy L. Thompson // LCOV_EXCL_STOP 7207a982d89SJeremy L. Thompson } 7217a982d89SJeremy L. Thompson 7227a982d89SJeremy L. Thompson /** 723ca94c3ddSJeremy L Thompson @brief Retrieve backend data for a `Ceed` context 7247a982d89SJeremy L. Thompson 725ca94c3ddSJeremy L Thompson @param[in] ceed `Ceed` context to retrieve data of 7267a982d89SJeremy L. Thompson @param[out] data Address to save data to 7277a982d89SJeremy L. Thompson 7287a982d89SJeremy L. Thompson @return An error code: 0 - success, otherwise - failure 7297a982d89SJeremy L. Thompson 7307a982d89SJeremy L. Thompson @ref Backend 7317a982d89SJeremy L. Thompson **/ 732777ff853SJeremy L Thompson int CeedGetData(Ceed ceed, void *data) { 733777ff853SJeremy L Thompson *(void **)data = ceed->data; 734e15f9bd0SJeremy L Thompson return CEED_ERROR_SUCCESS; 7357a982d89SJeremy L. Thompson } 7367a982d89SJeremy L. Thompson 7377a982d89SJeremy L. Thompson /** 738ca94c3ddSJeremy L Thompson @brief Set backend data for a `Ceed` context 7397a982d89SJeremy L. Thompson 740ca94c3ddSJeremy L Thompson @param[in,out] ceed `Ceed` context to set data of 741ea61e9acSJeremy L Thompson @param[in] data Address of data to set 7427a982d89SJeremy L. Thompson 7437a982d89SJeremy L. Thompson @return An error code: 0 - success, otherwise - failure 7447a982d89SJeremy L. Thompson 7457a982d89SJeremy L. Thompson @ref Backend 7467a982d89SJeremy L. Thompson **/ 747777ff853SJeremy L Thompson int CeedSetData(Ceed ceed, void *data) { 748777ff853SJeremy L Thompson ceed->data = data; 749e15f9bd0SJeremy L Thompson return CEED_ERROR_SUCCESS; 7507a982d89SJeremy L. Thompson } 7517a982d89SJeremy L. Thompson 75234359f16Sjeremylt /** 753ca94c3ddSJeremy L Thompson @brief Increment the reference counter for a `Ceed` context 75434359f16Sjeremylt 755ca94c3ddSJeremy L Thompson @param[in,out] ceed `Ceed` context to increment the reference counter 75634359f16Sjeremylt 75734359f16Sjeremylt @return An error code: 0 - success, otherwise - failure 75834359f16Sjeremylt 75934359f16Sjeremylt @ref Backend 76034359f16Sjeremylt **/ 7619560d06aSjeremylt int CeedReference(Ceed ceed) { 76234359f16Sjeremylt ceed->ref_count++; 76334359f16Sjeremylt return CEED_ERROR_SUCCESS; 76434359f16Sjeremylt } 76534359f16Sjeremylt 76673501bfeSJeremy L Thompson /** 7670b37c066SZach Atkins @brief Computes the current memory usage of the work vectors in a `Ceed` context and prints to debug.abort 7680b37c066SZach Atkins 7690b37c066SZach Atkins @param[in] ceed `Ceed` context 770ec4241e6SJeremy L Thompson @param[out] usage_mb Address of the variable where the MB of work vector usage will be stored 7710b37c066SZach Atkins 7720b37c066SZach Atkins @return An error code: 0 - success, otherwise - failure 7730b37c066SZach Atkins 7740b37c066SZach Atkins @ref Developer 7750b37c066SZach Atkins **/ 77655326fe7SZach Atkins int CeedGetWorkVectorMemoryUsage(Ceed ceed, CeedScalar *usage_mb) { 777fd326ce8SZach Atkins if (!ceed->VectorCreate) { 778fd326ce8SZach Atkins Ceed delegate; 779fd326ce8SZach Atkins 780fd326ce8SZach Atkins CeedCall(CeedGetObjectDelegate(ceed, &delegate, "Vector")); 781fd326ce8SZach Atkins CeedCheck(delegate, ceed, CEED_ERROR_UNSUPPORTED, "Backend does not implement VectorCreate"); 782fd326ce8SZach Atkins CeedCall(CeedGetWorkVectorMemoryUsage(delegate, usage_mb)); 783fd326ce8SZach Atkins CeedCall(CeedDestroy(&delegate)); 784fd326ce8SZach Atkins return CEED_ERROR_SUCCESS; 785fd326ce8SZach Atkins } 78655326fe7SZach Atkins *usage_mb = 0.0; 7870b37c066SZach Atkins if (ceed->work_vectors) { 7880b37c066SZach Atkins for (CeedInt i = 0; i < ceed->work_vectors->num_vecs; i++) { 7890b37c066SZach Atkins CeedSize vec_len; 7900b37c066SZach Atkins CeedCall(CeedVectorGetLength(ceed->work_vectors->vecs[i], &vec_len)); 79155326fe7SZach Atkins *usage_mb += vec_len; 7920b37c066SZach Atkins } 79355326fe7SZach Atkins *usage_mb *= sizeof(CeedScalar) * 1e-6; 79455326fe7SZach Atkins CeedDebug(ceed, "Resource {%s}: Work vectors memory usage: %" CeedInt_FMT " vectors, %g MB\n", ceed->resource, ceed->work_vectors->num_vecs, 79555326fe7SZach Atkins *usage_mb); 7960b37c066SZach Atkins } 7970b37c066SZach Atkins return CEED_ERROR_SUCCESS; 7980b37c066SZach Atkins } 7990b37c066SZach Atkins 8000b37c066SZach Atkins /** 8010b37c066SZach Atkins @brief Clear inactive work vectors in a `Ceed` context below a minimum length. 8020b37c066SZach Atkins 8030b37c066SZach Atkins @param[in,out] ceed `Ceed` context 8040b37c066SZach Atkins @param[in] min_len Minimum length of work vector to keep 8050b37c066SZach Atkins 8060b37c066SZach Atkins @return An error code: 0 - success, otherwise - failure 8070b37c066SZach Atkins 8080b37c066SZach Atkins @ref Backend 8090b37c066SZach Atkins **/ 8100b37c066SZach Atkins int CeedClearWorkVectors(Ceed ceed, CeedSize min_len) { 811fd326ce8SZach Atkins if (!ceed->VectorCreate) { 812fd326ce8SZach Atkins Ceed delegate; 813fd326ce8SZach Atkins 814fd326ce8SZach Atkins CeedCall(CeedGetObjectDelegate(ceed, &delegate, "Vector")); 815fd326ce8SZach Atkins CeedCheck(delegate, ceed, CEED_ERROR_UNSUPPORTED, "Backend does not implement VectorCreate"); 816fd326ce8SZach Atkins CeedCall(CeedClearWorkVectors(delegate, min_len)); 817fd326ce8SZach Atkins CeedCall(CeedDestroy(&delegate)); 818fd326ce8SZach Atkins return CEED_ERROR_SUCCESS; 819fd326ce8SZach Atkins } 8200b37c066SZach Atkins if (!ceed->work_vectors) return CEED_ERROR_SUCCESS; 8210b37c066SZach Atkins for (CeedInt i = 0; i < ceed->work_vectors->num_vecs; i++) { 8220b37c066SZach Atkins if (ceed->work_vectors->is_in_use[i]) continue; 8230b37c066SZach Atkins CeedSize vec_len; 8240b37c066SZach Atkins CeedCall(CeedVectorGetLength(ceed->work_vectors->vecs[i], &vec_len)); 8250b37c066SZach Atkins if (vec_len < min_len) { 82656a26733SZach Atkins ceed->ref_count += 2; // Note: increase ref_count to prevent Ceed destructor from triggering 8270b37c066SZach Atkins CeedCall(CeedVectorDestroy(&ceed->work_vectors->vecs[i])); 8280b37c066SZach Atkins ceed->ref_count -= 1; // Note: restore ref_count 8290b37c066SZach Atkins ceed->work_vectors->num_vecs--; 8300b37c066SZach Atkins if (ceed->work_vectors->num_vecs > 0) { 8310b37c066SZach Atkins ceed->work_vectors->vecs[i] = ceed->work_vectors->vecs[ceed->work_vectors->num_vecs]; 8320b37c066SZach Atkins ceed->work_vectors->is_in_use[i] = ceed->work_vectors->is_in_use[ceed->work_vectors->num_vecs]; 8330b37c066SZach Atkins ceed->work_vectors->is_in_use[ceed->work_vectors->num_vecs] = false; 8340b37c066SZach Atkins i--; 8350b37c066SZach Atkins } 8360b37c066SZach Atkins } 8370b37c066SZach Atkins } 8380b37c066SZach Atkins return CEED_ERROR_SUCCESS; 8390b37c066SZach Atkins } 8400b37c066SZach Atkins 8410b37c066SZach Atkins /** 84273501bfeSJeremy L Thompson @brief Get a `CeedVector` for scratch work from a `Ceed` context. 84373501bfeSJeremy L Thompson 84473501bfeSJeremy L Thompson Note: This vector must be restored with @ref CeedRestoreWorkVector(). 84573501bfeSJeremy L Thompson 84673501bfeSJeremy L Thompson @param[in] ceed `Ceed` context 84773501bfeSJeremy L Thompson @param[in] len Minimum length of work vector 84873501bfeSJeremy L Thompson @param[out] vec Address of the variable where `CeedVector` will be stored 84973501bfeSJeremy L Thompson 85073501bfeSJeremy L Thompson @return An error code: 0 - success, otherwise - failure 85173501bfeSJeremy L Thompson 85273501bfeSJeremy L Thompson @ref Backend 85373501bfeSJeremy L Thompson **/ 85473501bfeSJeremy L Thompson int CeedGetWorkVector(Ceed ceed, CeedSize len, CeedVector *vec) { 85573501bfeSJeremy L Thompson CeedInt i = 0; 85655326fe7SZach Atkins CeedScalar usage_mb; 85773501bfeSJeremy L Thompson 858fd326ce8SZach Atkins if (!ceed->VectorCreate) { 859fd326ce8SZach Atkins Ceed delegate; 860fd326ce8SZach Atkins 861fd326ce8SZach Atkins CeedCall(CeedGetObjectDelegate(ceed, &delegate, "Vector")); 862fd326ce8SZach Atkins CeedCheck(delegate, ceed, CEED_ERROR_UNSUPPORTED, "Backend does not implement VectorCreate"); 863fd326ce8SZach Atkins CeedCall(CeedGetWorkVector(delegate, len, vec)); 864fd326ce8SZach Atkins CeedCall(CeedDestroy(&delegate)); 865fd326ce8SZach Atkins return CEED_ERROR_SUCCESS; 866fd326ce8SZach Atkins } 867fd326ce8SZach Atkins 86873501bfeSJeremy L Thompson if (!ceed->work_vectors) CeedCall(CeedWorkVectorsCreate(ceed)); 86973501bfeSJeremy L Thompson 87073501bfeSJeremy L Thompson // Search for big enough work vector 87173501bfeSJeremy L Thompson for (i = 0; i < ceed->work_vectors->num_vecs; i++) { 87273501bfeSJeremy L Thompson if (!ceed->work_vectors->is_in_use[i]) { 87373501bfeSJeremy L Thompson CeedSize work_len; 87473501bfeSJeremy L Thompson 87573501bfeSJeremy L Thompson CeedCall(CeedVectorGetLength(ceed->work_vectors->vecs[i], &work_len)); 87673501bfeSJeremy L Thompson if (work_len >= len) break; 87773501bfeSJeremy L Thompson } 87873501bfeSJeremy L Thompson } 87973501bfeSJeremy L Thompson // Long enough vector was not found 88073501bfeSJeremy L Thompson if (i == ceed->work_vectors->num_vecs) { 88173501bfeSJeremy L Thompson if (ceed->work_vectors->max_vecs == 0) { 88273501bfeSJeremy L Thompson ceed->work_vectors->max_vecs = 1; 88373501bfeSJeremy L Thompson CeedCall(CeedCalloc(ceed->work_vectors->max_vecs, &ceed->work_vectors->vecs)); 88473501bfeSJeremy L Thompson CeedCall(CeedCalloc(ceed->work_vectors->max_vecs, &ceed->work_vectors->is_in_use)); 88573501bfeSJeremy L Thompson } else if (ceed->work_vectors->max_vecs == i) { 88673501bfeSJeremy L Thompson ceed->work_vectors->max_vecs *= 2; 88773501bfeSJeremy L Thompson CeedCall(CeedRealloc(ceed->work_vectors->max_vecs, &ceed->work_vectors->vecs)); 88873501bfeSJeremy L Thompson CeedCall(CeedRealloc(ceed->work_vectors->max_vecs, &ceed->work_vectors->is_in_use)); 88973501bfeSJeremy L Thompson } 89073501bfeSJeremy L Thompson ceed->work_vectors->num_vecs++; 89173501bfeSJeremy L Thompson CeedCallBackend(CeedVectorCreate(ceed, len, &ceed->work_vectors->vecs[i])); 89273501bfeSJeremy L Thompson ceed->ref_count--; // Note: ref_count manipulation to prevent a ref-loop 89355326fe7SZach Atkins if (ceed->is_debug) CeedGetWorkVectorMemoryUsage(ceed, &usage_mb); 89473501bfeSJeremy L Thompson } 89573501bfeSJeremy L Thompson // Return pointer to work vector 89673501bfeSJeremy L Thompson ceed->work_vectors->is_in_use[i] = true; 89773501bfeSJeremy L Thompson *vec = NULL; 89873501bfeSJeremy L Thompson CeedCall(CeedVectorReferenceCopy(ceed->work_vectors->vecs[i], vec)); 89973501bfeSJeremy L Thompson ceed->ref_count++; // Note: bump ref_count to account for external access 90073501bfeSJeremy L Thompson return CEED_ERROR_SUCCESS; 90173501bfeSJeremy L Thompson } 90273501bfeSJeremy L Thompson 90373501bfeSJeremy L Thompson /** 90473501bfeSJeremy L Thompson @brief Restore a `CeedVector` for scratch work from a `Ceed` context from @ref CeedGetWorkVector() 90573501bfeSJeremy L Thompson 90673501bfeSJeremy L Thompson @param[in] ceed `Ceed` context 90773501bfeSJeremy L Thompson @param[out] vec `CeedVector` to restore 90873501bfeSJeremy L Thompson 90973501bfeSJeremy L Thompson @return An error code: 0 - success, otherwise - failure 91073501bfeSJeremy L Thompson 91173501bfeSJeremy L Thompson @ref Backend 91273501bfeSJeremy L Thompson **/ 91373501bfeSJeremy L Thompson int CeedRestoreWorkVector(Ceed ceed, CeedVector *vec) { 914fd326ce8SZach Atkins if (!ceed->VectorCreate) { 915fd326ce8SZach Atkins Ceed delegate; 916fd326ce8SZach Atkins 917fd326ce8SZach Atkins CeedCall(CeedGetObjectDelegate(ceed, &delegate, "Vector")); 918fd326ce8SZach Atkins CeedCheck(delegate, ceed, CEED_ERROR_UNSUPPORTED, "Backend does not implement VectorCreate"); 919fd326ce8SZach Atkins CeedCall(CeedRestoreWorkVector(delegate, vec)); 920fd326ce8SZach Atkins CeedCall(CeedDestroy(&delegate)); 921fd326ce8SZach Atkins return CEED_ERROR_SUCCESS; 922fd326ce8SZach Atkins } 923fd326ce8SZach Atkins 92473501bfeSJeremy L Thompson for (CeedInt i = 0; i < ceed->work_vectors->num_vecs; i++) { 92573501bfeSJeremy L Thompson if (*vec == ceed->work_vectors->vecs[i]) { 92673501bfeSJeremy L Thompson CeedCheck(ceed->work_vectors->is_in_use[i], ceed, CEED_ERROR_ACCESS, "Work vector %" CeedSize_FMT " was not checked out but is being returned"); 92773501bfeSJeremy L Thompson CeedCall(CeedVectorDestroy(vec)); 92873501bfeSJeremy L Thompson ceed->work_vectors->is_in_use[i] = false; 92973501bfeSJeremy L Thompson ceed->ref_count--; // Note: reduce ref_count again to prevent a ref-loop 93073501bfeSJeremy L Thompson return CEED_ERROR_SUCCESS; 93173501bfeSJeremy L Thompson } 93273501bfeSJeremy L Thompson } 93373501bfeSJeremy L Thompson // LCOV_EXCL_START 93473501bfeSJeremy L Thompson return CeedError(ceed, CEED_ERROR_MAJOR, "vec was not checked out via CeedGetWorkVector()"); 93573501bfeSJeremy L Thompson // LCOV_EXCL_STOP 93673501bfeSJeremy L Thompson } 93773501bfeSJeremy L Thompson 938b13efd58SJeremy L Thompson /** 939b13efd58SJeremy L Thompson @brief Retrieve list of additional JiT source roots from `Ceed` context. 940b13efd58SJeremy L Thompson 941b13efd58SJeremy L Thompson Note: The caller is responsible for restoring `jit_source_roots` with @ref CeedRestoreJitSourceRoots(). 942b13efd58SJeremy L Thompson 943b13efd58SJeremy L Thompson @param[in] ceed `Ceed` context 944b13efd58SJeremy L Thompson @param[out] num_source_roots Number of JiT source directories 945b13efd58SJeremy L Thompson @param[out] jit_source_roots Absolute paths to additional JiT source directories 946b13efd58SJeremy L Thompson 947b13efd58SJeremy L Thompson @return An error code: 0 - success, otherwise - failure 948b13efd58SJeremy L Thompson 949b13efd58SJeremy L Thompson @ref Backend 950b13efd58SJeremy L Thompson **/ 951b13efd58SJeremy L Thompson int CeedGetJitSourceRoots(Ceed ceed, CeedInt *num_source_roots, const char ***jit_source_roots) { 952b13efd58SJeremy L Thompson Ceed ceed_parent; 953b13efd58SJeremy L Thompson 954b13efd58SJeremy L Thompson CeedCall(CeedGetParent(ceed, &ceed_parent)); 955b13efd58SJeremy L Thompson *num_source_roots = ceed_parent->num_jit_source_roots; 956b13efd58SJeremy L Thompson *jit_source_roots = (const char **)ceed_parent->jit_source_roots; 957aeb3a72dSJeremy L Thompson ceed_parent->num_jit_source_roots_readers++; 9589bc66399SJeremy L Thompson CeedCall(CeedDestroy(&ceed_parent)); 959b13efd58SJeremy L Thompson return CEED_ERROR_SUCCESS; 960b13efd58SJeremy L Thompson } 961b13efd58SJeremy L Thompson 962b13efd58SJeremy L Thompson /** 963b13efd58SJeremy L Thompson @brief Restore list of additional JiT source roots from with @ref CeedGetJitSourceRoots() 964b13efd58SJeremy L Thompson 965b13efd58SJeremy L Thompson @param[in] ceed `Ceed` context 966b13efd58SJeremy L Thompson @param[out] jit_source_roots Absolute paths to additional JiT source directories 967b13efd58SJeremy L Thompson 968b13efd58SJeremy L Thompson @return An error code: 0 - success, otherwise - failure 969b13efd58SJeremy L Thompson 970b13efd58SJeremy L Thompson @ref Backend 971b13efd58SJeremy L Thompson **/ 972b13efd58SJeremy L Thompson int CeedRestoreJitSourceRoots(Ceed ceed, const char ***jit_source_roots) { 973aeb3a72dSJeremy L Thompson Ceed ceed_parent; 974aeb3a72dSJeremy L Thompson 975aeb3a72dSJeremy L Thompson CeedCall(CeedGetParent(ceed, &ceed_parent)); 976b13efd58SJeremy L Thompson *jit_source_roots = NULL; 977aeb3a72dSJeremy L Thompson ceed_parent->num_jit_source_roots_readers--; 9789bc66399SJeremy L Thompson CeedCall(CeedDestroy(&ceed_parent)); 979b13efd58SJeremy L Thompson return CEED_ERROR_SUCCESS; 980b13efd58SJeremy L Thompson } 981b13efd58SJeremy L Thompson 9824753b775SJeremy L Thompson /** 9834753b775SJeremy L Thompson @brief Retrieve list of additional JiT defines from `Ceed` context. 9844753b775SJeremy L Thompson 9854753b775SJeremy L Thompson Note: The caller is responsible for restoring `jit_defines` with @ref CeedRestoreJitDefines(). 9864753b775SJeremy L Thompson 9874753b775SJeremy L Thompson @param[in] ceed `Ceed` context 9884753b775SJeremy L Thompson @param[out] num_jit_defines Number of JiT defines 9894753b775SJeremy L Thompson @param[out] jit_defines Strings such as `foo=bar`, used as `-Dfoo=bar` in JiT 9904753b775SJeremy L Thompson 9914753b775SJeremy L Thompson @return An error code: 0 - success, otherwise - failure 9924753b775SJeremy L Thompson 9934753b775SJeremy L Thompson @ref Backend 9944753b775SJeremy L Thompson **/ 9952686ebe6SJed Brown int CeedGetJitDefines(Ceed ceed, CeedInt *num_jit_defines, const char ***jit_defines) { 9964753b775SJeremy L Thompson Ceed ceed_parent; 9974753b775SJeremy L Thompson 9984753b775SJeremy L Thompson CeedCall(CeedGetParent(ceed, &ceed_parent)); 9992686ebe6SJed Brown *num_jit_defines = ceed_parent->num_jit_defines; 10004753b775SJeremy L Thompson *jit_defines = (const char **)ceed_parent->jit_defines; 1001aeb3a72dSJeremy L Thompson ceed_parent->num_jit_defines_readers++; 10029bc66399SJeremy L Thompson CeedCall(CeedDestroy(&ceed_parent)); 10034753b775SJeremy L Thompson return CEED_ERROR_SUCCESS; 10044753b775SJeremy L Thompson } 10054753b775SJeremy L Thompson 10064753b775SJeremy L Thompson /** 10074753b775SJeremy L Thompson @brief Restore list of additional JiT defines from with @ref CeedGetJitDefines() 10084753b775SJeremy L Thompson 10094753b775SJeremy L Thompson @param[in] ceed `Ceed` context 10104753b775SJeremy L Thompson @param[out] jit_defines String such as `foo=bar`, used as `-Dfoo=bar` in JiT 10114753b775SJeremy L Thompson 10124753b775SJeremy L Thompson @return An error code: 0 - success, otherwise - failure 10134753b775SJeremy L Thompson 10144753b775SJeremy L Thompson @ref Backend 10154753b775SJeremy L Thompson **/ 10164753b775SJeremy L Thompson int CeedRestoreJitDefines(Ceed ceed, const char ***jit_defines) { 1017aeb3a72dSJeremy L Thompson Ceed ceed_parent; 1018aeb3a72dSJeremy L Thompson 1019aeb3a72dSJeremy L Thompson CeedCall(CeedGetParent(ceed, &ceed_parent)); 10204753b775SJeremy L Thompson *jit_defines = NULL; 1021aeb3a72dSJeremy L Thompson ceed_parent->num_jit_defines_readers--; 10229bc66399SJeremy L Thompson CeedCall(CeedDestroy(&ceed_parent)); 10234753b775SJeremy L Thompson return CEED_ERROR_SUCCESS; 10244753b775SJeremy L Thompson } 10254753b775SJeremy L Thompson 10267a982d89SJeremy L. Thompson /// @} 10277a982d89SJeremy L. Thompson 10287a982d89SJeremy L. Thompson /// ---------------------------------------------------------------------------- 10297a982d89SJeremy L. Thompson /// Ceed Public API 10307a982d89SJeremy L. Thompson /// ---------------------------------------------------------------------------- 10317a982d89SJeremy L. Thompson /// @addtogroup CeedUser 10327a982d89SJeremy L. Thompson /// @{ 10337a982d89SJeremy L. Thompson 10347a982d89SJeremy L. Thompson /** 1035ca94c3ddSJeremy L Thompson @brief Get the list of available resource names for `Ceed` contexts 10364385fb7fSSebastian Grimberg 1037ca94c3ddSJeremy L Thompson Note: The caller is responsible for `free()`ing the resources and priorities arrays, but should not `free()` the contents of the resources array. 103822e44211Sjeremylt 103992ee7d1cSjeremylt @param[out] n Number of available resources 104092ee7d1cSjeremylt @param[out] resources List of available resource names 104122e44211Sjeremylt @param[out] priorities Resource name prioritization values, lower is better 104222e44211Sjeremylt 104322e44211Sjeremylt @return An error code: 0 - success, otherwise - failure 104422e44211Sjeremylt 104522e44211Sjeremylt @ref User 104622e44211Sjeremylt **/ 104722e44211Sjeremylt // LCOV_EXCL_START 10482b730f8bSJeremy L Thompson int CeedRegistryGetList(size_t *n, char ***const resources, CeedInt **priorities) { 1049d0c91ce9Sjeremylt *n = 0; 10509ff86846Sjeremylt *resources = malloc(num_backends * sizeof(**resources)); 10516574a04fSJeremy L Thompson CeedCheck(resources, NULL, CEED_ERROR_MAJOR, "malloc() failure"); 10529ff86846Sjeremylt if (priorities) { 10539ff86846Sjeremylt *priorities = malloc(num_backends * sizeof(**priorities)); 10546574a04fSJeremy L Thompson CeedCheck(priorities, NULL, CEED_ERROR_MAJOR, "malloc() failure"); 10559ff86846Sjeremylt } 105622e44211Sjeremylt for (size_t i = 0; i < num_backends; i++) { 1057d0c91ce9Sjeremylt // Only report compiled backends 1058d0c91ce9Sjeremylt if (backends[i].priority < CEED_MAX_BACKEND_PRIORITY) { 105922e44211Sjeremylt *resources[i] = backends[i].prefix; 10609ff86846Sjeremylt if (priorities) *priorities[i] = backends[i].priority; 1061d0c91ce9Sjeremylt *n += 1; 1062d0c91ce9Sjeremylt } 1063d0c91ce9Sjeremylt } 10646574a04fSJeremy L Thompson CeedCheck(*n, NULL, CEED_ERROR_MAJOR, "No backends installed"); 1065d0c91ce9Sjeremylt *resources = realloc(*resources, *n * sizeof(**resources)); 10666574a04fSJeremy L Thompson CeedCheck(resources, NULL, CEED_ERROR_MAJOR, "realloc() failure"); 1067d0c91ce9Sjeremylt if (priorities) { 1068d0c91ce9Sjeremylt *priorities = realloc(*priorities, *n * sizeof(**priorities)); 10696574a04fSJeremy L Thompson CeedCheck(priorities, NULL, CEED_ERROR_MAJOR, "realloc() failure"); 107022e44211Sjeremylt } 107122e44211Sjeremylt return CEED_ERROR_SUCCESS; 107245f1e315Sjeremylt } 107322e44211Sjeremylt // LCOV_EXCL_STOP 107422e44211Sjeremylt 107522e44211Sjeremylt /** 1076ca94c3ddSJeremy L Thompson @brief Initialize a `Ceed` context to use the specified resource. 10774385fb7fSSebastian Grimberg 1078ca94c3ddSJeremy L Thompson Note: Prefixing the resource with "help:" (e.g. "help:/cpu/self") will result in @ref CeedInt() printing the current libCEED version number and a list of current available backend resources to `stderr`. 1079b11c1e72Sjeremylt 1080ea61e9acSJeremy L Thompson @param[in] resource Resource to use, e.g., "/cpu/self" 1081ea61e9acSJeremy L Thompson @param[out] ceed The library context 1082b11c1e72Sjeremylt 1083b11c1e72Sjeremylt @return An error code: 0 - success, otherwise - failure 1084dfdf5a53Sjeremylt 10857a982d89SJeremy L. Thompson @ref User 1086ca94c3ddSJeremy L Thompson 1087ca94c3ddSJeremy L Thompson @sa CeedRegister() CeedDestroy() 1088b11c1e72Sjeremylt **/ 1089d7b241e6Sjeremylt int CeedInit(const char *resource, Ceed *ceed) { 10902b730f8bSJeremy L Thompson size_t match_len = 0, match_index = UINT_MAX, match_priority = CEED_MAX_BACKEND_PRIORITY, priority; 1091d7b241e6Sjeremylt 1092fe2413ffSjeremylt // Find matching backend 10936574a04fSJeremy L Thompson CeedCheck(resource, NULL, CEED_ERROR_MAJOR, "No resource provided"); 10942b730f8bSJeremy L Thompson CeedCall(CeedRegisterAll()); 109513873f79Sjeremylt 109622e44211Sjeremylt // Check for help request 109722e44211Sjeremylt const char *help_prefix = "help"; 10982b730f8bSJeremy L Thompson size_t match_help = 0; 10992b730f8bSJeremy L Thompson while (match_help < 4 && resource[match_help] == help_prefix[match_help]) match_help++; 110022e44211Sjeremylt if (match_help == 4) { 11012b730f8bSJeremy L Thompson fprintf(stderr, "libCEED version: %d.%d%d%s\n", CEED_VERSION_MAJOR, CEED_VERSION_MINOR, CEED_VERSION_PATCH, 110222e44211Sjeremylt CEED_VERSION_RELEASE ? "" : "+development"); 110392ee7d1cSjeremylt fprintf(stderr, "Available backend resources:\n"); 110422e44211Sjeremylt for (size_t i = 0; i < num_backends; i++) { 1105d0c91ce9Sjeremylt // Only report compiled backends 11062b730f8bSJeremy L Thompson if (backends[i].priority < CEED_MAX_BACKEND_PRIORITY) fprintf(stderr, " %s\n", backends[i].prefix); 110722e44211Sjeremylt } 110822e44211Sjeremylt fflush(stderr); 110922e44211Sjeremylt match_help = 5; // Delineating character expected 111022e44211Sjeremylt } else { 111122e44211Sjeremylt match_help = 0; 111222e44211Sjeremylt } 111322e44211Sjeremylt 1114ea61e9acSJeremy L Thompson // Find best match, computed as number of matching characters from requested resource stem 11152b730f8bSJeremy L Thompson size_t stem_length = 0; 11162b730f8bSJeremy L Thompson while (resource[stem_length + match_help] && resource[stem_length + match_help] != ':') stem_length++; 1117d7b241e6Sjeremylt for (size_t i = 0; i < num_backends; i++) { 11182b730f8bSJeremy L Thompson size_t n = 0; 1119d7b241e6Sjeremylt const char *prefix = backends[i].prefix; 11202b730f8bSJeremy L Thompson while (prefix[n] && prefix[n] == resource[n + match_help]) n++; 1121d7b241e6Sjeremylt priority = backends[i].priority; 1122d1d35e2fSjeremylt if (n > match_len || (n == match_len && match_priority > priority)) { 1123d1d35e2fSjeremylt match_len = n; 1124d1d35e2fSjeremylt match_priority = priority; 1125f7e22acaSJeremy L Thompson match_index = i; 1126d7b241e6Sjeremylt } 1127d7b241e6Sjeremylt } 11289c9a0587SLeila Ghaffari // Using Levenshtein distance to find closest match 11299c9a0587SLeila Ghaffari if (match_len <= 1 || match_len != stem_length) { 1130203015caSLeila Ghaffari // LCOV_EXCL_START 11319c9a0587SLeila Ghaffari size_t lev_dis = UINT_MAX; 1132f7e22acaSJeremy L Thompson size_t lev_index = UINT_MAX, lev_priority = CEED_MAX_BACKEND_PRIORITY; 11339c9a0587SLeila Ghaffari for (size_t i = 0; i < num_backends; i++) { 11349c9a0587SLeila Ghaffari const char *prefix = backends[i].prefix; 11359c9a0587SLeila Ghaffari size_t prefix_length = strlen(backends[i].prefix); 11369c9a0587SLeila Ghaffari size_t min_len = (prefix_length < stem_length) ? prefix_length : stem_length; 1137092904ddSLeila Ghaffari size_t column[min_len + 1]; 1138092904ddSLeila Ghaffari for (size_t j = 0; j <= min_len; j++) column[j] = j; 11399c9a0587SLeila Ghaffari for (size_t j = 1; j <= min_len; j++) { 11409c9a0587SLeila Ghaffari column[0] = j; 11419c9a0587SLeila Ghaffari for (size_t k = 1, last_diag = j - 1; k <= min_len; k++) { 1142092904ddSLeila Ghaffari size_t old_diag = column[k]; 11439c9a0587SLeila Ghaffari size_t min_1 = (column[k] < column[k - 1]) ? column[k] + 1 : column[k - 1] + 1; 11449c9a0587SLeila Ghaffari size_t min_2 = last_diag + (resource[k - 1] == prefix[j - 1] ? 0 : 1); 11459c9a0587SLeila Ghaffari column[k] = (min_1 < min_2) ? min_1 : min_2; 11469c9a0587SLeila Ghaffari last_diag = old_diag; 11479c9a0587SLeila Ghaffari } 11489c9a0587SLeila Ghaffari } 11499c9a0587SLeila Ghaffari size_t n = column[min_len]; 11509c9a0587SLeila Ghaffari priority = backends[i].priority; 11512b730f8bSJeremy L Thompson if (n < lev_dis || (n == lev_dis && lev_priority > priority)) { 11529c9a0587SLeila Ghaffari lev_dis = n; 11539c9a0587SLeila Ghaffari lev_priority = priority; 1154f7e22acaSJeremy L Thompson lev_index = i; 11559c9a0587SLeila Ghaffari } 11569c9a0587SLeila Ghaffari } 1157f7e22acaSJeremy L Thompson const char *prefix_lev = backends[lev_index].prefix; 11582b730f8bSJeremy L Thompson size_t lev_length = 0; 11592b730f8bSJeremy L Thompson while (prefix_lev[lev_length] && prefix_lev[lev_length] != '\0') lev_length++; 11609c9a0587SLeila Ghaffari size_t m = (lev_length < stem_length) ? lev_length : stem_length; 11616574a04fSJeremy L Thompson if (lev_dis + 1 >= m) return CeedError(NULL, CEED_ERROR_MAJOR, "No suitable backend: %s", resource); 11626574a04fSJeremy L Thompson else return CeedError(NULL, CEED_ERROR_MAJOR, "No suitable backend: %s\nClosest match: %s", resource, backends[lev_index].prefix); 1163203015caSLeila Ghaffari // LCOV_EXCL_STOP 11649c9a0587SLeila Ghaffari } 1165fe2413ffSjeremylt 1166fe2413ffSjeremylt // Setup Ceed 11672b730f8bSJeremy L Thompson CeedCall(CeedCalloc(1, ceed)); 11682b730f8bSJeremy L Thompson CeedCall(CeedCalloc(1, &(*ceed)->jit_source_roots)); 1169bc81ce41Sjeremylt const char *ceed_error_handler = getenv("CEED_ERROR_HANDLER"); 11702b730f8bSJeremy L Thompson if (!ceed_error_handler) ceed_error_handler = "abort"; 11712b730f8bSJeremy L Thompson if (!strcmp(ceed_error_handler, "exit")) (*ceed)->Error = CeedErrorExit; 11722b730f8bSJeremy L Thompson else if (!strcmp(ceed_error_handler, "store")) (*ceed)->Error = CeedErrorStore; 11732b730f8bSJeremy L Thompson else (*ceed)->Error = CeedErrorAbort; 1174d1d35e2fSjeremylt memcpy((*ceed)->err_msg, "No error message stored", 24); 1175d1d35e2fSjeremylt (*ceed)->ref_count = 1; 1176d7b241e6Sjeremylt (*ceed)->data = NULL; 1177fe2413ffSjeremylt 1178fe2413ffSjeremylt // Set lookup table 1179d1d35e2fSjeremylt FOffset f_offsets[] = { 11806e79d475Sjeremylt CEED_FTABLE_ENTRY(Ceed, Error), 11815ae360d4SJeremy L Thompson CEED_FTABLE_ENTRY(Ceed, SetStream), 11826e79d475Sjeremylt CEED_FTABLE_ENTRY(Ceed, GetPreferredMemType), 11836e79d475Sjeremylt CEED_FTABLE_ENTRY(Ceed, Destroy), 1184f8902d9eSjeremylt CEED_FTABLE_ENTRY(Ceed, VectorCreate), 11856e79d475Sjeremylt CEED_FTABLE_ENTRY(Ceed, ElemRestrictionCreate), 11863ac8f562SJeremy L Thompson CEED_FTABLE_ENTRY(Ceed, ElemRestrictionCreateAtPoints), 11876e79d475Sjeremylt CEED_FTABLE_ENTRY(Ceed, ElemRestrictionCreateBlocked), 11886e79d475Sjeremylt CEED_FTABLE_ENTRY(Ceed, BasisCreateTensorH1), 11896e79d475Sjeremylt CEED_FTABLE_ENTRY(Ceed, BasisCreateH1), 119050c301a5SRezgar Shakeri CEED_FTABLE_ENTRY(Ceed, BasisCreateHdiv), 1191c4e3f59bSSebastian Grimberg CEED_FTABLE_ENTRY(Ceed, BasisCreateHcurl), 11926e79d475Sjeremylt CEED_FTABLE_ENTRY(Ceed, TensorContractCreate), 11936e79d475Sjeremylt CEED_FTABLE_ENTRY(Ceed, QFunctionCreate), 1194777ff853SJeremy L Thompson CEED_FTABLE_ENTRY(Ceed, QFunctionContextCreate), 11956e79d475Sjeremylt CEED_FTABLE_ENTRY(Ceed, OperatorCreate), 119648acf710SJeremy L Thompson CEED_FTABLE_ENTRY(Ceed, OperatorCreateAtPoints), 11976e79d475Sjeremylt CEED_FTABLE_ENTRY(Ceed, CompositeOperatorCreate), 11989c774eddSJeremy L Thompson CEED_FTABLE_ENTRY(CeedVector, HasValidArray), 11999c774eddSJeremy L Thompson CEED_FTABLE_ENTRY(CeedVector, HasBorrowedArrayOfType), 12000b8f3c4eSJeremy L Thompson CEED_FTABLE_ENTRY(CeedVector, CopyStrided), 12016e79d475Sjeremylt CEED_FTABLE_ENTRY(CeedVector, SetArray), 12026a6c615bSJeremy L Thompson CEED_FTABLE_ENTRY(CeedVector, TakeArray), 12036e79d475Sjeremylt CEED_FTABLE_ENTRY(CeedVector, SetValue), 12040b8f3c4eSJeremy L Thompson CEED_FTABLE_ENTRY(CeedVector, SetValueStrided), 1205f48ed27dSnbeams CEED_FTABLE_ENTRY(CeedVector, SyncArray), 12066e79d475Sjeremylt CEED_FTABLE_ENTRY(CeedVector, GetArray), 12076e79d475Sjeremylt CEED_FTABLE_ENTRY(CeedVector, GetArrayRead), 12089c774eddSJeremy L Thompson CEED_FTABLE_ENTRY(CeedVector, GetArrayWrite), 12096e79d475Sjeremylt CEED_FTABLE_ENTRY(CeedVector, RestoreArray), 12106e79d475Sjeremylt CEED_FTABLE_ENTRY(CeedVector, RestoreArrayRead), 1211547d9b97Sjeremylt CEED_FTABLE_ENTRY(CeedVector, Norm), 1212e0dd3b27Sjeremylt CEED_FTABLE_ENTRY(CeedVector, Scale), 12130f7fd0f8Sjeremylt CEED_FTABLE_ENTRY(CeedVector, AXPY), 12145fb68f37SKaren (Ren) Stengel CEED_FTABLE_ENTRY(CeedVector, AXPBY), 12150f7fd0f8Sjeremylt CEED_FTABLE_ENTRY(CeedVector, PointwiseMult), 1216d99fa3c5SJeremy L Thompson CEED_FTABLE_ENTRY(CeedVector, Reciprocal), 12176e79d475Sjeremylt CEED_FTABLE_ENTRY(CeedVector, Destroy), 12186e79d475Sjeremylt CEED_FTABLE_ENTRY(CeedElemRestriction, Apply), 1219f30b1135SSebastian Grimberg CEED_FTABLE_ENTRY(CeedElemRestriction, ApplyUnsigned), 12207c1dbaffSSebastian Grimberg CEED_FTABLE_ENTRY(CeedElemRestriction, ApplyUnoriented), 122105fa913cSJeremy L Thompson CEED_FTABLE_ENTRY(CeedElemRestriction, ApplyAtPointsInElement), 12226e79d475Sjeremylt CEED_FTABLE_ENTRY(CeedElemRestriction, ApplyBlock), 1223bd33150aSjeremylt CEED_FTABLE_ENTRY(CeedElemRestriction, GetOffsets), 122477d1c127SSebastian Grimberg CEED_FTABLE_ENTRY(CeedElemRestriction, GetOrientations), 122577d1c127SSebastian Grimberg CEED_FTABLE_ENTRY(CeedElemRestriction, GetCurlOrientations), 122619605835SJeremy L Thompson CEED_FTABLE_ENTRY(CeedElemRestriction, GetAtPointsElementOffset), 12276e79d475Sjeremylt CEED_FTABLE_ENTRY(CeedElemRestriction, Destroy), 12286e79d475Sjeremylt CEED_FTABLE_ENTRY(CeedBasis, Apply), 1229db2becc9SJeremy L Thompson CEED_FTABLE_ENTRY(CeedBasis, ApplyAdd), 1230c8c3fa7dSJeremy L Thompson CEED_FTABLE_ENTRY(CeedBasis, ApplyAtPoints), 1231db2becc9SJeremy L Thompson CEED_FTABLE_ENTRY(CeedBasis, ApplyAddAtPoints), 12326e79d475Sjeremylt CEED_FTABLE_ENTRY(CeedBasis, Destroy), 12336e79d475Sjeremylt CEED_FTABLE_ENTRY(CeedTensorContract, Apply), 12346e79d475Sjeremylt CEED_FTABLE_ENTRY(CeedTensorContract, Destroy), 12356e79d475Sjeremylt CEED_FTABLE_ENTRY(CeedQFunction, Apply), 12368c84ac63Sjeremylt CEED_FTABLE_ENTRY(CeedQFunction, SetCUDAUserFunction), 12378c84ac63Sjeremylt CEED_FTABLE_ENTRY(CeedQFunction, SetHIPUserFunction), 12386e79d475Sjeremylt CEED_FTABLE_ENTRY(CeedQFunction, Destroy), 12399c774eddSJeremy L Thompson CEED_FTABLE_ENTRY(CeedQFunctionContext, HasValidData), 12409c774eddSJeremy L Thompson CEED_FTABLE_ENTRY(CeedQFunctionContext, HasBorrowedDataOfType), 1241777ff853SJeremy L Thompson CEED_FTABLE_ENTRY(CeedQFunctionContext, SetData), 1242891038deSjeremylt CEED_FTABLE_ENTRY(CeedQFunctionContext, TakeData), 1243777ff853SJeremy L Thompson CEED_FTABLE_ENTRY(CeedQFunctionContext, GetData), 124428bfd0b7SJeremy L Thompson CEED_FTABLE_ENTRY(CeedQFunctionContext, GetDataRead), 1245777ff853SJeremy L Thompson CEED_FTABLE_ENTRY(CeedQFunctionContext, RestoreData), 124628bfd0b7SJeremy L Thompson CEED_FTABLE_ENTRY(CeedQFunctionContext, RestoreDataRead), 12472e64a2b9SJeremy L Thompson CEED_FTABLE_ENTRY(CeedQFunctionContext, DataDestroy), 1248777ff853SJeremy L Thompson CEED_FTABLE_ENTRY(CeedQFunctionContext, Destroy), 124980ac2e43SJeremy L Thompson CEED_FTABLE_ENTRY(CeedOperator, LinearAssembleQFunction), 125070a7ffb3SJeremy L Thompson CEED_FTABLE_ENTRY(CeedOperator, LinearAssembleQFunctionUpdate), 125180ac2e43SJeremy L Thompson CEED_FTABLE_ENTRY(CeedOperator, LinearAssembleDiagonal), 12529e9210b8SJeremy L Thompson CEED_FTABLE_ENTRY(CeedOperator, LinearAssembleAddDiagonal), 125380ac2e43SJeremy L Thompson CEED_FTABLE_ENTRY(CeedOperator, LinearAssemblePointBlockDiagonal), 12549e9210b8SJeremy L Thompson CEED_FTABLE_ENTRY(CeedOperator, LinearAssembleAddPointBlockDiagonal), 1255e2f04181SAndrew T. Barker CEED_FTABLE_ENTRY(CeedOperator, LinearAssembleSymbolic), 1256e2f04181SAndrew T. Barker CEED_FTABLE_ENTRY(CeedOperator, LinearAssemble), 1257cefa2673SJeremy L Thompson CEED_FTABLE_ENTRY(CeedOperator, LinearAssembleSingle), 1258713f43c3Sjeremylt CEED_FTABLE_ENTRY(CeedOperator, CreateFDMElementInverse), 12596e79d475Sjeremylt CEED_FTABLE_ENTRY(CeedOperator, Apply), 1260250756a7Sjeremylt CEED_FTABLE_ENTRY(CeedOperator, ApplyComposite), 1261cae8b89aSjeremylt CEED_FTABLE_ENTRY(CeedOperator, ApplyAdd), 1262250756a7Sjeremylt CEED_FTABLE_ENTRY(CeedOperator, ApplyAddComposite), 12636e79d475Sjeremylt CEED_FTABLE_ENTRY(CeedOperator, ApplyJacobian), 12646e79d475Sjeremylt CEED_FTABLE_ENTRY(CeedOperator, Destroy), 12656e79d475Sjeremylt {NULL, 0} // End of lookup table - used in SetBackendFunction loop 12661dfeef1dSjeremylt }; 1267fe2413ffSjeremylt 12682b730f8bSJeremy L Thompson CeedCall(CeedCalloc(sizeof(f_offsets), &(*ceed)->f_offsets)); 1269d1d35e2fSjeremylt memcpy((*ceed)->f_offsets, f_offsets, sizeof(f_offsets)); 1270fe2413ffSjeremylt 127160f9e2d6SJeremy L Thompson // Record env variables CEED_DEBUG or DBG 12721c66c397SJeremy L Thompson (*ceed)->is_debug = getenv("CEED_DEBUG") || getenv("DEBUG") || getenv("DBG"); 127360f9e2d6SJeremy L Thompson 127422e44211Sjeremylt // Copy resource prefix, if backend setup successful 12752b730f8bSJeremy L Thompson CeedCall(CeedStringAllocCopy(backends[match_index].prefix, (char **)&(*ceed)->resource)); 1276ee5a26f2SJeremy L Thompson 1277ee5a26f2SJeremy L Thompson // Set default JiT source root 1278ea61e9acSJeremy L Thompson // Note: there will always be the default root for every Ceed but all additional paths are added to the top-most parent 12792b730f8bSJeremy L Thompson CeedCall(CeedAddJitSourceRoot(*ceed, (char *)CeedJitSourceRootDefault)); 1280ee5a26f2SJeremy L Thompson 1281d04bbc78SJeremy L Thompson // Backend specific setup 12822b730f8bSJeremy L Thompson CeedCall(backends[match_index].init(&resource[match_help], *ceed)); 1283e15f9bd0SJeremy L Thompson return CEED_ERROR_SUCCESS; 1284d7b241e6Sjeremylt } 1285d7b241e6Sjeremylt 1286d7b241e6Sjeremylt /** 1287ca94c3ddSJeremy L Thompson @brief Set the GPU stream for a `Ceed` context 12885ae360d4SJeremy L Thompson 1289ca94c3ddSJeremy L Thompson @param[in,out] ceed `Ceed` context to set the stream 12905ae360d4SJeremy L Thompson @param[in] handle Handle to GPU stream 12915ae360d4SJeremy L Thompson 12925ae360d4SJeremy L Thompson @return An error code: 0 - success, otherwise - failure 12935ae360d4SJeremy L Thompson 12945ae360d4SJeremy L Thompson @ref User 12955ae360d4SJeremy L Thompson **/ 12965ae360d4SJeremy L Thompson int CeedSetStream(Ceed ceed, void *handle) { 1297ca94c3ddSJeremy L Thompson CeedCheck(handle, ceed, CEED_ERROR_INCOMPATIBLE, "Stream handle must be non-NULL"); 12989ffb25e0SJames Wright if (ceed->SetStream) { 12995ae360d4SJeremy L Thompson CeedCall(ceed->SetStream(ceed, handle)); 13009ffb25e0SJames Wright } else { 13019ffb25e0SJames Wright Ceed delegate; 13029ffb25e0SJames Wright CeedCall(CeedGetDelegate(ceed, &delegate)); 13035ae360d4SJeremy L Thompson 130428ce3d2aSJeremy L Thompson if (delegate) CeedCall(CeedSetStream(delegate, handle)); 130528ce3d2aSJeremy L Thompson else return CeedError(ceed, CEED_ERROR_UNSUPPORTED, "Backend does not support setting stream"); 13069bc66399SJeremy L Thompson CeedCall(CeedDestroy(&delegate)); 13079ffb25e0SJames Wright } 13085ae360d4SJeremy L Thompson return CEED_ERROR_SUCCESS; 13095ae360d4SJeremy L Thompson } 13105ae360d4SJeremy L Thompson 13115ae360d4SJeremy L Thompson /** 1312ca94c3ddSJeremy L Thompson @brief Copy the pointer to a `Ceed` context. 13134385fb7fSSebastian Grimberg 1314ca94c3ddSJeremy L Thompson Both pointers should be destroyed with @ref CeedDestroy(). 1315512bb800SJeremy L Thompson 1316ca94c3ddSJeremy L Thompson Note: If the value of `*ceed_copy` passed to this function is non-`NULL`, then it is assumed that `*ceed_copy` is a pointer to a `Ceed` context. 1317ca94c3ddSJeremy L Thompson This `Ceed` context will be destroyed if `*ceed_copy` is the only reference to this `Ceed` context. 13189560d06aSjeremylt 1319ca94c3ddSJeremy L Thompson @param[in] ceed `Ceed` context to copy reference to 1320ea61e9acSJeremy L Thompson @param[in,out] ceed_copy Variable to store copied reference 13219560d06aSjeremylt 13229560d06aSjeremylt @return An error code: 0 - success, otherwise - failure 13239560d06aSjeremylt 13249560d06aSjeremylt @ref User 13259560d06aSjeremylt **/ 13269560d06aSjeremylt int CeedReferenceCopy(Ceed ceed, Ceed *ceed_copy) { 13272b730f8bSJeremy L Thompson CeedCall(CeedReference(ceed)); 13282b730f8bSJeremy L Thompson CeedCall(CeedDestroy(ceed_copy)); 13299560d06aSjeremylt *ceed_copy = ceed; 13309560d06aSjeremylt return CEED_ERROR_SUCCESS; 13319560d06aSjeremylt } 13329560d06aSjeremylt 13339560d06aSjeremylt /** 1334ca94c3ddSJeremy L Thompson @brief Get the full resource name for a `Ceed` context 13352f86a920SJeremy L Thompson 1336ca94c3ddSJeremy L Thompson @param[in] ceed `Ceed` context to get resource name of 13377a982d89SJeremy L. Thompson @param[out] resource Variable to store resource name 13382f86a920SJeremy L Thompson 13392f86a920SJeremy L Thompson @return An error code: 0 - success, otherwise - failure 13402f86a920SJeremy L Thompson 13417a982d89SJeremy L. Thompson @ref User 13425107b09fSJeremy L Thompson **/ 13437a982d89SJeremy L. Thompson int CeedGetResource(Ceed ceed, const char **resource) { 13447a982d89SJeremy L. Thompson *resource = (const char *)ceed->resource; 1345e15f9bd0SJeremy L Thompson return CEED_ERROR_SUCCESS; 13465107b09fSJeremy L Thompson } 13475107b09fSJeremy L Thompson 13485107b09fSJeremy L Thompson /** 1349ca94c3ddSJeremy L Thompson @brief Return `Ceed` context preferred memory type 1350c907536fSjeremylt 1351ca94c3ddSJeremy L Thompson @param[in] ceed `Ceed` context to get preferred memory type of 1352d1d35e2fSjeremylt @param[out] mem_type Address to save preferred memory type to 1353c907536fSjeremylt 1354c907536fSjeremylt @return An error code: 0 - success, otherwise - failure 1355c907536fSjeremylt 13567a982d89SJeremy L. Thompson @ref User 1357c907536fSjeremylt **/ 1358d1d35e2fSjeremylt int CeedGetPreferredMemType(Ceed ceed, CeedMemType *mem_type) { 1359c907536fSjeremylt if (ceed->GetPreferredMemType) { 13602b730f8bSJeremy L Thompson CeedCall(ceed->GetPreferredMemType(mem_type)); 1361c907536fSjeremylt } else { 1362c263cd57Sjeremylt Ceed delegate; 13632b730f8bSJeremy L Thompson CeedCall(CeedGetDelegate(ceed, &delegate)); 1364c263cd57Sjeremylt 1365c263cd57Sjeremylt if (delegate) { 13662b730f8bSJeremy L Thompson CeedCall(CeedGetPreferredMemType(delegate, mem_type)); 1367c263cd57Sjeremylt } else { 1368d1d35e2fSjeremylt *mem_type = CEED_MEM_HOST; 1369c907536fSjeremylt } 13709bc66399SJeremy L Thompson CeedCall(CeedDestroy(&delegate)); 1371c263cd57Sjeremylt } 1372e15f9bd0SJeremy L Thompson return CEED_ERROR_SUCCESS; 1373c907536fSjeremylt } 1374c907536fSjeremylt 1375c907536fSjeremylt /** 1376ca94c3ddSJeremy L Thompson @brief Get deterministic status of `Ceed` context 13779525855cSJeremy L Thompson 1378ca94c3ddSJeremy L Thompson @param[in] ceed `Ceed` context 1379d1d35e2fSjeremylt @param[out] is_deterministic Variable to store deterministic status 13809525855cSJeremy L Thompson 13819525855cSJeremy L Thompson @return An error code: 0 - success, otherwise - failure 13829525855cSJeremy L Thompson 13839525855cSJeremy L Thompson @ref User 13849525855cSJeremy L Thompson **/ 1385d1d35e2fSjeremylt int CeedIsDeterministic(Ceed ceed, bool *is_deterministic) { 1386d1d35e2fSjeremylt *is_deterministic = ceed->is_deterministic; 1387e15f9bd0SJeremy L Thompson return CEED_ERROR_SUCCESS; 13889525855cSJeremy L Thompson } 13899525855cSJeremy L Thompson 13909525855cSJeremy L Thompson /** 1391ca94c3ddSJeremy L Thompson @brief Set additional JiT source root for `Ceed` context 1392ee5a26f2SJeremy L Thompson 1393ca94c3ddSJeremy L Thompson @param[in,out] ceed `Ceed` context 1394ee5a26f2SJeremy L Thompson @param[in] jit_source_root Absolute path to additional JiT source directory 1395ee5a26f2SJeremy L Thompson 1396ee5a26f2SJeremy L Thompson @return An error code: 0 - success, otherwise - failure 1397ee5a26f2SJeremy L Thompson 1398ee5a26f2SJeremy L Thompson @ref User 1399ee5a26f2SJeremy L Thompson **/ 1400ee5a26f2SJeremy L Thompson int CeedAddJitSourceRoot(Ceed ceed, const char *jit_source_root) { 14016155f12fSJeremy L Thompson Ceed ceed_parent; 1402ee5a26f2SJeremy L Thompson 14032b730f8bSJeremy L Thompson CeedCall(CeedGetParent(ceed, &ceed_parent)); 1404830fc37bSJeremy L Thompson CeedCheck(!ceed_parent->num_jit_source_roots_readers, ceed, CEED_ERROR_ACCESS, "Cannot add JiT source root, read access has not been restored"); 14056155f12fSJeremy L Thompson 14066155f12fSJeremy L Thompson CeedInt index = ceed_parent->num_jit_source_roots; 1407ee5a26f2SJeremy L Thompson size_t path_length = strlen(jit_source_root); 14081c66c397SJeremy L Thompson 14094753b775SJeremy L Thompson if (ceed_parent->num_jit_source_roots == ceed_parent->max_jit_source_roots) { 14104753b775SJeremy L Thompson if (ceed_parent->max_jit_source_roots == 0) ceed_parent->max_jit_source_roots = 1; 14114753b775SJeremy L Thompson ceed_parent->max_jit_source_roots *= 2; 14124753b775SJeremy L Thompson CeedCall(CeedRealloc(ceed_parent->max_jit_source_roots, &ceed_parent->jit_source_roots)); 14134753b775SJeremy L Thompson } 14142b730f8bSJeremy L Thompson CeedCall(CeedCalloc(path_length + 1, &ceed_parent->jit_source_roots[index])); 1415d602d780SJeremy L Thompson memcpy(ceed_parent->jit_source_roots[index], jit_source_root, path_length); 14166155f12fSJeremy L Thompson ceed_parent->num_jit_source_roots++; 14179bc66399SJeremy L Thompson CeedCall(CeedDestroy(&ceed_parent)); 1418ee5a26f2SJeremy L Thompson return CEED_ERROR_SUCCESS; 1419ee5a26f2SJeremy L Thompson } 1420ee5a26f2SJeremy L Thompson 1421ee5a26f2SJeremy L Thompson /** 14224753b775SJeremy L Thompson @brief Set additional JiT compiler define for `Ceed` context 14234753b775SJeremy L Thompson 14244753b775SJeremy L Thompson @param[in,out] ceed `Ceed` context 14254753b775SJeremy L Thompson @param[in] jit_define String such as `foo=bar`, used as `-Dfoo=bar` in JiT 14264753b775SJeremy L Thompson 14274753b775SJeremy L Thompson @return An error code: 0 - success, otherwise - failure 14284753b775SJeremy L Thompson 14294753b775SJeremy L Thompson @ref User 14304753b775SJeremy L Thompson **/ 14314753b775SJeremy L Thompson int CeedAddJitDefine(Ceed ceed, const char *jit_define) { 14324753b775SJeremy L Thompson Ceed ceed_parent; 14334753b775SJeremy L Thompson 14344753b775SJeremy L Thompson CeedCall(CeedGetParent(ceed, &ceed_parent)); 1435830fc37bSJeremy L Thompson CeedCheck(!ceed_parent->num_jit_defines_readers, ceed, CEED_ERROR_ACCESS, "Cannot add JiT define, read access has not been restored"); 14364753b775SJeremy L Thompson 14374753b775SJeremy L Thompson CeedInt index = ceed_parent->num_jit_defines; 14384753b775SJeremy L Thompson size_t define_length = strlen(jit_define); 14394753b775SJeremy L Thompson 14404753b775SJeremy L Thompson if (ceed_parent->num_jit_defines == ceed_parent->max_jit_defines) { 14414753b775SJeremy L Thompson if (ceed_parent->max_jit_defines == 0) ceed_parent->max_jit_defines = 1; 14424753b775SJeremy L Thompson ceed_parent->max_jit_defines *= 2; 14434753b775SJeremy L Thompson CeedCall(CeedRealloc(ceed_parent->max_jit_defines, &ceed_parent->jit_defines)); 14444753b775SJeremy L Thompson } 14454753b775SJeremy L Thompson CeedCall(CeedCalloc(define_length + 1, &ceed_parent->jit_defines[index])); 14464753b775SJeremy L Thompson memcpy(ceed_parent->jit_defines[index], jit_define, define_length); 14474753b775SJeremy L Thompson ceed_parent->num_jit_defines++; 14489bc66399SJeremy L Thompson CeedCall(CeedDestroy(&ceed_parent)); 14494753b775SJeremy L Thompson return CEED_ERROR_SUCCESS; 14504753b775SJeremy L Thompson } 14514753b775SJeremy L Thompson 14524753b775SJeremy L Thompson /** 1453ca94c3ddSJeremy L Thompson @brief View a `Ceed` 14540a0da059Sjeremylt 1455ca94c3ddSJeremy L Thompson @param[in] ceed `Ceed` to view 14560a0da059Sjeremylt @param[in] stream Filestream to write to 14570a0da059Sjeremylt 14580a0da059Sjeremylt @return An error code: 0 - success, otherwise - failure 14590a0da059Sjeremylt 14600a0da059Sjeremylt @ref User 14610a0da059Sjeremylt **/ 14620a0da059Sjeremylt int CeedView(Ceed ceed, FILE *stream) { 1463d1d35e2fSjeremylt CeedMemType mem_type; 14640a0da059Sjeremylt 14652b730f8bSJeremy L Thompson CeedCall(CeedGetPreferredMemType(ceed, &mem_type)); 14660a0da059Sjeremylt 14672b730f8bSJeremy L Thompson fprintf(stream, 14682b730f8bSJeremy L Thompson "Ceed\n" 14690a0da059Sjeremylt " Ceed Resource: %s\n" 14700a0da059Sjeremylt " Preferred MemType: %s\n", 1471d1d35e2fSjeremylt ceed->resource, CeedMemTypes[mem_type]); 1472e15f9bd0SJeremy L Thompson return CEED_ERROR_SUCCESS; 14730a0da059Sjeremylt } 14740a0da059Sjeremylt 14750a0da059Sjeremylt /** 1476ca94c3ddSJeremy L Thompson @brief Destroy a `Ceed` 1477d7b241e6Sjeremylt 1478ca94c3ddSJeremy L Thompson @param[in,out] ceed Address of `Ceed` context to destroy 1479b11c1e72Sjeremylt 1480b11c1e72Sjeremylt @return An error code: 0 - success, otherwise - failure 1481dfdf5a53Sjeremylt 14827a982d89SJeremy L. Thompson @ref User 1483b11c1e72Sjeremylt **/ 1484d7b241e6Sjeremylt int CeedDestroy(Ceed *ceed) { 1485ad6481ceSJeremy L Thompson if (!*ceed || --(*ceed)->ref_count > 0) { 1486ad6481ceSJeremy L Thompson *ceed = NULL; 1487ad6481ceSJeremy L Thompson return CEED_ERROR_SUCCESS; 1488ad6481ceSJeremy L Thompson } 1489aeb3a72dSJeremy L Thompson 1490aeb3a72dSJeremy L Thompson CeedCheck(!(*ceed)->num_jit_source_roots_readers, *ceed, CEED_ERROR_ACCESS, 1491aeb3a72dSJeremy L Thompson "Cannot destroy ceed context, read access for JiT source roots has been granted"); 1492aeb3a72dSJeremy L Thompson CeedCheck(!(*ceed)->num_jit_defines_readers, *ceed, CEED_ERROR_ACCESS, "Cannot add JiT source root, read access for JiT defines has been granted"); 1493aeb3a72dSJeremy L Thompson 14942b730f8bSJeremy L Thompson if ((*ceed)->delegate) CeedCall(CeedDestroy(&(*ceed)->delegate)); 14950ace9bf2Sjeremylt 1496d1d35e2fSjeremylt if ((*ceed)->obj_delegate_count > 0) { 149792ae7e47SJeremy L Thompson for (CeedInt i = 0; i < (*ceed)->obj_delegate_count; i++) { 14982b730f8bSJeremy L Thompson CeedCall(CeedDestroy(&((*ceed)->obj_delegates[i].delegate))); 14992b730f8bSJeremy L Thompson CeedCall(CeedFree(&(*ceed)->obj_delegates[i].obj_name)); 1500aefd8378Sjeremylt } 15012b730f8bSJeremy L Thompson CeedCall(CeedFree(&(*ceed)->obj_delegates)); 1502aefd8378Sjeremylt } 15030ace9bf2Sjeremylt 15042b730f8bSJeremy L Thompson if ((*ceed)->Destroy) CeedCall((*ceed)->Destroy(*ceed)); 15050ace9bf2Sjeremylt 150692ae7e47SJeremy L Thompson for (CeedInt i = 0; i < (*ceed)->num_jit_source_roots; i++) { 15072b730f8bSJeremy L Thompson CeedCall(CeedFree(&(*ceed)->jit_source_roots[i])); 1508032e71eaSJeremy L Thompson } 15092b730f8bSJeremy L Thompson CeedCall(CeedFree(&(*ceed)->jit_source_roots)); 1510032e71eaSJeremy L Thompson 15114753b775SJeremy L Thompson for (CeedInt i = 0; i < (*ceed)->num_jit_defines; i++) { 15124753b775SJeremy L Thompson CeedCall(CeedFree(&(*ceed)->jit_defines[i])); 15134753b775SJeremy L Thompson } 15144753b775SJeremy L Thompson CeedCall(CeedFree(&(*ceed)->jit_defines)); 15154753b775SJeremy L Thompson 15162b730f8bSJeremy L Thompson CeedCall(CeedFree(&(*ceed)->f_offsets)); 15172b730f8bSJeremy L Thompson CeedCall(CeedFree(&(*ceed)->resource)); 15182b730f8bSJeremy L Thompson CeedCall(CeedDestroy(&(*ceed)->op_fallback_ceed)); 151973501bfeSJeremy L Thompson CeedCall(CeedWorkVectorsDestroy(*ceed)); 15202b730f8bSJeremy L Thompson CeedCall(CeedFree(ceed)); 1521e15f9bd0SJeremy L Thompson return CEED_ERROR_SUCCESS; 1522d7b241e6Sjeremylt } 1523d7b241e6Sjeremylt 1524f9982c62SWill Pazner // LCOV_EXCL_START 1525f9982c62SWill Pazner const char *CeedErrorFormat(Ceed ceed, const char *format, va_list *args) { 15262b730f8bSJeremy L Thompson if (ceed->parent) return CeedErrorFormat(ceed->parent, format, args); 152778464608Sjeremylt // Using pointer to va_list for better FFI, but clang-tidy can't verify va_list is initalized 152878464608Sjeremylt vsnprintf(ceed->err_msg, CEED_MAX_RESOURCE_LEN, format, *args); // NOLINT 1529d1d35e2fSjeremylt return ceed->err_msg; 1530f9982c62SWill Pazner } 1531f9982c62SWill Pazner // LCOV_EXCL_STOP 1532f9982c62SWill Pazner 15337a982d89SJeremy L. Thompson /** 1534ca94c3ddSJeremy L Thompson @brief Error handling implementation; use @ref CeedError() instead. 1535ca94c3ddSJeremy L Thompson 1536ca94c3ddSJeremy L Thompson @return An error code: 0 - success, otherwise - failure 15377a982d89SJeremy L. Thompson 15387a982d89SJeremy L. Thompson @ref Developer 15397a982d89SJeremy L. Thompson **/ 15402b730f8bSJeremy L Thompson int CeedErrorImpl(Ceed ceed, const char *filename, int lineno, const char *func, int ecode, const char *format, ...) { 15417a982d89SJeremy L. Thompson va_list args; 1542d1d35e2fSjeremylt int ret_val; 15431c66c397SJeremy L Thompson 15447a982d89SJeremy L. Thompson va_start(args, format); 15457a982d89SJeremy L. Thompson if (ceed) { 1546d1d35e2fSjeremylt ret_val = ceed->Error(ceed, filename, lineno, func, ecode, format, &args); 15477a982d89SJeremy L. Thompson } else { 1548b0d62198Sjeremylt // LCOV_EXCL_START 1549477729cfSJeremy L Thompson const char *ceed_error_handler = getenv("CEED_ERROR_HANDLER"); 15502b730f8bSJeremy L Thompson if (!ceed_error_handler) ceed_error_handler = "abort"; 1551bcbe1c99SJeremy L Thompson if (!strcmp(ceed_error_handler, "return")) { 1552bcbe1c99SJeremy L Thompson ret_val = CeedErrorReturn(ceed, filename, lineno, func, ecode, format, &args); 1553bcbe1c99SJeremy L Thompson } else { 1554477729cfSJeremy L Thompson // This function will not return 1555d1d35e2fSjeremylt ret_val = CeedErrorAbort(ceed, filename, lineno, func, ecode, format, &args); 15567a982d89SJeremy L. Thompson } 1557bcbe1c99SJeremy L Thompson } 15587a982d89SJeremy L. Thompson va_end(args); 1559d1d35e2fSjeremylt return ret_val; 1560b0d62198Sjeremylt // LCOV_EXCL_STOP 15617a982d89SJeremy L. Thompson } 15627a982d89SJeremy L. Thompson 1563477729cfSJeremy L Thompson /** 1564477729cfSJeremy L Thompson @brief Error handler that returns without printing anything. 1565477729cfSJeremy L Thompson 1566ca94c3ddSJeremy L Thompson Pass this to @ref CeedSetErrorHandler() to obtain this error handling behavior. 1567ca94c3ddSJeremy L Thompson 1568ca94c3ddSJeremy L Thompson @return An error code: 0 - success, otherwise - failure 1569477729cfSJeremy L Thompson 1570477729cfSJeremy L Thompson @ref Developer 1571477729cfSJeremy L Thompson **/ 1572477729cfSJeremy L Thompson // LCOV_EXCL_START 15732b730f8bSJeremy L Thompson int CeedErrorReturn(Ceed ceed, const char *filename, int line_no, const char *func, int err_code, const char *format, va_list *args) { 1574d1d35e2fSjeremylt return err_code; 1575477729cfSJeremy L Thompson } 1576477729cfSJeremy L Thompson // LCOV_EXCL_STOP 1577477729cfSJeremy L Thompson 1578477729cfSJeremy L Thompson /** 1579ea61e9acSJeremy L Thompson @brief Error handler that stores the error message for future use and returns the error. 1580477729cfSJeremy L Thompson 1581ca94c3ddSJeremy L Thompson Pass this to @ref CeedSetErrorHandler() to obtain this error handling behavior. 1582ca94c3ddSJeremy L Thompson 1583ca94c3ddSJeremy L Thompson @return An error code: 0 - success, otherwise - failure 1584477729cfSJeremy L Thompson 1585477729cfSJeremy L Thompson @ref Developer 1586477729cfSJeremy L Thompson **/ 1587477729cfSJeremy L Thompson // LCOV_EXCL_START 15882b730f8bSJeremy L Thompson int CeedErrorStore(Ceed ceed, const char *filename, int line_no, const char *func, int err_code, const char *format, va_list *args) { 15892b730f8bSJeremy L Thompson if (ceed->parent) return CeedErrorStore(ceed->parent, filename, line_no, func, err_code, format, args); 1590477729cfSJeremy L Thompson 1591477729cfSJeremy L Thompson // Build message 15921c66c397SJeremy L Thompson int len = snprintf(ceed->err_msg, CEED_MAX_RESOURCE_LEN, "%s:%d in %s(): ", filename, line_no, func); 159378464608Sjeremylt // Using pointer to va_list for better FFI, but clang-tidy can't verify va_list is initalized 159478464608Sjeremylt vsnprintf(ceed->err_msg + len, CEED_MAX_RESOURCE_LEN - len, format, *args); // NOLINT 1595d1d35e2fSjeremylt return err_code; 1596477729cfSJeremy L Thompson } 1597477729cfSJeremy L Thompson // LCOV_EXCL_STOP 1598477729cfSJeremy L Thompson 1599477729cfSJeremy L Thompson /** 1600ca94c3ddSJeremy L Thompson @brief Error handler that prints to `stderr` and aborts 1601477729cfSJeremy L Thompson 1602ca94c3ddSJeremy L Thompson Pass this to @ref CeedSetErrorHandler() to obtain this error handling behavior. 1603ca94c3ddSJeremy L Thompson 1604ca94c3ddSJeremy L Thompson @return An error code: 0 - success, otherwise - failure 1605477729cfSJeremy L Thompson 1606477729cfSJeremy L Thompson @ref Developer 1607477729cfSJeremy L Thompson **/ 1608477729cfSJeremy L Thompson // LCOV_EXCL_START 16092b730f8bSJeremy L Thompson int CeedErrorAbort(Ceed ceed, const char *filename, int line_no, const char *func, int err_code, const char *format, va_list *args) { 1610d1d35e2fSjeremylt fprintf(stderr, "%s:%d in %s(): ", filename, line_no, func); 1611f9982c62SWill Pazner vfprintf(stderr, format, *args); 1612477729cfSJeremy L Thompson fprintf(stderr, "\n"); 1613477729cfSJeremy L Thompson abort(); 1614d1d35e2fSjeremylt return err_code; 1615477729cfSJeremy L Thompson } 1616477729cfSJeremy L Thompson // LCOV_EXCL_STOP 1617477729cfSJeremy L Thompson 1618477729cfSJeremy L Thompson /** 1619ca94c3ddSJeremy L Thompson @brief Error handler that prints to `stderr` and exits. 1620477729cfSJeremy L Thompson 1621ca94c3ddSJeremy L Thompson Pass this to @ref CeedSetErrorHandler() to obtain this error handling behavior. 1622477729cfSJeremy L Thompson 1623ca94c3ddSJeremy L Thompson In contrast to @ref CeedErrorAbort(), this exits without a signal, so `atexit()` handlers (e.g., as used by gcov) are run. 1624ca94c3ddSJeremy L Thompson 1625ca94c3ddSJeremy L Thompson @return An error code: 0 - success, otherwise - failure 1626477729cfSJeremy L Thompson 1627477729cfSJeremy L Thompson @ref Developer 1628477729cfSJeremy L Thompson **/ 16292b730f8bSJeremy L Thompson int CeedErrorExit(Ceed ceed, const char *filename, int line_no, const char *func, int err_code, const char *format, va_list *args) { 1630d1d35e2fSjeremylt fprintf(stderr, "%s:%d in %s(): ", filename, line_no, func); 163178464608Sjeremylt // Using pointer to va_list for better FFI, but clang-tidy can't verify va_list is initalized 163278464608Sjeremylt vfprintf(stderr, format, *args); // NOLINT 1633477729cfSJeremy L Thompson fprintf(stderr, "\n"); 1634d1d35e2fSjeremylt exit(err_code); 1635d1d35e2fSjeremylt return err_code; 1636477729cfSJeremy L Thompson } 1637477729cfSJeremy L Thompson 1638477729cfSJeremy L Thompson /** 1639477729cfSJeremy L Thompson @brief Set error handler 1640477729cfSJeremy L Thompson 1641ca94c3ddSJeremy L Thompson A default error handler is set in @ref CeedInit(). 1642ca94c3ddSJeremy L Thompson Use this function to change the error handler to @ref CeedErrorReturn(), @ref CeedErrorAbort(), or a user-defined error handler. 1643ca94c3ddSJeremy L Thompson 1644ca94c3ddSJeremy L Thompson @return An error code: 0 - success, otherwise - failure 1645477729cfSJeremy L Thompson 1646477729cfSJeremy L Thompson @ref Developer 1647477729cfSJeremy L Thompson **/ 1648d1d35e2fSjeremylt int CeedSetErrorHandler(Ceed ceed, CeedErrorHandler handler) { 1649d1d35e2fSjeremylt ceed->Error = handler; 1650d1d35e2fSjeremylt if (ceed->delegate) CeedSetErrorHandler(ceed->delegate, handler); 16512b730f8bSJeremy L Thompson for (CeedInt i = 0; i < ceed->obj_delegate_count; i++) CeedSetErrorHandler(ceed->obj_delegates[i].delegate, handler); 1652e15f9bd0SJeremy L Thompson return CEED_ERROR_SUCCESS; 1653477729cfSJeremy L Thompson } 1654477729cfSJeremy L Thompson 1655477729cfSJeremy L Thompson /** 1656477729cfSJeremy L Thompson @brief Get error message 1657477729cfSJeremy L Thompson 1658ca94c3ddSJeremy L Thompson The error message is only stored when using the error handler @ref CeedErrorStore() 1659477729cfSJeremy L Thompson 1660ca94c3ddSJeremy L Thompson @param[in] ceed `Ceed` context to retrieve error message 1661d1d35e2fSjeremylt @param[out] err_msg Char pointer to hold error message 1662477729cfSJeremy L Thompson 1663ca94c3ddSJeremy L Thompson @return An error code: 0 - success, otherwise - failure 1664ca94c3ddSJeremy L Thompson 1665477729cfSJeremy L Thompson @ref Developer 1666477729cfSJeremy L Thompson **/ 1667d1d35e2fSjeremylt int CeedGetErrorMessage(Ceed ceed, const char **err_msg) { 16682b730f8bSJeremy L Thompson if (ceed->parent) return CeedGetErrorMessage(ceed->parent, err_msg); 1669d1d35e2fSjeremylt *err_msg = ceed->err_msg; 1670e15f9bd0SJeremy L Thompson return CEED_ERROR_SUCCESS; 1671477729cfSJeremy L Thompson } 1672477729cfSJeremy L Thompson 1673477729cfSJeremy L Thompson /** 1674ca94c3ddSJeremy L Thompson @brief Restore error message. 1675477729cfSJeremy L Thompson 1676ca94c3ddSJeremy L Thompson The error message is only stored when using the error handler @ref CeedErrorStore(). 1677477729cfSJeremy L Thompson 1678ca94c3ddSJeremy L Thompson @param[in] ceed `Ceed` context to restore error message 1679d1d35e2fSjeremylt @param[out] err_msg Char pointer that holds error message 1680477729cfSJeremy L Thompson 1681ca94c3ddSJeremy L Thompson @return An error code: 0 - success, otherwise - failure 1682ca94c3ddSJeremy L Thompson 1683477729cfSJeremy L Thompson @ref Developer 1684477729cfSJeremy L Thompson **/ 1685d1d35e2fSjeremylt int CeedResetErrorMessage(Ceed ceed, const char **err_msg) { 16862b730f8bSJeremy L Thompson if (ceed->parent) return CeedResetErrorMessage(ceed->parent, err_msg); 1687d1d35e2fSjeremylt *err_msg = NULL; 1688d1d35e2fSjeremylt memcpy(ceed->err_msg, "No error message stored", 24); 1689e15f9bd0SJeremy L Thompson return CEED_ERROR_SUCCESS; 1690477729cfSJeremy L Thompson } 1691477729cfSJeremy L Thompson 16921070991dSJed Brown /** 1693ca94c3ddSJeremy L Thompson @brief Get libCEED library version information. 16941070991dSJed Brown 1695ea61e9acSJeremy L Thompson libCEED version numbers have the form major.minor.patch. 1696ea61e9acSJeremy L Thompson Non-release versions may contain unstable interfaces. 16971070991dSJed Brown 16981070991dSJed Brown @param[out] major Major version of the library 16991070991dSJed Brown @param[out] minor Minor version of the library 17001070991dSJed Brown @param[out] patch Patch (subminor) version of the library 170153cbfc38SJeremy L Thompson @param[out] release True for releases; false for development branches 17021070991dSJed Brown 1703ca94c3ddSJeremy L Thompson The caller may pass `NULL` for any arguments that are not needed. 17041070991dSJed Brown 1705ca94c3ddSJeremy L Thompson @return An error code: 0 - success, otherwise - failure 17061070991dSJed Brown 17071070991dSJed Brown @ref Developer 1708ca94c3ddSJeremy L Thompson 17094f69910bSJed Brown @sa CEED_VERSION_GE() CeedGetGitVersion() CeedGetBuildConfiguration() 17101070991dSJed Brown */ 17111070991dSJed Brown int CeedGetVersion(int *major, int *minor, int *patch, bool *release) { 17121070991dSJed Brown if (major) *major = CEED_VERSION_MAJOR; 17131070991dSJed Brown if (minor) *minor = CEED_VERSION_MINOR; 17141070991dSJed Brown if (patch) *patch = CEED_VERSION_PATCH; 17151070991dSJed Brown if (release) *release = CEED_VERSION_RELEASE; 1716ca94c3ddSJeremy L Thompson return CEED_ERROR_SUCCESS; 17171070991dSJed Brown } 17181070991dSJed Brown 171953cbfc38SJeremy L Thompson /** 172053cbfc38SJeremy L Thompson @brief Get libCEED scalar type, such as F64 or F32 172153cbfc38SJeremy L Thompson 172253cbfc38SJeremy L Thompson @param[out] scalar_type Type of libCEED scalars 172353cbfc38SJeremy L Thompson 1724ca94c3ddSJeremy L Thompson @return An error code: 0 - success, otherwise - failure 1725ca94c3ddSJeremy L Thompson 172653cbfc38SJeremy L Thompson @ref Developer 172753cbfc38SJeremy L Thompson */ 172880a9ef05SNatalie Beams int CeedGetScalarType(CeedScalarType *scalar_type) { 172980a9ef05SNatalie Beams *scalar_type = CEED_SCALAR_TYPE; 1730ca94c3ddSJeremy L Thompson return CEED_ERROR_SUCCESS; 173180a9ef05SNatalie Beams } 173280a9ef05SNatalie Beams 1733d7b241e6Sjeremylt /// @} 1734