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 CeedInterlaceMode imode = CEED_NONINTERLACED; 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 31 CeedElemRestrictionCreateBlocked(ceed, imode, ne, elemsize, blksize, ne+1, 1, 32 CEED_MEM_HOST, CEED_USE_POINTER, ind, &r); 33 34 CeedVectorCreate(ceed, blksize*elemsize, &y); 35 CeedVectorSetValue(y, 0); // Allocates array 36 37 // NoTranspose 38 CeedElemRestrictionApplyBlock(r, 1, CEED_NOTRANSPOSE, x, y, 39 CEED_REQUEST_IMMEDIATE); 40 41 // Zero padded entries 42 CeedVectorGetArray(y, CEED_MEM_HOST, &y_array); 43 for (CeedInt i = (elemsize*ne - blksize*elemsize); i < blksize*elemsize; ++i) { 44 y_array[i] = 0; 45 } 46 CeedVectorRestoreArray(y, &y_array); 47 CeedVectorView(y, "%12.8f", stdout); 48 49 // Transpose 50 CeedVectorGetArray(x, CEED_MEM_HOST, (CeedScalar **)&a); 51 for (CeedInt i=0; i<ne+1; i++) 52 a[i] = 0; 53 CeedVectorRestoreArray(x, (CeedScalar **)&a); 54 CeedElemRestrictionApplyBlock(r, 1, CEED_TRANSPOSE, y, x, 55 CEED_REQUEST_IMMEDIATE); 56 CeedVectorView(x, "%12.8f", stdout); 57 58 CeedVectorDestroy(&x); 59 CeedVectorDestroy(&y); 60 CeedElemRestrictionDestroy(&r); 61 CeedDestroy(&ceed); 62 return 0; 63 } 64