Line data Source code
1 : /*
2 : - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
3 : SLEPc - Scalable Library for Eigenvalue Problem Computations
4 : Copyright (c) 2002-, Universitat Politecnica de Valencia, Spain
5 :
6 : This file is part of SLEPc.
7 : SLEPc is distributed under a 2-clause BSD license (see LICENSE).
8 : - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
9 : */
10 :
11 : static char help[] = "Test BV operations using internal buffer instead of array arguments.\n\n";
12 :
13 : #include <slepcbv.h>
14 :
15 4 : int main(int argc,char **argv)
16 : {
17 4 : Vec t,v,z;
18 4 : BV X;
19 4 : PetscInt i,j,n=10,k=5,l=3;
20 4 : PetscReal nrm;
21 4 : PetscViewer view;
22 4 : PetscBool verbose;
23 :
24 4 : PetscFunctionBeginUser;
25 4 : PetscCall(SlepcInitialize(&argc,&argv,NULL,help));
26 4 : PetscCall(PetscOptionsGetInt(NULL,NULL,"-n",&n,NULL));
27 4 : PetscCall(PetscOptionsGetInt(NULL,NULL,"-k",&k,NULL));
28 4 : PetscCall(PetscOptionsGetInt(NULL,NULL,"-l",&l,NULL));
29 4 : PetscCall(PetscOptionsHasName(NULL,NULL,"-verbose",&verbose));
30 4 : PetscCall(PetscPrintf(PETSC_COMM_WORLD,"Test BV with %" PetscInt_FMT " columns of dimension %" PetscInt_FMT ".\n",k,n));
31 :
32 : /* Create template vector */
33 4 : PetscCall(VecCreate(PETSC_COMM_WORLD,&t));
34 4 : PetscCall(VecSetSizes(t,PETSC_DECIDE,n));
35 4 : PetscCall(VecSetFromOptions(t));
36 :
37 : /* Create BV object X */
38 4 : PetscCall(BVCreate(PETSC_COMM_WORLD,&X));
39 4 : PetscCall(PetscObjectSetName((PetscObject)X,"X"));
40 4 : PetscCall(BVSetSizesFromVec(X,t,k));
41 4 : PetscCall(BVSetFromOptions(X));
42 :
43 : /* Set up viewer */
44 4 : PetscCall(PetscViewerASCIIGetStdout(PETSC_COMM_WORLD,&view));
45 4 : if (verbose) PetscCall(PetscViewerPushFormat(view,PETSC_VIEWER_ASCII_MATLAB));
46 :
47 : /* Fill X entries */
48 24 : for (j=0;j<k;j++) {
49 20 : PetscCall(BVGetColumn(X,j,&v));
50 20 : PetscCall(VecSet(v,0.0));
51 100 : for (i=0;i<4;i++) {
52 80 : if (i+j<n) PetscCall(VecSetValue(v,i+j,(PetscScalar)(3*i+j-2),INSERT_VALUES));
53 : }
54 20 : PetscCall(VecAssemblyBegin(v));
55 20 : PetscCall(VecAssemblyEnd(v));
56 20 : PetscCall(BVRestoreColumn(X,j,&v));
57 : }
58 4 : if (verbose) PetscCall(BVView(X,view));
59 :
60 : /* Test BVDotColumn */
61 4 : PetscCall(BVDotColumn(X,2,NULL));
62 4 : if (verbose) {
63 0 : PetscCall(PetscPrintf(PETSC_COMM_WORLD,"After BVDotColumn - - - - - - -\n"));
64 0 : PetscCall(BVGetBufferVec(X,&z));
65 0 : PetscCall(VecView(z,view));
66 : }
67 : /* Test BVMultColumn */
68 4 : PetscCall(BVMultColumn(X,-1.0,1.0,2,NULL));
69 4 : if (verbose) {
70 0 : PetscCall(PetscPrintf(PETSC_COMM_WORLD,"After BVMultColumn - - - - - - - - -\n"));
71 0 : PetscCall(BVView(X,view));
72 : }
73 :
74 4 : PetscCall(BVNorm(X,NORM_FROBENIUS,&nrm));
75 4 : PetscCall(PetscPrintf(PETSC_COMM_WORLD,"Frobenius Norm or X = %g\n",(double)nrm));
76 :
77 4 : PetscCall(BVDestroy(&X));
78 4 : PetscCall(VecDestroy(&t));
79 4 : PetscCall(SlepcFinalize());
80 : return 0;
81 : }
82 :
83 : /*TEST
84 :
85 : testset:
86 : output_file: output/test13_1.out
87 : test:
88 : suffix: 1
89 : args: -bv_type {{vecs contiguous svec mat}shared output}
90 : test:
91 : suffix: 1_cuda
92 : args: -bv_type {{svec mat}} -vec_type cuda
93 : requires: cuda
94 : test:
95 : suffix: 1_hip
96 : args: -bv_type {{svec mat}} -vec_type hip
97 : requires: hip
98 :
99 : TEST*/
|