Dear Barry,
the code is attached.
Just to let you know. When I commented out MatShellSetContext() in
FormJacobianShell(), then the code seems to work, meaning that the base
vector is passed to shell matrix context behind the scene.
Best regards,
Yi
On 2/5/24 19:09, Barry Smith wrote:
Send the entire code.
On Feb 4, 2024, at 4:43 PM, Yi Hu <y...@mpie.de> wrote:
Thanks for your response. You are correct. I overlooked this step.
Now I am trying to correct my "shell matrix approach" for ex1f.F90 of
snes solver
(https://github.com/hyharry/small_petsc_test/blob/master/test_shell_jac/ex1f.F90).
I realized that I need to record the base vector X in the context of
shell matrix and then use this info to carry MyMult. However, the
context cannot be obtained through MatShellGetContext(). Here are the
critical parts of my code.
INTERFACE MatCreateShell
SUBROUTINE MatCreateShell(comm,mloc,nloc,m,n,ctx,mat,ierr)
USE solver_context
MPI_Comm :: comm
PetscInt :: mloc,nloc,m,n
Vec :: ctx
Mat :: mat
PetscErrorCode :: ierr
END SUBROUTINE MatCreateShell
END INTERFACE MatCreateShell
INTERFACE MatShellSetContext
SUBROUTINE MatShellSetContext(mat,ctx,ierr)
USE solver_context
Mat :: mat
!TYPE(MatCtx) :: ctx
Vec :: ctx
PetscErrorCode :: ierr
END SUBROUTINE MatShellSetContext
END INTERFACE MatShellSetContext
INTERFACE MatShellGetContext
SUBROUTINE MatShellGetContext(mat,ctx,ierr)
USE solver_context
Mat :: mat
Vec, Pointer :: ctx
PetscErrorCode :: ierr
END SUBROUTINE MatShellGetContext
END INTERFACE MatShellGetContext
in my FormShellJacobian() I did
subroutine FormJacobianShell(snes,X,jac,B,dummy,ierr)
......
call MatShellSetContext(jac,X,ierr)
......
Then in MyMult() I tried to recover this context by
call MatShellGetContext(J,x,ierr)
call VecView(x,PETSC_VIEWER_STDOUT_WORLD,ierr)
Then the program failed with
[0]PETSC ERROR: --------------------- Error Message
--------------------------------------------------------------
[0]PETSC ERROR: Null argument, when expecting valid pointer
[0]PETSC ERROR: Null Pointer: Parameter # 1
In MyMult, I actually defined x to be a pointer. So I am confused here.
Best regards,
Yi
On 1/31/24 03:18, Barry Smith wrote:
It is not running an extra KSP iteration. This "extra" matmult is
normal and occurs in many of the SNESLineSearchApply_* functions,
for example,
https://petsc.org/release/src/snes/linesearch/impls/bt/linesearchbt.c.html#SNESLineSearchApply_BT It
is used to decide if the Newton step results in sufficient decrease
of the function value.
Barry
On Jan 30, 2024, at 3:19 PM, Yi Hu <y...@mpie.de> wrote:
Hello Barry,
Thanks for your reply. The monitor options are fine. I actually
meant my modification of snes tutorial ex1f.F90 does not work and
has some unexpected behavior. I basically wanted to test if I can
use a shell matrix as my jacobian (code is here
https://github.com/hyharry/small_petsc_test/blob/master/test_shell_jac/ex1f.F90).
After compile my modified version and run with these monitor
options, it gives me the following,
( in rhs )
( leave rhs )
0 SNES Function norm 6.041522986797e+00
++++++++++++ in jac shell +++++++++++
0 KSP Residual norm 6.041522986797e+00
=== start mymult ===
=== done mymult ===
1 KSP Residual norm 5.065392549852e-16
Linear solve converged due to CONVERGED_RTOL iterations 1
=== start mymult ===
=== done mymult ===
( in rhs )
( leave rhs )
1 SNES Function norm 3.512662245652e+00
++++++++++++ in jac shell +++++++++++
0 KSP Residual norm 3.512662245652e+00
=== start mymult ===
=== done mymult ===
1 KSP Residual norm 6.230314124713e-16
Linear solve converged due to CONVERGED_RTOL iterations 1
=== start mymult ===
=== done mymult ===
( in rhs )
( leave rhs )
( in rhs )
( leave rhs )
2 SNES Function norm 8.969285922373e-01
++++++++++++ in jac shell +++++++++++
0 KSP Residual norm 8.969285922373e-01
=== start mymult ===
=== done mymult ===
1 KSP Residual norm 0.000000000000e+00
Linear solve converged due to CONVERGED_ATOL iterations 1
=== start mymult ===
=== done mymult ===
( in rhs )
( leave rhs )
( in rhs )
( leave rhs )
3 SNES Function norm 4.863816734540e-01
++++++++++++ in jac shell +++++++++++
0 KSP Residual norm 4.863816734540e-01
=== start mymult ===
=== done mymult ===
1 KSP Residual norm 0.000000000000e+00
Linear solve converged due to CONVERGED_ATOL iterations 1
=== start mymult ===
=== done mymult ===
( in rhs )
( leave rhs )
( in rhs )
( leave rhs )
4 SNES Function norm 3.512070785520e-01
++++++++++++ in jac shell +++++++++++
0 KSP Residual norm 3.512070785520e-01
=== start mymult ===
=== done mymult ===
1 KSP Residual norm 0.000000000000e+00
Linear solve converged due to CONVERGED_ATOL iterations 1
=== start mymult ===
=== done mymult ===
( in rhs )
( leave rhs )
( in rhs )
( leave rhs )
5 SNES Function norm 2.769700293115e-01
++++++++++++ in jac shell +++++++++++
0 KSP Residual norm 2.769700293115e-01
=== start mymult ===
=== done mymult ===
1 KSP Residual norm 1.104778916974e-16
Linear solve converged due to CONVERGED_RTOL iterations 1
=== start mymult ===
=== done mymult ===
( in rhs )
( leave rhs )
( in rhs )
( leave rhs )
6 SNES Function norm 2.055345318150e-01
++++++++++++ in jac shell +++++++++++
0 KSP Residual norm 2.055345318150e-01
=== start mymult ===
=== done mymult ===
1 KSP Residual norm 0.000000000000e+00
Linear solve converged due to CONVERGED_ATOL iterations 1
=== start mymult ===
=== done mymult ===
( in rhs )
( leave rhs )
( in rhs )
( leave rhs )
7 SNES Function norm 1.267482220786e-01
++++++++++++ in jac shell +++++++++++
0 KSP Residual norm 1.267482220786e-01
=== start mymult ===
=== done mymult ===
1 KSP Residual norm 1.498679601680e-17
Linear solve converged due to CONVERGED_RTOL iterations 1
=== start mymult ===
=== done mymult ===
( in rhs )
( leave rhs )
( in rhs )
( leave rhs )
8 SNES Function norm 3.468150619264e-02
++++++++++++ in jac shell +++++++++++
0 KSP Residual norm 3.468150619264e-02
=== start mymult ===
=== done mymult ===
1 KSP Residual norm 5.944160522951e-18
Linear solve converged due to CONVERGED_RTOL iterations 1
=== start mymult ===
=== done mymult ===
( in rhs )
( leave rhs )
( in rhs )
( leave rhs )
( in rhs )
( leave rhs )
( in rhs )
( leave rhs )
( in rhs )
( leave rhs )
( in rhs )
( leave rhs )
( in rhs )
( leave rhs )
( in rhs )
( leave rhs )
( in rhs )
( leave rhs )
( in rhs )
( leave rhs )
( in rhs )
( leave rhs )
( in rhs )
( leave rhs )
( in rhs )
( leave rhs )
( in rhs )
( leave rhs )
=== start mymult ===
=== done mymult ===
Nonlinear solve did not converge due to DIVERGED_LINE_SEARCH
iterations 8
Number of SNES iterations = 8
After each "Linear solve converged due to CONVERGED_ATOL
iterations", the code starts to do mymult again. So I thought it
did an extra (unwanted) KSP iteration. I would like to ask if this
extra iteration could be disabled, or maybe I am wrong about it.
Best regards,
Yi
On 1/30/24 18:35, Barry Smith wrote:
How do I see a difference? What does "hence ruin my previous
converged KSP result" mean? A different answer at the end of the
KSP solve?
$ ./joe > joe.basic
~/Src/petsc/src/ksp/ksp/tutorials*(barry/2023-09-15/fix-log-pcmpi=)*arch-fix-log-pcmpi
$ ./joe -ksp_monitor -ksp_converged_reason -snes_monitor > joe.monitor
~/Src/petsc/src/ksp/ksp/tutorials*(barry/2023-09-15/fix-log-pcmpi=)*arch-fix-log-pcmpi
$ diff joe.basic joe.monitor
0a1,36
> 0 SNES Function norm 6.041522986797e+00
> 0 KSP Residual norm 6.041522986797e+00
> 1 KSP Residual norm 5.065392549852e-16
> Linear solve converged due to CONVERGED_RTOL_NORMAL iterations 1
> 1 SNES Function norm 3.512662245652e+00
> 0 KSP Residual norm 3.512662245652e+00
> 1 KSP Residual norm 6.230314124713e-16
> Linear solve converged due to CONVERGED_RTOL_NORMAL iterations 1
> 2 SNES Function norm 8.969285922373e-01
> 0 KSP Residual norm 8.969285922373e-01
> 1 KSP Residual norm 0.000000000000e+00
> Linear solve converged due to CONVERGED_RTOL_NORMAL iterations 1
> 3 SNES Function norm 4.863816734540e-01
> 0 KSP Residual norm 4.863816734540e-01
> 1 KSP Residual norm 0.000000000000e+00
> Linear solve converged due to CONVERGED_RTOL_NORMAL iterations 1
> 4 SNES Function norm 3.512070785520e-01
> 0 KSP Residual norm 3.512070785520e-01
> 1 KSP Residual norm 0.000000000000e+00
> Linear solve converged due to CONVERGED_RTOL_NORMAL iterations 1
> 5 SNES Function norm 2.769700293115e-01
> 0 KSP Residual norm 2.769700293115e-01
> 1 KSP Residual norm 1.104778916974e-16
> Linear solve converged due to CONVERGED_RTOL_NORMAL iterations 1
> 6 SNES Function norm 2.055345318150e-01
> 0 KSP Residual norm 2.055345318150e-01
> 1 KSP Residual norm 1.535110861002e-17
> Linear solve converged due to CONVERGED_RTOL_NORMAL iterations 1
> 7 SNES Function norm 1.267482220786e-01
> 0 KSP Residual norm 1.267482220786e-01
> 1 KSP Residual norm 1.498679601680e-17
> Linear solve converged due to CONVERGED_RTOL_NORMAL iterations 1
> 8 SNES Function norm 3.468150619264e-02
> 0 KSP Residual norm 3.468150619264e-02
> 1 KSP Residual norm 5.944160522951e-18
> Linear solve converged due to CONVERGED_RTOL_NORMAL iterations 1
On Jan 30, 2024, at 11:19 AM, Yi Hu <y...@mpie.de> wrote:
Dear PETSc team,
I am still trying to sort out my previous
threadhttps://lists.mcs.anl.gov/pipermail/petsc-users/2024-January/050079.htmlusing
a minimal working example. However, I encountered another
problem. Basically I combined the basic usage of SNES solver and
shell matrix and tried to make it work. The jacobian of my snes
is replaced by a customized MATOP_MULT. The minimal example code
can be viewed
herehttps://github.com/hyharry/small_petsc_test/blob/master/test_shell_jac/ex1f.F90
When running with -ksp_monitor -ksp_converged_reason, it shows an
extra mymult step, and hence ruin my previous converged KSP
result. Implement a customized converged call-back also does not
help. I am wondering how to skip this extra ksp iteration. Could
anyone help me on this?
Thanks for your help.
Best wishes,
Yi
------------------------------------------------------------------------
-------------------------------------------------
Stay up to date and follow us on LinkedIn, Twitter and YouTube.
Max-Planck-Institut für Eisenforschung GmbH
Max-Planck-Straße 1
D-40237 Düsseldorf
Handelsregister B 2533
Amtsgericht Düsseldorf
Geschäftsführung
Prof. Dr. Gerhard Dehm
Prof. Dr. Jörg Neugebauer
Prof. Dr. Dierk Raabe
Dr. Kai de Weldige
Ust.-Id.-Nr.: DE 11 93 58 514
Steuernummer: 105 5891 1000
Please consider that invitations and e-mails of our institute are
only valid if they end with …@mpie.de.
If you are not sure of the validity please contact r...@mpie.de
Bitte beachten Sie, dass Einladungen zu Veranstaltungen und E-Mails
aus unserem Haus nur mit der Endung …@mpie.de gültig sind.
In Zweifelsfällen wenden Sie sich bitte an r...@mpie.de
-------------------------------------------------
------------------------------------------------------------------------
-------------------------------------------------
Stay up to date and follow us on LinkedIn, Twitter and YouTube.
Max-Planck-Institut für Eisenforschung GmbH
Max-Planck-Straße 1
D-40237 Düsseldorf
Handelsregister B 2533
Amtsgericht Düsseldorf
Geschäftsführung
Prof. Dr. Gerhard Dehm
Prof. Dr. Jörg Neugebauer
Prof. Dr. Dierk Raabe
Dr. Kai de Weldige
Ust.-Id.-Nr.: DE 11 93 58 514
Steuernummer: 105 5891 1000
Please consider that invitations and e-mails of our institute are
only valid if they end with …@mpie.de.
If you are not sure of the validity please contact r...@mpie.de
Bitte beachten Sie, dass Einladungen zu Veranstaltungen und E-Mails
aus unserem Haus nur mit der Endung …@mpie.de gültig sind.
In Zweifelsfällen wenden Sie sich bitte an r...@mpie.de
-------------------------------------------------
------------------------------------------------------------------------
-------------------------------------------------
Stay up to date and follow us on LinkedIn, Twitter and YouTube.
Max-Planck-Institut für Eisenforschung GmbH
Max-Planck-Straße 1
D-40237 Düsseldorf
Handelsregister B 2533
Amtsgericht Düsseldorf
Geschäftsführung
Prof. Dr. Gerhard Dehm
Prof. Dr. Jörg Neugebauer
Prof. Dr. Dierk Raabe
Dr. Kai de Weldige
Ust.-Id.-Nr.: DE 11 93 58 514
Steuernummer: 105 5891 1000
Please consider that invitations and e-mails of our institute are
only valid if they end with …@mpie.de.
If you are not sure of the validity please contact r...@mpie.de
Bitte beachten Sie, dass Einladungen zu Veranstaltungen und E-Mails
aus unserem Haus nur mit der Endung …@mpie.de gültig sind.
In Zweifelsfällen wenden Sie sich bitte an r...@mpie.de
-------------------------------------------------
-------------------------------------------------
Stay up to date and follow us on LinkedIn, Twitter and YouTube.
Max-Planck-Institut für Eisenforschung GmbH
Max-Planck-Straße 1
D-40237 Düsseldorf
Handelsregister B 2533
Amtsgericht Düsseldorf
Geschäftsführung
Prof. Dr. Gerhard Dehm
Prof. Dr. Jörg Neugebauer
Prof. Dr. Dierk Raabe
Dr. Kai de Weldige
Ust.-Id.-Nr.: DE 11 93 58 514
Steuernummer: 105 5891 1000
Please consider that invitations and e-mails of our institute are
only valid if they end with …@mpie.de.
If you are not sure of the validity please contact r...@mpie.de
Bitte beachten Sie, dass Einladungen zu Veranstaltungen und E-Mails
aus unserem Haus nur mit der Endung …@mpie.de gültig sind.
In Zweifelsfällen wenden Sie sich bitte an r...@mpie.de
-------------------------------------------------
!
!
! Description: Uses the Newton method to solve a two-variable system.
!
!!/*T
! Concepts: SNES^basic uniprocessor example
! Processors: 1
!T*/
! =============================================================================
!
! Demonstrates use of MatShellSetContext() and MatShellGetContext()
!
! Contributed by: Samuel Lanthaler
!
MODULE solver_context
#include "petsc/finclude/petsc.h"
USE petscsys
USE petscmat
IMPLICIT NONE
TYPE :: MatCtx
PetscReal :: lambda,kappa
PetscReal :: h
END TYPE MatCtx
END MODULE solver_context
MODULE solver_context_interfaces
USE solver_context
IMPLICIT NONE
! ----------------------------------------------------
INTERFACE MatCreateShell
SUBROUTINE MatCreateShell(comm,mloc,nloc,m,n,ctx,mat,ierr)
USE solver_context
MPI_Comm :: comm
PetscInt :: mloc,nloc,m,n
!TYPE(MatCtx) :: ctx
Vec :: ctx
!PetscReal, dimension(2) :: ctx
Mat :: mat
PetscErrorCode :: ierr
END SUBROUTINE MatCreateShell
END INTERFACE MatCreateShell
! ----------------------------------------------------
! ----------------------------------------------------
INTERFACE MatShellSetContext
SUBROUTINE MatShellSetContext(mat,ctx,ierr)
USE solver_context
Mat :: mat
!TYPE(MatCtx) :: ctx
Vec :: ctx
!PetscReal :: ctx
PetscErrorCode :: ierr
END SUBROUTINE MatShellSetContext
END INTERFACE MatShellSetContext
! ----------------------------------------------------
! ----------------------------------------------------
INTERFACE MatShellGetContext
SUBROUTINE MatShellGetContext(mat,ctx,ierr)
USE solver_context
Mat :: mat
!TYPE(MatCtx), POINTER :: ctx
Vec, Pointer :: ctx
PetscErrorCode :: ierr
END SUBROUTINE MatShellGetContext
END INTERFACE MatShellGetContext
END MODULE solver_context_interfaces
! =============================================================================
program main
#include <petsc/finclude/petsc.h>
use petsc
use solver_context_interfaces
implicit none
! - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
! Variable declarations
! - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
!
! Variables:
! snes - nonlinear solver
! ksp - linear solver
! pc - preconditioner context
! ksp - Krylov subspace method context
! x, r - solution, residual vectors
! J - Jacobian matrix
! its - iterations for convergence
!
SNES snes
PC pc
KSP ksp
Vec x,r
Mat J
SNESLineSearch linesearch
PetscErrorCode ierr
PetscInt its,i2,i20
PetscMPIInt size,rank
PetscScalar pfive
PetscReal tol
PetscBool setls
#if defined(PETSC_USE_LOG)
PetscViewer viewer
#endif
double precision threshold,oldthreshold
! ====== Yi: Shell Mat ======
TYPE(MatCtx) :: ctxF
TYPE(MatCtx),POINTER :: ctxF_pt
Mat :: F
PetscInt :: n=128
! ===== Yi: record X as ctx for MyMult ====
Vec :: X_rec
Vec, Pointer :: tmp_get
! Note: Any user-defined Fortran routines (such as FormJacobian)
! MUST be declared as external.
external FormFunction, FormJacobian, MyLineSearch
external FormJacobianShell, MyMult ! ==== Yi ====
external converge_test_ksp ! ==== Yi ====
! - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
! Macro definitions
! - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
!
! Macros to make clearer the process of setting values in vectors and
! getting values from vectors. These vectors are used in the routines
! FormFunction() and FormJacobian().
! - The element lx_a(ib) is element ib in the vector x
!
#define lx_a(ib) lx_v(lx_i + (ib))
#define lf_a(ib) lf_v(lf_i + (ib))
!
! - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
! Beginning of program
! - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
call PetscInitialize(PETSC_NULL_CHARACTER,ierr)
if (ierr .ne. 0) then
print*,'Unable to initialize PETSc'
stop
endif
call PetscLogNestedBegin(ierr);CHKERRA(ierr)
threshold = 1.0
call PetscLogSetThreshold(threshold,oldthreshold,ierr)
! dummy test of logging a reduction
#if defined(PETSC_USE_LOG)
ierr = PetscAReduce()
#endif
call MPI_Comm_size(PETSC_COMM_WORLD,size,ierr)
call MPI_Comm_rank(PETSC_COMM_WORLD,rank,ierr)
!if (size .ne. 1) then
! SETERRA(PETSC_COMM_SELF,PETSC_ERR_WRONG_MPI_SIZE,'Uniprocessor example')
!endif
! ====== Yi: Shell Mat ======
!ctxF%lambda = 3.14d0
!CALL MatCreateShell(PETSC_COMM_WORLD,n,n,n,n,ctxF,F,ierr)
!CALL MatShellSetContext(F,ctxF,ierr)
!PRINT*,'ctxF%lambda = ',ctxF%lambda
!CALL MatShellGetContext(F,ctxF_pt,ierr)
!PRINT*,'ctxF_pt%lambda = ',ctxF_pt%lambda
!call MatDestroy(F,ierr)
i2 = 2
i20 = 20
! - - - - - - - - - -- - - - - - - - - - - - - - - - - - - - - - - - - -
! Create nonlinear solver context
! - - - - - - - - - -- - - - - - - - - - - - - - - - - - - - - - - - - -
call SNESCreate(PETSC_COMM_WORLD,snes,ierr)
! - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
! Create matrix and vector data structures; set corresponding routines
! - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
! Create vectors for solution and nonlinear function
call VecCreateSeq(PETSC_COMM_SELF,i2,x,ierr)
call VecDuplicate(x,r,ierr)
! Yi: X_rec
call VecDuplicate(x,X_rec,ierr)
! Set function evaluation routine and vector
call SNESSetFunction(snes,r,FormFunction,0,ierr)
! Create Jacobian matrix data structure
! call MatCreate(PETSC_COMM_SELF,J,ierr)
! call MatSetSizes(J,PETSC_DECIDE,PETSC_DECIDE,i2,i2,ierr)
! call MatSetFromOptions(J,ierr)
! call MatSetUp(J,ierr)
! Set Jacobian matrix data structure and Jacobian evaluation routine
! call SNESSetJacobian(snes,J,J,FormJacobian,0,ierr)
! ====== Yi: Shell Mat ======
CALL MatCreateShell(PETSC_COMM_WORLD,PETSC_DECIDE,PETSC_DECIDE,&
i2,i2,x,J,ierr)
call MatShellSetOperation(J,MATOP_MULT,MyMult,ierr)
call SNESSetJacobian(snes,J,J,FormJacobianShell,0,ierr)
call MatAssemblyBegin(J,MAT_FINAL_ASSEMBLY,ierr)
call MatAssemblyEnd(J,MAT_FINAL_ASSEMBLY,ierr)
call MatShellGetContext(J,tmp_get,ierr)
call VecView(tmp_get,PETSC_VIEWER_STDOUT_WORLD,ierr)
print*, 'get in main'
! - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
! Customize nonlinear solver; set runtime options
! - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
! Set linear solver defaults for this problem. By extracting the
! KSP, KSP, and PC contexts from the SNES context, we can then
! directly call any KSP, KSP, and PC routines to set various options.
call SNESGetKSP(snes,ksp,ierr)
call KSPGetPC(ksp,pc,ierr)
call PCSetType(pc,PCNONE,ierr)
tol = 1.e-4
call KSPSetTolerances(ksp,tol,PETSC_DEFAULT_REAL, &
& PETSC_DEFAULT_REAL,i20,ierr)
! call KSPSetConvergenceTest(ksp,converge_test_ksp,0,PETSC_NULL_FUNCTION,ierr)
! Set SNES/KSP/KSP/PC runtime options, e.g.,
! -snes_view -snes_monitor -ksp_type <ksp> -pc_type <pc>
! These options will override those specified above as long as
! SNESSetFromOptions() is called _after_ any other customization
! routines.
call SNESSetFromOptions(snes,ierr)
call PetscOptionsHasName(PETSC_NULL_OPTIONS,PETSC_NULL_CHARACTER, &
& '-setls',setls,ierr)
if (setls) then
call SNESGetLineSearch(snes, linesearch, ierr)
call SNESLineSearchSetType(linesearch, 'shell', ierr)
call SNESLineSearchShellSetUserFunc(linesearch, MyLineSearch, &
& 0, ierr)
endif
! - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
! Evaluate initial guess; then solve nonlinear system
! - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
! Note: The user should initialize the vector, x, with the initial guess
! for the nonlinear solver prior to calling SNESSolve(). In particular,
! to employ an initial guess of zero, the user should explicitly set
! this vector to zero by calling VecSet().
pfive = 0.5
call VecSet(x,pfive,ierr)
call SNESSolve(snes,PETSC_NULL_VEC,x,ierr)
! View solver converged reason; we could instead use the option -snes_converged_reason
call SNESConvergedReasonView(snes,PETSC_VIEWER_STDOUT_WORLD,ierr)
call SNESGetIterationNumber(snes,its,ierr);
if (rank .eq. 0) then
write(6,100) its
endif
100 format('Number of SNES iterations = ',i5)
! - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
! Free work space. All PETSc objects should be destroyed when they
! are no longer needed.
! - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
call VecDestroy(X_rec,ierr)
call VecDestroy(x,ierr)
call VecDestroy(r,ierr)
call MatDestroy(J,ierr)
call SNESDestroy(snes,ierr)
#if defined(PETSC_USE_LOG)
call PetscViewerASCIIOpen(PETSC_COMM_WORLD,'filename.xml',viewer,ierr)
call PetscViewerPushFormat(viewer,PETSC_VIEWER_ASCII_XML,ierr)
call PetscLogView(viewer,ierr)
call PetscViewerDestroy(viewer,ierr)
#endif
call PetscFinalize(ierr)
end
!
! ------------------------------------------------------------------------
!
! FormFunction - Evaluates nonlinear function, F(x).
!
! Input Parameters:
! snes - the SNES context
! x - input vector
! dummy - optional user-defined context (not used here)
!
! Output Parameter:
! f - function vector
!
subroutine FormFunction(snes,x,f,dummy,ierr)
use petscsnes
implicit none
SNES snes
Vec x,f
PetscErrorCode ierr
integer dummy(*)
! Declarations for use with local arrays
PetscScalar lx_v(2),lf_v(2)
PetscOffset lx_i,lf_i
! Get pointers to vector data.
! - For default PETSc vectors, VecGetArray() returns a pointer to
! the data array. Otherwise, the routine is implementation dependent.
! - You MUST call VecRestoreArray() when you no longer need access to
! the array.
! - Note that the Fortran interface to VecGetArray() differs from the
! C version. See the Fortran chapter of the users manual for details.
! print*, '( in rhs )'
call VecGetArrayRead(x,lx_v,lx_i,ierr)
call VecGetArray(f,lf_v,lf_i,ierr)
! Compute function
lf_a(1) = lx_a(1)*lx_a(1) &
& + lx_a(1)*lx_a(2) - 3.0
lf_a(2) = lx_a(1)*lx_a(2) &
& + lx_a(2)*lx_a(2) - 6.0
! Restore vectors
call VecRestoreArrayRead(x,lx_v,lx_i,ierr)
call VecRestoreArray(f,lf_v,lf_i,ierr)
! print*, '( leave rhs )'
return
end
! ---------------------------------------------------------------------
!
! FormJacobian - Evaluates Jacobian matrix.
!
! Input Parameters:
! snes - the SNES context
! x - input vector
! dummy - optional user-defined context (not used here)
!
! Output Parameters:
! A - Jacobian matrix
! B - optionally different preconditioning matrix
!
subroutine FormJacobian(snes,X,jac,B,dummy,ierr)
use petscsnes
implicit none
SNES snes
Vec X
Mat jac,B
PetscScalar A(4)
PetscErrorCode ierr
PetscInt idx(2),i2
integer dummy(*)
! Declarations for use with local arrays
PetscScalar lx_v(2)
PetscOffset lx_i
! Get pointer to vector data
i2 = 2
call VecGetArrayRead(x,lx_v,lx_i,ierr)
! Compute Jacobian entries and insert into matrix.
! - Since this is such a small problem, we set all entries for
! the matrix at once.
! - Note that MatSetValues() uses 0-based row and column numbers
! in Fortran as well as in C (as set here in the array idx).
idx(1) = 0
idx(2) = 1
A(1) = 2.0*lx_a(1) + lx_a(2)
A(2) = lx_a(1)
A(3) = lx_a(2)
A(4) = lx_a(1) + 2.0*lx_a(2)
call MatSetValues(B,i2,idx,i2,idx,A,INSERT_VALUES,ierr)
! Restore vector
call VecRestoreArrayRead(x,lx_v,lx_i,ierr)
! Assemble matrix
call MatAssemblyBegin(B,MAT_FINAL_ASSEMBLY,ierr)
call MatAssemblyEnd(B,MAT_FINAL_ASSEMBLY,ierr)
if (B .ne. jac) then
call MatAssemblyBegin(jac,MAT_FINAL_ASSEMBLY,ierr)
call MatAssemblyEnd(jac,MAT_FINAL_ASSEMBLY,ierr)
endif
return
end
subroutine MyLineSearch(linesearch, lctx, ierr)
use petscsnes
implicit none
SNESLineSearch linesearch
SNES snes
integer lctx
Vec x, f,g, y, w
PetscReal ynorm,gnorm,xnorm
PetscErrorCode ierr
PetscScalar mone
mone = -1.0
call SNESLineSearchGetSNES(linesearch, snes, ierr)
call SNESLineSearchGetVecs(linesearch, x, f, y, w, g, ierr)
call VecNorm(y,NORM_2,ynorm,ierr)
call VecAXPY(x,mone,y,ierr)
call SNESComputeFunction(snes,x,f,ierr)
call VecNorm(f,NORM_2,gnorm,ierr)
call VecNorm(x,NORM_2,xnorm,ierr)
call VecNorm(y,NORM_2,ynorm,ierr)
call SNESLineSearchSetNorms(linesearch, xnorm, gnorm, ynorm, &
& ierr)
return
end
! ======== Yi: shell mat ========
subroutine FormJacobianShell(snes,X,jac,B,dummy,ierr)
use petscsnes
use petscmat
use solver_context_interfaces
implicit none
SNES snes
Vec X
Vec, Pointer :: X_get
Mat jac,B
PetscErrorCode ierr
integer dummy(*)
!call MatShellGetContext(jac,X_get,ierr)
!call VecView(X_get,PETSC_VIEWER_STDOUT_SELF,ierr)
!print*, 'above should be same as main'
call MatShellSetContext(jac,X,ierr)
print*, 'ctx changed'
call MatShellGetContext(jac,X_get,ierr)
call VecView(X_get,PETSC_VIEWER_STDOUT_WORLD,ierr)
call MatAssemblyBegin(jac,MAT_FINAL_ASSEMBLY,ierr)
call MatAssemblyEnd(jac,MAT_FINAL_ASSEMBLY,ierr)
end subroutine FormJacobianShell
! Yi Note:
! customized action is J(X)dX
! so J should know the current X (in rhs or formFunction)
! dX is the sought direction (solved by ksp)
! X should be recorded by ctx of shell matrix
subroutine MyMult(J,dX,F,ierr)
use petscsnes
use solver_context_interfaces
implicit none
SNES snes
Vec dX
Mat B
PetscScalar A(4)
PetscErrorCode ierr
PetscInt idx(2),i2
Vec F
Mat J
! Declarations for use with local arrays
PetscScalar lx_v(2)
PetscOffset lx_i
Vec, Pointer :: x
! Get pointer to vector data
! print*, '=== start mymult ==='
i2 = 2
call MatView(J,PETSC_VIEWER_STDOUT_WORLD,ierr)
print*, 'ready to get ctx?'
call MatShellGetContext(J,x,ierr)
print*, 'done get ctx'
call VecView(x,PETSC_VIEWER_STDOUT_WORLD,ierr)
call VecGetArrayRead(x,lx_v,lx_i,ierr)
! Yi: create tmp B
! call MatCreateDense(PETSC_COMM_WORLD,i2,i2,i2,i2,B,ierr)
call MatCreate(PETSC_COMM_SELF,b,ierr)
call MatSetSizes(B,PETSC_DECIDE,PETSC_DECIDE,i2,i2,ierr)
call MatSetUp(B,ierr)
! Compute Jacobian entries and insert into matrix.
! - Since this is such a small problem, we set all entries for
! the matrix at once.
! - Note that MatSetValues() uses 0-based row and column numbers
! in Fortran as well as in C (as set here in the array idx).
idx(1) = 0
idx(2) = 1
A(1) = 2.0*lx_a(1) + lx_a(2)
A(2) = lx_a(1)
A(3) = lx_a(2)
A(4) = lx_a(1) + 2.0*lx_a(2)
call MatSetValues(B,i2,idx,i2,idx,A,INSERT_VALUES,ierr)
! Restore vector
call VecRestoreArrayRead(x,lx_v,lx_i,ierr)
! Assemble matrix
call MatAssemblyBegin(B,MAT_FINAL_ASSEMBLY,ierr)
call MatAssemblyEnd(B,MAT_FINAL_ASSEMBLY,ierr)
call MatMult(B,dX,F,ierr)
call MatDestroy(B,ierr)
! print*, '=== done mymult ==='
return
end subroutine MyMult
subroutine converge_test_ksp(ksp, it, rnorm, reason, ctx, ierr)
use petsc
KSP :: ksp
PetscInt :: it
PetscReal :: rnorm
KSPConvergedReason :: reason
type(PetscObject), pointer :: ctx
PetscErrorCode :: ierr
!print *, '!!!!!!!!!!!!!!!!!!!!!!my ksp test'
call KSPGetResidualNorm(ksp, rnorm, ierr)
print *, rnorm
if ( rnorm < 1.0e-5 ) then
reason = 1
endif
end subroutine converge_test_ksp
!/*TEST
!
! test:
! args: -ksp_gmres_cgs_refinement_type refine_always -snes_monitor_short
! requires: !single
!
!TEST*/