1 2 static char help[] = "Tests MatMPIBAIJ format in sequential run \n"; 3 4 #include <petscmat.h> 5 int main(int argc,char **args) 6 { 7 Mat A,B; 8 PetscInt i,rstart,rend; 9 PetscMPIInt rank,size; 10 PetscErrorCode ierr; 11 PetscScalar v; 12 13 ierr = PetscInitialize(&argc,&args,(char*)0,help);if (ierr) return ierr; 14 CHKERRMPI(MPI_Comm_rank(PETSC_COMM_WORLD,&rank)); 15 CHKERRMPI(MPI_Comm_size(PETSC_COMM_WORLD,&size)); 16 17 /* Create a MPIBAIJ matrix */ 18 CHKERRQ(MatCreate(PETSC_COMM_WORLD,&A)); 19 CHKERRQ(MatSetSizes(A,PETSC_DECIDE,PETSC_DECIDE,32,32)); 20 CHKERRQ(MatSetType(A,MATMPIBAIJ)); 21 CHKERRQ(MatSeqBAIJSetPreallocation(A,2,2,NULL)); 22 CHKERRQ(MatMPIBAIJSetPreallocation(A,2,2,NULL,2,NULL)); 23 24 v = 1.0; 25 CHKERRQ(MatGetOwnershipRange(A,&rstart,&rend)); 26 for (i=rstart; i<rend; i++) { 27 CHKERRQ(MatSetValues(A,1,&i,1,&i,&v,INSERT_VALUES)); 28 } 29 CHKERRQ(MatAssemblyBegin(A,MAT_FINAL_ASSEMBLY)); 30 CHKERRQ(MatAssemblyEnd(A,MAT_FINAL_ASSEMBLY)); 31 32 /* Convert A to AIJ format */ 33 CHKERRQ(MatConvert(A,MATAIJ,MAT_INITIAL_MATRIX,&B)); 34 35 CHKERRQ(MatDestroy(&A)); 36 CHKERRQ(MatDestroy(&B)); 37 ierr = PetscFinalize(); 38 return ierr; 39 } 40 41 /*TEST 42 43 test: 44 output_file: output/ex160.out 45 46 TEST*/ 47