| Line | Branch | Exec | Source |
|---|---|---|---|
| 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[] = "Eigenproblem for the 1-D Laplacian with constraints.\n\n" | ||
| 12 | "Based on ex1.\n" | ||
| 13 | "The command line options are:\n" | ||
| 14 | " -n <n>, where <n> = number of grid subdivisions = matrix dimension.\n\n"; | ||
| 15 | |||
| 16 | #include <slepceps.h> | ||
| 17 | |||
| 18 | 14 | int main(int argc,char **argv) | |
| 19 | { | ||
| 20 | 14 | Mat A; | |
| 21 | 14 | EPS eps; | |
| 22 | 14 | EPSType type; | |
| 23 | 14 | Vec *vi=NULL,*vc=NULL,t; | |
| 24 | 14 | PetscInt n=30,nev=4,i,j,Istart,Iend,nini=0,ncon=0,bs; | |
| 25 | 14 | PetscReal alpha,beta,restart; | |
| 26 | 14 | PetscBool flg,lock; | |
| 27 | |||
| 28 |
1/2✓ Branch 0 taken 2 times.
✗ Branch 1 not taken.
|
14 | PetscFunctionBeginUser; |
| 29 |
4/6✓ Branch 0 taken 2 times.
✓ Branch 1 taken 8 times.
✓ Branch 2 taken 2 times.
✗ Branch 3 not taken.
✗ Branch 4 not taken.
✓ Branch 5 taken 2 times.
|
14 | PetscCall(SlepcInitialize(&argc,&argv,NULL,help)); |
| 30 |
4/6✓ Branch 0 taken 2 times.
✓ Branch 1 taken 8 times.
✓ Branch 2 taken 2 times.
✗ Branch 3 not taken.
✗ Branch 4 not taken.
✓ Branch 5 taken 2 times.
|
14 | PetscCall(PetscOptionsGetInt(NULL,NULL,"-n",&n,NULL)); |
| 31 |
4/6✓ Branch 0 taken 2 times.
✓ Branch 1 taken 8 times.
✓ Branch 2 taken 2 times.
✗ Branch 3 not taken.
✗ Branch 4 not taken.
✓ Branch 5 taken 2 times.
|
14 | PetscCall(PetscOptionsGetInt(NULL,NULL,"-nini",&nini,NULL)); |
| 32 |
4/6✓ Branch 0 taken 2 times.
✓ Branch 1 taken 8 times.
✓ Branch 2 taken 2 times.
✗ Branch 3 not taken.
✗ Branch 4 not taken.
✓ Branch 5 taken 2 times.
|
14 | PetscCall(PetscOptionsGetInt(NULL,NULL,"-ncon",&ncon,NULL)); |
| 33 |
4/6✓ Branch 0 taken 2 times.
✓ Branch 1 taken 8 times.
✓ Branch 2 taken 2 times.
✗ Branch 3 not taken.
✗ Branch 4 not taken.
✓ Branch 5 taken 2 times.
|
14 | PetscCall(PetscPrintf(PETSC_COMM_WORLD,"\n1-D Laplacian Eigenproblem, n=%" PetscInt_FMT " nini=%" PetscInt_FMT " ncon=%" PetscInt_FMT "\n\n",n,nini,ncon)); |
| 34 | |||
| 35 | /* - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - | ||
| 36 | Compute the operator matrix that defines the eigensystem, Ax=kx | ||
| 37 | - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - */ | ||
| 38 | |||
| 39 |
4/6✓ Branch 0 taken 2 times.
✓ Branch 1 taken 8 times.
✓ Branch 2 taken 2 times.
✗ Branch 3 not taken.
✗ Branch 4 not taken.
✓ Branch 5 taken 2 times.
|
14 | PetscCall(MatCreate(PETSC_COMM_WORLD,&A)); |
| 40 |
4/6✓ Branch 0 taken 2 times.
✓ Branch 1 taken 8 times.
✓ Branch 2 taken 2 times.
✗ Branch 3 not taken.
✗ Branch 4 not taken.
✓ Branch 5 taken 2 times.
|
14 | PetscCall(MatSetSizes(A,PETSC_DECIDE,PETSC_DECIDE,n,n)); |
| 41 |
4/6✓ Branch 0 taken 2 times.
✓ Branch 1 taken 8 times.
✓ Branch 2 taken 2 times.
✗ Branch 3 not taken.
✗ Branch 4 not taken.
✓ Branch 5 taken 2 times.
|
14 | PetscCall(MatSetFromOptions(A)); |
| 42 | |||
| 43 |
4/6✓ Branch 0 taken 2 times.
✓ Branch 1 taken 8 times.
✓ Branch 2 taken 2 times.
✗ Branch 3 not taken.
✗ Branch 4 not taken.
✓ Branch 5 taken 2 times.
|
14 | PetscCall(MatGetOwnershipRange(A,&Istart,&Iend)); |
| 44 |
2/2✓ Branch 0 taken 10 times.
✓ Branch 1 taken 10 times.
|
434 | for (i=Istart;i<Iend;i++) { |
| 45 |
6/8✓ Branch 0 taken 10 times.
✓ Branch 1 taken 10 times.
✓ Branch 2 taken 2 times.
✓ Branch 3 taken 8 times.
✓ Branch 4 taken 2 times.
✗ Branch 5 not taken.
✗ Branch 6 not taken.
✓ Branch 7 taken 2 times.
|
420 | if (i>0) PetscCall(MatSetValue(A,i,i-1,-1.0,INSERT_VALUES)); |
| 46 |
6/8✓ Branch 0 taken 10 times.
✓ Branch 1 taken 10 times.
✓ Branch 2 taken 2 times.
✓ Branch 3 taken 8 times.
✓ Branch 4 taken 2 times.
✗ Branch 5 not taken.
✗ Branch 6 not taken.
✓ Branch 7 taken 2 times.
|
420 | if (i<n-1) PetscCall(MatSetValue(A,i,i+1,-1.0,INSERT_VALUES)); |
| 47 |
4/6✓ Branch 0 taken 2 times.
✓ Branch 1 taken 8 times.
✓ Branch 2 taken 2 times.
✗ Branch 3 not taken.
✗ Branch 4 not taken.
✓ Branch 5 taken 2 times.
|
420 | PetscCall(MatSetValue(A,i,i,2.0,INSERT_VALUES)); |
| 48 | } | ||
| 49 |
4/6✓ Branch 0 taken 2 times.
✓ Branch 1 taken 8 times.
✓ Branch 2 taken 2 times.
✗ Branch 3 not taken.
✗ Branch 4 not taken.
✓ Branch 5 taken 2 times.
|
14 | PetscCall(MatAssemblyBegin(A,MAT_FINAL_ASSEMBLY)); |
| 50 |
4/6✓ Branch 0 taken 2 times.
✓ Branch 1 taken 8 times.
✓ Branch 2 taken 2 times.
✗ Branch 3 not taken.
✗ Branch 4 not taken.
✓ Branch 5 taken 2 times.
|
14 | PetscCall(MatAssemblyEnd(A,MAT_FINAL_ASSEMBLY)); |
| 51 | |||
| 52 | /* - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - | ||
| 53 | Create the eigensolver and set various options | ||
| 54 | - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - */ | ||
| 55 |
4/6✓ Branch 0 taken 2 times.
✓ Branch 1 taken 8 times.
✓ Branch 2 taken 2 times.
✗ Branch 3 not taken.
✗ Branch 4 not taken.
✓ Branch 5 taken 2 times.
|
14 | PetscCall(EPSCreate(PETSC_COMM_WORLD,&eps)); |
| 56 |
4/6✓ Branch 0 taken 2 times.
✓ Branch 1 taken 8 times.
✓ Branch 2 taken 2 times.
✗ Branch 3 not taken.
✗ Branch 4 not taken.
✓ Branch 5 taken 2 times.
|
14 | PetscCall(EPSSetOperators(eps,A,NULL)); |
| 57 |
4/6✓ Branch 0 taken 2 times.
✓ Branch 1 taken 8 times.
✓ Branch 2 taken 2 times.
✗ Branch 3 not taken.
✗ Branch 4 not taken.
✓ Branch 5 taken 2 times.
|
14 | PetscCall(EPSSetProblemType(eps,EPS_HEP)); |
| 58 |
4/6✓ Branch 0 taken 2 times.
✓ Branch 1 taken 8 times.
✓ Branch 2 taken 2 times.
✗ Branch 3 not taken.
✗ Branch 4 not taken.
✓ Branch 5 taken 2 times.
|
14 | PetscCall(EPSSetType(eps,EPSLOBPCG)); |
| 59 |
4/6✓ Branch 0 taken 2 times.
✓ Branch 1 taken 8 times.
✓ Branch 2 taken 2 times.
✗ Branch 3 not taken.
✗ Branch 4 not taken.
✓ Branch 5 taken 2 times.
|
14 | PetscCall(EPSSetWhichEigenpairs(eps,EPS_SMALLEST_REAL)); |
| 60 |
4/6✓ Branch 0 taken 2 times.
✓ Branch 1 taken 8 times.
✓ Branch 2 taken 2 times.
✗ Branch 3 not taken.
✗ Branch 4 not taken.
✓ Branch 5 taken 2 times.
|
14 | PetscCall(EPSSetConvergenceTest(eps,EPS_CONV_ABS)); |
| 61 |
4/6✓ Branch 0 taken 2 times.
✓ Branch 1 taken 8 times.
✓ Branch 2 taken 2 times.
✗ Branch 3 not taken.
✗ Branch 4 not taken.
✓ Branch 5 taken 2 times.
|
14 | PetscCall(EPSSetDimensions(eps,nev,PETSC_DETERMINE,PETSC_DETERMINE)); |
| 62 |
4/6✓ Branch 0 taken 2 times.
✓ Branch 1 taken 8 times.
✓ Branch 2 taken 2 times.
✗ Branch 3 not taken.
✗ Branch 4 not taken.
✓ Branch 5 taken 2 times.
|
14 | PetscCall(EPSLOBPCGSetBlockSize(eps,nev)); |
| 63 |
4/6✓ Branch 0 taken 2 times.
✓ Branch 1 taken 8 times.
✓ Branch 2 taken 2 times.
✗ Branch 3 not taken.
✗ Branch 4 not taken.
✓ Branch 5 taken 2 times.
|
14 | PetscCall(EPSLOBPCGSetRestart(eps,0.7)); |
| 64 |
4/6✓ Branch 0 taken 2 times.
✓ Branch 1 taken 8 times.
✓ Branch 2 taken 2 times.
✗ Branch 3 not taken.
✗ Branch 4 not taken.
✓ Branch 5 taken 2 times.
|
14 | PetscCall(EPSSetTolerances(eps,1e-8,1200)); |
| 65 |
4/6✓ Branch 0 taken 2 times.
✓ Branch 1 taken 8 times.
✓ Branch 2 taken 2 times.
✗ Branch 3 not taken.
✗ Branch 4 not taken.
✓ Branch 5 taken 2 times.
|
14 | PetscCall(EPSSetFromOptions(eps)); |
| 66 | |||
| 67 |
4/6✓ Branch 0 taken 2 times.
✓ Branch 1 taken 8 times.
✓ Branch 2 taken 2 times.
✗ Branch 3 not taken.
✗ Branch 4 not taken.
✓ Branch 5 taken 2 times.
|
14 | PetscCall(MatCreateVecs(A,&t,NULL)); |
| 68 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 10 times.
|
14 | if (nini) { |
| 69 | ✗ | PetscCall(VecDuplicateVecs(t,nini,&vi)); | |
| 70 | ✗ | for (i=0;i<nini;i++) PetscCall(VecSetRandom(vi[i],NULL)); | |
| 71 | ✗ | PetscCall(EPSSetInitialSpace(eps,nini,vi)); | |
| 72 | } | ||
| 73 |
1/2✓ Branch 0 taken 10 times.
✗ Branch 1 not taken.
|
14 | if (ncon) { /* constraints are exact eigenvectors of lowest eigenvalues */ |
| 74 | 14 | alpha = PETSC_PI/(n+1); | |
| 75 | 14 | beta = PetscSqrtReal(2.0/(n+1)); | |
| 76 |
4/6✓ Branch 0 taken 2 times.
✓ Branch 1 taken 8 times.
✓ Branch 2 taken 2 times.
✗ Branch 3 not taken.
✗ Branch 4 not taken.
✓ Branch 5 taken 2 times.
|
14 | PetscCall(VecDuplicateVecs(t,ncon,&vc)); |
| 77 |
2/2✓ Branch 0 taken 10 times.
✓ Branch 1 taken 10 times.
|
42 | for (i=0;i<ncon;i++) { |
| 78 |
7/8✓ Branch 0 taken 2 times.
✓ Branch 1 taken 8 times.
✓ Branch 2 taken 10 times.
✓ Branch 3 taken 8 times.
✗ Branch 4 not taken.
✓ Branch 5 taken 2 times.
✓ Branch 6 taken 2 times.
✓ Branch 7 taken 2 times.
|
868 | for (j=0;j<n;j++) PetscCall(VecSetValue(vc[i],j,PetscSinReal(alpha*(j+1)*(i+1))*beta,INSERT_VALUES)); |
| 79 |
4/6✓ Branch 0 taken 2 times.
✓ Branch 1 taken 8 times.
✓ Branch 2 taken 2 times.
✗ Branch 3 not taken.
✗ Branch 4 not taken.
✓ Branch 5 taken 2 times.
|
28 | PetscCall(VecAssemblyBegin(vc[i])); |
| 80 |
4/6✓ Branch 0 taken 2 times.
✓ Branch 1 taken 8 times.
✓ Branch 2 taken 2 times.
✗ Branch 3 not taken.
✗ Branch 4 not taken.
✓ Branch 5 taken 2 times.
|
28 | PetscCall(VecAssemblyEnd(vc[i])); |
| 81 | } | ||
| 82 |
4/6✓ Branch 0 taken 2 times.
✓ Branch 1 taken 8 times.
✓ Branch 2 taken 2 times.
✗ Branch 3 not taken.
✗ Branch 4 not taken.
✓ Branch 5 taken 2 times.
|
14 | PetscCall(EPSSetDeflationSpace(eps,ncon,vc)); |
| 83 | } | ||
| 84 | |||
| 85 | /* - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - | ||
| 86 | Solve the eigensystem | ||
| 87 | - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - */ | ||
| 88 | |||
| 89 |
4/6✓ Branch 0 taken 2 times.
✓ Branch 1 taken 8 times.
✓ Branch 2 taken 2 times.
✗ Branch 3 not taken.
✗ Branch 4 not taken.
✓ Branch 5 taken 2 times.
|
14 | PetscCall(EPSSolve(eps)); |
| 90 |
4/6✓ Branch 0 taken 2 times.
✓ Branch 1 taken 8 times.
✓ Branch 2 taken 2 times.
✗ Branch 3 not taken.
✗ Branch 4 not taken.
✓ Branch 5 taken 2 times.
|
14 | PetscCall(EPSGetType(eps,&type)); |
| 91 |
4/6✓ Branch 0 taken 2 times.
✓ Branch 1 taken 8 times.
✓ Branch 2 taken 2 times.
✗ Branch 3 not taken.
✗ Branch 4 not taken.
✓ Branch 5 taken 2 times.
|
14 | PetscCall(PetscPrintf(PETSC_COMM_WORLD," Solution method: %s\n",type)); |
| 92 |
4/6✓ Branch 0 taken 2 times.
✓ Branch 1 taken 8 times.
✓ Branch 2 taken 2 times.
✗ Branch 3 not taken.
✗ Branch 4 not taken.
✓ Branch 5 taken 2 times.
|
14 | PetscCall(PetscObjectTypeCompare((PetscObject)eps,EPSLOBPCG,&flg)); |
| 93 |
1/2✓ Branch 0 taken 10 times.
✗ Branch 1 not taken.
|
14 | if (flg) { |
| 94 |
4/6✓ Branch 0 taken 2 times.
✓ Branch 1 taken 8 times.
✓ Branch 2 taken 2 times.
✗ Branch 3 not taken.
✗ Branch 4 not taken.
✓ Branch 5 taken 2 times.
|
14 | PetscCall(EPSLOBPCGGetLocking(eps,&lock)); |
| 95 |
5/8✓ Branch 0 taken 10 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 2 times.
✓ Branch 3 taken 8 times.
✓ Branch 4 taken 2 times.
✗ Branch 5 not taken.
✗ Branch 6 not taken.
✓ Branch 7 taken 2 times.
|
14 | if (lock) PetscCall(PetscPrintf(PETSC_COMM_WORLD," Using soft locking\n")); |
| 96 |
4/6✓ Branch 0 taken 2 times.
✓ Branch 1 taken 8 times.
✓ Branch 2 taken 2 times.
✗ Branch 3 not taken.
✗ Branch 4 not taken.
✓ Branch 5 taken 2 times.
|
14 | PetscCall(EPSLOBPCGGetRestart(eps,&restart)); |
| 97 |
4/6✓ Branch 0 taken 2 times.
✓ Branch 1 taken 8 times.
✓ Branch 2 taken 2 times.
✗ Branch 3 not taken.
✗ Branch 4 not taken.
✓ Branch 5 taken 2 times.
|
14 | PetscCall(PetscPrintf(PETSC_COMM_WORLD," LOBPCG Restart parameter=%.4g\n",(double)restart)); |
| 98 |
4/6✓ Branch 0 taken 2 times.
✓ Branch 1 taken 8 times.
✓ Branch 2 taken 2 times.
✗ Branch 3 not taken.
✗ Branch 4 not taken.
✓ Branch 5 taken 2 times.
|
14 | PetscCall(EPSLOBPCGGetBlockSize(eps,&bs)); |
| 99 |
4/6✓ Branch 0 taken 2 times.
✓ Branch 1 taken 8 times.
✓ Branch 2 taken 2 times.
✗ Branch 3 not taken.
✗ Branch 4 not taken.
✓ Branch 5 taken 2 times.
|
14 | PetscCall(PetscPrintf(PETSC_COMM_WORLD," LOBPCG Block size=%" PetscInt_FMT "\n",bs)); |
| 100 | } | ||
| 101 | |||
| 102 | /* - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - | ||
| 103 | Display solution and clean up | ||
| 104 | - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - */ | ||
| 105 | |||
| 106 |
4/6✓ Branch 0 taken 2 times.
✓ Branch 1 taken 8 times.
✓ Branch 2 taken 2 times.
✗ Branch 3 not taken.
✗ Branch 4 not taken.
✓ Branch 5 taken 2 times.
|
14 | PetscCall(EPSErrorView(eps,EPS_ERROR_RELATIVE,NULL)); |
| 107 |
4/6✓ Branch 0 taken 2 times.
✓ Branch 1 taken 8 times.
✓ Branch 2 taken 2 times.
✗ Branch 3 not taken.
✗ Branch 4 not taken.
✓ Branch 5 taken 2 times.
|
14 | PetscCall(EPSDestroy(&eps)); |
| 108 |
4/6✓ Branch 0 taken 2 times.
✓ Branch 1 taken 8 times.
✓ Branch 2 taken 2 times.
✗ Branch 3 not taken.
✗ Branch 4 not taken.
✓ Branch 5 taken 2 times.
|
14 | PetscCall(MatDestroy(&A)); |
| 109 |
4/6✓ Branch 0 taken 2 times.
✓ Branch 1 taken 8 times.
✓ Branch 2 taken 2 times.
✗ Branch 3 not taken.
✗ Branch 4 not taken.
✓ Branch 5 taken 2 times.
|
14 | PetscCall(VecDestroyVecs(nini,&vi)); |
| 110 |
4/6✓ Branch 0 taken 2 times.
✓ Branch 1 taken 8 times.
✓ Branch 2 taken 2 times.
✗ Branch 3 not taken.
✗ Branch 4 not taken.
✓ Branch 5 taken 2 times.
|
14 | PetscCall(VecDestroyVecs(ncon,&vc)); |
| 111 |
4/6✓ Branch 0 taken 2 times.
✓ Branch 1 taken 8 times.
✓ Branch 2 taken 2 times.
✗ Branch 3 not taken.
✗ Branch 4 not taken.
✓ Branch 5 taken 2 times.
|
14 | PetscCall(VecDestroy(&t)); |
| 112 |
3/6✓ Branch 0 taken 2 times.
✓ Branch 1 taken 2 times.
✓ Branch 2 taken 2 times.
✗ Branch 3 not taken.
✗ Branch 4 not taken.
✗ Branch 5 not taken.
|
14 | PetscCall(SlepcFinalize()); |
| 113 | return 0; | ||
| 114 | } | ||
| 115 | |||
| 116 | /*TEST | ||
| 117 | |||
| 118 | testset: | ||
| 119 | args: -ncon 2 | ||
| 120 | output_file: output/test24_1.out | ||
| 121 | test: | ||
| 122 | suffix: 1 | ||
| 123 | requires: !single | ||
| 124 | test: | ||
| 125 | suffix: 1_cuda | ||
| 126 | args: -mat_type aijcusparse | ||
| 127 | requires: cuda !single | ||
| 128 | test: | ||
| 129 | suffix: 1_hip | ||
| 130 | args: -mat_type aijhipsparse | ||
| 131 | requires: hip !single | ||
| 132 | |||
| 133 | TEST*/ | ||
| 134 |