1c4762a1bSJed Brown static char help[] = "Tests using MatShift() to create a constant diagonal matrix\n\n"; 2c4762a1bSJed Brown 3c4762a1bSJed Brown #include <petscmat.h> 4c4762a1bSJed Brown 5c4762a1bSJed Brown int main(int argc,char **argv) 6c4762a1bSJed Brown { 7c4762a1bSJed Brown Mat A,F; 8c4762a1bSJed Brown MatFactorInfo info; 9c4762a1bSJed Brown PetscErrorCode ierr; 10c4762a1bSJed Brown PetscInt m = 10; 11c4762a1bSJed Brown IS perm; 12c4762a1bSJed Brown PetscMPIInt size; 13c4762a1bSJed Brown PetscBool issbaij; 14c4762a1bSJed Brown 15c4762a1bSJed Brown ierr = PetscInitialize(&argc,&argv,(char*) 0,help);if (ierr) return ierr; 16ffc4695bSBarry Smith ierr = MPI_Comm_size(PETSC_COMM_WORLD,&size);CHKERRMPI(ierr); 17c4762a1bSJed Brown 18c4762a1bSJed Brown ierr = MatCreate(PETSC_COMM_WORLD,&A);CHKERRQ(ierr); 19c4762a1bSJed Brown ierr = MatSetSizes(A,PETSC_DECIDE,PETSC_DECIDE,m,m);CHKERRQ(ierr); 20c4762a1bSJed Brown ierr = MatSetFromOptions(A);CHKERRQ(ierr); 21c4762a1bSJed Brown ierr = MatSetUp(A);CHKERRQ(ierr); 22c4762a1bSJed Brown ierr = MatAssemblyBegin(A,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 23c4762a1bSJed Brown ierr = MatAssemblyEnd(A,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 24c4762a1bSJed Brown 25c4762a1bSJed Brown ierr = MatShift(A,1.0);CHKERRQ(ierr); 26c4762a1bSJed Brown 27c4762a1bSJed Brown ierr = PetscObjectTypeCompare((PetscObject)A,MATSEQSBAIJ,&issbaij);CHKERRQ(ierr); 28c4762a1bSJed Brown if (size == 1 && !issbaij) { 29c4762a1bSJed Brown ierr = MatGetFactor(A,MATSOLVERPETSC,MAT_FACTOR_LU,&F);CHKERRQ(ierr); 30c4762a1bSJed Brown ierr = MatFactorInfoInitialize(&info);CHKERRQ(ierr); 31c4762a1bSJed Brown ierr = ISCreateStride(PETSC_COMM_SELF,m,0,1,&perm);CHKERRQ(ierr); 32c4762a1bSJed Brown ierr = MatLUFactorSymbolic(F,A,perm,perm,&info);CHKERRQ(ierr); 33c4762a1bSJed Brown ierr = MatLUFactorNumeric(F,A,&info);CHKERRQ(ierr); 34c4762a1bSJed Brown ierr = MatDestroy(&F);CHKERRQ(ierr); 35c4762a1bSJed Brown ierr = ISDestroy(&perm);CHKERRQ(ierr); 36c4762a1bSJed Brown } 37c4762a1bSJed Brown ierr = MatDestroy(&A);CHKERRQ(ierr); 38c4762a1bSJed Brown ierr = PetscFinalize(); 39c4762a1bSJed Brown return ierr; 40c4762a1bSJed Brown } 41c4762a1bSJed Brown 42c4762a1bSJed Brown /*TEST 43c4762a1bSJed Brown 44c4762a1bSJed Brown test: 45*dfd57a17SPierre Jolivet requires: defined(PETSC_USE_INFO) 46c4762a1bSJed Brown args: -info 47c4762a1bSJed Brown filter: grep malloc | sort -b 48c4762a1bSJed Brown 49c4762a1bSJed Brown test: 50c4762a1bSJed Brown suffix: 2 51c4762a1bSJed Brown nsize: 2 52*dfd57a17SPierre Jolivet requires: defined(PETSC_USE_INFO) 53c4762a1bSJed Brown args: -info ex182info 54c4762a1bSJed Brown filter: grep -h malloc "ex182info.0" | sort -b 55c4762a1bSJed Brown 56c4762a1bSJed Brown test: 57c4762a1bSJed Brown suffix: 3 58*dfd57a17SPierre Jolivet requires: defined(PETSC_USE_INFO) 59c4762a1bSJed Brown args: -info -mat_type baij 60c4762a1bSJed Brown filter: grep malloc | sort -b 61c4762a1bSJed Brown 62c4762a1bSJed Brown test: 63c4762a1bSJed Brown suffix: 4 64c4762a1bSJed Brown nsize: 2 65*dfd57a17SPierre Jolivet requires: defined(PETSC_USE_INFO) 66c4762a1bSJed Brown args: -info ex182info -mat_type baij 67c4762a1bSJed Brown filter: grep -h malloc "ex182info.1" | sort -b 68c4762a1bSJed Brown 69c4762a1bSJed Brown test: 70c4762a1bSJed Brown suffix: 5 71*dfd57a17SPierre Jolivet requires: defined(PETSC_USE_INFO) 72c4762a1bSJed Brown args: -info -mat_type sbaij 73c4762a1bSJed Brown filter: grep malloc | sort -b 74c4762a1bSJed Brown 75c4762a1bSJed Brown test: 76c4762a1bSJed Brown suffix: 6 77c4762a1bSJed Brown nsize: 2 78*dfd57a17SPierre Jolivet requires: defined(PETSC_USE_INFO) 79c4762a1bSJed Brown args: -info ex182info -mat_type sbaij 80c4762a1bSJed Brown filter: grep -h malloc "ex182info.0" | sort -b 81c4762a1bSJed Brown 82c20d7725SJed Brown test: 83c20d7725SJed Brown suffix: 7 84c20d7725SJed Brown nsize: 1 85*dfd57a17SPierre Jolivet requires: defined(PETSC_USE_INFO) 86c20d7725SJed Brown args: -info ex182info 87c20d7725SJed Brown filter: grep -h malloc "ex182info.0" | grep -v Running | sort -b 88c20d7725SJed Brown 89c20d7725SJed Brown test: 90c20d7725SJed Brown suffix: 8 91c20d7725SJed Brown nsize: 2 92*dfd57a17SPierre Jolivet requires: defined(PETSC_USE_INFO) 93c20d7725SJed Brown args: -info ex182info:mat 94c20d7725SJed Brown filter: grep -h malloc "ex182info.1" | sort -b 95c20d7725SJed Brown 96c20d7725SJed Brown test: 97c20d7725SJed Brown suffix: 9 98c20d7725SJed Brown nsize: 1 99*dfd57a17SPierre Jolivet requires: defined(PETSC_USE_INFO) 100c20d7725SJed Brown args: -info ex182info:sys 101d6837840SMatthew G. Knepley filter: grep -h -ve Running -ve MPI_Comm -ve Initialize -ve communicator -ve HostName -ve PetscDetermineInitialFPTrap -ve libpetscbamg "ex182info.0" | sort -b 102c20d7725SJed Brown 103c20d7725SJed Brown test: 104c20d7725SJed Brown suffix: 10 105c20d7725SJed Brown nsize: 1 106*dfd57a17SPierre Jolivet requires: defined(PETSC_USE_INFO) 107c20d7725SJed Brown args: -info :~sys 108c20d7725SJed Brown filter: grep -h malloc | sort -b 109c20d7725SJed Brown 110c20d7725SJed Brown test: 111c20d7725SJed Brown suffix: 11 112c20d7725SJed Brown nsize: 2 113*dfd57a17SPierre Jolivet requires: defined(PETSC_USE_INFO) 114c20d7725SJed Brown args: -info :~sys,mat 115c20d7725SJed Brown filter: sort -b 116c20d7725SJed Brown 117c20d7725SJed Brown test: 118c20d7725SJed Brown suffix: 12 119c20d7725SJed Brown nsize: 2 120*dfd57a17SPierre Jolivet requires: defined(PETSC_USE_INFO) 121c20d7725SJed Brown args: -info ex182info:sys,mat 122d6837840SMatthew G. Knepley filter: grep -h -ve Running -ve MPI_Comm -ve Initialize -ve communicator -ve HostName -ve PetscDetermineInitialFPTrap -ve libpetscbamg "ex182info.1" | sort -b 123c20d7725SJed Brown 124c20d7725SJed Brown test: 125c20d7725SJed Brown suffix: 13 126c20d7725SJed Brown nsize: 2 127*dfd57a17SPierre Jolivet requires: defined(PETSC_USE_INFO) 128c20d7725SJed Brown args: -info ex182info:mat:~self 129c20d7725SJed Brown filter: grep -h "ex182info.1" | sort -b 130c20d7725SJed Brown 131c20d7725SJed Brown test: 132c20d7725SJed Brown suffix: 14 133c20d7725SJed Brown nsize: 2 134*dfd57a17SPierre Jolivet requires: defined(PETSC_USE_INFO) 135c20d7725SJed Brown args: -info ex182info::~self 136aba4c478SBarry Smith filter: grep -h -ve Running -ve MPI_Comm -ve Initialize -ve communicator -ve HostName -ve PetscDetermineInitialFPTrap "ex182info.1" | sort -b 137c20d7725SJed Brown 138c20d7725SJed Brown test: 139c20d7725SJed Brown suffix: 15 140c20d7725SJed Brown nsize: 2 141*dfd57a17SPierre Jolivet requires: defined(PETSC_USE_INFO) 142c20d7725SJed Brown args: -info ex182info::self 143d6837840SMatthew G. Knepley filter: grep -h -ve Running -ve MPI_Comm -ve Initialize -ve communicator -ve HostName -ve PetscDetermineInitialFPTrap -ve libpetscbamg "ex182info.1" | sort -b 144c20d7725SJed Brown 145c4762a1bSJed Brown TEST*/ 146