Enable multiple precisions in CeedVector

backends/hip-ref/ceed-hip-ref-vector.c

@@ -35,6 +35,47 @@ static inline int CeedScalarTypeGetSize_Hip(Ceed ceed, CeedScalarType prec,
   return CEED_ERROR_SUCCESS;
 }
 
+//------------------------------------------------------------------------------
+// Get info about the current status of the different precisions in the
+// valid, borrowed, and owned arrays, for a specific mem_type
+//------------------------------------------------------------------------------
+static int CeedVectorCheckArrayStatus_Hip(CeedVector vec,
+    CeedMemType mem_type,
+    unsigned int *valid_status,
+    unsigned int *borrowed_status,
+    unsigned int *owned_status) {
+
+  int ierr;
+  CeedVector_Hip *impl;
+  ierr = CeedVectorGetData(vec, &impl); CeedChkBackend(ierr);
+  *valid_status = 0;
+  *borrowed_status = 0;
+  *owned_status = 0;
+  switch(mem_type) {
+  case CEED_MEM_HOST:
+    for (int i = 0; i < CEED_NUM_PRECISIONS; i++) {
+      if (!!impl->h_array.values[i])
+        *valid_status += 1 << i;
+      if (!!impl->h_array_borrowed.values[i])
+        *borrowed_status += 1 << i;
+      if (!!impl->h_array_owned.values[i])
+        *owned_status += 1 << i;
+    }
+    break;
+  case CEED_MEM_DEVICE:
+    for (int i = 0; i < CEED_NUM_PRECISIONS; i++) {
+      if (!!impl->d_array.values[i])
+        *valid_status += 1 << i;
+      if (!!impl->d_array_borrowed.values[i])
+        *borrowed_status += 1 << i;
+      if (!!impl->d_array_owned.values[i])
+        *owned_status += 1 << i;
+    }
+    break;
+  }
+  return CEED_ERROR_SUCCESS;
+}
+
 //------------------------------------------------------------------------------
 // Set all pointers as invalid
 //------------------------------------------------------------------------------
@@ -1171,6 +1212,8 @@ int CeedVectorCreate_Hip(CeedSize n, CeedVector vec) {
   ierr = CeedSetBackendFunction(ceed, "Vector", vec, "HasBorrowedArrayOfType",
                                 CeedVectorHasBorrowedArrayOfType_Hip);
   CeedChkBackend(ierr);
+  ierr = CeedSetBackendFunction(ceed, "Vector", vec, "CheckArrayStatus",
+                                CeedVectorCheckArrayStatus_Hip); CeedChkBackend(ierr);
   ierr = CeedSetBackendFunction(ceed, "Vector", vec, "SetArrayGeneric",
                                 CeedVectorSetArrayGeneric_Hip); CeedChkBackend(ierr);
   ierr = CeedSetBackendFunction(ceed, "Vector", vec, "TakeArrayGeneric",

include/ceed-impl.h

@@ -129,6 +129,8 @@ struct CeedVector_private {
   Ceed ceed;
   int (*HasValidArray)(CeedVector, bool *);
   int (*HasBorrowedArrayOfType)(CeedVector, CeedMemType, CeedScalarType, bool *);
+  int (*CheckArrayStatus)(CeedVector, CeedMemType, unsigned int *, unsigned int *,
+                          unsigned int *);
   int (*SetArrayGeneric)(CeedVector, CeedMemType, CeedScalarType, CeedCopyMode,
                          void *);
   int (*SetValue)(CeedVector, CeedScalar);

include/ceed/ceed.h

@@ -398,6 +398,9 @@ CEED_EXTERN int CeedVectorRestoreArrayReadFP32(CeedVector vec,
     const float **array);
 CEED_EXTERN int CeedVectorRestoreArrayReadFP64(CeedVector vec,
     const double **array);
+CEED_EXTERN int CeedVectorCheckArrayStatus(CeedVector vec, CeedMemType mem_type, 
+    unsigned int *valid_status, unsigned int *borrowed_status, 
+    unsigned int *owned_status);
 CEED_EXTERN int CeedVectorNorm(CeedVector vec, CeedNormType type,
                                CeedScalar *norm);
 CEED_EXTERN int CeedVectorScale(CeedVector x, CeedScalar alpha);

interface/ceed-vector.c

@@ -1070,6 +1070,39 @@ int CeedVectorRestoreArrayReadFP64(CeedVector vec, const double **array) {
   return CeedVectorRestoreArrayReadGeneric(vec, (const void **) array);
 }
 
+/**
+  @brief Check the current status of the CeedVector's data arrays.
+
+  This function sets unsigned int parameters indicating for which precisions
+  the data for mem_type is not NULL, for borrowed and owned arrays, and also
+  which precisions are currently valid.
+  The values can be checked bitwise, such that, e.g.,
+  (valid_status & (1 << CEED_SCALAR_FP32)) will be true if the FP32 array is
+  currently valid, and false otherwise.
+
+  @param vec              CeedVector for which to check status
+  @param mem_type         Mem type for which to check status
+  @param valid_status     Status indicator for valid data
+  @param borrowed_status  Status indicator for borrowed arrays
+  @param owned_status     Status indicator for owned arrays
+**/
+int CeedVectorCheckArrayStatus(CeedVector vec, CeedMemType mem_type,
+                               unsigned int *valid_status,
+                               unsigned int *borrowed_status,
+                               unsigned int *owned_status) {
+  int ierr;
+
+  if (!vec->CheckArrayStatus)
+    // LCOV_EXCL_START
+    return CeedError(vec->ceed, CEED_ERROR_UNSUPPORTED,
+                     "Backend does not support CheckArrayStatus");
+  // LCOV_EXCL_STOP
+
+  ierr = vec->CheckArrayStatus(vec, mem_type, valid_status, borrowed_status,
+                               owned_status); CeedChk(ierr);
+  return CEED_ERROR_SUCCESS;
+}
+
 /**
   @brief Get the norm of a CeedVector.
 

interface/ceed.c

@@ -848,6 +848,7 @@ int CeedInit(const char *resource, Ceed *ceed) {
     CEED_FTABLE_ENTRY(CeedVector, GetArrayWriteGeneric),
     CEED_FTABLE_ENTRY(CeedVector, RestoreArrayGeneric),
     CEED_FTABLE_ENTRY(CeedVector, RestoreArrayReadGeneric),
+    CEED_FTABLE_ENTRY(CeedVector, CheckArrayStatus),
     CEED_FTABLE_ENTRY(CeedVector, Norm),
     CEED_FTABLE_ENTRY(CeedVector, Scale),
     CEED_FTABLE_ENTRY(CeedVector, AXPY),