1 /// @file 2 /// Test QFunction helper macro 3 /// \test Test QFunction helper macro 4 #include "t406-qfunction.h" 5 6 #include <ceed.h> 7 #include <math.h> 8 #include <stdio.h> 9 #include <stdlib.h> 10 #include <string.h> 11 12 int main(int argc, char **argv) { 13 Ceed ceed; 14 CeedVector in[16], out[16]; 15 CeedVector q_data, w, u, v; 16 CeedQFunction qf_setup, qf_mass; 17 CeedInt q = 8; 18 CeedScalar v_true[q]; 19 20 CeedInit(argv[1], &ceed); 21 { 22 char file_path[2056] = __FILE__; 23 char *last_slash = strrchr(file_path, '/'); 24 25 memcpy(&file_path[last_slash - file_path], "/test-include/", 15); 26 CeedAddJitSourceRoot(ceed, file_path); 27 CeedAddJitDefine(ceed, "COMPILER_DEFINED_SCALE=42"); 28 } 29 30 CeedVectorCreate(ceed, q, &w); 31 CeedVectorCreate(ceed, q, &u); 32 { 33 CeedScalar w_array[q], u_array[q]; 34 35 for (CeedInt i = 0; i < q; i++) { 36 CeedScalar x = 2. * i / (q - 1) - 1; 37 w_array[i] = 1 - x * x; 38 u_array[i] = 2 + 3 * x + 5 * x * x; 39 v_true[i] = w_array[i] * u_array[i]; 40 } 41 CeedVectorSetArray(w, CEED_MEM_HOST, CEED_COPY_VALUES, w_array); 42 CeedVectorSetArray(u, CEED_MEM_HOST, CEED_COPY_VALUES, u_array); 43 } 44 CeedVectorCreate(ceed, q, &v); 45 CeedVectorSetValue(v, 0); 46 CeedVectorCreate(ceed, q, &q_data); 47 CeedVectorSetValue(q_data, 0); 48 49 CeedQFunctionCreateInterior(ceed, 1, setup, setup_loc, &qf_setup); 50 CeedQFunctionAddInput(qf_setup, "w", 1, CEED_EVAL_WEIGHT); 51 CeedQFunctionAddOutput(qf_setup, "q data", 1, CEED_EVAL_NONE); 52 { 53 in[0] = w; 54 out[0] = q_data; 55 CeedQFunctionApply(qf_setup, q, in, out); 56 } 57 58 CeedQFunctionCreateInterior(ceed, 1, mass, mass_loc, &qf_mass); 59 CeedQFunctionAddInput(qf_mass, "q data", 1, CEED_EVAL_NONE); 60 CeedQFunctionAddInput(qf_mass, "u", 1, CEED_EVAL_INTERP); 61 CeedQFunctionAddOutput(qf_mass, "v", 1, CEED_EVAL_INTERP); 62 { 63 in[0] = w; 64 in[1] = u; 65 out[0] = v; 66 CeedQFunctionApply(qf_mass, q, in, out); 67 } 68 69 // Verify results 70 { 71 const CeedScalar *v_array; 72 73 CeedVectorGetArrayRead(v, CEED_MEM_HOST, &v_array); 74 for (CeedInt i = 0; i < q; i++) { 75 if (fabs(5 * COMPILER_DEFINED_SCALE * v_true[i] * sqrt(2.) - v_array[i]) > 5E3 * CEED_EPSILON) { 76 // LCOV_EXCL_START 77 printf("[%" CeedInt_FMT "] v_true %f != v %f\n", i, 5 * COMPILER_DEFINED_SCALE * v_true[i] * sqrt(2.), v_array[i]); 78 // LCOV_EXCL_STOP 79 } 80 } 81 CeedVectorRestoreArrayRead(v, &v_array); 82 } 83 84 CeedVectorDestroy(&w); 85 CeedVectorDestroy(&u); 86 CeedVectorDestroy(&v); 87 CeedVectorDestroy(&q_data); 88 CeedQFunctionDestroy(&qf_setup); 89 CeedQFunctionDestroy(&qf_mass); 90 CeedDestroy(&ceed); 91 return 0; 92 } 93