xref: /libCEED/tests/t123-vector.c (revision 49aac155e7a09736f56fb3abac0f57dab29f7cbf)
1 /// @file
2 /// Test scaling a vector
3 /// \test Test scaling of a vector
4 #include <ceed.h>
5 #include <math.h>
6 #include <stdio.h>
7 
8 int main(int argc, char **argv) {
9   Ceed       ceed;
10   CeedVector x;
11   CeedInt    len = 10;
12 
13   CeedInit(argv[1], &ceed);
14 
15   CeedVectorCreate(ceed, len, &x);
16   {
17     CeedScalar array[len];
18 
19     for (CeedInt i = 0; i < len; i++) array[i] = 10 + i;
20     CeedVectorSetArray(x, CEED_MEM_HOST, CEED_COPY_VALUES, array);
21   }
22   {
23     // Sync memtype to device for GPU backends
24     CeedMemType type = CEED_MEM_HOST;
25     CeedGetPreferredMemType(ceed, &type);
26     CeedVectorSyncArray(x, type);
27   }
28   CeedVectorScale(x, -0.5);
29 
30   {
31     const CeedScalar *read_array;
32 
33     CeedVectorGetArrayRead(x, CEED_MEM_HOST, &read_array);
34     for (CeedInt i = 0; i < len; i++) {
35       if (fabs(read_array[i] + (10.0 + i) / 2) > 1e-14) {
36         // LCOV_EXCL_START
37         printf("Error in alpha x at index %" CeedInt_FMT ", computed: %f actual: %f\n", i, read_array[i], -(10.0 + i) / 2);
38         // LCOV_EXCL_STOP
39       }
40     }
41     CeedVectorRestoreArrayRead(x, &read_array);
42   }
43 
44   CeedVectorDestroy(&x);
45   CeedDestroy(&ceed);
46   return 0;
47 }
48