xref: /petsc/src/binding/petsc4py/demo/legacy/kspsolve/petsc-ksp.py (revision 5a48edb989d3ea10d6aff6c0e26d581c18691deb)
1*55a74a43SLisandro Dalcin# create linear solver
2*55a74a43SLisandro Dalcinksp = PETSc.KSP()
3*55a74a43SLisandro Dalcinksp.create(PETSc.COMM_WORLD)
4*55a74a43SLisandro Dalcin# use conjugate gradients
5*55a74a43SLisandro Dalcinksp.setType('cg')
6*55a74a43SLisandro Dalcin# and incomplete Cholesky
7*55a74a43SLisandro Dalcinksp.getPC().setType('icc')
8*55a74a43SLisandro Dalcin# obtain sol & rhs vectors
9*55a74a43SLisandro Dalcinx, b = A.createVecs()
10*55a74a43SLisandro Dalcinx.set(0)
11*55a74a43SLisandro Dalcinb.set(1)
12*55a74a43SLisandro Dalcin# and next solve
13*55a74a43SLisandro Dalcinksp.setOperators(A)
14*55a74a43SLisandro Dalcinksp.setFromOptions()
15*55a74a43SLisandro Dalcinksp.solve(b, x)
16