1 /// @file 2 /// Test scaling a vector 3 /// \test Test scaling of a vector 4 #include <ceed.h> 5 #include <math.h> 6 7 int main(int argc, char **argv) { 8 Ceed ceed; 9 CeedVector x; 10 CeedInt len = 10; 11 12 CeedInit(argv[1], &ceed); 13 14 CeedVectorCreate(ceed, len, &x); 15 { 16 CeedScalar array[len]; 17 18 for (CeedInt i = 0; i < len; i++) array[i] = 10 + i; 19 CeedVectorSetArray(x, CEED_MEM_HOST, CEED_COPY_VALUES, array); 20 } 21 { 22 // Sync memtype to device for GPU backends 23 CeedMemType type = CEED_MEM_HOST; 24 CeedGetPreferredMemType(ceed, &type); 25 CeedVectorSyncArray(x, type); 26 } 27 CeedVectorScale(x, -0.5); 28 29 { 30 const CeedScalar *read_array; 31 32 CeedVectorGetArrayRead(x, CEED_MEM_HOST, &read_array); 33 for (CeedInt i = 0; i < len; i++) { 34 if (fabs(read_array[i] + (10.0 + i) / 2) > 1e-14) { 35 // LCOV_EXCL_START 36 printf("Error in alpha x at index %" CeedInt_FMT ", computed: %f actual: %f\n", i, read_array[i], -(10.0 + i) / 2); 37 // LCOV_EXCL_STOP 38 } 39 } 40 CeedVectorRestoreArrayRead(x, &read_array); 41 } 42 43 CeedVectorDestroy(&x); 44 CeedDestroy(&ceed); 45 return 0; 46 } 47