1 // Copyright (c) 2017-2022, Lawrence Livermore National Security, LLC and other CEED contributors. 2 // All Rights Reserved. See the top-level LICENSE and NOTICE files for details. 3 // 4 // SPDX-License-Identifier: BSD-2-Clause 5 // 6 // This file is part of CEED: http://github.com/ceed 7 8 #include <ceed.h> 9 #include <ceed/backend.h> 10 #include <ceed/jit-source/gallery/ceed-identity.h> 11 #include <stddef.h> 12 #include <string.h> 13 14 /** 15 @brief Set fields identity QFunction that copies inputs directly into outputs 16 **/ 17 static int CeedQFunctionInit_Identity(Ceed ceed, const char *requested, CeedQFunction qf) { 18 // Check QFunction name 19 const char *name = "Identity"; 20 CeedCheck(!strcmp(name, requested), ceed, CEED_ERROR_UNSUPPORTED, "QFunction '%s' does not match requested name: %s", name, requested); 21 22 // QFunction fields 'input' and 'output' with requested emodes added by the library rather than being added here 23 24 CeedCall(CeedQFunctionSetUserFlopsEstimate(qf, 0)); 25 26 // Context data 27 CeedQFunctionContext ctx; 28 IdentityCtx ctx_data = {.size = 1}; 29 CeedCall(CeedQFunctionContextCreate(ceed, &ctx)); 30 CeedCall(CeedQFunctionContextSetData(ctx, CEED_MEM_HOST, CEED_COPY_VALUES, sizeof(ctx_data), (void *)&ctx_data)); 31 CeedCall(CeedQFunctionContextRegisterInt32(ctx, "size", offsetof(IdentityCtx, size), 1, "field size of identity QFunction")); 32 CeedCall(CeedQFunctionSetContext(qf, ctx)); 33 CeedCall(CeedQFunctionContextDestroy(&ctx)); 34 35 return CEED_ERROR_SUCCESS; 36 } 37 38 /** 39 @brief Register identity QFunction that copies inputs directly into outputs 40 **/ 41 CEED_INTERN int CeedQFunctionRegister_Identity(void) { 42 return CeedQFunctionRegister("Identity", Identity_loc, 1, Identity, CeedQFunctionInit_Identity); 43 } 44