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) { 63746b50f9eSZach Atkins if (ceed->op_fallback_ceed) { 638ca38d01dSJeremy L Thompson CeedDebug256(ceed, CEED_DEBUG_COLOR_SUCCESS, "---------- Ceed Fallback ----------\n"); 63946b50f9eSZach Atkins CeedDebug(ceed, "Falling back from Ceed with backend %s at address %p to Ceed with backend %s at address %p", ceed->resource, ceed, 64046b50f9eSZach 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 65146b50f9eSZach Atkins The current fallback, if any, is freed by calling this function. 6527a982d89SJeremy L. Thompson 653ca94c3ddSJeremy L Thompson @param[in,out] ceed `Ceed` context 65446b50f9eSZach 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 **/ 66046b50f9eSZach Atkins int CeedSetOperatorFallbackCeed(Ceed ceed, Ceed fallback_ceed) { 66146b50f9eSZach Atkins CeedCall(CeedReferenceCopy(fallback_ceed, &ceed->op_fallback_ceed)); 66246b50f9eSZach 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 /** 9632027fb9dSSirAlienTheGreat @brief Retrieve list of additional Rust source roots from `Ceed` context. 9642027fb9dSSirAlienTheGreat 9652027fb9dSSirAlienTheGreat Note: The caller is responsible for restoring `rust_source_roots` with @ref CeedRestoreRustSourceRoots(). 9662027fb9dSSirAlienTheGreat 9672027fb9dSSirAlienTheGreat @param[in] ceed `Ceed` context 9682027fb9dSSirAlienTheGreat @param[out] num_source_roots Number of JiT source directories 9692027fb9dSSirAlienTheGreat @param[out] rust_source_roots Absolute paths to additional Rust source directories 9702027fb9dSSirAlienTheGreat 9712027fb9dSSirAlienTheGreat @return An error code: 0 - success, otherwise - failure 9722027fb9dSSirAlienTheGreat 9732027fb9dSSirAlienTheGreat @ref Backend 9742027fb9dSSirAlienTheGreat **/ 9752027fb9dSSirAlienTheGreat int CeedGetRustSourceRoots(Ceed ceed, CeedInt *num_source_roots, const char ***rust_source_roots) { 9762027fb9dSSirAlienTheGreat Ceed ceed_parent; 9772027fb9dSSirAlienTheGreat 9782027fb9dSSirAlienTheGreat CeedCall(CeedGetParent(ceed, &ceed_parent)); 9792027fb9dSSirAlienTheGreat *num_source_roots = ceed_parent->num_rust_source_roots; 9802027fb9dSSirAlienTheGreat *rust_source_roots = (const char **)ceed_parent->rust_source_roots; 9812027fb9dSSirAlienTheGreat ceed_parent->num_rust_source_roots_readers++; 9822027fb9dSSirAlienTheGreat CeedCall(CeedDestroy(&ceed_parent)); 9832027fb9dSSirAlienTheGreat return CEED_ERROR_SUCCESS; 9842027fb9dSSirAlienTheGreat } 9852027fb9dSSirAlienTheGreat 9862027fb9dSSirAlienTheGreat /** 987b13efd58SJeremy L Thompson @brief Restore list of additional JiT source roots from with @ref CeedGetJitSourceRoots() 988b13efd58SJeremy L Thompson 989b13efd58SJeremy L Thompson @param[in] ceed `Ceed` context 990b13efd58SJeremy L Thompson @param[out] jit_source_roots Absolute paths to additional JiT source directories 991b13efd58SJeremy L Thompson 992b13efd58SJeremy L Thompson @return An error code: 0 - success, otherwise - failure 993b13efd58SJeremy L Thompson 994b13efd58SJeremy L Thompson @ref Backend 995b13efd58SJeremy L Thompson **/ 996b13efd58SJeremy L Thompson int CeedRestoreJitSourceRoots(Ceed ceed, const char ***jit_source_roots) { 997aeb3a72dSJeremy L Thompson Ceed ceed_parent; 998aeb3a72dSJeremy L Thompson 999aeb3a72dSJeremy L Thompson CeedCall(CeedGetParent(ceed, &ceed_parent)); 1000b13efd58SJeremy L Thompson *jit_source_roots = NULL; 1001aeb3a72dSJeremy L Thompson ceed_parent->num_jit_source_roots_readers--; 10029bc66399SJeremy L Thompson CeedCall(CeedDestroy(&ceed_parent)); 1003b13efd58SJeremy L Thompson return CEED_ERROR_SUCCESS; 1004b13efd58SJeremy L Thompson } 1005b13efd58SJeremy L Thompson 10064753b775SJeremy L Thompson /** 10072027fb9dSSirAlienTheGreat @brief Restore list of additional Rust source roots from with @ref CeedGetJitSourceRoots() 10082027fb9dSSirAlienTheGreat 10092027fb9dSSirAlienTheGreat @param[in] ceed `Ceed` context 10102027fb9dSSirAlienTheGreat @param[out] rust_source_roots Absolute paths to additional Rust source directories 10112027fb9dSSirAlienTheGreat 10122027fb9dSSirAlienTheGreat @return An error code: 0 - success, otherwise - failure 10132027fb9dSSirAlienTheGreat 10142027fb9dSSirAlienTheGreat @ref Backend 10152027fb9dSSirAlienTheGreat **/ 10162027fb9dSSirAlienTheGreat int CeedRestoreRustSourceRoots(Ceed ceed, const char ***rust_source_roots) { 10172027fb9dSSirAlienTheGreat Ceed ceed_parent; 10182027fb9dSSirAlienTheGreat 10192027fb9dSSirAlienTheGreat CeedCall(CeedGetParent(ceed, &ceed_parent)); 10202027fb9dSSirAlienTheGreat *rust_source_roots = NULL; 10212027fb9dSSirAlienTheGreat ceed_parent->num_rust_source_roots_readers--; 10222027fb9dSSirAlienTheGreat CeedCall(CeedDestroy(&ceed_parent)); 10232027fb9dSSirAlienTheGreat return CEED_ERROR_SUCCESS; 10242027fb9dSSirAlienTheGreat } 10252027fb9dSSirAlienTheGreat 10262027fb9dSSirAlienTheGreat /** 10274753b775SJeremy L Thompson @brief Retrieve list of additional JiT defines from `Ceed` context. 10284753b775SJeremy L Thompson 10294753b775SJeremy L Thompson Note: The caller is responsible for restoring `jit_defines` with @ref CeedRestoreJitDefines(). 10304753b775SJeremy L Thompson 10314753b775SJeremy L Thompson @param[in] ceed `Ceed` context 10324753b775SJeremy L Thompson @param[out] num_jit_defines Number of JiT defines 10334753b775SJeremy L Thompson @param[out] jit_defines Strings such as `foo=bar`, used as `-Dfoo=bar` in JiT 10344753b775SJeremy L Thompson 10354753b775SJeremy L Thompson @return An error code: 0 - success, otherwise - failure 10364753b775SJeremy L Thompson 10374753b775SJeremy L Thompson @ref Backend 10384753b775SJeremy L Thompson **/ 10392686ebe6SJed Brown int CeedGetJitDefines(Ceed ceed, CeedInt *num_jit_defines, const char ***jit_defines) { 10404753b775SJeremy L Thompson Ceed ceed_parent; 10414753b775SJeremy L Thompson 10424753b775SJeremy L Thompson CeedCall(CeedGetParent(ceed, &ceed_parent)); 10432686ebe6SJed Brown *num_jit_defines = ceed_parent->num_jit_defines; 10444753b775SJeremy L Thompson *jit_defines = (const char **)ceed_parent->jit_defines; 1045aeb3a72dSJeremy L Thompson ceed_parent->num_jit_defines_readers++; 10469bc66399SJeremy L Thompson CeedCall(CeedDestroy(&ceed_parent)); 10474753b775SJeremy L Thompson return CEED_ERROR_SUCCESS; 10484753b775SJeremy L Thompson } 10494753b775SJeremy L Thompson 10504753b775SJeremy L Thompson /** 10514753b775SJeremy L Thompson @brief Restore list of additional JiT defines from with @ref CeedGetJitDefines() 10524753b775SJeremy L Thompson 10534753b775SJeremy L Thompson @param[in] ceed `Ceed` context 10544753b775SJeremy L Thompson @param[out] jit_defines String such as `foo=bar`, used as `-Dfoo=bar` in JiT 10554753b775SJeremy L Thompson 10564753b775SJeremy L Thompson @return An error code: 0 - success, otherwise - failure 10574753b775SJeremy L Thompson 10584753b775SJeremy L Thompson @ref Backend 10594753b775SJeremy L Thompson **/ 10604753b775SJeremy L Thompson int CeedRestoreJitDefines(Ceed ceed, const char ***jit_defines) { 1061aeb3a72dSJeremy L Thompson Ceed ceed_parent; 1062aeb3a72dSJeremy L Thompson 1063aeb3a72dSJeremy L Thompson CeedCall(CeedGetParent(ceed, &ceed_parent)); 10644753b775SJeremy L Thompson *jit_defines = NULL; 1065aeb3a72dSJeremy L Thompson ceed_parent->num_jit_defines_readers--; 10669bc66399SJeremy L Thompson CeedCall(CeedDestroy(&ceed_parent)); 10674753b775SJeremy L Thompson return CEED_ERROR_SUCCESS; 10684753b775SJeremy L Thompson } 10694753b775SJeremy L Thompson 10707a982d89SJeremy L. Thompson /// @} 10717a982d89SJeremy L. Thompson 10727a982d89SJeremy L. Thompson /// ---------------------------------------------------------------------------- 10737a982d89SJeremy L. Thompson /// Ceed Public API 10747a982d89SJeremy L. Thompson /// ---------------------------------------------------------------------------- 10757a982d89SJeremy L. Thompson /// @addtogroup CeedUser 10767a982d89SJeremy L. Thompson /// @{ 10777a982d89SJeremy L. Thompson 10787a982d89SJeremy L. Thompson /** 1079ca94c3ddSJeremy L Thompson @brief Get the list of available resource names for `Ceed` contexts 10804385fb7fSSebastian Grimberg 1081ca94c3ddSJeremy 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. 108222e44211Sjeremylt 108392ee7d1cSjeremylt @param[out] n Number of available resources 108492ee7d1cSjeremylt @param[out] resources List of available resource names 108522e44211Sjeremylt @param[out] priorities Resource name prioritization values, lower is better 108622e44211Sjeremylt 108722e44211Sjeremylt @return An error code: 0 - success, otherwise - failure 108822e44211Sjeremylt 108922e44211Sjeremylt @ref User 109022e44211Sjeremylt **/ 109122e44211Sjeremylt // LCOV_EXCL_START 10922b730f8bSJeremy L Thompson int CeedRegistryGetList(size_t *n, char ***const resources, CeedInt **priorities) { 1093d0c91ce9Sjeremylt *n = 0; 10949ff86846Sjeremylt *resources = malloc(num_backends * sizeof(**resources)); 10956574a04fSJeremy L Thompson CeedCheck(resources, NULL, CEED_ERROR_MAJOR, "malloc() failure"); 10969ff86846Sjeremylt if (priorities) { 10979ff86846Sjeremylt *priorities = malloc(num_backends * sizeof(**priorities)); 10986574a04fSJeremy L Thompson CeedCheck(priorities, NULL, CEED_ERROR_MAJOR, "malloc() failure"); 10999ff86846Sjeremylt } 110022e44211Sjeremylt for (size_t i = 0; i < num_backends; i++) { 1101d0c91ce9Sjeremylt // Only report compiled backends 1102d0c91ce9Sjeremylt if (backends[i].priority < CEED_MAX_BACKEND_PRIORITY) { 110322e44211Sjeremylt *resources[i] = backends[i].prefix; 11049ff86846Sjeremylt if (priorities) *priorities[i] = backends[i].priority; 1105d0c91ce9Sjeremylt *n += 1; 1106d0c91ce9Sjeremylt } 1107d0c91ce9Sjeremylt } 11086574a04fSJeremy L Thompson CeedCheck(*n, NULL, CEED_ERROR_MAJOR, "No backends installed"); 1109d0c91ce9Sjeremylt *resources = realloc(*resources, *n * sizeof(**resources)); 11106574a04fSJeremy L Thompson CeedCheck(resources, NULL, CEED_ERROR_MAJOR, "realloc() failure"); 1111d0c91ce9Sjeremylt if (priorities) { 1112d0c91ce9Sjeremylt *priorities = realloc(*priorities, *n * sizeof(**priorities)); 11136574a04fSJeremy L Thompson CeedCheck(priorities, NULL, CEED_ERROR_MAJOR, "realloc() failure"); 111422e44211Sjeremylt } 111522e44211Sjeremylt return CEED_ERROR_SUCCESS; 111645f1e315Sjeremylt } 111722e44211Sjeremylt // LCOV_EXCL_STOP 111822e44211Sjeremylt 111922e44211Sjeremylt /** 1120ca94c3ddSJeremy L Thompson @brief Initialize a `Ceed` context to use the specified resource. 11214385fb7fSSebastian Grimberg 1122ca94c3ddSJeremy 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`. 1123b11c1e72Sjeremylt 1124ea61e9acSJeremy L Thompson @param[in] resource Resource to use, e.g., "/cpu/self" 1125ea61e9acSJeremy L Thompson @param[out] ceed The library context 1126b11c1e72Sjeremylt 1127b11c1e72Sjeremylt @return An error code: 0 - success, otherwise - failure 1128dfdf5a53Sjeremylt 11297a982d89SJeremy L. Thompson @ref User 1130ca94c3ddSJeremy L Thompson 1131ca94c3ddSJeremy L Thompson @sa CeedRegister() CeedDestroy() 1132b11c1e72Sjeremylt **/ 1133d7b241e6Sjeremylt int CeedInit(const char *resource, Ceed *ceed) { 11342b730f8bSJeremy L Thompson size_t match_len = 0, match_index = UINT_MAX, match_priority = CEED_MAX_BACKEND_PRIORITY, priority; 1135d7b241e6Sjeremylt 1136fe2413ffSjeremylt // Find matching backend 11376574a04fSJeremy L Thompson CeedCheck(resource, NULL, CEED_ERROR_MAJOR, "No resource provided"); 11382b730f8bSJeremy L Thompson CeedCall(CeedRegisterAll()); 113913873f79Sjeremylt 114022e44211Sjeremylt // Check for help request 114122e44211Sjeremylt const char *help_prefix = "help"; 11422b730f8bSJeremy L Thompson size_t match_help = 0; 11432b730f8bSJeremy L Thompson while (match_help < 4 && resource[match_help] == help_prefix[match_help]) match_help++; 114422e44211Sjeremylt if (match_help == 4) { 11452b730f8bSJeremy L Thompson fprintf(stderr, "libCEED version: %d.%d%d%s\n", CEED_VERSION_MAJOR, CEED_VERSION_MINOR, CEED_VERSION_PATCH, 114622e44211Sjeremylt CEED_VERSION_RELEASE ? "" : "+development"); 114792ee7d1cSjeremylt fprintf(stderr, "Available backend resources:\n"); 114822e44211Sjeremylt for (size_t i = 0; i < num_backends; i++) { 1149d0c91ce9Sjeremylt // Only report compiled backends 11502b730f8bSJeremy L Thompson if (backends[i].priority < CEED_MAX_BACKEND_PRIORITY) fprintf(stderr, " %s\n", backends[i].prefix); 115122e44211Sjeremylt } 115222e44211Sjeremylt fflush(stderr); 115322e44211Sjeremylt match_help = 5; // Delineating character expected 115422e44211Sjeremylt } else { 115522e44211Sjeremylt match_help = 0; 115622e44211Sjeremylt } 115722e44211Sjeremylt 1158ea61e9acSJeremy L Thompson // Find best match, computed as number of matching characters from requested resource stem 11592b730f8bSJeremy L Thompson size_t stem_length = 0; 11602b730f8bSJeremy L Thompson while (resource[stem_length + match_help] && resource[stem_length + match_help] != ':') stem_length++; 1161d7b241e6Sjeremylt for (size_t i = 0; i < num_backends; i++) { 11622b730f8bSJeremy L Thompson size_t n = 0; 1163d7b241e6Sjeremylt const char *prefix = backends[i].prefix; 11642b730f8bSJeremy L Thompson while (prefix[n] && prefix[n] == resource[n + match_help]) n++; 1165d7b241e6Sjeremylt priority = backends[i].priority; 1166d1d35e2fSjeremylt if (n > match_len || (n == match_len && match_priority > priority)) { 1167d1d35e2fSjeremylt match_len = n; 1168d1d35e2fSjeremylt match_priority = priority; 1169f7e22acaSJeremy L Thompson match_index = i; 1170d7b241e6Sjeremylt } 1171d7b241e6Sjeremylt } 11729c9a0587SLeila Ghaffari // Using Levenshtein distance to find closest match 11739c9a0587SLeila Ghaffari if (match_len <= 1 || match_len != stem_length) { 1174203015caSLeila Ghaffari // LCOV_EXCL_START 11759c9a0587SLeila Ghaffari size_t lev_dis = UINT_MAX; 1176f7e22acaSJeremy L Thompson size_t lev_index = UINT_MAX, lev_priority = CEED_MAX_BACKEND_PRIORITY; 11779c9a0587SLeila Ghaffari for (size_t i = 0; i < num_backends; i++) { 11789c9a0587SLeila Ghaffari const char *prefix = backends[i].prefix; 11799c9a0587SLeila Ghaffari size_t prefix_length = strlen(backends[i].prefix); 11809c9a0587SLeila Ghaffari size_t min_len = (prefix_length < stem_length) ? prefix_length : stem_length; 1181092904ddSLeila Ghaffari size_t column[min_len + 1]; 1182092904ddSLeila Ghaffari for (size_t j = 0; j <= min_len; j++) column[j] = j; 11839c9a0587SLeila Ghaffari for (size_t j = 1; j <= min_len; j++) { 11849c9a0587SLeila Ghaffari column[0] = j; 11859c9a0587SLeila Ghaffari for (size_t k = 1, last_diag = j - 1; k <= min_len; k++) { 1186092904ddSLeila Ghaffari size_t old_diag = column[k]; 11879c9a0587SLeila Ghaffari size_t min_1 = (column[k] < column[k - 1]) ? column[k] + 1 : column[k - 1] + 1; 11889c9a0587SLeila Ghaffari size_t min_2 = last_diag + (resource[k - 1] == prefix[j - 1] ? 0 : 1); 11899c9a0587SLeila Ghaffari column[k] = (min_1 < min_2) ? min_1 : min_2; 11909c9a0587SLeila Ghaffari last_diag = old_diag; 11919c9a0587SLeila Ghaffari } 11929c9a0587SLeila Ghaffari } 11939c9a0587SLeila Ghaffari size_t n = column[min_len]; 11949c9a0587SLeila Ghaffari priority = backends[i].priority; 11952b730f8bSJeremy L Thompson if (n < lev_dis || (n == lev_dis && lev_priority > priority)) { 11969c9a0587SLeila Ghaffari lev_dis = n; 11979c9a0587SLeila Ghaffari lev_priority = priority; 1198f7e22acaSJeremy L Thompson lev_index = i; 11999c9a0587SLeila Ghaffari } 12009c9a0587SLeila Ghaffari } 1201f7e22acaSJeremy L Thompson const char *prefix_lev = backends[lev_index].prefix; 12022b730f8bSJeremy L Thompson size_t lev_length = 0; 12032b730f8bSJeremy L Thompson while (prefix_lev[lev_length] && prefix_lev[lev_length] != '\0') lev_length++; 12049c9a0587SLeila Ghaffari size_t m = (lev_length < stem_length) ? lev_length : stem_length; 12056574a04fSJeremy L Thompson if (lev_dis + 1 >= m) return CeedError(NULL, CEED_ERROR_MAJOR, "No suitable backend: %s", resource); 12066574a04fSJeremy L Thompson else return CeedError(NULL, CEED_ERROR_MAJOR, "No suitable backend: %s\nClosest match: %s", resource, backends[lev_index].prefix); 1207203015caSLeila Ghaffari // LCOV_EXCL_STOP 12089c9a0587SLeila Ghaffari } 1209fe2413ffSjeremylt 1210fe2413ffSjeremylt // Setup Ceed 12112b730f8bSJeremy L Thompson CeedCall(CeedCalloc(1, ceed)); 12122b730f8bSJeremy L Thompson CeedCall(CeedCalloc(1, &(*ceed)->jit_source_roots)); 12132027fb9dSSirAlienTheGreat CeedCall(CeedCalloc(1, &(*ceed)->rust_source_roots)); 1214bc81ce41Sjeremylt const char *ceed_error_handler = getenv("CEED_ERROR_HANDLER"); 12152b730f8bSJeremy L Thompson if (!ceed_error_handler) ceed_error_handler = "abort"; 12162b730f8bSJeremy L Thompson if (!strcmp(ceed_error_handler, "exit")) (*ceed)->Error = CeedErrorExit; 12172b730f8bSJeremy L Thompson else if (!strcmp(ceed_error_handler, "store")) (*ceed)->Error = CeedErrorStore; 12182b730f8bSJeremy L Thompson else (*ceed)->Error = CeedErrorAbort; 1219d1d35e2fSjeremylt memcpy((*ceed)->err_msg, "No error message stored", 24); 1220d1d35e2fSjeremylt (*ceed)->ref_count = 1; 1221d7b241e6Sjeremylt (*ceed)->data = NULL; 1222fe2413ffSjeremylt 1223fe2413ffSjeremylt // Set lookup table 1224d1d35e2fSjeremylt FOffset f_offsets[] = { 12256e79d475Sjeremylt CEED_FTABLE_ENTRY(Ceed, Error), 12265ae360d4SJeremy L Thompson CEED_FTABLE_ENTRY(Ceed, SetStream), 12276e79d475Sjeremylt CEED_FTABLE_ENTRY(Ceed, GetPreferredMemType), 12286e79d475Sjeremylt CEED_FTABLE_ENTRY(Ceed, Destroy), 1229f8902d9eSjeremylt CEED_FTABLE_ENTRY(Ceed, VectorCreate), 12306e79d475Sjeremylt CEED_FTABLE_ENTRY(Ceed, ElemRestrictionCreate), 12313ac8f562SJeremy L Thompson CEED_FTABLE_ENTRY(Ceed, ElemRestrictionCreateAtPoints), 12326e79d475Sjeremylt CEED_FTABLE_ENTRY(Ceed, ElemRestrictionCreateBlocked), 12336e79d475Sjeremylt CEED_FTABLE_ENTRY(Ceed, BasisCreateTensorH1), 12346e79d475Sjeremylt CEED_FTABLE_ENTRY(Ceed, BasisCreateH1), 123550c301a5SRezgar Shakeri CEED_FTABLE_ENTRY(Ceed, BasisCreateHdiv), 1236c4e3f59bSSebastian Grimberg CEED_FTABLE_ENTRY(Ceed, BasisCreateHcurl), 12376e79d475Sjeremylt CEED_FTABLE_ENTRY(Ceed, TensorContractCreate), 12386e79d475Sjeremylt CEED_FTABLE_ENTRY(Ceed, QFunctionCreate), 1239777ff853SJeremy L Thompson CEED_FTABLE_ENTRY(Ceed, QFunctionContextCreate), 12406e79d475Sjeremylt CEED_FTABLE_ENTRY(Ceed, OperatorCreate), 124148acf710SJeremy L Thompson CEED_FTABLE_ENTRY(Ceed, OperatorCreateAtPoints), 12426e79d475Sjeremylt CEED_FTABLE_ENTRY(Ceed, CompositeOperatorCreate), 12439c774eddSJeremy L Thompson CEED_FTABLE_ENTRY(CeedVector, HasValidArray), 12449c774eddSJeremy L Thompson CEED_FTABLE_ENTRY(CeedVector, HasBorrowedArrayOfType), 12450b8f3c4eSJeremy L Thompson CEED_FTABLE_ENTRY(CeedVector, CopyStrided), 12466e79d475Sjeremylt CEED_FTABLE_ENTRY(CeedVector, SetArray), 12476a6c615bSJeremy L Thompson CEED_FTABLE_ENTRY(CeedVector, TakeArray), 12486e79d475Sjeremylt CEED_FTABLE_ENTRY(CeedVector, SetValue), 12490b8f3c4eSJeremy L Thompson CEED_FTABLE_ENTRY(CeedVector, SetValueStrided), 1250f48ed27dSnbeams CEED_FTABLE_ENTRY(CeedVector, SyncArray), 12516e79d475Sjeremylt CEED_FTABLE_ENTRY(CeedVector, GetArray), 12526e79d475Sjeremylt CEED_FTABLE_ENTRY(CeedVector, GetArrayRead), 12539c774eddSJeremy L Thompson CEED_FTABLE_ENTRY(CeedVector, GetArrayWrite), 12546e79d475Sjeremylt CEED_FTABLE_ENTRY(CeedVector, RestoreArray), 12556e79d475Sjeremylt CEED_FTABLE_ENTRY(CeedVector, RestoreArrayRead), 1256547d9b97Sjeremylt CEED_FTABLE_ENTRY(CeedVector, Norm), 1257e0dd3b27Sjeremylt CEED_FTABLE_ENTRY(CeedVector, Scale), 12580f7fd0f8Sjeremylt CEED_FTABLE_ENTRY(CeedVector, AXPY), 12595fb68f37SKaren (Ren) Stengel CEED_FTABLE_ENTRY(CeedVector, AXPBY), 12600f7fd0f8Sjeremylt CEED_FTABLE_ENTRY(CeedVector, PointwiseMult), 1261d99fa3c5SJeremy L Thompson CEED_FTABLE_ENTRY(CeedVector, Reciprocal), 12626e79d475Sjeremylt CEED_FTABLE_ENTRY(CeedVector, Destroy), 12636e79d475Sjeremylt CEED_FTABLE_ENTRY(CeedElemRestriction, Apply), 1264f30b1135SSebastian Grimberg CEED_FTABLE_ENTRY(CeedElemRestriction, ApplyUnsigned), 12657c1dbaffSSebastian Grimberg CEED_FTABLE_ENTRY(CeedElemRestriction, ApplyUnoriented), 126605fa913cSJeremy L Thompson CEED_FTABLE_ENTRY(CeedElemRestriction, ApplyAtPointsInElement), 12676e79d475Sjeremylt CEED_FTABLE_ENTRY(CeedElemRestriction, ApplyBlock), 1268bd33150aSjeremylt CEED_FTABLE_ENTRY(CeedElemRestriction, GetOffsets), 126977d1c127SSebastian Grimberg CEED_FTABLE_ENTRY(CeedElemRestriction, GetOrientations), 127077d1c127SSebastian Grimberg CEED_FTABLE_ENTRY(CeedElemRestriction, GetCurlOrientations), 127119605835SJeremy L Thompson CEED_FTABLE_ENTRY(CeedElemRestriction, GetAtPointsElementOffset), 12726e79d475Sjeremylt CEED_FTABLE_ENTRY(CeedElemRestriction, Destroy), 12736e79d475Sjeremylt CEED_FTABLE_ENTRY(CeedBasis, Apply), 1274db2becc9SJeremy L Thompson CEED_FTABLE_ENTRY(CeedBasis, ApplyAdd), 1275c8c3fa7dSJeremy L Thompson CEED_FTABLE_ENTRY(CeedBasis, ApplyAtPoints), 1276db2becc9SJeremy L Thompson CEED_FTABLE_ENTRY(CeedBasis, ApplyAddAtPoints), 12776e79d475Sjeremylt CEED_FTABLE_ENTRY(CeedBasis, Destroy), 12786e79d475Sjeremylt CEED_FTABLE_ENTRY(CeedTensorContract, Apply), 12796e79d475Sjeremylt CEED_FTABLE_ENTRY(CeedTensorContract, Destroy), 12806e79d475Sjeremylt CEED_FTABLE_ENTRY(CeedQFunction, Apply), 12818c84ac63Sjeremylt CEED_FTABLE_ENTRY(CeedQFunction, SetCUDAUserFunction), 12828c84ac63Sjeremylt CEED_FTABLE_ENTRY(CeedQFunction, SetHIPUserFunction), 12836e79d475Sjeremylt CEED_FTABLE_ENTRY(CeedQFunction, Destroy), 12849c774eddSJeremy L Thompson CEED_FTABLE_ENTRY(CeedQFunctionContext, HasValidData), 12859c774eddSJeremy L Thompson CEED_FTABLE_ENTRY(CeedQFunctionContext, HasBorrowedDataOfType), 1286777ff853SJeremy L Thompson CEED_FTABLE_ENTRY(CeedQFunctionContext, SetData), 1287891038deSjeremylt CEED_FTABLE_ENTRY(CeedQFunctionContext, TakeData), 1288777ff853SJeremy L Thompson CEED_FTABLE_ENTRY(CeedQFunctionContext, GetData), 128928bfd0b7SJeremy L Thompson CEED_FTABLE_ENTRY(CeedQFunctionContext, GetDataRead), 1290777ff853SJeremy L Thompson CEED_FTABLE_ENTRY(CeedQFunctionContext, RestoreData), 129128bfd0b7SJeremy L Thompson CEED_FTABLE_ENTRY(CeedQFunctionContext, RestoreDataRead), 12922e64a2b9SJeremy L Thompson CEED_FTABLE_ENTRY(CeedQFunctionContext, DataDestroy), 1293777ff853SJeremy L Thompson CEED_FTABLE_ENTRY(CeedQFunctionContext, Destroy), 129480ac2e43SJeremy L Thompson CEED_FTABLE_ENTRY(CeedOperator, LinearAssembleQFunction), 129570a7ffb3SJeremy L Thompson CEED_FTABLE_ENTRY(CeedOperator, LinearAssembleQFunctionUpdate), 129680ac2e43SJeremy L Thompson CEED_FTABLE_ENTRY(CeedOperator, LinearAssembleDiagonal), 12979e9210b8SJeremy L Thompson CEED_FTABLE_ENTRY(CeedOperator, LinearAssembleAddDiagonal), 129880ac2e43SJeremy L Thompson CEED_FTABLE_ENTRY(CeedOperator, LinearAssemblePointBlockDiagonal), 12999e9210b8SJeremy L Thompson CEED_FTABLE_ENTRY(CeedOperator, LinearAssembleAddPointBlockDiagonal), 1300e2f04181SAndrew T. Barker CEED_FTABLE_ENTRY(CeedOperator, LinearAssembleSymbolic), 1301e2f04181SAndrew T. Barker CEED_FTABLE_ENTRY(CeedOperator, LinearAssemble), 1302cefa2673SJeremy L Thompson CEED_FTABLE_ENTRY(CeedOperator, LinearAssembleSingle), 1303713f43c3Sjeremylt CEED_FTABLE_ENTRY(CeedOperator, CreateFDMElementInverse), 13046e79d475Sjeremylt CEED_FTABLE_ENTRY(CeedOperator, Apply), 1305250756a7Sjeremylt CEED_FTABLE_ENTRY(CeedOperator, ApplyComposite), 1306cae8b89aSjeremylt CEED_FTABLE_ENTRY(CeedOperator, ApplyAdd), 1307250756a7Sjeremylt CEED_FTABLE_ENTRY(CeedOperator, ApplyAddComposite), 13086e79d475Sjeremylt CEED_FTABLE_ENTRY(CeedOperator, ApplyJacobian), 13096e79d475Sjeremylt CEED_FTABLE_ENTRY(CeedOperator, Destroy), 13106e79d475Sjeremylt {NULL, 0} // End of lookup table - used in SetBackendFunction loop 13111dfeef1dSjeremylt }; 1312fe2413ffSjeremylt 13132b730f8bSJeremy L Thompson CeedCall(CeedCalloc(sizeof(f_offsets), &(*ceed)->f_offsets)); 1314d1d35e2fSjeremylt memcpy((*ceed)->f_offsets, f_offsets, sizeof(f_offsets)); 1315fe2413ffSjeremylt 131660f9e2d6SJeremy L Thompson // Record env variables CEED_DEBUG or DBG 13171c66c397SJeremy L Thompson (*ceed)->is_debug = getenv("CEED_DEBUG") || getenv("DEBUG") || getenv("DBG"); 131860f9e2d6SJeremy L Thompson 131922e44211Sjeremylt // Copy resource prefix, if backend setup successful 13202b730f8bSJeremy L Thompson CeedCall(CeedStringAllocCopy(backends[match_index].prefix, (char **)&(*ceed)->resource)); 1321ee5a26f2SJeremy L Thompson 1322ee5a26f2SJeremy L Thompson // Set default JiT source root 1323ea61e9acSJeremy L Thompson // Note: there will always be the default root for every Ceed but all additional paths are added to the top-most parent 13242b730f8bSJeremy L Thompson CeedCall(CeedAddJitSourceRoot(*ceed, (char *)CeedJitSourceRootDefault)); 1325ee5a26f2SJeremy L Thompson 13262027fb9dSSirAlienTheGreat // By default, make cuda compile without clang, use nvrtc instead 13272027fb9dSSirAlienTheGreat // Note that this is overridden if a rust file is included (rust requires clang) 13282027fb9dSSirAlienTheGreat const char *env = getenv("GPU_CLANG"); 13292027fb9dSSirAlienTheGreat 13302027fb9dSSirAlienTheGreat if (env && strcmp(env, "1") == 0) { 13312027fb9dSSirAlienTheGreat (*ceed)->cuda_compile_with_clang = true; 13322027fb9dSSirAlienTheGreat } else { 13332027fb9dSSirAlienTheGreat (*ceed)->cuda_compile_with_clang = false; 13342027fb9dSSirAlienTheGreat } 13352027fb9dSSirAlienTheGreat 1336d04bbc78SJeremy L Thompson // Backend specific setup 13372b730f8bSJeremy L Thompson CeedCall(backends[match_index].init(&resource[match_help], *ceed)); 1338e15f9bd0SJeremy L Thompson return CEED_ERROR_SUCCESS; 1339d7b241e6Sjeremylt } 1340d7b241e6Sjeremylt 1341d7b241e6Sjeremylt /** 1342ca94c3ddSJeremy L Thompson @brief Set the GPU stream for a `Ceed` context 13435ae360d4SJeremy L Thompson 1344ca94c3ddSJeremy L Thompson @param[in,out] ceed `Ceed` context to set the stream 13455ae360d4SJeremy L Thompson @param[in] handle Handle to GPU stream 13465ae360d4SJeremy L Thompson 13475ae360d4SJeremy L Thompson @return An error code: 0 - success, otherwise - failure 13485ae360d4SJeremy L Thompson 13495ae360d4SJeremy L Thompson @ref User 13505ae360d4SJeremy L Thompson **/ 13515ae360d4SJeremy L Thompson int CeedSetStream(Ceed ceed, void *handle) { 1352ca94c3ddSJeremy L Thompson CeedCheck(handle, ceed, CEED_ERROR_INCOMPATIBLE, "Stream handle must be non-NULL"); 13539ffb25e0SJames Wright if (ceed->SetStream) { 13545ae360d4SJeremy L Thompson CeedCall(ceed->SetStream(ceed, handle)); 13559ffb25e0SJames Wright } else { 13569ffb25e0SJames Wright Ceed delegate; 13579ffb25e0SJames Wright CeedCall(CeedGetDelegate(ceed, &delegate)); 13585ae360d4SJeremy L Thompson 135928ce3d2aSJeremy L Thompson if (delegate) CeedCall(CeedSetStream(delegate, handle)); 136028ce3d2aSJeremy L Thompson else return CeedError(ceed, CEED_ERROR_UNSUPPORTED, "Backend does not support setting stream"); 13619bc66399SJeremy L Thompson CeedCall(CeedDestroy(&delegate)); 13629ffb25e0SJames Wright } 13635ae360d4SJeremy L Thompson return CEED_ERROR_SUCCESS; 13645ae360d4SJeremy L Thompson } 13655ae360d4SJeremy L Thompson 13665ae360d4SJeremy L Thompson /** 1367ca94c3ddSJeremy L Thompson @brief Copy the pointer to a `Ceed` context. 13684385fb7fSSebastian Grimberg 1369ca94c3ddSJeremy L Thompson Both pointers should be destroyed with @ref CeedDestroy(). 1370512bb800SJeremy L Thompson 1371ca94c3ddSJeremy 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. 1372ca94c3ddSJeremy L Thompson This `Ceed` context will be destroyed if `*ceed_copy` is the only reference to this `Ceed` context. 13739560d06aSjeremylt 1374ca94c3ddSJeremy L Thompson @param[in] ceed `Ceed` context to copy reference to 1375ea61e9acSJeremy L Thompson @param[in,out] ceed_copy Variable to store copied reference 13769560d06aSjeremylt 13779560d06aSjeremylt @return An error code: 0 - success, otherwise - failure 13789560d06aSjeremylt 13799560d06aSjeremylt @ref User 13809560d06aSjeremylt **/ 13819560d06aSjeremylt int CeedReferenceCopy(Ceed ceed, Ceed *ceed_copy) { 13822b730f8bSJeremy L Thompson CeedCall(CeedReference(ceed)); 13832b730f8bSJeremy L Thompson CeedCall(CeedDestroy(ceed_copy)); 13849560d06aSjeremylt *ceed_copy = ceed; 13859560d06aSjeremylt return CEED_ERROR_SUCCESS; 13869560d06aSjeremylt } 13879560d06aSjeremylt 13889560d06aSjeremylt /** 1389ca94c3ddSJeremy L Thompson @brief Get the full resource name for a `Ceed` context 13902f86a920SJeremy L Thompson 1391ca94c3ddSJeremy L Thompson @param[in] ceed `Ceed` context to get resource name of 13927a982d89SJeremy L. Thompson @param[out] resource Variable to store resource name 13932f86a920SJeremy L Thompson 13942f86a920SJeremy L Thompson @return An error code: 0 - success, otherwise - failure 13952f86a920SJeremy L Thompson 13967a982d89SJeremy L. Thompson @ref User 13975107b09fSJeremy L Thompson **/ 13987a982d89SJeremy L. Thompson int CeedGetResource(Ceed ceed, const char **resource) { 13997a982d89SJeremy L. Thompson *resource = (const char *)ceed->resource; 1400e15f9bd0SJeremy L Thompson return CEED_ERROR_SUCCESS; 14015107b09fSJeremy L Thompson } 14025107b09fSJeremy L Thompson 14035107b09fSJeremy L Thompson /** 1404ca94c3ddSJeremy L Thompson @brief Return `Ceed` context preferred memory type 1405c907536fSjeremylt 1406ca94c3ddSJeremy L Thompson @param[in] ceed `Ceed` context to get preferred memory type of 1407d1d35e2fSjeremylt @param[out] mem_type Address to save preferred memory type to 1408c907536fSjeremylt 1409c907536fSjeremylt @return An error code: 0 - success, otherwise - failure 1410c907536fSjeremylt 14117a982d89SJeremy L. Thompson @ref User 1412c907536fSjeremylt **/ 1413d1d35e2fSjeremylt int CeedGetPreferredMemType(Ceed ceed, CeedMemType *mem_type) { 1414c907536fSjeremylt if (ceed->GetPreferredMemType) { 14152b730f8bSJeremy L Thompson CeedCall(ceed->GetPreferredMemType(mem_type)); 1416c907536fSjeremylt } else { 1417c263cd57Sjeremylt Ceed delegate; 14182b730f8bSJeremy L Thompson CeedCall(CeedGetDelegate(ceed, &delegate)); 1419c263cd57Sjeremylt 1420c263cd57Sjeremylt if (delegate) { 14212b730f8bSJeremy L Thompson CeedCall(CeedGetPreferredMemType(delegate, mem_type)); 1422c263cd57Sjeremylt } else { 1423d1d35e2fSjeremylt *mem_type = CEED_MEM_HOST; 1424c907536fSjeremylt } 14259bc66399SJeremy L Thompson CeedCall(CeedDestroy(&delegate)); 1426c263cd57Sjeremylt } 1427e15f9bd0SJeremy L Thompson return CEED_ERROR_SUCCESS; 1428c907536fSjeremylt } 1429c907536fSjeremylt 1430c907536fSjeremylt /** 1431ca94c3ddSJeremy L Thompson @brief Get deterministic status of `Ceed` context 14329525855cSJeremy L Thompson 1433ca94c3ddSJeremy L Thompson @param[in] ceed `Ceed` context 1434d1d35e2fSjeremylt @param[out] is_deterministic Variable to store deterministic status 14359525855cSJeremy L Thompson 14369525855cSJeremy L Thompson @return An error code: 0 - success, otherwise - failure 14379525855cSJeremy L Thompson 14389525855cSJeremy L Thompson @ref User 14399525855cSJeremy L Thompson **/ 1440d1d35e2fSjeremylt int CeedIsDeterministic(Ceed ceed, bool *is_deterministic) { 1441d1d35e2fSjeremylt *is_deterministic = ceed->is_deterministic; 1442e15f9bd0SJeremy L Thompson return CEED_ERROR_SUCCESS; 14439525855cSJeremy L Thompson } 14449525855cSJeremy L Thompson 14459525855cSJeremy L Thompson /** 1446ca94c3ddSJeremy L Thompson @brief Set additional JiT source root for `Ceed` context 1447ee5a26f2SJeremy L Thompson 1448ca94c3ddSJeremy L Thompson @param[in,out] ceed `Ceed` context 1449ee5a26f2SJeremy L Thompson @param[in] jit_source_root Absolute path to additional JiT source directory 1450ee5a26f2SJeremy L Thompson 1451ee5a26f2SJeremy L Thompson @return An error code: 0 - success, otherwise - failure 1452ee5a26f2SJeremy L Thompson 1453ee5a26f2SJeremy L Thompson @ref User 1454ee5a26f2SJeremy L Thompson **/ 1455ee5a26f2SJeremy L Thompson int CeedAddJitSourceRoot(Ceed ceed, const char *jit_source_root) { 14566155f12fSJeremy L Thompson Ceed ceed_parent; 1457ee5a26f2SJeremy L Thompson 14582b730f8bSJeremy L Thompson CeedCall(CeedGetParent(ceed, &ceed_parent)); 1459830fc37bSJeremy 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"); 14606155f12fSJeremy L Thompson 14616155f12fSJeremy L Thompson CeedInt index = ceed_parent->num_jit_source_roots; 1462ee5a26f2SJeremy L Thompson size_t path_length = strlen(jit_source_root); 14631c66c397SJeremy L Thompson 14644753b775SJeremy L Thompson if (ceed_parent->num_jit_source_roots == ceed_parent->max_jit_source_roots) { 14654753b775SJeremy L Thompson if (ceed_parent->max_jit_source_roots == 0) ceed_parent->max_jit_source_roots = 1; 14664753b775SJeremy L Thompson ceed_parent->max_jit_source_roots *= 2; 14674753b775SJeremy L Thompson CeedCall(CeedRealloc(ceed_parent->max_jit_source_roots, &ceed_parent->jit_source_roots)); 14684753b775SJeremy L Thompson } 14692b730f8bSJeremy L Thompson CeedCall(CeedCalloc(path_length + 1, &ceed_parent->jit_source_roots[index])); 1470d602d780SJeremy L Thompson memcpy(ceed_parent->jit_source_roots[index], jit_source_root, path_length); 14716155f12fSJeremy L Thompson ceed_parent->num_jit_source_roots++; 14729bc66399SJeremy L Thompson CeedCall(CeedDestroy(&ceed_parent)); 1473ee5a26f2SJeremy L Thompson return CEED_ERROR_SUCCESS; 1474ee5a26f2SJeremy L Thompson } 1475ee5a26f2SJeremy L Thompson 1476ee5a26f2SJeremy L Thompson /** 14772027fb9dSSirAlienTheGreat @brief Set additional Rust source root for `Ceed` context for use in QFunction 14782027fb9dSSirAlienTheGreat 14792027fb9dSSirAlienTheGreat @param[in,out] ceed `Ceed` context 14802027fb9dSSirAlienTheGreat @param[in] rust_source_root Absolute path to additional Rust source directory 14812027fb9dSSirAlienTheGreat 14822027fb9dSSirAlienTheGreat @return An error code: 0 - success, otherwise - failure 14832027fb9dSSirAlienTheGreat 14842027fb9dSSirAlienTheGreat @ref User 14852027fb9dSSirAlienTheGreat **/ 14862027fb9dSSirAlienTheGreat int CeedAddRustSourceRoot(Ceed ceed, const char *rust_source_root) { 14872027fb9dSSirAlienTheGreat Ceed ceed_parent; 14882027fb9dSSirAlienTheGreat 14892027fb9dSSirAlienTheGreat CeedCall(CeedGetParent(ceed, &ceed_parent)); 14902027fb9dSSirAlienTheGreat CeedCheck(!ceed_parent->num_rust_source_roots_readers, ceed, CEED_ERROR_ACCESS, "Cannot add Rust source root, read access has not been restored"); 14912027fb9dSSirAlienTheGreat 14922027fb9dSSirAlienTheGreat CeedInt index = ceed_parent->num_rust_source_roots; 14932027fb9dSSirAlienTheGreat size_t path_length = strlen(rust_source_root); 14942027fb9dSSirAlienTheGreat 14952027fb9dSSirAlienTheGreat if (ceed_parent->num_rust_source_roots == ceed_parent->max_rust_source_roots) { 14962027fb9dSSirAlienTheGreat if (ceed_parent->max_rust_source_roots == 0) ceed_parent->max_rust_source_roots = 1; 14972027fb9dSSirAlienTheGreat ceed_parent->max_rust_source_roots *= 2; 14982027fb9dSSirAlienTheGreat CeedCall(CeedRealloc(ceed_parent->max_rust_source_roots, &ceed_parent->rust_source_roots)); 14992027fb9dSSirAlienTheGreat } 15002027fb9dSSirAlienTheGreat CeedCall(CeedCalloc(path_length + 1, &ceed_parent->rust_source_roots[index])); 15012027fb9dSSirAlienTheGreat memcpy(ceed_parent->rust_source_roots[index], rust_source_root, path_length); 15022027fb9dSSirAlienTheGreat ceed_parent->num_rust_source_roots++; 15032027fb9dSSirAlienTheGreat ceed_parent->cuda_compile_with_clang = true; 15042027fb9dSSirAlienTheGreat ceed->cuda_compile_with_clang = true; 15052027fb9dSSirAlienTheGreat CeedCall(CeedDestroy(&ceed_parent)); 15062027fb9dSSirAlienTheGreat return CEED_ERROR_SUCCESS; 15072027fb9dSSirAlienTheGreat } 15082027fb9dSSirAlienTheGreat 15092027fb9dSSirAlienTheGreat /** 15104753b775SJeremy L Thompson @brief Set additional JiT compiler define for `Ceed` context 15114753b775SJeremy L Thompson 15124753b775SJeremy L Thompson @param[in,out] ceed `Ceed` context 15134753b775SJeremy L Thompson @param[in] jit_define String such as `foo=bar`, used as `-Dfoo=bar` in JiT 15144753b775SJeremy L Thompson 15154753b775SJeremy L Thompson @return An error code: 0 - success, otherwise - failure 15164753b775SJeremy L Thompson 15174753b775SJeremy L Thompson @ref User 15184753b775SJeremy L Thompson **/ 15194753b775SJeremy L Thompson int CeedAddJitDefine(Ceed ceed, const char *jit_define) { 15204753b775SJeremy L Thompson Ceed ceed_parent; 15214753b775SJeremy L Thompson 15224753b775SJeremy L Thompson CeedCall(CeedGetParent(ceed, &ceed_parent)); 1523830fc37bSJeremy L Thompson CeedCheck(!ceed_parent->num_jit_defines_readers, ceed, CEED_ERROR_ACCESS, "Cannot add JiT define, read access has not been restored"); 15244753b775SJeremy L Thompson 15254753b775SJeremy L Thompson CeedInt index = ceed_parent->num_jit_defines; 15264753b775SJeremy L Thompson size_t define_length = strlen(jit_define); 15274753b775SJeremy L Thompson 15284753b775SJeremy L Thompson if (ceed_parent->num_jit_defines == ceed_parent->max_jit_defines) { 15294753b775SJeremy L Thompson if (ceed_parent->max_jit_defines == 0) ceed_parent->max_jit_defines = 1; 15304753b775SJeremy L Thompson ceed_parent->max_jit_defines *= 2; 15314753b775SJeremy L Thompson CeedCall(CeedRealloc(ceed_parent->max_jit_defines, &ceed_parent->jit_defines)); 15324753b775SJeremy L Thompson } 15334753b775SJeremy L Thompson CeedCall(CeedCalloc(define_length + 1, &ceed_parent->jit_defines[index])); 15344753b775SJeremy L Thompson memcpy(ceed_parent->jit_defines[index], jit_define, define_length); 15354753b775SJeremy L Thompson ceed_parent->num_jit_defines++; 15369bc66399SJeremy L Thompson CeedCall(CeedDestroy(&ceed_parent)); 15374753b775SJeremy L Thompson return CEED_ERROR_SUCCESS; 15384753b775SJeremy L Thompson } 15394753b775SJeremy L Thompson 15404753b775SJeremy L Thompson /** 1541ca94c3ddSJeremy L Thompson @brief View a `Ceed` 15420a0da059Sjeremylt 1543ca94c3ddSJeremy L Thompson @param[in] ceed `Ceed` to view 15440a0da059Sjeremylt @param[in] stream Filestream to write to 15450a0da059Sjeremylt 15460a0da059Sjeremylt @return An error code: 0 - success, otherwise - failure 15470a0da059Sjeremylt 15480a0da059Sjeremylt @ref User 15490a0da059Sjeremylt **/ 15500a0da059Sjeremylt int CeedView(Ceed ceed, FILE *stream) { 1551d1d35e2fSjeremylt CeedMemType mem_type; 15520a0da059Sjeremylt 15532b730f8bSJeremy L Thompson CeedCall(CeedGetPreferredMemType(ceed, &mem_type)); 15540a0da059Sjeremylt 15552b730f8bSJeremy L Thompson fprintf(stream, 15562b730f8bSJeremy L Thompson "Ceed\n" 15570a0da059Sjeremylt " Ceed Resource: %s\n" 15580a0da059Sjeremylt " Preferred MemType: %s\n", 1559d1d35e2fSjeremylt ceed->resource, CeedMemTypes[mem_type]); 1560e15f9bd0SJeremy L Thompson return CEED_ERROR_SUCCESS; 15610a0da059Sjeremylt } 15620a0da059Sjeremylt 15630a0da059Sjeremylt /** 1564ca94c3ddSJeremy L Thompson @brief Destroy a `Ceed` 1565d7b241e6Sjeremylt 1566ca94c3ddSJeremy L Thompson @param[in,out] ceed Address of `Ceed` context to destroy 1567b11c1e72Sjeremylt 1568b11c1e72Sjeremylt @return An error code: 0 - success, otherwise - failure 1569dfdf5a53Sjeremylt 15707a982d89SJeremy L. Thompson @ref User 1571b11c1e72Sjeremylt **/ 1572d7b241e6Sjeremylt int CeedDestroy(Ceed *ceed) { 1573ad6481ceSJeremy L Thompson if (!*ceed || --(*ceed)->ref_count > 0) { 1574ad6481ceSJeremy L Thompson *ceed = NULL; 1575ad6481ceSJeremy L Thompson return CEED_ERROR_SUCCESS; 1576ad6481ceSJeremy L Thompson } 1577aeb3a72dSJeremy L Thompson 1578aeb3a72dSJeremy L Thompson CeedCheck(!(*ceed)->num_jit_source_roots_readers, *ceed, CEED_ERROR_ACCESS, 1579aeb3a72dSJeremy L Thompson "Cannot destroy ceed context, read access for JiT source roots has been granted"); 1580aeb3a72dSJeremy 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"); 1581aeb3a72dSJeremy L Thompson 15822b730f8bSJeremy L Thompson if ((*ceed)->delegate) CeedCall(CeedDestroy(&(*ceed)->delegate)); 15830ace9bf2Sjeremylt 1584d1d35e2fSjeremylt if ((*ceed)->obj_delegate_count > 0) { 158592ae7e47SJeremy L Thompson for (CeedInt i = 0; i < (*ceed)->obj_delegate_count; i++) { 15862b730f8bSJeremy L Thompson CeedCall(CeedDestroy(&((*ceed)->obj_delegates[i].delegate))); 15872b730f8bSJeremy L Thompson CeedCall(CeedFree(&(*ceed)->obj_delegates[i].obj_name)); 1588aefd8378Sjeremylt } 15892b730f8bSJeremy L Thompson CeedCall(CeedFree(&(*ceed)->obj_delegates)); 1590aefd8378Sjeremylt } 15910ace9bf2Sjeremylt 15922b730f8bSJeremy L Thompson if ((*ceed)->Destroy) CeedCall((*ceed)->Destroy(*ceed)); 15930ace9bf2Sjeremylt 159492ae7e47SJeremy L Thompson for (CeedInt i = 0; i < (*ceed)->num_jit_source_roots; i++) { 15952b730f8bSJeremy L Thompson CeedCall(CeedFree(&(*ceed)->jit_source_roots[i])); 1596032e71eaSJeremy L Thompson } 15972b730f8bSJeremy L Thompson CeedCall(CeedFree(&(*ceed)->jit_source_roots)); 1598032e71eaSJeremy L Thompson 15994753b775SJeremy L Thompson for (CeedInt i = 0; i < (*ceed)->num_jit_defines; i++) { 16004753b775SJeremy L Thompson CeedCall(CeedFree(&(*ceed)->jit_defines[i])); 16014753b775SJeremy L Thompson } 16024753b775SJeremy L Thompson CeedCall(CeedFree(&(*ceed)->jit_defines)); 16034753b775SJeremy L Thompson 1604*898eb931SJeremy L Thompson for (CeedInt i = 0; i < (*ceed)->num_rust_source_roots; i++) { 1605*898eb931SJeremy L Thompson CeedCall(CeedFree(&(*ceed)->rust_source_roots[i])); 1606*898eb931SJeremy L Thompson } 1607*898eb931SJeremy L Thompson CeedCall(CeedFree(&(*ceed)->rust_source_roots)); 1608*898eb931SJeremy L Thompson 16092b730f8bSJeremy L Thompson CeedCall(CeedFree(&(*ceed)->f_offsets)); 16102b730f8bSJeremy L Thompson CeedCall(CeedFree(&(*ceed)->resource)); 16112b730f8bSJeremy L Thompson CeedCall(CeedDestroy(&(*ceed)->op_fallback_ceed)); 161273501bfeSJeremy L Thompson CeedCall(CeedWorkVectorsDestroy(*ceed)); 16132b730f8bSJeremy L Thompson CeedCall(CeedFree(ceed)); 1614e15f9bd0SJeremy L Thompson return CEED_ERROR_SUCCESS; 1615d7b241e6Sjeremylt } 1616d7b241e6Sjeremylt 1617f9982c62SWill Pazner // LCOV_EXCL_START 1618f9982c62SWill Pazner const char *CeedErrorFormat(Ceed ceed, const char *format, va_list *args) { 16192b730f8bSJeremy L Thompson if (ceed->parent) return CeedErrorFormat(ceed->parent, format, args); 162078464608Sjeremylt // Using pointer to va_list for better FFI, but clang-tidy can't verify va_list is initalized 162178464608Sjeremylt vsnprintf(ceed->err_msg, CEED_MAX_RESOURCE_LEN, format, *args); // NOLINT 1622d1d35e2fSjeremylt return ceed->err_msg; 1623f9982c62SWill Pazner } 1624f9982c62SWill Pazner // LCOV_EXCL_STOP 1625f9982c62SWill Pazner 16267a982d89SJeremy L. Thompson /** 1627ca94c3ddSJeremy L Thompson @brief Error handling implementation; use @ref CeedError() instead. 1628ca94c3ddSJeremy L Thompson 1629ca94c3ddSJeremy L Thompson @return An error code: 0 - success, otherwise - failure 16307a982d89SJeremy L. Thompson 16317a982d89SJeremy L. Thompson @ref Developer 16327a982d89SJeremy L. Thompson **/ 16332b730f8bSJeremy L Thompson int CeedErrorImpl(Ceed ceed, const char *filename, int lineno, const char *func, int ecode, const char *format, ...) { 16347a982d89SJeremy L. Thompson va_list args; 1635d1d35e2fSjeremylt int ret_val; 16361c66c397SJeremy L Thompson 16377a982d89SJeremy L. Thompson va_start(args, format); 16387a982d89SJeremy L. Thompson if (ceed) { 1639d1d35e2fSjeremylt ret_val = ceed->Error(ceed, filename, lineno, func, ecode, format, &args); 16407a982d89SJeremy L. Thompson } else { 1641b0d62198Sjeremylt // LCOV_EXCL_START 1642477729cfSJeremy L Thompson const char *ceed_error_handler = getenv("CEED_ERROR_HANDLER"); 16432b730f8bSJeremy L Thompson if (!ceed_error_handler) ceed_error_handler = "abort"; 1644bcbe1c99SJeremy L Thompson if (!strcmp(ceed_error_handler, "return")) { 1645bcbe1c99SJeremy L Thompson ret_val = CeedErrorReturn(ceed, filename, lineno, func, ecode, format, &args); 1646bcbe1c99SJeremy L Thompson } else { 1647477729cfSJeremy L Thompson // This function will not return 1648d1d35e2fSjeremylt ret_val = CeedErrorAbort(ceed, filename, lineno, func, ecode, format, &args); 16497a982d89SJeremy L. Thompson } 1650bcbe1c99SJeremy L Thompson } 16517a982d89SJeremy L. Thompson va_end(args); 1652d1d35e2fSjeremylt return ret_val; 1653b0d62198Sjeremylt // LCOV_EXCL_STOP 16547a982d89SJeremy L. Thompson } 16557a982d89SJeremy L. Thompson 1656477729cfSJeremy L Thompson /** 1657477729cfSJeremy L Thompson @brief Error handler that returns without printing anything. 1658477729cfSJeremy L Thompson 1659ca94c3ddSJeremy L Thompson Pass this to @ref CeedSetErrorHandler() to obtain this error handling behavior. 1660ca94c3ddSJeremy L Thompson 1661ca94c3ddSJeremy L Thompson @return An error code: 0 - success, otherwise - failure 1662477729cfSJeremy L Thompson 1663477729cfSJeremy L Thompson @ref Developer 1664477729cfSJeremy L Thompson **/ 1665477729cfSJeremy L Thompson // LCOV_EXCL_START 16662b730f8bSJeremy L Thompson int CeedErrorReturn(Ceed ceed, const char *filename, int line_no, const char *func, int err_code, const char *format, va_list *args) { 1667d1d35e2fSjeremylt return err_code; 1668477729cfSJeremy L Thompson } 1669477729cfSJeremy L Thompson // LCOV_EXCL_STOP 1670477729cfSJeremy L Thompson 1671477729cfSJeremy L Thompson /** 1672ea61e9acSJeremy L Thompson @brief Error handler that stores the error message for future use and returns the error. 1673477729cfSJeremy L Thompson 1674ca94c3ddSJeremy L Thompson Pass this to @ref CeedSetErrorHandler() to obtain this error handling behavior. 1675ca94c3ddSJeremy L Thompson 1676ca94c3ddSJeremy L Thompson @return An error code: 0 - success, otherwise - failure 1677477729cfSJeremy L Thompson 1678477729cfSJeremy L Thompson @ref Developer 1679477729cfSJeremy L Thompson **/ 1680477729cfSJeremy L Thompson // LCOV_EXCL_START 16812b730f8bSJeremy L Thompson int CeedErrorStore(Ceed ceed, const char *filename, int line_no, const char *func, int err_code, const char *format, va_list *args) { 16822b730f8bSJeremy L Thompson if (ceed->parent) return CeedErrorStore(ceed->parent, filename, line_no, func, err_code, format, args); 1683477729cfSJeremy L Thompson 1684477729cfSJeremy L Thompson // Build message 16851c66c397SJeremy L Thompson int len = snprintf(ceed->err_msg, CEED_MAX_RESOURCE_LEN, "%s:%d in %s(): ", filename, line_no, func); 168678464608Sjeremylt // Using pointer to va_list for better FFI, but clang-tidy can't verify va_list is initalized 168778464608Sjeremylt vsnprintf(ceed->err_msg + len, CEED_MAX_RESOURCE_LEN - len, format, *args); // NOLINT 1688d1d35e2fSjeremylt return err_code; 1689477729cfSJeremy L Thompson } 1690477729cfSJeremy L Thompson // LCOV_EXCL_STOP 1691477729cfSJeremy L Thompson 1692477729cfSJeremy L Thompson /** 1693ca94c3ddSJeremy L Thompson @brief Error handler that prints to `stderr` and aborts 1694477729cfSJeremy L Thompson 1695ca94c3ddSJeremy L Thompson Pass this to @ref CeedSetErrorHandler() to obtain this error handling behavior. 1696ca94c3ddSJeremy L Thompson 1697ca94c3ddSJeremy L Thompson @return An error code: 0 - success, otherwise - failure 1698477729cfSJeremy L Thompson 1699477729cfSJeremy L Thompson @ref Developer 1700477729cfSJeremy L Thompson **/ 1701477729cfSJeremy L Thompson // LCOV_EXCL_START 17022b730f8bSJeremy L Thompson int CeedErrorAbort(Ceed ceed, const char *filename, int line_no, const char *func, int err_code, const char *format, va_list *args) { 1703d1d35e2fSjeremylt fprintf(stderr, "%s:%d in %s(): ", filename, line_no, func); 1704f9982c62SWill Pazner vfprintf(stderr, format, *args); 1705477729cfSJeremy L Thompson fprintf(stderr, "\n"); 1706477729cfSJeremy L Thompson abort(); 1707d1d35e2fSjeremylt return err_code; 1708477729cfSJeremy L Thompson } 1709477729cfSJeremy L Thompson // LCOV_EXCL_STOP 1710477729cfSJeremy L Thompson 1711477729cfSJeremy L Thompson /** 1712ca94c3ddSJeremy L Thompson @brief Error handler that prints to `stderr` and exits. 1713477729cfSJeremy L Thompson 1714ca94c3ddSJeremy L Thompson Pass this to @ref CeedSetErrorHandler() to obtain this error handling behavior. 1715477729cfSJeremy L Thompson 1716ca94c3ddSJeremy L Thompson In contrast to @ref CeedErrorAbort(), this exits without a signal, so `atexit()` handlers (e.g., as used by gcov) are run. 1717ca94c3ddSJeremy L Thompson 1718ca94c3ddSJeremy L Thompson @return An error code: 0 - success, otherwise - failure 1719477729cfSJeremy L Thompson 1720477729cfSJeremy L Thompson @ref Developer 1721477729cfSJeremy L Thompson **/ 17222b730f8bSJeremy L Thompson int CeedErrorExit(Ceed ceed, const char *filename, int line_no, const char *func, int err_code, const char *format, va_list *args) { 1723d1d35e2fSjeremylt fprintf(stderr, "%s:%d in %s(): ", filename, line_no, func); 172478464608Sjeremylt // Using pointer to va_list for better FFI, but clang-tidy can't verify va_list is initalized 172578464608Sjeremylt vfprintf(stderr, format, *args); // NOLINT 1726477729cfSJeremy L Thompson fprintf(stderr, "\n"); 1727d1d35e2fSjeremylt exit(err_code); 1728d1d35e2fSjeremylt return err_code; 1729477729cfSJeremy L Thompson } 1730477729cfSJeremy L Thompson 1731477729cfSJeremy L Thompson /** 1732477729cfSJeremy L Thompson @brief Set error handler 1733477729cfSJeremy L Thompson 1734ca94c3ddSJeremy L Thompson A default error handler is set in @ref CeedInit(). 1735ca94c3ddSJeremy L Thompson Use this function to change the error handler to @ref CeedErrorReturn(), @ref CeedErrorAbort(), or a user-defined error handler. 1736ca94c3ddSJeremy L Thompson 1737ca94c3ddSJeremy L Thompson @return An error code: 0 - success, otherwise - failure 1738477729cfSJeremy L Thompson 1739477729cfSJeremy L Thompson @ref Developer 1740477729cfSJeremy L Thompson **/ 1741d1d35e2fSjeremylt int CeedSetErrorHandler(Ceed ceed, CeedErrorHandler handler) { 1742d1d35e2fSjeremylt ceed->Error = handler; 1743d1d35e2fSjeremylt if (ceed->delegate) CeedSetErrorHandler(ceed->delegate, handler); 17442b730f8bSJeremy L Thompson for (CeedInt i = 0; i < ceed->obj_delegate_count; i++) CeedSetErrorHandler(ceed->obj_delegates[i].delegate, handler); 1745e15f9bd0SJeremy L Thompson return CEED_ERROR_SUCCESS; 1746477729cfSJeremy L Thompson } 1747477729cfSJeremy L Thompson 1748477729cfSJeremy L Thompson /** 1749477729cfSJeremy L Thompson @brief Get error message 1750477729cfSJeremy L Thompson 1751ca94c3ddSJeremy L Thompson The error message is only stored when using the error handler @ref CeedErrorStore() 1752477729cfSJeremy L Thompson 1753ca94c3ddSJeremy L Thompson @param[in] ceed `Ceed` context to retrieve error message 1754d1d35e2fSjeremylt @param[out] err_msg Char pointer to hold error message 1755477729cfSJeremy L Thompson 1756ca94c3ddSJeremy L Thompson @return An error code: 0 - success, otherwise - failure 1757ca94c3ddSJeremy L Thompson 1758477729cfSJeremy L Thompson @ref Developer 1759477729cfSJeremy L Thompson **/ 1760d1d35e2fSjeremylt int CeedGetErrorMessage(Ceed ceed, const char **err_msg) { 17612b730f8bSJeremy L Thompson if (ceed->parent) return CeedGetErrorMessage(ceed->parent, err_msg); 1762d1d35e2fSjeremylt *err_msg = ceed->err_msg; 1763e15f9bd0SJeremy L Thompson return CEED_ERROR_SUCCESS; 1764477729cfSJeremy L Thompson } 1765477729cfSJeremy L Thompson 1766477729cfSJeremy L Thompson /** 1767ca94c3ddSJeremy L Thompson @brief Restore error message. 1768477729cfSJeremy L Thompson 1769ca94c3ddSJeremy L Thompson The error message is only stored when using the error handler @ref CeedErrorStore(). 1770477729cfSJeremy L Thompson 1771ca94c3ddSJeremy L Thompson @param[in] ceed `Ceed` context to restore error message 1772d1d35e2fSjeremylt @param[out] err_msg Char pointer that holds error message 1773477729cfSJeremy L Thompson 1774ca94c3ddSJeremy L Thompson @return An error code: 0 - success, otherwise - failure 1775ca94c3ddSJeremy L Thompson 1776477729cfSJeremy L Thompson @ref Developer 1777477729cfSJeremy L Thompson **/ 1778d1d35e2fSjeremylt int CeedResetErrorMessage(Ceed ceed, const char **err_msg) { 17792b730f8bSJeremy L Thompson if (ceed->parent) return CeedResetErrorMessage(ceed->parent, err_msg); 1780d1d35e2fSjeremylt *err_msg = NULL; 1781d1d35e2fSjeremylt memcpy(ceed->err_msg, "No error message stored", 24); 1782e15f9bd0SJeremy L Thompson return CEED_ERROR_SUCCESS; 1783477729cfSJeremy L Thompson } 1784477729cfSJeremy L Thompson 17851070991dSJed Brown /** 1786ca94c3ddSJeremy L Thompson @brief Get libCEED library version information. 17871070991dSJed Brown 1788ea61e9acSJeremy L Thompson libCEED version numbers have the form major.minor.patch. 1789ea61e9acSJeremy L Thompson Non-release versions may contain unstable interfaces. 17901070991dSJed Brown 17911070991dSJed Brown @param[out] major Major version of the library 17921070991dSJed Brown @param[out] minor Minor version of the library 17931070991dSJed Brown @param[out] patch Patch (subminor) version of the library 179453cbfc38SJeremy L Thompson @param[out] release True for releases; false for development branches 17951070991dSJed Brown 1796ca94c3ddSJeremy L Thompson The caller may pass `NULL` for any arguments that are not needed. 17971070991dSJed Brown 1798ca94c3ddSJeremy L Thompson @return An error code: 0 - success, otherwise - failure 17991070991dSJed Brown 18001070991dSJed Brown @ref Developer 1801ca94c3ddSJeremy L Thompson 18024f69910bSJed Brown @sa CEED_VERSION_GE() CeedGetGitVersion() CeedGetBuildConfiguration() 18031070991dSJed Brown */ 18041070991dSJed Brown int CeedGetVersion(int *major, int *minor, int *patch, bool *release) { 18051070991dSJed Brown if (major) *major = CEED_VERSION_MAJOR; 18061070991dSJed Brown if (minor) *minor = CEED_VERSION_MINOR; 18071070991dSJed Brown if (patch) *patch = CEED_VERSION_PATCH; 18081070991dSJed Brown if (release) *release = CEED_VERSION_RELEASE; 1809ca94c3ddSJeremy L Thompson return CEED_ERROR_SUCCESS; 18101070991dSJed Brown } 18111070991dSJed Brown 181253cbfc38SJeremy L Thompson /** 181353cbfc38SJeremy L Thompson @brief Get libCEED scalar type, such as F64 or F32 181453cbfc38SJeremy L Thompson 181553cbfc38SJeremy L Thompson @param[out] scalar_type Type of libCEED scalars 181653cbfc38SJeremy L Thompson 1817ca94c3ddSJeremy L Thompson @return An error code: 0 - success, otherwise - failure 1818ca94c3ddSJeremy L Thompson 181953cbfc38SJeremy L Thompson @ref Developer 182053cbfc38SJeremy L Thompson */ 182180a9ef05SNatalie Beams int CeedGetScalarType(CeedScalarType *scalar_type) { 182280a9ef05SNatalie Beams *scalar_type = CEED_SCALAR_TYPE; 1823ca94c3ddSJeremy L Thompson return CEED_ERROR_SUCCESS; 182480a9ef05SNatalie Beams } 182580a9ef05SNatalie Beams 1826d7b241e6Sjeremylt /// @} 1827