xref: /libCEED/tests/t108-vector.c (revision 2b730f8b5a9c809740a0b3b302db43a719c636b1)
1547d9b97Sjeremylt /// @file
2547d9b97Sjeremylt /// Test vector norms
3547d9b97Sjeremylt /// \test Test vector norms
4547d9b97Sjeremylt #include <ceed.h>
5547d9b97Sjeremylt #include <math.h>
6547d9b97Sjeremylt 
7547d9b97Sjeremylt int main(int argc, char **argv) {
8547d9b97Sjeremylt   Ceed       ceed;
9547d9b97Sjeremylt   CeedVector x;
10547d9b97Sjeremylt   CeedInt    n = 10;
11547d9b97Sjeremylt   CeedScalar a[10];
12547d9b97Sjeremylt 
13547d9b97Sjeremylt   CeedInit(argv[1], &ceed);
14547d9b97Sjeremylt 
15547d9b97Sjeremylt   CeedVectorCreate(ceed, n, &x);
16*2b730f8bSJeremy L Thompson   for (CeedInt i = 0; i < n; i++) a[i] = i * (i % 2 ? 1 : -1);
17547d9b97Sjeremylt   CeedVectorSetArray(x, CEED_MEM_HOST, CEED_USE_POINTER, a);
1816e0f512Sjeremylt   {
1916e0f512Sjeremylt     // Sync memtype to device for GPU backends
2016e0f512Sjeremylt     CeedMemType type = CEED_MEM_HOST;
2116e0f512Sjeremylt     CeedGetPreferredMemType(ceed, &type);
2216e0f512Sjeremylt     CeedVectorSyncArray(x, type);
2316e0f512Sjeremylt   }
24547d9b97Sjeremylt 
25547d9b97Sjeremylt   CeedScalar norm;
26547d9b97Sjeremylt   CeedVectorNorm(x, CEED_NORM_1, &norm);
27*2b730f8bSJeremy L Thompson   if (fabs(norm - 45.) > 100. * CEED_EPSILON) printf("Error: L1 norm %f != 45.\n", norm);
28547d9b97Sjeremylt 
29547d9b97Sjeremylt   CeedVectorNorm(x, CEED_NORM_2, &norm);
30*2b730f8bSJeremy L Thompson   if (fabs(norm - sqrt(285.)) > 100. * CEED_EPSILON) printf("Error: L2 norm %f != sqrt(285.)\n", norm);
31547d9b97Sjeremylt 
32547d9b97Sjeremylt   CeedVectorNorm(x, CEED_NORM_MAX, &norm);
33*2b730f8bSJeremy L Thompson   if (fabs(norm - 9.) > 100. * CEED_EPSILON) printf("Error: Max norm %f != 9.\n", norm);
34547d9b97Sjeremylt 
35547d9b97Sjeremylt   CeedVectorDestroy(&x);
36547d9b97Sjeremylt   CeedDestroy(&ceed);
37547d9b97Sjeremylt   return 0;
38547d9b97Sjeremylt }
39