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/backend.h> 9 #include <ceed/ceed.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 if (strcmp(name, requested)) { 21 // LCOV_EXCL_START 22 return CeedError(ceed, CEED_ERROR_UNSUPPORTED, "QFunction '%s' does not match requested name: %s", name, requested); 23 // LCOV_EXCL_STOP 24 } 25 26 // QFunction fields 'input' and 'output' with requested emodes added 27 // by the library rather than being added here 28 29 CeedCall(CeedQFunctionSetUserFlopsEstimate(qf, 0)); 30 31 // Context data 32 CeedQFunctionContext ctx; 33 IdentityCtx ctx_data = {.size = 1}; 34 CeedCall(CeedQFunctionContextCreate(ceed, &ctx)); 35 CeedCall(CeedQFunctionContextSetData(ctx, CEED_MEM_HOST, CEED_COPY_VALUES, sizeof(ctx_data), (void *)&ctx_data)); 36 CeedCall(CeedQFunctionContextRegisterInt32(ctx, "size", offsetof(IdentityCtx, size), 1, "field size of identity QFunction")); 37 CeedCall(CeedQFunctionSetContext(qf, ctx)); 38 CeedCall(CeedQFunctionContextDestroy(&ctx)); 39 40 return CEED_ERROR_SUCCESS; 41 } 42 43 /** 44 @brief Register identity QFunction that copies inputs directly into outputs 45 **/ 46 CEED_INTERN int CeedQFunctionRegister_Identity(void) { 47 return CeedQFunctionRegister("Identity", Identity_loc, 1, Identity, CeedQFunctionInit_Identity); 48 } 49