Merge pull request #1927 from CEED/jeremy/scalar-multiplicity

Scalar Multigrid Multiplicity

Author: jeremylt

gallery/ceed-gallery-list.h

@@ -13,6 +13,7 @@
 // At the time of this writing, all the gallery functions are defined, but we're adopting the same strategy here as for the backends because future gallery @ref CeedQFunction might depend on external libraries.
 
 CEED_GALLERY_QFUNCTION(CeedQFunctionRegister_Identity)
+CEED_GALLERY_QFUNCTION(CeedQFunctionRegister_IdentityScalar)
 CEED_GALLERY_QFUNCTION(CeedQFunctionRegister_Mass1DBuild)
 CEED_GALLERY_QFUNCTION(CeedQFunctionRegister_Mass2DBuild)
 CEED_GALLERY_QFUNCTION(CeedQFunctionRegister_Mass3DBuild)
@@ -28,3 +29,4 @@ CEED_GALLERY_QFUNCTION(CeedQFunctionRegister_Vector3Poisson1DApply)
 CEED_GALLERY_QFUNCTION(CeedQFunctionRegister_Vector3Poisson2DApply)
 CEED_GALLERY_QFUNCTION(CeedQFunctionRegister_Vector3Poisson3DApply)
 CEED_GALLERY_QFUNCTION(CeedQFunctionRegister_Scale)
+CEED_GALLERY_QFUNCTION(CeedQFunctionRegister_ScaleScalar)

gallery/identity/ceed-identity-to-scalar.c

@@ -0,0 +1,34 @@
+// Copyright (c) 2017-2026, Lawrence Livermore National Security, LLC and other CEED contributors.
+// All Rights Reserved. See the top-level LICENSE and NOTICE files for details.
+//
+// SPDX-License-Identifier: BSD-2-Clause
+//
+// This file is part of CEED:  http://github.com/ceed
+
+#include <ceed.h>
+#include <ceed/backend.h>
+#include <ceed/jit-source/gallery/ceed-identity-to-scalar.h>
+#include <stddef.h>
+#include <string.h>
+
+/**
+  @brief Set fields identity `CeedQFunction` that copies first input component directly into output
+**/
+static int CeedQFunctionInit_IdentityScalar(Ceed ceed, const char *requested, CeedQFunction qf) {
+  // Check QFunction name
+  const char *name = "Identity to scalar";
+
+  CeedCheck(!strcmp(name, requested), ceed, CEED_ERROR_UNSUPPORTED, "QFunction '%s' does not match requested name: %s", name, requested);
+
+  // QFunction fields 'input' and 'output' with requested emodes added by the library rather than being added here
+
+  CeedCall(CeedQFunctionSetUserFlopsEstimate(qf, 0));
+  return CEED_ERROR_SUCCESS;
+}
+
+/**
+  @brief Register identity `CeedQFunction` that copies first input component directly into output
+**/
+CEED_INTERN int CeedQFunctionRegister_IdentityScalar(void) {
+  return CeedQFunctionRegister("Identity to scalar", IdentityScalar_loc, 1, IdentityScalar, CeedQFunctionInit_IdentityScalar);
+}

gallery/identity/ceed-identity.c

@@ -17,21 +17,22 @@
 static int CeedQFunctionInit_Identity(Ceed ceed, const char *requested, CeedQFunction qf) {
   // Check QFunction name
   const char *name = "Identity";
+
   CeedCheck(!strcmp(name, requested), ceed, CEED_ERROR_UNSUPPORTED, "QFunction '%s' does not match requested name: %s", name, requested);
 
   // QFunction fields 'input' and 'output' with requested emodes added by the library rather than being added here
 
-  CeedCall(CeedQFunctionSetUserFlopsEstimate(qf, 0));
-
   // Context data
   CeedQFunctionContext ctx;
   IdentityCtx          ctx_data = {.size = 1};
+
   CeedCall(CeedQFunctionContextCreate(ceed, &ctx));
   CeedCall(CeedQFunctionContextSetData(ctx, CEED_MEM_HOST, CEED_COPY_VALUES, sizeof(ctx_data), &ctx_data));
   CeedCall(CeedQFunctionContextRegisterInt32(ctx, "size", offsetof(IdentityCtx, size), 1, "field size of identity QFunction"));
   CeedCall(CeedQFunctionSetContext(qf, ctx));
   CeedCall(CeedQFunctionContextDestroy(&ctx));
 
+  CeedCall(CeedQFunctionSetUserFlopsEstimate(qf, 0));
   return CEED_ERROR_SUCCESS;
 }
 

