xref: /libCEED/tests/t119-vector.c (revision 2b730f8b5a9c809740a0b3b302db43a719c636b1)
1d99fa3c5SJeremy L Thompson /// @file
2d99fa3c5SJeremy L Thompson /// Test taking the reciprocal of a vector
3d99fa3c5SJeremy L Thompson /// \test Test taking the reciprocal of a vector
4d99fa3c5SJeremy L Thompson #include <ceed.h>
5*2b730f8bSJeremy L Thompson #include <math.h>
6d99fa3c5SJeremy L Thompson 
7d99fa3c5SJeremy L Thompson int main(int argc, char **argv) {
8d99fa3c5SJeremy L Thompson   Ceed              ceed;
9d99fa3c5SJeremy L Thompson   CeedVector        x;
10d99fa3c5SJeremy L Thompson   CeedInt           n;
11d99fa3c5SJeremy L Thompson   CeedScalar        a[10];
12d99fa3c5SJeremy L Thompson   const CeedScalar *b;
13d99fa3c5SJeremy L Thompson 
14d99fa3c5SJeremy L Thompson   CeedInit(argv[1], &ceed);
15d99fa3c5SJeremy L Thompson 
16d99fa3c5SJeremy L Thompson   n = 10;
17d99fa3c5SJeremy L Thompson   CeedVectorCreate(ceed, n, &x);
18*2b730f8bSJeremy L Thompson   for (CeedInt i = 0; i < n; i++) a[i] = 10 + i;
19d99fa3c5SJeremy L Thompson   CeedVectorSetArray(x, CEED_MEM_HOST, CEED_USE_POINTER, a);
2016e0f512Sjeremylt   {
2116e0f512Sjeremylt     // Sync memtype to device for GPU backends
2216e0f512Sjeremylt     CeedMemType type = CEED_MEM_HOST;
2316e0f512Sjeremylt     CeedGetPreferredMemType(ceed, &type);
2416e0f512Sjeremylt     CeedVectorSyncArray(x, type);
2516e0f512Sjeremylt   }
26d99fa3c5SJeremy L Thompson   CeedVectorReciprocal(x);
27d99fa3c5SJeremy L Thompson 
28d99fa3c5SJeremy L Thompson   CeedVectorGetArrayRead(x, CEED_MEM_HOST, &b);
29*2b730f8bSJeremy L Thompson   for (CeedInt i = 0; i < n; i++) {
30*2b730f8bSJeremy L Thompson     if (fabs(b[i] - 1. / (10 + i)) > 10. * CEED_EPSILON) printf("Error reading array b[%" CeedInt_FMT "] = %f\n", i, (CeedScalar)b[i]);
31*2b730f8bSJeremy L Thompson   }
32d99fa3c5SJeremy L Thompson   CeedVectorRestoreArrayRead(x, &b);
33d99fa3c5SJeremy L Thompson 
34d99fa3c5SJeremy L Thompson   CeedVectorDestroy(&x);
35d99fa3c5SJeremy L Thompson   CeedDestroy(&ceed);
36d99fa3c5SJeremy L Thompson   return 0;
37d99fa3c5SJeremy L Thompson }
38