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 CeedTransposeMode lmode = CEED_NOTRANSPOSE; 17 18 CeedInit(argv[1], &ceed); 19 20 CeedVectorCreate(ceed, ne+1, &x); 21 for (CeedInt i=0; i<ne+1; i++) 22 a[i] = 10 + i; 23 CeedVectorSetArray(x, CEED_MEM_HOST, CEED_USE_POINTER, a); 24 25 for (CeedInt i=0; i<ne; i++) { 26 for (CeedInt k=0; k<elemsize; k++) { 27 ind[elemsize*i+k] = i+k; 28 } 29 } 30 CeedElemRestrictionCreateBlocked(ceed, lmode, ne, elemsize, blksize, ne+1, 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