gallery/mass-vector/ceed-vectormassapply.c

@@ -16,16 +16,17 @@
 static int CeedQFunctionInit_Vector3MassApply(Ceed ceed, const char *requested, CeedQFunction qf) {
   // Check QFunction name
   const char *name = "Vector3MassApply";
+
   CeedCheck(!strcmp(name, requested), ceed, CEED_ERROR_UNSUPPORTED, "QFunction '%s' does not match requested name: %s", name, requested);
 
   // Add QFunction fields
   const CeedInt num_comp = 3;
+
   CeedCall(CeedQFunctionAddInput(qf, "u", num_comp, CEED_EVAL_INTERP));
   CeedCall(CeedQFunctionAddInput(qf, "qdata", 1, CEED_EVAL_NONE));
   CeedCall(CeedQFunctionAddOutput(qf, "v", num_comp, CEED_EVAL_INTERP));
 
   CeedCall(CeedQFunctionSetUserFlopsEstimate(qf, num_comp));
-
   return CEED_ERROR_SUCCESS;
 }
 

gallery/mass/ceed-mass1dbuild.c

@@ -16,16 +16,17 @@
 static int CeedQFunctionInit_Mass1DBuild(Ceed ceed, const char *requested, CeedQFunction qf) {
   // Check QFunction name
   const char *name = "Mass1DBuild";
+
   CeedCheck(!strcmp(name, requested), ceed, CEED_ERROR_UNSUPPORTED, "QFunction '%s' does not match requested name: %s", name, requested);
 
   // Add QFunction fields
   const CeedInt dim = 1;
+
   CeedCall(CeedQFunctionAddInput(qf, "dx", dim * dim, CEED_EVAL_GRAD));
   CeedCall(CeedQFunctionAddInput(qf, "weights", 1, CEED_EVAL_WEIGHT));
   CeedCall(CeedQFunctionAddOutput(qf, "qdata", 1, CEED_EVAL_NONE));
 
   CeedCall(CeedQFunctionSetUserFlopsEstimate(qf, 1));
-
   return CEED_ERROR_SUCCESS;
 }
 

gallery/mass/ceed-mass2dbuild.c

@@ -16,16 +16,17 @@
 static int CeedQFunctionInit_Mass2DBuild(Ceed ceed, const char *requested, CeedQFunction qf) {
   // Check QFunction name
   const char *name = "Mass2DBuild";
+
   CeedCheck(!strcmp(name, requested), ceed, CEED_ERROR_UNSUPPORTED, "QFunction '%s' does not match requested name: %s", name, requested);
 
   // Add QFunction fields
   const CeedInt dim = 2;
+
   CeedCall(CeedQFunctionAddInput(qf, "dx", dim * dim, CEED_EVAL_GRAD));
   CeedCall(CeedQFunctionAddInput(qf, "weights", 1, CEED_EVAL_WEIGHT));
   CeedCall(CeedQFunctionAddOutput(qf, "qdata", 1, CEED_EVAL_NONE));
 
   CeedCall(CeedQFunctionSetUserFlopsEstimate(qf, 4));
-
   return CEED_ERROR_SUCCESS;
 }
 

gallery/mass/ceed-mass3dbuild.c

