xref: /libCEED/rust/libceed-sys/c-src/interface/ceed-vector.c (revision 96b902e2a709bcaf161c65c2cb076b00357cddfa)
10436c2adSjeremylt // Copyright (c) 2017, Lawrence Livermore National Security, LLC. Produced at
20436c2adSjeremylt // the Lawrence Livermore National Laboratory. LLNL-CODE-734707. All Rights
30436c2adSjeremylt // reserved. See files LICENSE and NOTICE for details.
40436c2adSjeremylt //
50436c2adSjeremylt // This file is part of CEED, a collection of benchmarks, miniapps, software
60436c2adSjeremylt // libraries and APIs for efficient high-order finite element and spectral
70436c2adSjeremylt // element discretizations for exascale applications. For more information and
80436c2adSjeremylt // source code availability see http://github.com/ceed.
90436c2adSjeremylt //
100436c2adSjeremylt // The CEED research is supported by the Exascale Computing Project 17-SC-20-SC,
110436c2adSjeremylt // a collaborative effort of two U.S. Department of Energy organizations (Office
120436c2adSjeremylt // of Science and the National Nuclear Security Administration) responsible for
130436c2adSjeremylt // the planning and preparation of a capable exascale ecosystem, including
140436c2adSjeremylt // software, applications, hardware, advanced system engineering and early
150436c2adSjeremylt // testbed platforms, in support of the nation's exascale computing imperative.
160436c2adSjeremylt 
17ec3da8bcSJed Brown #include <ceed/ceed.h>
18ec3da8bcSJed Brown #include <ceed/backend.h>
193d576824SJeremy L Thompson #include <ceed-impl.h>
20547d9b97Sjeremylt #include <math.h>
213d576824SJeremy L Thompson #include <stdint.h>
223d576824SJeremy L Thompson #include <stdio.h>
230436c2adSjeremylt 
247a982d89SJeremy L. Thompson /// @file
257a982d89SJeremy L. Thompson /// Implementation of public CeedVector interfaces
267a982d89SJeremy L. Thompson 
270436c2adSjeremylt /// @cond DOXYGEN_SKIP
280436c2adSjeremylt static struct CeedVector_private ceed_vector_active;
290436c2adSjeremylt static struct CeedVector_private ceed_vector_none;
300436c2adSjeremylt /// @endcond
310436c2adSjeremylt 
327a982d89SJeremy L. Thompson /// @addtogroup CeedVectorUser
337a982d89SJeremy L. Thompson /// @{
347a982d89SJeremy L. Thompson 
357a982d89SJeremy L. Thompson /// Indicate that vector will be provided as an explicit argument to
367a982d89SJeremy L. Thompson ///   CeedOperatorApply().
377a982d89SJeremy L. Thompson const CeedVector CEED_VECTOR_ACTIVE = &ceed_vector_active;
387a982d89SJeremy L. Thompson 
39*96b902e2Sjeremylt /// Indicate that no vector is applicable (i.e., for @ref CEED_EVAL_WEIGHT).
407a982d89SJeremy L. Thompson const CeedVector CEED_VECTOR_NONE = &ceed_vector_none;
417a982d89SJeremy L. Thompson 
427a982d89SJeremy L. Thompson /// @}
437a982d89SJeremy L. Thompson 
447a982d89SJeremy L. Thompson /// ----------------------------------------------------------------------------
457a982d89SJeremy L. Thompson /// CeedVector Backend API
467a982d89SJeremy L. Thompson /// ----------------------------------------------------------------------------
477a982d89SJeremy L. Thompson /// @addtogroup CeedVectorBackend
487a982d89SJeremy L. Thompson /// @{
497a982d89SJeremy L. Thompson 
507a982d89SJeremy L. Thompson /**
517a982d89SJeremy L. Thompson   @brief Get the Ceed associated with a CeedVector
527a982d89SJeremy L. Thompson 
537a982d89SJeremy L. Thompson   @param vec        CeedVector to retrieve state
547a982d89SJeremy L. Thompson   @param[out] ceed  Variable to store ceed
557a982d89SJeremy L. Thompson 
567a982d89SJeremy L. Thompson   @return An error code: 0 - success, otherwise - failure
577a982d89SJeremy L. Thompson 
587a982d89SJeremy L. Thompson   @ref Backend
597a982d89SJeremy L. Thompson **/
607a982d89SJeremy L. Thompson int CeedVectorGetCeed(CeedVector vec, Ceed *ceed) {
617a982d89SJeremy L. Thompson   *ceed = vec->ceed;
62e15f9bd0SJeremy L Thompson   return CEED_ERROR_SUCCESS;
637a982d89SJeremy L. Thompson }
647a982d89SJeremy L. Thompson 
657a982d89SJeremy L. Thompson /**
667a982d89SJeremy L. Thompson   @brief Get the state of a CeedVector
677a982d89SJeremy L. Thompson 
687a982d89SJeremy L. Thompson   @param vec         CeedVector to retrieve state
697a982d89SJeremy L. Thompson   @param[out] state  Variable to store state
707a982d89SJeremy L. Thompson 
717a982d89SJeremy L. Thompson   @return An error code: 0 - success, otherwise - failure
727a982d89SJeremy L. Thompson 
737a982d89SJeremy L. Thompson   @ref Backend
747a982d89SJeremy L. Thompson **/
757a982d89SJeremy L. Thompson int CeedVectorGetState(CeedVector vec, uint64_t *state) {
767a982d89SJeremy L. Thompson   *state = vec->state;
77e15f9bd0SJeremy L Thompson   return CEED_ERROR_SUCCESS;
787a982d89SJeremy L. Thompson }
797a982d89SJeremy L. Thompson 
807a982d89SJeremy L. Thompson /**
8127312b0fSJed Brown   @brief Add a reference to a CeedVector
827a982d89SJeremy L. Thompson 
835f67fadeSJeremy L Thompson   @param[out] vec  CeedVector to increment reference counter
847a982d89SJeremy L. Thompson 
857a982d89SJeremy L. Thompson   @return An error code: 0 - success, otherwise - failure
867a982d89SJeremy L. Thompson 
877a982d89SJeremy L. Thompson   @ref Backend
887a982d89SJeremy L. Thompson **/
897a982d89SJeremy L. Thompson int CeedVectorAddReference(CeedVector vec) {
90d1d35e2fSjeremylt   vec->ref_count++;
91e15f9bd0SJeremy L Thompson   return CEED_ERROR_SUCCESS;
927a982d89SJeremy L. Thompson }
937a982d89SJeremy L. Thompson 
947a982d89SJeremy L. Thompson /**
957a982d89SJeremy L. Thompson   @brief Get the backend data of a CeedVector
967a982d89SJeremy L. Thompson 
977a982d89SJeremy L. Thompson   @param vec        CeedVector to retrieve state
987a982d89SJeremy L. Thompson   @param[out] data  Variable to store data
997a982d89SJeremy L. Thompson 
1007a982d89SJeremy L. Thompson   @return An error code: 0 - success, otherwise - failure
1017a982d89SJeremy L. Thompson 
1027a982d89SJeremy L. Thompson   @ref Backend
1037a982d89SJeremy L. Thompson **/
104777ff853SJeremy L Thompson int CeedVectorGetData(CeedVector vec, void *data) {
105777ff853SJeremy L Thompson   *(void **)data = vec->data;
106e15f9bd0SJeremy L Thompson   return CEED_ERROR_SUCCESS;
1077a982d89SJeremy L. Thompson }
1087a982d89SJeremy L. Thompson 
1097a982d89SJeremy L. Thompson /**
1107a982d89SJeremy L. Thompson   @brief Set the backend data of a CeedVector
1117a982d89SJeremy L. Thompson 
1127a982d89SJeremy L. Thompson   @param[out] vec  CeedVector to retrieve state
1137a982d89SJeremy L. Thompson   @param data      Data to set
1147a982d89SJeremy L. Thompson 
1157a982d89SJeremy L. Thompson   @return An error code: 0 - success, otherwise - failure
1167a982d89SJeremy L. Thompson 
1177a982d89SJeremy L. Thompson   @ref Backend
1187a982d89SJeremy L. Thompson **/
119777ff853SJeremy L Thompson int CeedVectorSetData(CeedVector vec, void *data) {
120777ff853SJeremy L Thompson   vec->data = data;
121e15f9bd0SJeremy L Thompson   return CEED_ERROR_SUCCESS;
1227a982d89SJeremy L. Thompson }
1237a982d89SJeremy L. Thompson 
12434359f16Sjeremylt /**
12534359f16Sjeremylt   @brief Increment the reference counter for a CeedVector
12634359f16Sjeremylt 
12734359f16Sjeremylt   @param vec  CeedVector to increment the reference counter
12834359f16Sjeremylt 
12934359f16Sjeremylt   @return An error code: 0 - success, otherwise - failure
13034359f16Sjeremylt 
13134359f16Sjeremylt   @ref Backend
13234359f16Sjeremylt **/
1339560d06aSjeremylt int CeedVectorReference(CeedVector vec) {
13434359f16Sjeremylt   vec->ref_count++;
13534359f16Sjeremylt   return CEED_ERROR_SUCCESS;
13634359f16Sjeremylt }
13734359f16Sjeremylt 
1387a982d89SJeremy L. Thompson /// @}
1397a982d89SJeremy L. Thompson 
1407a982d89SJeremy L. Thompson /// ----------------------------------------------------------------------------
1417a982d89SJeremy L. Thompson /// CeedVector Public API
1427a982d89SJeremy L. Thompson /// ----------------------------------------------------------------------------
1437a982d89SJeremy L. Thompson /// @addtogroup CeedVectorUser
1440436c2adSjeremylt /// @{
1450436c2adSjeremylt 
1460436c2adSjeremylt /**
1470436c2adSjeremylt   @brief Create a CeedVector of the specified length (does not allocate memory)
1480436c2adSjeremylt 
1490436c2adSjeremylt   @param ceed      Ceed object where the CeedVector will be created
1500436c2adSjeremylt   @param length    Length of vector
1510436c2adSjeremylt   @param[out] vec  Address of the variable where the newly created
1520436c2adSjeremylt                      CeedVector will be stored
1530436c2adSjeremylt 
1540436c2adSjeremylt   @return An error code: 0 - success, otherwise - failure
1550436c2adSjeremylt 
1567a982d89SJeremy L. Thompson   @ref User
1570436c2adSjeremylt **/
1580436c2adSjeremylt int CeedVectorCreate(Ceed ceed, CeedInt length, CeedVector *vec) {
1590436c2adSjeremylt   int ierr;
1600436c2adSjeremylt 
1610436c2adSjeremylt   if (!ceed->VectorCreate) {
1620436c2adSjeremylt     Ceed delegate;
1630436c2adSjeremylt     ierr = CeedGetObjectDelegate(ceed, &delegate, "Vector"); CeedChk(ierr);
1640436c2adSjeremylt 
1650436c2adSjeremylt     if (!delegate)
1660436c2adSjeremylt       // LCOV_EXCL_START
167e15f9bd0SJeremy L Thompson       return CeedError(ceed, CEED_ERROR_UNSUPPORTED,
168e15f9bd0SJeremy L Thompson                        "Backend does not support VectorCreate");
1690436c2adSjeremylt     // LCOV_EXCL_STOP
1700436c2adSjeremylt 
1710436c2adSjeremylt     ierr = CeedVectorCreate(delegate, length, vec); CeedChk(ierr);
172e15f9bd0SJeremy L Thompson     return CEED_ERROR_SUCCESS;
1730436c2adSjeremylt   }
1740436c2adSjeremylt 
1750436c2adSjeremylt   ierr = CeedCalloc(1, vec); CeedChk(ierr);
1760436c2adSjeremylt   (*vec)->ceed = ceed;
1779560d06aSjeremylt   ierr = CeedReference(ceed); CeedChk(ierr);
178d1d35e2fSjeremylt   (*vec)->ref_count = 1;
1790436c2adSjeremylt   (*vec)->length = length;
1800436c2adSjeremylt   (*vec)->state = 0;
1810436c2adSjeremylt   ierr = ceed->VectorCreate(length, *vec); CeedChk(ierr);
182e15f9bd0SJeremy L Thompson   return CEED_ERROR_SUCCESS;
1830436c2adSjeremylt }
1840436c2adSjeremylt 
1850436c2adSjeremylt /**
1869560d06aSjeremylt   @brief Copy the pointer to a CeedVector. Both pointers should
1879560d06aSjeremylt            be destroyed with `CeedVectorDestroy()`;
1889560d06aSjeremylt            Note: If `*vec_copy` is non-NULL, then it is assumed that
1899560d06aSjeremylt            `*vec_copy` is a pointer to a CeedVector. This
1909560d06aSjeremylt            CeedVector will be destroyed if `*vec_copy` is the only
1919560d06aSjeremylt            reference to this CeedVector.
1929560d06aSjeremylt 
1939560d06aSjeremylt   @param vec            CeedVector to copy reference to
1949560d06aSjeremylt   @param[out] vec_copy  Variable to store copied reference
1959560d06aSjeremylt 
1969560d06aSjeremylt   @return An error code: 0 - success, otherwise - failure
1979560d06aSjeremylt 
1989560d06aSjeremylt   @ref User
1999560d06aSjeremylt **/
2009560d06aSjeremylt int CeedVectorReferenceCopy(CeedVector vec, CeedVector *vec_copy) {
2019560d06aSjeremylt   int ierr;
2029560d06aSjeremylt 
2039560d06aSjeremylt   ierr = CeedVectorReference(vec); CeedChk(ierr);
2049560d06aSjeremylt   ierr = CeedVectorDestroy(vec_copy); CeedChk(ierr);
2059560d06aSjeremylt   *vec_copy = vec;
2069560d06aSjeremylt   return CEED_ERROR_SUCCESS;
2079560d06aSjeremylt }
2089560d06aSjeremylt 
2099560d06aSjeremylt /**
2100436c2adSjeremylt   @brief Set the array used by a CeedVector, freeing any previously allocated
211b3cf021fSjeremylt            array if applicable. The backend may copy values to a different
212b3cf021fSjeremylt            memtype, such as during @ref CeedOperatorApply().
2136a6c615bSJeremy L Thompson            See also @ref CeedVectorSyncArray() and @ref CeedVectorTakeArray().
2140436c2adSjeremylt 
2150436c2adSjeremylt   @param vec        CeedVector
216d1d35e2fSjeremylt   @param mem_type   Memory type of the array being passed
217d1d35e2fSjeremylt   @param copy_mode  Copy mode for the array
2184cc79fe7SJed Brown   @param array      Array to be used, or NULL with @ref CEED_COPY_VALUES to have the
2190436c2adSjeremylt                       library allocate
2200436c2adSjeremylt 
2210436c2adSjeremylt   @return An error code: 0 - success, otherwise - failure
2220436c2adSjeremylt 
2237a982d89SJeremy L. Thompson   @ref User
2240436c2adSjeremylt **/
225d1d35e2fSjeremylt int CeedVectorSetArray(CeedVector vec, CeedMemType mem_type,
226d1d35e2fSjeremylt                        CeedCopyMode copy_mode,
2270436c2adSjeremylt                        CeedScalar *array) {
2280436c2adSjeremylt   int ierr;
2290436c2adSjeremylt 
2300436c2adSjeremylt   if (!vec->SetArray)
2310436c2adSjeremylt     // LCOV_EXCL_START
232e15f9bd0SJeremy L Thompson     return CeedError(vec->ceed, CEED_ERROR_UNSUPPORTED,
233e15f9bd0SJeremy L Thompson                      "Backend does not support VectorSetArray");
2340436c2adSjeremylt   // LCOV_EXCL_STOP
2350436c2adSjeremylt 
2360436c2adSjeremylt   if (vec->state % 2 == 1)
237e15f9bd0SJeremy L Thompson     return CeedError(vec->ceed, CEED_ERROR_ACCESS,
238e15f9bd0SJeremy L Thompson                      "Cannot grant CeedVector array access, the "
2390436c2adSjeremylt                      "access lock is already in use");
2400436c2adSjeremylt 
241d1d35e2fSjeremylt   if (vec->num_readers > 0)
242e15f9bd0SJeremy L Thompson     return CeedError(vec->ceed, CEED_ERROR_ACCESS,
243e15f9bd0SJeremy L Thompson                      "Cannot grant CeedVector array access, a "
2440436c2adSjeremylt                      "process has read access");
2450436c2adSjeremylt 
246d1d35e2fSjeremylt   ierr = vec->SetArray(vec, mem_type, copy_mode, array); CeedChk(ierr);
2470436c2adSjeremylt   vec->state += 2;
248e15f9bd0SJeremy L Thompson   return CEED_ERROR_SUCCESS;
2490436c2adSjeremylt }
2500436c2adSjeremylt 
2510436c2adSjeremylt /**
2520436c2adSjeremylt   @brief Set the CeedVector to a constant value
2530436c2adSjeremylt 
2540436c2adSjeremylt   @param vec        CeedVector
2550436c2adSjeremylt   @param[in] value  Value to be used
2560436c2adSjeremylt 
2570436c2adSjeremylt   @return An error code: 0 - success, otherwise - failure
2580436c2adSjeremylt 
2597a982d89SJeremy L. Thompson   @ref User
2600436c2adSjeremylt **/
2610436c2adSjeremylt int CeedVectorSetValue(CeedVector vec, CeedScalar value) {
2620436c2adSjeremylt   int ierr;
2630436c2adSjeremylt 
2640436c2adSjeremylt   if (vec->state % 2 == 1)
265e15f9bd0SJeremy L Thompson     return CeedError(vec->ceed, CEED_ERROR_ACCESS,
266e15f9bd0SJeremy L Thompson                      "Cannot grant CeedVector array access, the "
2670436c2adSjeremylt                      "access lock is already in use");
2680436c2adSjeremylt 
2690436c2adSjeremylt   if (vec->SetValue) {
2700436c2adSjeremylt     ierr = vec->SetValue(vec, value); CeedChk(ierr);
2710436c2adSjeremylt   } else {
2720436c2adSjeremylt     CeedScalar *array;
2730436c2adSjeremylt     ierr = CeedVectorGetArray(vec, CEED_MEM_HOST, &array); CeedChk(ierr);
2740436c2adSjeremylt     for (int i=0; i<vec->length; i++) array[i] = value;
2750436c2adSjeremylt     ierr = CeedVectorRestoreArray(vec, &array); CeedChk(ierr);
2760436c2adSjeremylt   }
2770436c2adSjeremylt   vec->state += 2;
278e15f9bd0SJeremy L Thompson   return CEED_ERROR_SUCCESS;
2790436c2adSjeremylt }
2800436c2adSjeremylt 
2810436c2adSjeremylt /**
282b3cf021fSjeremylt   @brief Sync the CeedVector to a specified memtype. This function is used to
283b3cf021fSjeremylt            force synchronization of arrays set with @ref CeedVectorSetArray().
284b3cf021fSjeremylt            If the requested memtype is already synchronized, this function
285b3cf021fSjeremylt            results in a no-op.
2860436c2adSjeremylt 
2870436c2adSjeremylt   @param vec       CeedVector
288d1d35e2fSjeremylt   @param mem_type  Memtype to be synced
2890436c2adSjeremylt 
2900436c2adSjeremylt   @return An error code: 0 - success, otherwise - failure
2910436c2adSjeremylt 
2927a982d89SJeremy L. Thompson   @ref User
2930436c2adSjeremylt **/
294d1d35e2fSjeremylt int CeedVectorSyncArray(CeedVector vec, CeedMemType mem_type) {
2950436c2adSjeremylt   int ierr;
2960436c2adSjeremylt 
2970436c2adSjeremylt   if (vec->state % 2 == 1)
298e15f9bd0SJeremy L Thompson     return CeedError(vec->ceed, CEED_ERROR_ACCESS,
299e15f9bd0SJeremy L Thompson                      "Cannot sync CeedVector, the access lock is "
3000436c2adSjeremylt                      "already in use");
3010436c2adSjeremylt 
3020436c2adSjeremylt   if (vec->SyncArray) {
303d1d35e2fSjeremylt     ierr = vec->SyncArray(vec, mem_type); CeedChk(ierr);
3040436c2adSjeremylt   } else {
3050436c2adSjeremylt     const CeedScalar *array;
306d1d35e2fSjeremylt     ierr = CeedVectorGetArrayRead(vec, mem_type, &array); CeedChk(ierr);
3070436c2adSjeremylt     ierr = CeedVectorRestoreArrayRead(vec, &array); CeedChk(ierr);
3080436c2adSjeremylt   }
309e15f9bd0SJeremy L Thompson   return CEED_ERROR_SUCCESS;
3100436c2adSjeremylt }
3110436c2adSjeremylt 
3120436c2adSjeremylt /**
3136a6c615bSJeremy L Thompson   @brief Take ownership of the CeedVector array and remove the array from the
31458921e9fSJeremy L Thompson            CeedVector. The caller is responsible for managing and freeing
31558921e9fSJeremy L Thompson            the array.
3166a6c615bSJeremy L Thompson 
3176a6c615bSJeremy L Thompson   @param vec         CeedVector
318d1d35e2fSjeremylt   @param mem_type    Memory type on which to take the array. If the backend
3196a6c615bSJeremy L Thompson                        uses a different memory type, this will perform a copy.
320d1d35e2fSjeremylt   @param[out] array  Array on memory type mem_type, or NULL if array pointer is
3216a6c615bSJeremy L Thompson                        not required
3226a6c615bSJeremy L Thompson 
3236a6c615bSJeremy L Thompson   @return An error code: 0 - success, otherwise - failure
3246a6c615bSJeremy L Thompson 
3256a6c615bSJeremy L Thompson   @ref User
3266a6c615bSJeremy L Thompson **/
327d1d35e2fSjeremylt int CeedVectorTakeArray(CeedVector vec, CeedMemType mem_type,
328d1d35e2fSjeremylt                         CeedScalar **array) {
3296a6c615bSJeremy L Thompson   int ierr;
3306a6c615bSJeremy L Thompson 
3316a6c615bSJeremy L Thompson   if (vec->state % 2 == 1)
3326a6c615bSJeremy L Thompson     // LCOV_EXCL_START
333e15f9bd0SJeremy L Thompson     return CeedError(vec->ceed, CEED_ERROR_ACCESS,
334e15f9bd0SJeremy L Thompson                      "Cannot take CeedVector array, the access "
3356a6c615bSJeremy L Thompson                      "lock is already in use");
3366a6c615bSJeremy L Thompson   // LCOV_EXCL_STOP
337d1d35e2fSjeremylt   if (vec->num_readers > 0)
3386a6c615bSJeremy L Thompson     // LCOV_EXCL_START
339e15f9bd0SJeremy L Thompson     return CeedError(vec->ceed, CEED_ERROR_ACCESS,
340e15f9bd0SJeremy L Thompson                      "Cannot take CeedVector array, a process "
3416a6c615bSJeremy L Thompson                      "has read access");
3426a6c615bSJeremy L Thompson   // LCOV_EXCL_STOP
3436a6c615bSJeremy L Thompson 
344d1d35e2fSjeremylt   CeedScalar *temp_array = NULL;
345d1d35e2fSjeremylt   ierr = vec->TakeArray(vec, mem_type, &temp_array); CeedChk(ierr);
346d1d35e2fSjeremylt   if (array) (*array) = temp_array;
347e15f9bd0SJeremy L Thompson   return CEED_ERROR_SUCCESS;
3486a6c615bSJeremy L Thompson }
3496a6c615bSJeremy L Thompson 
3506a6c615bSJeremy L Thompson /**
351b3cf021fSjeremylt   @brief Get read/write access to a CeedVector via the specified memory type.
352b3cf021fSjeremylt            Restore access with @ref CeedVectorRestoreArray().
3530436c2adSjeremylt 
3540436c2adSjeremylt   @param vec         CeedVector to access
355d1d35e2fSjeremylt   @param mem_type    Memory type on which to access the array. If the backend
356b3cf021fSjeremylt                        uses a different memory type, this will perform a copy.
357d1d35e2fSjeremylt   @param[out] array  Array on memory type mem_type
3580436c2adSjeremylt 
3590436c2adSjeremylt   @note The CeedVectorGetArray* and CeedVectorRestoreArray* functions provide
3600436c2adSjeremylt     access to array pointers in the desired memory space. Pairing get/restore
3610436c2adSjeremylt     allows the Vector to track access, thus knowing if norms or other
3620436c2adSjeremylt     operations may need to be recomputed.
3630436c2adSjeremylt 
3640436c2adSjeremylt   @return An error code: 0 - success, otherwise - failure
3650436c2adSjeremylt 
3667a982d89SJeremy L. Thompson   @ref User
3670436c2adSjeremylt **/
368d1d35e2fSjeremylt int CeedVectorGetArray(CeedVector vec, CeedMemType mem_type,
369d1d35e2fSjeremylt                        CeedScalar **array) {
3700436c2adSjeremylt   int ierr;
3710436c2adSjeremylt 
3720436c2adSjeremylt   if (!vec->GetArray)
3730436c2adSjeremylt     // LCOV_EXCL_START
374e15f9bd0SJeremy L Thompson     return CeedError(vec->ceed, CEED_ERROR_UNSUPPORTED,
375e15f9bd0SJeremy L Thompson                      "Backend does not support GetArray");
3760436c2adSjeremylt   // LCOV_EXCL_STOP
3770436c2adSjeremylt 
3780436c2adSjeremylt   if (vec->state % 2 == 1)
379e15f9bd0SJeremy L Thompson     return CeedError(vec->ceed, CEED_ERROR_ACCESS,
380e15f9bd0SJeremy L Thompson                      "Cannot grant CeedVector array access, the "
3810436c2adSjeremylt                      "access lock is already in use");
3820436c2adSjeremylt 
383d1d35e2fSjeremylt   if (vec->num_readers > 0)
384e15f9bd0SJeremy L Thompson     return CeedError(vec->ceed, CEED_ERROR_ACCESS,
385e15f9bd0SJeremy L Thompson                      "Cannot grant CeedVector array access, a "
3860436c2adSjeremylt                      "process has read access");
3870436c2adSjeremylt 
388d1d35e2fSjeremylt   ierr = vec->GetArray(vec, mem_type, array); CeedChk(ierr);
3890436c2adSjeremylt   vec->state += 1;
390e15f9bd0SJeremy L Thompson   return CEED_ERROR_SUCCESS;
3910436c2adSjeremylt }
3920436c2adSjeremylt 
3930436c2adSjeremylt /**
394b3cf021fSjeremylt   @brief Get read-only access to a CeedVector via the specified memory type.
395b3cf021fSjeremylt            Restore access with @ref CeedVectorRestoreArrayRead().
3960436c2adSjeremylt 
3970436c2adSjeremylt   @param vec         CeedVector to access
398d1d35e2fSjeremylt   @param mem_type    Memory type on which to access the array.  If the backend
3990436c2adSjeremylt                        uses a different memory type, this will perform a copy
4000436c2adSjeremylt                        (possibly cached).
401d1d35e2fSjeremylt   @param[out] array  Array on memory type mem_type
4020436c2adSjeremylt 
4030436c2adSjeremylt   @return An error code: 0 - success, otherwise - failure
4040436c2adSjeremylt 
4057a982d89SJeremy L. Thompson   @ref User
4060436c2adSjeremylt **/
407d1d35e2fSjeremylt int CeedVectorGetArrayRead(CeedVector vec, CeedMemType mem_type,
4080436c2adSjeremylt                            const CeedScalar **array) {
4090436c2adSjeremylt   int ierr;
4100436c2adSjeremylt 
4110436c2adSjeremylt   if (!vec->GetArrayRead)
4120436c2adSjeremylt     // LCOV_EXCL_START
413e15f9bd0SJeremy L Thompson     return CeedError(vec->ceed, CEED_ERROR_UNSUPPORTED,
414e15f9bd0SJeremy L Thompson                      "Backend does not support GetArrayRead");
4150436c2adSjeremylt   // LCOV_EXCL_STOP
4160436c2adSjeremylt 
4170436c2adSjeremylt   if (vec->state % 2 == 1)
418e15f9bd0SJeremy L Thompson     return CeedError(vec->ceed, CEED_ERROR_ACCESS,
419e15f9bd0SJeremy L Thompson                      "Cannot grant CeedVector read-only array "
4200436c2adSjeremylt                      "access, the access lock is already in use");
4210436c2adSjeremylt 
422d1d35e2fSjeremylt   ierr = vec->GetArrayRead(vec, mem_type, array); CeedChk(ierr);
423d1d35e2fSjeremylt   vec->num_readers++;
424e15f9bd0SJeremy L Thompson   return CEED_ERROR_SUCCESS;
4250436c2adSjeremylt }
4260436c2adSjeremylt 
4270436c2adSjeremylt /**
428b3cf021fSjeremylt   @brief Restore an array obtained using @ref CeedVectorGetArray()
4290436c2adSjeremylt 
4300436c2adSjeremylt   @param vec    CeedVector to restore
4310436c2adSjeremylt   @param array  Array of vector data
4320436c2adSjeremylt 
4330436c2adSjeremylt   @return An error code: 0 - success, otherwise - failure
4340436c2adSjeremylt 
4357a982d89SJeremy L. Thompson   @ref User
4360436c2adSjeremylt **/
4370436c2adSjeremylt int CeedVectorRestoreArray(CeedVector vec, CeedScalar **array) {
4380436c2adSjeremylt   int ierr;
4390436c2adSjeremylt 
4400436c2adSjeremylt   if (!vec->RestoreArray)
4410436c2adSjeremylt     // LCOV_EXCL_START
442e15f9bd0SJeremy L Thompson     return CeedError(vec->ceed, CEED_ERROR_UNSUPPORTED,
443e15f9bd0SJeremy L Thompson                      "Backend does not support RestoreArray");
4440436c2adSjeremylt   // LCOV_EXCL_STOP
4450436c2adSjeremylt 
4460436c2adSjeremylt   if (vec->state % 2 != 1)
447e15f9bd0SJeremy L Thompson     return CeedError(vec->ceed, CEED_ERROR_ACCESS,
448e15f9bd0SJeremy L Thompson                      "Cannot restore CeedVector array access, "
4490436c2adSjeremylt                      "access was not granted");
4500436c2adSjeremylt 
4510436c2adSjeremylt   ierr = vec->RestoreArray(vec); CeedChk(ierr);
4520436c2adSjeremylt   *array = NULL;
4530436c2adSjeremylt   vec->state += 1;
454e15f9bd0SJeremy L Thompson   return CEED_ERROR_SUCCESS;
4550436c2adSjeremylt }
4560436c2adSjeremylt 
4570436c2adSjeremylt /**
458b3cf021fSjeremylt   @brief Restore an array obtained using @ref CeedVectorGetArrayRead()
4590436c2adSjeremylt 
4600436c2adSjeremylt   @param vec    CeedVector to restore
4610436c2adSjeremylt   @param array  Array of vector data
4620436c2adSjeremylt 
4630436c2adSjeremylt   @return An error code: 0 - success, otherwise - failure
4640436c2adSjeremylt 
4657a982d89SJeremy L. Thompson   @ref User
4660436c2adSjeremylt **/
4670436c2adSjeremylt int CeedVectorRestoreArrayRead(CeedVector vec, const CeedScalar **array) {
4680436c2adSjeremylt   int ierr;
4690436c2adSjeremylt 
4700436c2adSjeremylt   if (!vec->RestoreArrayRead)
4710436c2adSjeremylt     // LCOV_EXCL_START
472e15f9bd0SJeremy L Thompson     return CeedError(vec->ceed, CEED_ERROR_UNSUPPORTED,
473e15f9bd0SJeremy L Thompson                      "Backend does not support RestoreArrayRead");
4740436c2adSjeremylt   // LCOV_EXCL_STOP
4750436c2adSjeremylt 
4760436c2adSjeremylt   ierr = vec->RestoreArrayRead(vec); CeedChk(ierr);
4770436c2adSjeremylt   *array = NULL;
478d1d35e2fSjeremylt   vec->num_readers--;
479e15f9bd0SJeremy L Thompson   return CEED_ERROR_SUCCESS;
4800436c2adSjeremylt }
4810436c2adSjeremylt 
4820436c2adSjeremylt /**
483171f8ca9Sjeremylt   @brief Get the norm of a CeedVector.
484171f8ca9Sjeremylt 
485171f8ca9Sjeremylt   Note: This operation is local to the CeedVector. This function will likely
486171f8ca9Sjeremylt           not provide the desired results for the norm of the libCEED portion
487171f8ca9Sjeremylt           of a parallel vector or a CeedVector with duplicated or hanging nodes.
488547d9b97Sjeremylt 
489547d9b97Sjeremylt   @param vec        CeedVector to retrieve maximum value
490d1d35e2fSjeremylt   @param norm_type  Norm type @ref CEED_NORM_1, @ref CEED_NORM_2, or @ref CEED_NORM_MAX
491547d9b97Sjeremylt   @param[out] norm  Variable to store norm value
492547d9b97Sjeremylt 
493547d9b97Sjeremylt   @return An error code: 0 - success, otherwise - failure
494547d9b97Sjeremylt 
4957a982d89SJeremy L. Thompson   @ref User
496547d9b97Sjeremylt **/
497d1d35e2fSjeremylt int CeedVectorNorm(CeedVector vec, CeedNormType norm_type, CeedScalar *norm) {
498547d9b97Sjeremylt   int ierr;
499547d9b97Sjeremylt 
500547d9b97Sjeremylt   // Backend impl for GPU, if added
501547d9b97Sjeremylt   if (vec->Norm) {
502d1d35e2fSjeremylt     ierr = vec->Norm(vec, norm_type, norm); CeedChk(ierr);
503e15f9bd0SJeremy L Thompson     return CEED_ERROR_SUCCESS;
504547d9b97Sjeremylt   }
505547d9b97Sjeremylt 
506547d9b97Sjeremylt   const CeedScalar *array;
507547d9b97Sjeremylt   ierr = CeedVectorGetArrayRead(vec, CEED_MEM_HOST, &array); CeedChk(ierr);
508547d9b97Sjeremylt 
509547d9b97Sjeremylt   *norm = 0.;
510d1d35e2fSjeremylt   switch (norm_type) {
511547d9b97Sjeremylt   case CEED_NORM_1:
512547d9b97Sjeremylt     for (int i=0; i<vec->length; i++) {
513547d9b97Sjeremylt       *norm += fabs(array[i]);
514547d9b97Sjeremylt     }
515547d9b97Sjeremylt     break;
516547d9b97Sjeremylt   case CEED_NORM_2:
517547d9b97Sjeremylt     for (int i=0; i<vec->length; i++) {
518547d9b97Sjeremylt       *norm += fabs(array[i])*fabs(array[i]);
519547d9b97Sjeremylt     }
520547d9b97Sjeremylt     break;
521547d9b97Sjeremylt   case CEED_NORM_MAX:
522547d9b97Sjeremylt     for (int i=0; i<vec->length; i++) {
523d1d35e2fSjeremylt       const CeedScalar abs_v_i = fabs(array[i]);
524d1d35e2fSjeremylt       *norm = *norm > abs_v_i ? *norm : abs_v_i;
525547d9b97Sjeremylt     }
526547d9b97Sjeremylt   }
527d1d35e2fSjeremylt   if (norm_type == CEED_NORM_2)
528547d9b97Sjeremylt     *norm = sqrt(*norm);
529547d9b97Sjeremylt 
530547d9b97Sjeremylt   ierr = CeedVectorRestoreArrayRead(vec, &array); CeedChk(ierr);
531e15f9bd0SJeremy L Thompson   return CEED_ERROR_SUCCESS;
532547d9b97Sjeremylt }
533547d9b97Sjeremylt 
534547d9b97Sjeremylt /**
535e0dd3b27Sjeremylt   @brief Compute x = alpha x
536e0dd3b27Sjeremylt 
537*96b902e2Sjeremylt   @param[in,out] x  vector for scaling
538*96b902e2Sjeremylt   @param[in] alpha  scaling factor
539e0dd3b27Sjeremylt 
540e0dd3b27Sjeremylt   @return An error code: 0 - success, otherwise - failure
541e0dd3b27Sjeremylt 
542e0dd3b27Sjeremylt   @ref User
543e0dd3b27Sjeremylt **/
544e0dd3b27Sjeremylt int CeedVectorScale(CeedVector x, CeedScalar alpha) {
545e0dd3b27Sjeremylt   int ierr;
546e0dd3b27Sjeremylt   CeedScalar *x_array;
547e0dd3b27Sjeremylt   CeedInt n_x;
548e0dd3b27Sjeremylt 
549e0dd3b27Sjeremylt   ierr = CeedVectorGetLength(x, &n_x); CeedChk(ierr);
550e0dd3b27Sjeremylt 
551e0dd3b27Sjeremylt   // Backend implementation
552e0dd3b27Sjeremylt   if (x->Scale)
553e0dd3b27Sjeremylt     return x->Scale(x, alpha);
554e0dd3b27Sjeremylt 
555e0dd3b27Sjeremylt   // Default implementation
556e0dd3b27Sjeremylt   ierr = CeedVectorGetArray(x, CEED_MEM_HOST, &x_array); CeedChk(ierr);
557e0dd3b27Sjeremylt   for (CeedInt i=0; i<n_x; i++)
558e0dd3b27Sjeremylt     x_array[i] *= alpha;
559e0dd3b27Sjeremylt   ierr = CeedVectorRestoreArray(x, &x_array); CeedChk(ierr);
560e0dd3b27Sjeremylt 
561e0dd3b27Sjeremylt   return CEED_ERROR_SUCCESS;
562e0dd3b27Sjeremylt }
563e0dd3b27Sjeremylt 
564e0dd3b27Sjeremylt /**
5650f7fd0f8Sjeremylt   @brief Compute y = alpha x + y
5660f7fd0f8Sjeremylt 
567*96b902e2Sjeremylt   @param[in,out] y  target vector for sum
568*96b902e2Sjeremylt   @param[in] alpha  scaling factor
569*96b902e2Sjeremylt   @param[in] x      second vector, must be different than y
5700f7fd0f8Sjeremylt 
5710f7fd0f8Sjeremylt   @return An error code: 0 - success, otherwise - failure
5720f7fd0f8Sjeremylt 
5730f7fd0f8Sjeremylt   @ref User
5740f7fd0f8Sjeremylt **/
5750f7fd0f8Sjeremylt int CeedVectorAXPY(CeedVector y, CeedScalar alpha, CeedVector x) {
5760f7fd0f8Sjeremylt   int ierr;
5770f7fd0f8Sjeremylt   CeedScalar *y_array;
5780f7fd0f8Sjeremylt   CeedScalar const *x_array;
5790f7fd0f8Sjeremylt   CeedInt n_x, n_y;
5800f7fd0f8Sjeremylt 
5810f7fd0f8Sjeremylt   ierr = CeedVectorGetLength(y, &n_y); CeedChk(ierr);
5820f7fd0f8Sjeremylt   ierr = CeedVectorGetLength(x, &n_x); CeedChk(ierr);
5830f7fd0f8Sjeremylt   if (n_x != n_y)
5840f7fd0f8Sjeremylt     // LCOV_EXCL_START
5850f7fd0f8Sjeremylt     return CeedError(y->ceed, CEED_ERROR_UNSUPPORTED,
5860f7fd0f8Sjeremylt                      "Cannot add vector of different lengths");
5870f7fd0f8Sjeremylt   // LCOV_EXCL_STOP
5880f7fd0f8Sjeremylt   if (x == y)
5890f7fd0f8Sjeremylt     // LCOV_EXCL_START
5900f7fd0f8Sjeremylt     return CeedError(y->ceed, CEED_ERROR_UNSUPPORTED,
5910f7fd0f8Sjeremylt                      "Cannot use same vector for x and y in CeedVectorAXPY");
5920f7fd0f8Sjeremylt   // LCOV_EXCL_STOP
5930f7fd0f8Sjeremylt 
5942d04630dSjeremylt   Ceed ceed_parent_x, ceed_parent_y;
5952d04630dSjeremylt   ierr = CeedGetParent(x->ceed, &ceed_parent_x); CeedChk(ierr);
5962d04630dSjeremylt   ierr = CeedGetParent(y->ceed, &ceed_parent_y); CeedChk(ierr);
5972d04630dSjeremylt   if (ceed_parent_x != ceed_parent_y)
5982d04630dSjeremylt     // LCOV_EXCL_START
5992d04630dSjeremylt     return CeedError(y->ceed, CEED_ERROR_INCOMPATIBLE,
6002d04630dSjeremylt                      "Vectors x and y must be created by the same Ceed context");
6012d04630dSjeremylt   // LCOV_EXCL_STOP
6022d04630dSjeremylt 
6030f7fd0f8Sjeremylt   // Backend implementation
6040f7fd0f8Sjeremylt   if (y->AXPY)
6050f7fd0f8Sjeremylt     return y->AXPY(y, alpha, x);
6060f7fd0f8Sjeremylt 
6070f7fd0f8Sjeremylt   // Default implementation
6080f7fd0f8Sjeremylt   ierr = CeedVectorGetArray(y, CEED_MEM_HOST, &y_array); CeedChk(ierr);
6090f7fd0f8Sjeremylt   ierr = CeedVectorGetArrayRead(x, CEED_MEM_HOST, &x_array); CeedChk(ierr);
6100f7fd0f8Sjeremylt 
6110f7fd0f8Sjeremylt   for (CeedInt i=0; i<n_y; i++)
6120f7fd0f8Sjeremylt     y_array[i] += alpha * x_array[i];
6130f7fd0f8Sjeremylt 
6140f7fd0f8Sjeremylt   ierr = CeedVectorRestoreArray(y, &y_array); CeedChk(ierr);
6150f7fd0f8Sjeremylt   ierr = CeedVectorRestoreArrayRead(x, &x_array); CeedChk(ierr);
6160f7fd0f8Sjeremylt 
6170f7fd0f8Sjeremylt   return CEED_ERROR_SUCCESS;
6180f7fd0f8Sjeremylt }
6190f7fd0f8Sjeremylt 
6200f7fd0f8Sjeremylt /**
6210c1bc3c2Sjeremylt   @brief Compute the pointwise multiplication w = x .* y. Any
6220f7fd0f8Sjeremylt            subset of x, y, and w may be the same vector.
6230f7fd0f8Sjeremylt 
624*96b902e2Sjeremylt   @param[out] w  target vector for the product
625*96b902e2Sjeremylt   @param[in] x   first vector for product
626*96b902e2Sjeremylt   @param[in] y   second vector for the product
6270f7fd0f8Sjeremylt 
6280f7fd0f8Sjeremylt   @return An error code: 0 - success, otherwise - failure
6290f7fd0f8Sjeremylt 
6300f7fd0f8Sjeremylt   @ ref User
6310f7fd0f8Sjeremylt **/
6320f7fd0f8Sjeremylt int CeedVectorPointwiseMult(CeedVector w, CeedVector x, CeedVector y) {
6330f7fd0f8Sjeremylt   int ierr;
6340f7fd0f8Sjeremylt   CeedScalar *w_array;
6350f7fd0f8Sjeremylt   CeedScalar const *x_array, *y_array;
6362d04630dSjeremylt   CeedInt n_w, n_x, n_y;
6370f7fd0f8Sjeremylt 
6380f7fd0f8Sjeremylt   ierr = CeedVectorGetLength(w, &n_w); CeedChk(ierr);
6390f7fd0f8Sjeremylt   ierr = CeedVectorGetLength(x, &n_x); CeedChk(ierr);
6400f7fd0f8Sjeremylt   ierr = CeedVectorGetLength(y, &n_y); CeedChk(ierr);
6410f7fd0f8Sjeremylt   if (n_w != n_x || n_w != n_y)
6420f7fd0f8Sjeremylt     // LCOV_EXCL_START
6430f7fd0f8Sjeremylt     return CeedError(w->ceed, CEED_ERROR_UNSUPPORTED,
6440f7fd0f8Sjeremylt                      "Cannot multiply vectors of different lengths");
6450f7fd0f8Sjeremylt   // LCOV_EXCL_STOP
6460f7fd0f8Sjeremylt 
6472d04630dSjeremylt   Ceed ceed_parent_w, ceed_parent_x, ceed_parent_y;
6482d04630dSjeremylt   ierr = CeedGetParent(w->ceed, &ceed_parent_w); CeedChk(ierr);
6492d04630dSjeremylt   ierr = CeedGetParent(x->ceed, &ceed_parent_x); CeedChk(ierr);
6502d04630dSjeremylt   ierr = CeedGetParent(y->ceed, &ceed_parent_y); CeedChk(ierr);
6512d04630dSjeremylt   if ((ceed_parent_w != ceed_parent_y) ||
6522d04630dSjeremylt       (ceed_parent_w != ceed_parent_y))
6532d04630dSjeremylt     // LCOV_EXCL_START
6542d04630dSjeremylt     return CeedError(w->ceed, CEED_ERROR_INCOMPATIBLE,
6552d04630dSjeremylt                      "Vectors w, x, and y must be created by the same Ceed context");
6562d04630dSjeremylt   // LCOV_EXCL_STOP
6572d04630dSjeremylt 
6580f7fd0f8Sjeremylt   // Backend implementation
6590f7fd0f8Sjeremylt   if (w->PointwiseMult)
6600f7fd0f8Sjeremylt     return w->PointwiseMult(w, x, y);
6610f7fd0f8Sjeremylt 
6620f7fd0f8Sjeremylt   // Default implementation
6630f7fd0f8Sjeremylt   ierr = CeedVectorGetArray(w, CEED_MEM_HOST, &w_array); CeedChk(ierr);
6640f7fd0f8Sjeremylt   if (x != w) {
6650f7fd0f8Sjeremylt     ierr = CeedVectorGetArrayRead(x, CEED_MEM_HOST, &x_array); CeedChk(ierr);
6660f7fd0f8Sjeremylt   } else {
6670f7fd0f8Sjeremylt     x_array = w_array;
6680f7fd0f8Sjeremylt   }
6690f7fd0f8Sjeremylt   if (y != w && y != x) {
6700f7fd0f8Sjeremylt     ierr = CeedVectorGetArrayRead(y, CEED_MEM_HOST, &y_array); CeedChk(ierr);
6710f7fd0f8Sjeremylt   } else if (y != x) {
6720f7fd0f8Sjeremylt     y_array = w_array;
6730f7fd0f8Sjeremylt   } else {
6740f7fd0f8Sjeremylt     y_array = x_array;
6750f7fd0f8Sjeremylt   }
6760f7fd0f8Sjeremylt 
6770f7fd0f8Sjeremylt   for (CeedInt i=0; i<n_w; i++)
6780f7fd0f8Sjeremylt     w_array[i] = x_array[i] * y_array[i];
6790f7fd0f8Sjeremylt 
6800f7fd0f8Sjeremylt   if (y != w && y != x) {
6810f7fd0f8Sjeremylt     ierr = CeedVectorRestoreArrayRead(y, &y_array); CeedChk(ierr);
6820f7fd0f8Sjeremylt   }
6830f7fd0f8Sjeremylt   if (x != w) {
6840f7fd0f8Sjeremylt     ierr = CeedVectorRestoreArrayRead(x, &x_array); CeedChk(ierr);
6850f7fd0f8Sjeremylt   }
6860f7fd0f8Sjeremylt   ierr = CeedVectorRestoreArray(w, &w_array); CeedChk(ierr);
6870f7fd0f8Sjeremylt   return CEED_ERROR_SUCCESS;
6880f7fd0f8Sjeremylt }
6890f7fd0f8Sjeremylt 
6900f7fd0f8Sjeremylt /**
691d99fa3c5SJeremy L Thompson   @brief Take the reciprocal of a CeedVector.
692d99fa3c5SJeremy L Thompson 
693d99fa3c5SJeremy L Thompson   @param vec  CeedVector to take reciprocal
694d99fa3c5SJeremy L Thompson 
695d99fa3c5SJeremy L Thompson   @return An error code: 0 - success, otherwise - failure
696d99fa3c5SJeremy L Thompson 
697d99fa3c5SJeremy L Thompson   @ref User
698d99fa3c5SJeremy L Thompson **/
699d99fa3c5SJeremy L Thompson int CeedVectorReciprocal(CeedVector vec) {
700d99fa3c5SJeremy L Thompson   int ierr;
701d99fa3c5SJeremy L Thompson 
702d99fa3c5SJeremy L Thompson   // Check if vector data set
703d99fa3c5SJeremy L Thompson   if (!vec->state)
704d99fa3c5SJeremy L Thompson     // LCOV_EXCL_START
705e15f9bd0SJeremy L Thompson     return CeedError(vec->ceed, CEED_ERROR_INCOMPLETE,
706d99fa3c5SJeremy L Thompson                      "CeedVector must have data set to take reciprocal");
707d99fa3c5SJeremy L Thompson   // LCOV_EXCL_STOP
708d99fa3c5SJeremy L Thompson 
709d99fa3c5SJeremy L Thompson   // Backend impl for GPU, if added
710d99fa3c5SJeremy L Thompson   if (vec->Reciprocal) {
711d99fa3c5SJeremy L Thompson     ierr = vec->Reciprocal(vec); CeedChk(ierr);
712e15f9bd0SJeremy L Thompson     return CEED_ERROR_SUCCESS;
713d99fa3c5SJeremy L Thompson   }
714d99fa3c5SJeremy L Thompson 
715d99fa3c5SJeremy L Thompson   CeedInt len;
716d99fa3c5SJeremy L Thompson   ierr = CeedVectorGetLength(vec, &len); CeedChk(ierr);
717d99fa3c5SJeremy L Thompson   CeedScalar *array;
718d99fa3c5SJeremy L Thompson   ierr = CeedVectorGetArray(vec, CEED_MEM_HOST, &array); CeedChk(ierr);
719d99fa3c5SJeremy L Thompson   for (CeedInt i=0; i<len; i++)
720d99fa3c5SJeremy L Thompson     if (fabs(array[i]) > CEED_EPSILON)
721d99fa3c5SJeremy L Thompson       array[i] = 1./array[i];
722d99fa3c5SJeremy L Thompson 
723e15f9bd0SJeremy L Thompson   ierr = CeedVectorRestoreArray(vec, &array); CeedChk(ierr);
724e15f9bd0SJeremy L Thompson   return CEED_ERROR_SUCCESS;
725d99fa3c5SJeremy L Thompson }
726d99fa3c5SJeremy L Thompson 
727d99fa3c5SJeremy L Thompson /**
7287a982d89SJeremy L. Thompson   @brief View a CeedVector
7290436c2adSjeremylt 
7300a0da059Sjeremylt   @param[in] vec     CeedVector to view
731d1d35e2fSjeremylt   @param[in] fp_fmt  Printing format
7320a0da059Sjeremylt   @param[in] stream  Filestream to write to
7330a0da059Sjeremylt 
7340436c2adSjeremylt   @return An error code: 0 - success, otherwise - failure
7350436c2adSjeremylt 
7367a982d89SJeremy L. Thompson   @ref User
7370436c2adSjeremylt **/
738d1d35e2fSjeremylt int CeedVectorView(CeedVector vec, const char *fp_fmt, FILE *stream) {
7397a982d89SJeremy L. Thompson   const CeedScalar *x;
7407a982d89SJeremy L. Thompson 
7417a982d89SJeremy L. Thompson   int ierr = CeedVectorGetArrayRead(vec, CEED_MEM_HOST, &x); CeedChk(ierr);
7427a982d89SJeremy L. Thompson 
7437a982d89SJeremy L. Thompson   char fmt[1024];
7447a982d89SJeremy L. Thompson   fprintf(stream, "CeedVector length %ld\n", (long)vec->length);
745d1d35e2fSjeremylt   snprintf(fmt, sizeof fmt, "  %s\n", fp_fmt ? fp_fmt : "%g");
7467a982d89SJeremy L. Thompson   for (CeedInt i=0; i<vec->length; i++)
7477a982d89SJeremy L. Thompson     fprintf(stream, fmt, x[i]);
7487a982d89SJeremy L. Thompson 
7497a982d89SJeremy L. Thompson   ierr = CeedVectorRestoreArrayRead(vec, &x); CeedChk(ierr);
750e15f9bd0SJeremy L Thompson   return CEED_ERROR_SUCCESS;
7510436c2adSjeremylt }
7520436c2adSjeremylt 
7530436c2adSjeremylt /**
7547a982d89SJeremy L. Thompson   @brief Get the length of a CeedVector
7550436c2adSjeremylt 
7567a982d89SJeremy L. Thompson   @param vec          CeedVector to retrieve length
7577a982d89SJeremy L. Thompson   @param[out] length  Variable to store length
7580436c2adSjeremylt 
7590436c2adSjeremylt   @return An error code: 0 - success, otherwise - failure
7600436c2adSjeremylt 
7617a982d89SJeremy L. Thompson   @ref User
7620436c2adSjeremylt **/
7637a982d89SJeremy L. Thompson int CeedVectorGetLength(CeedVector vec, CeedInt *length) {
7647a982d89SJeremy L. Thompson   *length = vec->length;
765e15f9bd0SJeremy L Thompson   return CEED_ERROR_SUCCESS;
7660436c2adSjeremylt }
7670436c2adSjeremylt 
7680436c2adSjeremylt /**
7690436c2adSjeremylt   @brief Destroy a CeedVector
7700436c2adSjeremylt 
7710436c2adSjeremylt   @param vec  CeedVector to destroy
7720436c2adSjeremylt 
7730436c2adSjeremylt   @return An error code: 0 - success, otherwise - failure
7740436c2adSjeremylt 
7757a982d89SJeremy L. Thompson   @ref User
7760436c2adSjeremylt **/
7770436c2adSjeremylt int CeedVectorDestroy(CeedVector *vec) {
7780436c2adSjeremylt   int ierr;
7790436c2adSjeremylt 
780d1d35e2fSjeremylt   if (!*vec || --(*vec)->ref_count > 0) return CEED_ERROR_SUCCESS;
7810436c2adSjeremylt 
782187168c7SJeremy L Thompson   if (((*vec)->state % 2) == 1)
783e15f9bd0SJeremy L Thompson     return CeedError((*vec)->ceed, CEED_ERROR_ACCESS,
784187168c7SJeremy L Thompson                      "Cannot destroy CeedVector, the writable access "
7850436c2adSjeremylt                      "lock is in use");
7860436c2adSjeremylt 
787d1d35e2fSjeremylt   if ((*vec)->num_readers > 0)
788e15f9bd0SJeremy L Thompson     return CeedError((*vec)->ceed, CEED_ERROR_ACCESS,
789e15f9bd0SJeremy L Thompson                      "Cannot destroy CeedVector, a process has "
790187168c7SJeremy L Thompson                      "read access");
791187168c7SJeremy L Thompson 
7920436c2adSjeremylt   if ((*vec)->Destroy) {
7930436c2adSjeremylt     ierr = (*vec)->Destroy(*vec); CeedChk(ierr);
7940436c2adSjeremylt   }
7950436c2adSjeremylt 
7960436c2adSjeremylt   ierr = CeedDestroy(&(*vec)->ceed); CeedChk(ierr);
7970436c2adSjeremylt   ierr = CeedFree(vec); CeedChk(ierr);
798e15f9bd0SJeremy L Thompson   return CEED_ERROR_SUCCESS;
7990436c2adSjeremylt }
8000436c2adSjeremylt 
8010436c2adSjeremylt /// @}
802