1 2 /* tests MatSeqSBAIJSetPreallocationCSR() and MatMPISBAIJSetPreallocationCSR() */ 3 4 #include <petsc.h> 5 6 int main(int argc, char **args) 7 { 8 PetscInt ia[3] = { 0, 2, 4}; 9 PetscInt ja[4] = { 0, 1, 0, 1}; 10 PetscScalar c[16] = { 0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15 }; 11 Mat ssbaij; 12 Mat msbaij; 13 14 CHKERRQ(PetscInitialize(&argc,&args,(char*)0,(char*)0)); 15 16 CHKERRQ(MatCreate(PETSC_COMM_SELF, &ssbaij)); 17 CHKERRQ(MatCreate(PETSC_COMM_SELF, &msbaij)); 18 CHKERRQ(MatSetType(ssbaij, MATSEQSBAIJ)); 19 CHKERRQ(MatSetType(msbaij, MATMPISBAIJ)); 20 CHKERRQ(MatSetBlockSize(ssbaij, 2)); 21 CHKERRQ(MatSetSizes(ssbaij, 4, 4, 4, 4)); 22 CHKERRQ(MatSetBlockSize(msbaij, 2)); 23 CHKERRQ(MatSetSizes(msbaij, 4, 4, 4, 4)); 24 CHKERRQ(MatSeqSBAIJSetPreallocationCSR(ssbaij, 2, ia, ja, c)); 25 CHKERRQ(MatMPISBAIJSetPreallocationCSR(msbaij, 2, ia, ja, c)); 26 CHKERRQ(MatView(ssbaij, PETSC_VIEWER_STDOUT_(PETSC_COMM_SELF))); 27 CHKERRQ(MatView(msbaij, PETSC_VIEWER_STDOUT_(PETSC_COMM_SELF))); 28 CHKERRQ(MatDestroy(&ssbaij)); 29 CHKERRQ(MatDestroy(&msbaij)); 30 CHKERRQ(PetscFinalize()); 31 return 0; 32 } 33 34 /*TEST 35 36 test: 37 filter: sed "s?\.??g" 38 39 TEST*/ 40