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