xref: /libCEED/tests/t105-vector.c (revision 4fee36f0a30516a0b5ad51bf7eb3b32d83efd623)
10436c2adSjeremylt /// @file
20436c2adSjeremylt /// Test creation, setting, reading, restoring, and destroying of a vector using CEED_MEM_DEVICE
30436c2adSjeremylt /// \test Test creation, setting, reading, restoring, and destroying of a vector
40436c2adSjeremylt #include <ceed.h>
50436c2adSjeremylt 
60436c2adSjeremylt int main(int argc, char **argv) {
70436c2adSjeremylt   Ceed       ceed;
8*4fee36f0SJeremy L Thompson   CeedVector x, y;
9*4fee36f0SJeremy L Thompson   CeedInt    len = 10;
100436c2adSjeremylt 
110436c2adSjeremylt   CeedInit(argv[1], &ceed);
120436c2adSjeremylt 
13*4fee36f0SJeremy L Thompson   CeedVectorCreate(ceed, len, &x);
14*4fee36f0SJeremy L Thompson   CeedVectorCreate(ceed, len, &y);
15*4fee36f0SJeremy L Thompson   {
16*4fee36f0SJeremy L Thompson     CeedScalar array[len];
170436c2adSjeremylt 
18*4fee36f0SJeremy L Thompson     for (CeedInt i = 0; i < len; i++) array[i] = len + i;
19*4fee36f0SJeremy L Thompson     CeedVectorSetArray(x, CEED_MEM_HOST, CEED_COPY_VALUES, array);
202b730f8bSJeremy L Thompson   }
21*4fee36f0SJeremy L Thompson   {
22*4fee36f0SJeremy L Thompson     const CeedScalar *read_array;
23*4fee36f0SJeremy L Thompson 
24*4fee36f0SJeremy L Thompson     CeedVectorGetArrayRead(x, CEED_MEM_DEVICE, &read_array);
25*4fee36f0SJeremy L Thompson     CeedVectorSetArray(y, CEED_MEM_DEVICE, CEED_COPY_VALUES, (CeedScalar *)read_array);
26*4fee36f0SJeremy L Thompson     CeedVectorRestoreArrayRead(x, &read_array);
27*4fee36f0SJeremy L Thompson   }
28*4fee36f0SJeremy L Thompson   {
29*4fee36f0SJeremy L Thompson     const CeedScalar *read_array;
30*4fee36f0SJeremy L Thompson 
31*4fee36f0SJeremy L Thompson     CeedVectorGetArrayRead(y, CEED_MEM_HOST, &read_array);
32*4fee36f0SJeremy L Thompson     for (CeedInt i = 0; i < len; i++) {
33*4fee36f0SJeremy L Thompson       if (read_array[i] != len + i) printf("Error reading array[%" CeedInt_FMT "] = %f\n", i, (CeedScalar)read_array[i]);
34*4fee36f0SJeremy L Thompson     }
35*4fee36f0SJeremy L Thompson     CeedVectorRestoreArrayRead(y, &read_array);
36*4fee36f0SJeremy L Thompson   }
370436c2adSjeremylt 
380436c2adSjeremylt   CeedVectorDestroy(&x);
390436c2adSjeremylt   CeedVectorDestroy(&y);
400436c2adSjeremylt   CeedDestroy(&ceed);
410436c2adSjeremylt   return 0;
420436c2adSjeremylt }
43