1 // Copyright (c) 2017-2026, 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 `CeedQFunction` that copies inputs directly into outputs
16 **/
CeedQFunctionInit_Identity(Ceed ceed,const char * requested,CeedQFunction qf)17 static int CeedQFunctionInit_Identity(Ceed ceed, const char *requested, CeedQFunction qf) {
18 // Check QFunction name
19 const char *name = "Identity";
20
21 CeedCheck(!strcmp(name, requested), ceed, CEED_ERROR_UNSUPPORTED, "QFunction '%s' does not match requested name: %s", name, requested);
22
23 // QFunction fields 'input' and 'output' with requested emodes added by the library rather than being added here
24
25 // Context data
26 CeedQFunctionContext ctx;
27 IdentityCtx ctx_data = {.size = 1};
28
29 CeedCall(CeedQFunctionContextCreate(ceed, &ctx));
30 CeedCall(CeedQFunctionContextSetData(ctx, CEED_MEM_HOST, CEED_COPY_VALUES, sizeof(ctx_data), &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 CeedCall(CeedQFunctionSetUserFlopsEstimate(qf, 0));
36 return CEED_ERROR_SUCCESS;
37 }
38
39 /**
40 @brief Register identity `CeedQFunction` that copies inputs directly into outputs
41 **/
CeedQFunctionRegister_Identity(void)42 CEED_INTERN int CeedQFunctionRegister_Identity(void) {
43 return CeedQFunctionRegister("Identity", Identity_loc, 1, Identity, CeedQFunctionInit_Identity);
44 }
45