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[] = "Symmetric-indefinite eigenproblem.\n\n"
      12             :   "The command line options are:\n"
      13             :   "  -n <n>, where <n> = number of grid subdivisions in x dimension.\n"
      14             :   "  -m <m>, where <m> = number of grid subdivisions in y dimension.\n\n";
      15             : 
      16             : #include <slepceps.h>
      17             : 
      18           4 : int main(int argc,char **argv)
      19             : {
      20           4 :   Mat            A,B;             /* problem matrices */
      21           4 :   EPS            eps;             /* eigenproblem solver context */
      22           4 :   PetscInt       N,n=10,m,Istart,Iend,II,nev,i,j;
      23           4 :   PetscBool      flag,terse;
      24             : 
      25           4 :   PetscFunctionBeginUser;
      26           4 :   PetscCall(SlepcInitialize(&argc,&argv,(char*)0,help));
      27           4 :   PetscCall(PetscOptionsGetInt(NULL,NULL,"-n",&n,NULL));
      28           4 :   PetscCall(PetscOptionsGetInt(NULL,NULL,"-m",&m,&flag));
      29           4 :   if (!flag) m=n;
      30           4 :   N = n*m;
      31           4 :   PetscCall(PetscPrintf(PETSC_COMM_WORLD,"\nSymmetric-indefinite eigenproblem, N=%" PetscInt_FMT "\n\n",N));
      32             : 
      33             :   /* - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
      34             :           Compute the matrices that define the eigensystem, Ax=kBx
      35             :      - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - */
      36             : 
      37           4 :   PetscCall(MatCreate(PETSC_COMM_WORLD,&A));
      38           4 :   PetscCall(MatSetSizes(A,PETSC_DECIDE,PETSC_DECIDE,N,N));
      39           4 :   PetscCall(MatSetFromOptions(A));
      40             : 
      41           4 :   PetscCall(MatCreate(PETSC_COMM_WORLD,&B));
      42           4 :   PetscCall(MatSetSizes(B,PETSC_DECIDE,PETSC_DECIDE,N,N));
      43           4 :   PetscCall(MatSetFromOptions(B));
      44             : 
      45           4 :   PetscCall(MatGetOwnershipRange(A,&Istart,&Iend));
      46         424 :   for (II=Istart;II<Iend;II++) {
      47         420 :     i = II/n; j = II-i*n;
      48         420 :     if (i>0) PetscCall(MatSetValue(A,II,II-n,-1.0,INSERT_VALUES));
      49         420 :     if (i<m-1) PetscCall(MatSetValue(A,II,II+n,-1.0,INSERT_VALUES));
      50         420 :     if (j>0) PetscCall(MatSetValue(A,II,II-1,-1.0,INSERT_VALUES));
      51         420 :     if (j<n-1) PetscCall(MatSetValue(A,II,II+1,-1.0,INSERT_VALUES));
      52         420 :     PetscCall(MatSetValue(A,II,II,4.0,INSERT_VALUES));
      53         420 :     PetscCall(MatSetValue(B,II,N-II-1,1.0,INSERT_VALUES));
      54             :   }
      55             : 
      56           4 :   PetscCall(MatAssemblyBegin(A,MAT_FINAL_ASSEMBLY));
      57           4 :   PetscCall(MatAssemblyBegin(B,MAT_FINAL_ASSEMBLY));
      58           4 :   PetscCall(MatAssemblyEnd(A,MAT_FINAL_ASSEMBLY));
      59           4 :   PetscCall(MatAssemblyEnd(B,MAT_FINAL_ASSEMBLY));
      60             : 
      61             :   /* - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
      62             :                 Create the eigensolver and set various options
      63             :      - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - */
      64             : 
      65           4 :   PetscCall(EPSCreate(PETSC_COMM_WORLD,&eps));
      66           4 :   PetscCall(EPSSetOperators(eps,A,B));
      67           4 :   PetscCall(EPSSetProblemType(eps,EPS_GHIEP));
      68           4 :   PetscCall(EPSSetFromOptions(eps));
      69             : 
      70             :   /* - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
      71             :                       Solve the eigensystem
      72             :      - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - */
      73             : 
      74           4 :   PetscCall(EPSSolve(eps));
      75           4 :   PetscCall(EPSGetDimensions(eps,&nev,NULL,NULL));
      76           4 :   PetscCall(PetscPrintf(PETSC_COMM_WORLD," Number of requested eigenvalues: %" PetscInt_FMT "\n",nev));
      77             : 
      78             :   /* - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
      79             :                     Display solution and clean up
      80             :      - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - */
      81             : 
      82             :   /* show detailed info unless -terse option is given by user */
      83           4 :   PetscCall(PetscOptionsHasName(NULL,NULL,"-terse",&terse));
      84           4 :   if (terse) PetscCall(EPSErrorView(eps,EPS_ERROR_RELATIVE,NULL));
      85             :   else {
      86           0 :     PetscCall(PetscViewerPushFormat(PETSC_VIEWER_STDOUT_WORLD,PETSC_VIEWER_ASCII_INFO_DETAIL));
      87           0 :     PetscCall(EPSConvergedReasonView(eps,PETSC_VIEWER_STDOUT_WORLD));
      88           0 :     PetscCall(EPSErrorView(eps,EPS_ERROR_RELATIVE,PETSC_VIEWER_STDOUT_WORLD));
      89           0 :     PetscCall(PetscViewerPopFormat(PETSC_VIEWER_STDOUT_WORLD));
      90             :   }
      91             : 
      92           4 :   PetscCall(EPSDestroy(&eps));
      93           4 :   PetscCall(MatDestroy(&A));
      94           4 :   PetscCall(MatDestroy(&B));
      95           4 :   PetscCall(SlepcFinalize());
      96             :   return 0;
      97             : }
      98             : 
      99             : /*TEST
     100             : 
     101             :    testset:
     102             :       args: -eps_nev 4 -eps_ncv 12 -terse -st_type sinvert -eps_krylovschur_restart .3
     103             :       requires: !single
     104             :       output_file: output/test18_1.out
     105             :       test:
     106             :          suffix: 1_ks
     107             :       test:
     108             :          suffix: 1_ks_gnhep
     109             :          args: -eps_gen_non_hermitian
     110             :          requires: !__float128
     111             :       test:
     112             :          suffix: 2_cuda_ks
     113             :          args: -mat_type aijcusparse
     114             :          requires: cuda
     115             :       test:
     116             :          suffix: 2_cuda_ks_gnhep
     117             :          args: -eps_gen_non_hermitian -mat_type aijcusparse
     118             :          requires: cuda
     119             : 
     120             :    testset:
     121             :       args: -n 10 -m 11 -eps_target 0.2 -eps_harmonic -eps_nev 2 -eps_ncv 11 -terse
     122             :       requires: !single
     123             :       filter: sed -e "s/[+-]0\.0*i//g"
     124             :       output_file: output/test18_2.out
     125             :       test:
     126             :          suffix: 2_gd
     127             :          args: -eps_type gd
     128             :       test:
     129             :          suffix: 2_jd
     130             :          args: -eps_type jd -st_ksp_type bcgs
     131             : 
     132             : TEST*/