static char help[] = "Tests MatMPI{AIJ,BAIJ,SBAIJ}SetPreallocationCSR\n\n"; #include int main(int argc,char **argv) { PetscErrorCode ierr; PetscInt ia[3]={0,2,4}; PetscInt ja[4]={0,1,0,1}; PetscScalar c[16]={0,1,2,3,4,5,6,7,8,9,10,11,12,13,14,15}; PetscInt ia2[5]={0,4,8,12,16}; PetscInt ja2[16]={0,1,2,3,0,1,2,3,0,1,2,3,0,1,2,3}; PetscScalar c2[16]={0,1,4,5,2,3,6,7,8,9,12,13,10,11,14,15}; PetscMPIInt size,rank; Mat ssbaij; PetscBool rect = PETSC_FALSE; ierr = PetscInitialize(&argc,&argv,NULL,help);if (ierr) return ierr; ierr = MPI_Comm_size(PETSC_COMM_WORLD,&size);CHKERRMPI(ierr); ierr = MPI_Comm_rank(PETSC_COMM_WORLD,&rank);CHKERRMPI(ierr); if (size < 2) SETERRQ(PETSC_COMM_WORLD,PETSC_ERR_SUP,"This is an example with more then one processors"); if (rank) { PetscInt i; for (i = 0; i < 3; i++) ia[i] = 0; for (i = 0; i < 5; i++) ia2[i] = 0; } ierr = PetscOptionsGetBool(NULL,NULL,"-rect",&rect,NULL);CHKERRQ(ierr); ierr = MatCreate(PETSC_COMM_WORLD,&ssbaij);CHKERRQ(ierr); ierr = MatSetBlockSize(ssbaij,2);CHKERRQ(ierr); if (rect) { ierr = MatSetType(ssbaij,MATMPIBAIJ);CHKERRQ(ierr); ierr = MatSetSizes(ssbaij,4,6,PETSC_DECIDE,PETSC_DECIDE);CHKERRQ(ierr); } else { ierr = MatSetType(ssbaij,MATMPISBAIJ);CHKERRQ(ierr); ierr = MatSetSizes(ssbaij,4,4,PETSC_DECIDE,PETSC_DECIDE);CHKERRQ(ierr); } ierr = MatSetFromOptions(ssbaij);CHKERRQ(ierr); ierr = MatMPIAIJSetPreallocationCSR(ssbaij,ia2,ja2,c2);CHKERRQ(ierr); ierr = MatMPIBAIJSetPreallocationCSR(ssbaij,2,ia,ja,c);CHKERRQ(ierr); ierr = MatMPISBAIJSetPreallocationCSR(ssbaij,2,ia,ja,c);CHKERRQ(ierr); ierr = MatViewFromOptions(ssbaij,NULL,"-view");CHKERRQ(ierr); ierr = MatDestroy(&ssbaij);CHKERRQ(ierr); ierr = PetscFinalize(); return ierr; } /*TEST test: filter: grep -v type | sed -e "s/\.//g" suffix: aijbaij_csr nsize: 2 args: -mat_type {{aij baij}} -view -rect {{0 1}} test: filter: sed -e "s/\.//g" suffix: sbaij_csr nsize: 2 args: -mat_type sbaij -view -rect {{0 1}} TEST*/