中科院超级计算中心.pptVIP

* 　　　　 /*$Id: ex2.c,v 1.94 2001/08/07 21:30:54 bsmith Exp $*/ /* 运行方式: mpirun -np procs ex2 [-help] [all PETSc options] */ static char help[] = Solves a linear system in parallel with SLES.\n\ Input parameters include: \n\ -random_exact_sol : use a random exact solution vector\n\ -view_exact_sol : write exact solution vector to stdout\n\ -m mesh_x : number of mesh points in x-direction\n\ -n mesh_n : number of mesh points in y-direction\n\n; /* Include petscsles.h so that we can use SLES solvers. Note that this file automatically includes: petsc.h - base PETSc routines petscvec.h - vectors petscsys.h - system routines petscmat.h - matrices petscis.h - index sets petscksp.h - Krylov subspace methods petscviewer.h - viewers petscpc.h - preconditioners */ #include petscsles.h #undef __FUNCT__ #define __FUNCT__ main int main(int argc,char **args) { Vec x,b,u; /* 近似解，右端向量和分析解*/ Mat A; /* 线性算子/矩阵*/ SLES sles; /* 线性解法器*/ PetscRandom rctx; /* 随机数发生器环境*/ PetscReal norm; /* 解误差的范数*/ int i, j, I, J, Istart, Iend, ierr, m = 8, n = 7, its; PetscTruth flg; PetscScalar v, one = 1.0, neg_one = -1.0; KSP ksp; PetscInitialize(argc, args, (char *)0, help); ierr =PetscOptionsGetInt(PETSC_NULL,-m,m,PETSC_NULL); CHKERRQ(ierr); ierr =PetscOptionsGetInt(PETSC_NULL,-n,n,PETSC_NULL); CHKERRQ(ierr); * 26 /* - - - - - - - - - - - - - - - -- - - - - - - - - - -- - - - - - - - - - -- - - - - - - - - - -- - - - - - - - - - -- - 计算矩阵算子和右端向量 - - - - - - - - - - - - - - - - - - -- - - - - - - - - - - - - - - - - - - -- - - - - - - - - - - - - - - - - - - -- - - - - - - - - - - - - - - - - -- */ /*创建矩阵对象*/ ierr=MatCreate(PETSC_COMM_WORLD,PETSC_DECIDE,PETSC_DECIDE,m*n,m*n,A); CHKERRQ(ierr); ierr=MatSetFromOptions(A); CHKERRQ(ierr); /*获得局部划分的上下界*/ ierr = MatGetOwnershipRange(A,Istart,Iend); CHKERRQ(ierr); /*给矩阵的每个元素赋值*/ for (I=Istart; IIend; I++) { v = -1.0; i = I/n; j = I - i*n; if (i0) {J = I - n; ierr = MatSetValues(A,1,I,1,J,v,INSERT_VALUES);CHKERRQ(ierr);} if (im-1) {J = I + n; ierr = MatSetValues