1 /// @file 2 /// Test creation, use, and destruction of a blocked element restriction 3 /// \test Test creation, use, and destruction of a blocked element restriction 4 #include <ceed.h> 5 6 int main(int argc, char **argv) { 7 Ceed ceed; 8 CeedVector x, y; 9 CeedInt ne = 8; 10 CeedInt blksize = 5; 11 CeedInt elemsize = 2; 12 CeedInt ind[elemsize*ne]; 13 CeedScalar a[ne+1]; 14 CeedElemRestriction r; 15 CeedScalar *y_array; 16 17 CeedInit(argv[1], &ceed); 18 19 CeedVectorCreate(ceed, ne+1, &x); 20 for (CeedInt i=0; i<ne+1; i++) 21 a[i] = 10 + i; 22 CeedVectorSetArray(x, CEED_MEM_HOST, CEED_USE_POINTER, a); 23 24 for (CeedInt i=0; i<ne; i++) { 25 for (CeedInt k=0; k<elemsize; k++) { 26 ind[elemsize*i+k] = i+k; 27 } 28 } 29 30 CeedElemRestrictionCreateBlocked(ceed, ne, elemsize, blksize, 1, 1, ne+1, 31 CEED_MEM_HOST, CEED_USE_POINTER, ind, &r); 32 33 CeedVectorCreate(ceed, blksize*elemsize, &y); 34 CeedVectorSetValue(y, 0); // Allocates array 35 36 // NoTranspose 37 CeedElemRestrictionApplyBlock(r, 1, CEED_NOTRANSPOSE, x, y, 38 CEED_REQUEST_IMMEDIATE); 39 40 // Zero padded entries 41 CeedVectorGetArray(y, CEED_MEM_HOST, &y_array); 42 for (CeedInt i = (elemsize*ne - blksize*elemsize); i < blksize*elemsize; ++i) { 43 y_array[i] = 0; 44 } 45 CeedVectorRestoreArray(y, &y_array); 46 CeedVectorView(y, "%12.8f", stdout); 47 48 // Transpose 49 CeedVectorGetArray(x, CEED_MEM_HOST, (CeedScalar **)&a); 50 for (CeedInt i=0; i<ne+1; i++) 51 a[i] = 0; 52 CeedVectorRestoreArray(x, (CeedScalar **)&a); 53 CeedElemRestrictionApplyBlock(r, 1, CEED_TRANSPOSE, y, x, 54 CEED_REQUEST_IMMEDIATE); 55 CeedVectorView(x, "%12.8f", stdout); 56 57 CeedVectorDestroy(&x); 58 CeedVectorDestroy(&y); 59 CeedElemRestrictionDestroy(&r); 60 CeedDestroy(&ceed); 61 return 0; 62 } 63