Actual source code: dvdschm.c

      1: /*
      2:    - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
      3:    SLEPc - Scalable Library for Eigenvalue Problem Computations
      4:    Copyright (c) 2002-, Universitat Politecnica de Valencia, Spain

      6:    This file is part of SLEPc.
      7:    SLEPc is distributed under a 2-clause BSD license (see LICENSE).
      8:    - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
      9: */

     11: #include "davidson.h"

     13: #define DVD_CHECKSUM(b) ((b)->max_size_V + (b)->max_size_oldX)

     15: PetscErrorCode dvd_schm_basic_preconf(dvdDashboard *d,dvdBlackboard *b,PetscInt mpd,PetscInt min_size_V,PetscInt bs,PetscInt ini_size_V,PetscInt size_initV,PetscInt plusk,HarmType_t harmMode,KSP ksp,InitType_t init,PetscBool allResiduals,PetscBool orth,PetscBool doubleexp)
     16: {
     17:   PetscInt       check_sum0,check_sum1;

     19:   PetscFunctionBegin;
     20:   PetscCall(PetscMemzero(b,sizeof(dvdBlackboard)));
     21:   b->state = DVD_STATE_PRECONF;

     23:   for (check_sum0=-1,check_sum1=DVD_CHECKSUM(b); check_sum0 != check_sum1; check_sum0 = check_sum1, check_sum1 = DVD_CHECKSUM(b)) {

     25:     /* Setup basic management of V */
     26:     PetscCall(dvd_managementV_basic(d,b,bs,mpd,min_size_V,plusk,PetscNot(harmMode==DVD_HARM_NONE),allResiduals));

     28:     /* Setup the initial subspace for V */
     29:     PetscCall(dvd_initV(d,b,ini_size_V,size_initV,(init==DVD_INITV_KRYLOV)?PETSC_TRUE:PETSC_FALSE));

     31:     /* Setup the convergence in order to use the SLEPc convergence test */
     32:     PetscCall(dvd_testconv_slepc(d,b));

     34:     /* Setup Raileigh-Ritz for selecting the best eigenpairs in V */
     35:     PetscCall(dvd_calcpairs_qz(d,b,orth,PetscNot(harmMode==DVD_HARM_NONE)));
     36:     if (harmMode != DVD_HARM_NONE) PetscCall(dvd_harm_conf(d,b,harmMode,PETSC_FALSE,0.0));

     38:     /* Setup the method for improving the eigenvectors */
     39:     if (doubleexp) PetscCall(dvd_improvex_gd2(d,b,ksp,bs));
     40:     else {
     41:       PetscCall(dvd_improvex_jd(d,b,ksp,bs,PETSC_FALSE));
     42:       PetscCall(dvd_improvex_jd_proj_uv(d,b));
     43:       PetscCall(dvd_improvex_jd_lit_const(d,b,0,0.0,0.0));
     44:     }
     45:   }
     46:   PetscFunctionReturn(PETSC_SUCCESS);
     47: }

     49: PetscErrorCode dvd_schm_basic_conf(dvdDashboard *d,dvdBlackboard *b,PetscInt mpd,PetscInt min_size_V,PetscInt bs,PetscInt ini_size_V,PetscInt size_initV,PetscInt plusk,HarmType_t harmMode,PetscBool fixedTarget,PetscScalar t,KSP ksp,PetscReal fix,InitType_t init,PetscBool allResiduals,PetscBool orth,PetscBool dynamic,PetscBool doubleexp)
     50: {
     51:   PetscInt       check_sum0,check_sum1,maxits;
     52:   PetscReal      tol;

     54:   PetscFunctionBegin;
     55:   b->state = DVD_STATE_CONF;
     56:   check_sum0 = DVD_CHECKSUM(b);

     58:   /* Setup basic management of V */
     59:   PetscCall(dvd_managementV_basic(d,b,bs,mpd,min_size_V,plusk,PetscNot(harmMode==DVD_HARM_NONE),allResiduals));

     61:   /* Setup the initial subspace for V */
     62:   PetscCall(dvd_initV(d,b,ini_size_V,size_initV,(init==DVD_INITV_KRYLOV)?PETSC_TRUE:PETSC_FALSE));

     64:   /* Setup the convergence in order to use the SLEPc convergence test */
     65:   PetscCall(dvd_testconv_slepc(d,b));

     67:   /* Setup Raileigh-Ritz for selecting the best eigenpairs in V */
     68:   PetscCall(dvd_calcpairs_qz(d,b,orth,PetscNot(harmMode==DVD_HARM_NONE)));
     69:   if (harmMode != DVD_HARM_NONE) PetscCall(dvd_harm_conf(d,b,harmMode,fixedTarget,t));

     71:   /* Setup the method for improving the eigenvectors */
     72:   if (doubleexp) PetscCall(dvd_improvex_gd2(d,b,ksp,bs));
     73:   else {
     74:     PetscCall(dvd_improvex_jd(d,b,ksp,bs,dynamic));
     75:     PetscCall(dvd_improvex_jd_proj_uv(d,b));
     76:     PetscCall(KSPGetTolerances(ksp,&tol,NULL,NULL,&maxits));
     77:     PetscCall(dvd_improvex_jd_lit_const(d,b,maxits,tol,fix));
     78:   }

     80:   check_sum1 = DVD_CHECKSUM(b);
     81:   PetscAssert(check_sum0==check_sum1,PETSC_COMM_SELF,PETSC_ERR_PLIB,"Something awful happened");
     82:   PetscFunctionReturn(PETSC_SUCCESS);
     83: }