@@ -16,16 +16,17 @@
 static int CeedQFunctionInit_Mass3DBuild(Ceed ceed, const char *requested, CeedQFunction qf) {
   // Check QFunction name
   const char *name = "Mass3DBuild";
+
   CeedCheck(!strcmp(name, requested), ceed, CEED_ERROR_UNSUPPORTED, "QFunction '%s' does not match requested name: %s", name, requested);
 
   // Add QFunction fields
   const CeedInt dim = 3;
+
   CeedCall(CeedQFunctionAddInput(qf, "dx", dim * dim, CEED_EVAL_GRAD));
   CeedCall(CeedQFunctionAddInput(qf, "weights", 1, CEED_EVAL_WEIGHT));
   CeedCall(CeedQFunctionAddOutput(qf, "qdata", 1, CEED_EVAL_NONE));
 
   CeedCall(CeedQFunctionSetUserFlopsEstimate(qf, 15));
-
   return CEED_ERROR_SUCCESS;
 }
 

gallery/mass/ceed-massapply.c

@@ -16,6 +16,7 @@
 static int CeedQFunctionInit_MassApply(Ceed ceed, const char *requested, CeedQFunction qf) {
   // Check QFunction name
   const char *name = "MassApply";
+
   CeedCheck(!strcmp(name, requested), ceed, CEED_ERROR_UNSUPPORTED, "QFunction '%s' does not match requested name: %s", name, requested);
 
   // Add QFunction fields
@@ -24,7 +25,6 @@ static int CeedQFunctionInit_MassApply(Ceed ceed, const char *requested, CeedQFu
   CeedCall(CeedQFunctionAddOutput(qf, "v", 1, CEED_EVAL_INTERP));
 
   CeedCall(CeedQFunctionSetUserFlopsEstimate(qf, 1));
-
   return CEED_ERROR_SUCCESS;
 }
 

gallery/poisson-vector/ceed-vectorpoisson1dapply.c

@@ -16,16 +16,17 @@
 static int CeedQFunctionInit_Vector3Poisson1DApply(Ceed ceed, const char *requested, CeedQFunction qf) {
   // Check QFunction name
   const char *name = "Vector3Poisson1DApply";
+
   CeedCheck(!strcmp(name, requested), ceed, CEED_ERROR_UNSUPPORTED, "QFunction '%s' does not match requested name: %s", name, requested);
 
   // Add QFunction fields
   const CeedInt dim = 1, num_comp = 3;
+
   CeedCall(CeedQFunctionAddInput(qf, "du", num_comp * dim, CEED_EVAL_GRAD));
   CeedCall(CeedQFunctionAddInput(qf, "qdata", dim * (dim + 1) / 2, CEED_EVAL_NONE));
   CeedCall(CeedQFunctionAddOutput(qf, "dv", num_comp * dim, CEED_EVAL_GRAD));
 
   CeedCall(CeedQFunctionSetUserFlopsEstimate(qf, num_comp));
-
   return CEED_ERROR_SUCCESS;
 }
 

gallery/poisson-vector/ceed-vectorpoisson2dapply.c

@@ -16,16 +16,17 @@
 static int CeedQFunctionInit_Vector3Poisson2DApply(Ceed ceed, const char *requested, CeedQFunction qf) {
   // Check QFunction name
   const char *name = "Vector3Poisson2DApply";
+
   CeedCheck(!strcmp(name, requested), ceed, CEED_ERROR_UNSUPPORTED, "QFunction '%s' does not match requested name: %s", name, requested);
 
   // Add QFunction fields
   const CeedInt dim = 2, num_comp = 3;
+
   CeedCall(CeedQFunctionAddInput(qf, "du", num_comp * dim, CEED_EVAL_GRAD));
   CeedCall(CeedQFunctionAddInput(qf, "qdata", dim * (dim + 1) / 2, CEED_EVAL_NONE));
   CeedCall(CeedQFunctionAddOutput(qf, "dv", num_comp * dim, CEED_EVAL_GRAD));
 
   CeedCall(CeedQFunctionSetUserFlopsEstimate(qf, num_comp * 6));
-
   return CEED_ERROR_SUCCESS;
 }