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