1 #if defined(PETSC_HAVE_LIBMKL_INTEL_ILP64) 2 #define MKL_ILP64 3 #endif 4 5 #include <../src/mat/impls/aij/seq/aij.h> /*I "petscmat.h" I*/ 6 #include <../src/mat/impls/dense/seq/dense.h> 7 #include <petscblaslapack.h> 8 9 #include <stdio.h> 10 #include <stdlib.h> 11 #include <math.h> 12 #include <mkl.h> 13 14 /* 15 * Possible mkl_pardiso phases that controls the execution of the solver. 16 * For more information check mkl_pardiso manual. 17 */ 18 #define JOB_ANALYSIS 11 19 #define JOB_ANALYSIS_NUMERICAL_FACTORIZATION 12 20 #define JOB_ANALYSIS_NUMERICAL_FACTORIZATION_SOLVE_ITERATIVE_REFINEMENT 13 21 #define JOB_NUMERICAL_FACTORIZATION 22 22 #define JOB_NUMERICAL_FACTORIZATION_SOLVE_ITERATIVE_REFINEMENT 23 23 #define JOB_SOLVE_ITERATIVE_REFINEMENT 33 24 #define JOB_SOLVE_FORWARD_SUBSTITUTION 331 25 #define JOB_SOLVE_DIAGONAL_SUBSTITUTION 332 26 #define JOB_SOLVE_BACKWARD_SUBSTITUTION 333 27 #define JOB_RELEASE_OF_LU_MEMORY 0 28 #define JOB_RELEASE_OF_ALL_MEMORY -1 29 30 #define IPARM_SIZE 64 31 32 #if defined(PETSC_USE_64BIT_INDICES) 33 #if defined(PETSC_HAVE_LIBMKL_INTEL_ILP64) 34 /* sizeof(MKL_INT) == sizeof(long long int) if ilp64*/ 35 #define INT_TYPE long long int 36 #define MKL_PARDISO pardiso 37 #define MKL_PARDISO_INIT pardisoinit 38 #else 39 #define INT_TYPE long long int 40 #define MKL_PARDISO pardiso_64 41 #define MKL_PARDISO_INIT pardiso_64init 42 #endif 43 #else 44 #define INT_TYPE int 45 #define MKL_PARDISO pardiso 46 #define MKL_PARDISO_INIT pardisoinit 47 #endif 48 49 50 /* 51 * Internal data structure. 52 * For more information check mkl_pardiso manual. 53 */ 54 typedef struct { 55 56 /* Configuration vector*/ 57 INT_TYPE iparm[IPARM_SIZE]; 58 59 /* 60 * Internal mkl_pardiso memory location. 61 * After the first call to mkl_pardiso do not modify pt, as that could cause a serious memory leak. 62 */ 63 void *pt[IPARM_SIZE]; 64 65 /* Basic mkl_pardiso info*/ 66 INT_TYPE phase, maxfct, mnum, mtype, n, nrhs, msglvl, err; 67 68 /* Matrix structure*/ 69 void *a; 70 INT_TYPE *ia, *ja; 71 72 /* Number of non-zero elements*/ 73 INT_TYPE nz; 74 75 /* Row permutaton vector*/ 76 INT_TYPE *perm; 77 78 /* Define if matrix preserves sparse structure.*/ 79 MatStructure matstruc; 80 81 /* Schur complement */ 82 PetscScalar *schur; 83 PetscInt schur_size; 84 PetscInt *schur_idxs; 85 PetscScalar *schur_work; 86 PetscBLASInt schur_work_size; 87 PetscInt schur_solver_type; 88 PetscInt *schur_pivots; 89 PetscBool schur_factored; 90 PetscBool schur_inverted; 91 92 /* True if mkl_pardiso function have been used.*/ 93 PetscBool CleanUp; 94 } Mat_MKL_PARDISO; 95 96 97 void pardiso_64init(void *pt, INT_TYPE *mtype, INT_TYPE iparm []) 98 { 99 int iparm_copy[IPARM_SIZE], mtype_copy, i; 100 101 mtype_copy = *mtype; 102 pardisoinit(pt, &mtype_copy, iparm_copy); 103 for(i = 0; i < IPARM_SIZE; i++){ 104 iparm[i] = iparm_copy[i]; 105 } 106 } 107 108 #undef __FUNCT__ 109 #define __FUNCT__ "MatMKLPardisoFactorSchur_Private" 110 static PetscErrorCode MatMKLPardisoFactorSchur_Private(Mat_MKL_PARDISO* mpardiso) 111 { 112 PetscBLASInt B_N,B_ierr; 113 PetscErrorCode ierr; 114 115 PetscFunctionBegin; 116 if (mpardiso->schur_factored) { 117 PetscFunctionReturn(0); 118 } 119 ierr = PetscBLASIntCast(mpardiso->schur_size,&B_N);CHKERRQ(ierr); 120 switch (mpardiso->schur_solver_type) { 121 case 1: /* symmetric */ 122 if (!mpardiso->schur_pivots) { 123 ierr = PetscMalloc1(B_N,&mpardiso->schur_pivots);CHKERRQ(ierr); 124 } 125 ierr = PetscFPTrapPush(PETSC_FP_TRAP_OFF);CHKERRQ(ierr); 126 PetscStackCallBLAS("LAPACKsytrf",LAPACKsytrf_("L",&B_N,mpardiso->schur,&B_N,mpardiso->schur_pivots,mpardiso->schur_work,&mpardiso->schur_work_size,&B_ierr)); 127 ierr = PetscFPTrapPop();CHKERRQ(ierr); 128 if (B_ierr) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_LIB,"Error in SYTRF Lapack routine %d",(int)B_ierr); 129 break; 130 case 2: /* hermitian solver */ 131 ierr = PetscFPTrapPush(PETSC_FP_TRAP_OFF);CHKERRQ(ierr); 132 PetscStackCallBLAS("LAPACKpotrf",LAPACKpotrf_("L",&B_N,mpardiso->schur,&B_N,&B_ierr)); 133 ierr = PetscFPTrapPop();CHKERRQ(ierr); 134 if (B_ierr) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_LIB,"Error in POTRF Lapack routine %d",(int)B_ierr); 135 break; 136 default: /* general */ 137 if (!mpardiso->schur_pivots) { 138 ierr = PetscMalloc1(B_N,&mpardiso->schur_pivots);CHKERRQ(ierr); 139 } 140 ierr = PetscFPTrapPush(PETSC_FP_TRAP_OFF);CHKERRQ(ierr); 141 PetscStackCallBLAS("LAPACKgetrf",LAPACKgetrf_(&B_N,&B_N,mpardiso->schur,&B_N,mpardiso->schur_pivots,&B_ierr)); 142 ierr = PetscFPTrapPop();CHKERRQ(ierr); 143 if (B_ierr) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_LIB,"Error in GETRF Lapack routine %d",(int)B_ierr); 144 break; 145 } 146 mpardiso->schur_factored = PETSC_TRUE; 147 PetscFunctionReturn(0); 148 } 149 150 #undef __FUNCT__ 151 #define __FUNCT__ "MatMKLPardisoInvertSchur_Private" 152 static PetscErrorCode MatMKLPardisoInvertSchur_Private(Mat_MKL_PARDISO* mpardiso) 153 { 154 PetscBLASInt B_N,B_ierr; 155 PetscErrorCode ierr; 156 157 PetscFunctionBegin; 158 ierr = MatMKLPardisoFactorSchur_Private(mpardiso);CHKERRQ(ierr); 159 ierr = PetscBLASIntCast(mpardiso->schur_size,&B_N);CHKERRQ(ierr); 160 switch (mpardiso->schur_solver_type) { 161 case 1: /* symmetric */ 162 ierr = PetscFPTrapPush(PETSC_FP_TRAP_OFF);CHKERRQ(ierr); 163 PetscStackCallBLAS("LAPACKsytri",LAPACKsytri_("L",&B_N,mpardiso->schur,&B_N,mpardiso->schur_pivots,mpardiso->schur_work,&B_ierr)); 164 ierr = PetscFPTrapPop();CHKERRQ(ierr); 165 if (B_ierr) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_LIB,"Error in SYTRI Lapack routine %d",(int)B_ierr); 166 break; 167 case 2: /* hermitian solver */ 168 ierr = PetscFPTrapPush(PETSC_FP_TRAP_OFF);CHKERRQ(ierr); 169 PetscStackCallBLAS("LAPACKpotri",LAPACKpotri_("L",&B_N,mpardiso->schur,&B_N,&B_ierr)); 170 ierr = PetscFPTrapPop();CHKERRQ(ierr); 171 if (B_ierr) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_LIB,"Error in POTRI Lapack routine %d",(int)B_ierr); 172 break; 173 default: /* general */ 174 ierr = PetscFPTrapPush(PETSC_FP_TRAP_OFF);CHKERRQ(ierr); 175 PetscStackCallBLAS("LAPACKgetri",LAPACKgetri_(&B_N,mpardiso->schur,&B_N,mpardiso->schur_pivots,mpardiso->schur_work,&mpardiso->schur_work_size,&B_ierr)); 176 ierr = PetscFPTrapPop();CHKERRQ(ierr); 177 if (B_ierr) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_LIB,"Error in GETRI Lapack routine %d",(int)B_ierr); 178 break; 179 } 180 mpardiso->schur_inverted = PETSC_TRUE; 181 PetscFunctionReturn(0); 182 } 183 184 #undef __FUNCT__ 185 #define __FUNCT__ "MatMKLPardisoSolveSchur_Private" 186 static PetscErrorCode MatMKLPardisoSolveSchur_Private(Mat_MKL_PARDISO* mpardiso, PetscScalar *B, PetscScalar *X) 187 { 188 PetscScalar one=1.,zero=0.,*schur_rhs,*schur_sol; 189 PetscBLASInt B_N,B_Nrhs,B_ierr; 190 char type[2]; 191 PetscErrorCode ierr; 192 193 PetscFunctionBegin; 194 ierr = MatMKLPardisoFactorSchur_Private(mpardiso);CHKERRQ(ierr); 195 ierr = PetscBLASIntCast(mpardiso->schur_size,&B_N);CHKERRQ(ierr); 196 ierr = PetscBLASIntCast(mpardiso->nrhs,&B_Nrhs);CHKERRQ(ierr); 197 if (X == B && mpardiso->schur_inverted) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"X and B cannot point to the same address"); 198 if (X != B) { /* using LAPACK *TRS subroutines */ 199 ierr = PetscMemcpy(X,B,B_N*B_Nrhs*sizeof(PetscScalar));CHKERRQ(ierr); 200 } 201 schur_rhs = B; 202 schur_sol = X; 203 switch (mpardiso->schur_solver_type) { 204 case 1: /* symmetric solver */ 205 if (mpardiso->schur_inverted) { 206 PetscStackCallBLAS("BLASsymm",BLASsymm_("L","L",&B_N,&B_Nrhs,&one,mpardiso->schur,&B_N,schur_rhs,&B_N,&zero,schur_sol,&B_N)); 207 } else { 208 ierr = PetscFPTrapPush(PETSC_FP_TRAP_OFF);CHKERRQ(ierr); 209 PetscStackCallBLAS("LAPACKsytrs",LAPACKsytrs_("L",&B_N,&B_Nrhs,mpardiso->schur,&B_N,mpardiso->schur_pivots,schur_sol,&B_N,&B_ierr)); 210 ierr = PetscFPTrapPop();CHKERRQ(ierr); 211 if (B_ierr) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_LIB,"Error in SYTRS Lapack routine %d",(int)B_ierr); 212 } 213 break; 214 case 2: /* hermitian solver */ 215 if (mpardiso->schur_inverted) { /* BLAShemm should go here */ 216 PetscStackCallBLAS("BLASsymm",BLASsymm_("L","L",&B_N,&B_Nrhs,&one,mpardiso->schur,&B_N,schur_rhs,&B_N,&zero,schur_sol,&B_N)); 217 } else { 218 ierr = PetscFPTrapPush(PETSC_FP_TRAP_OFF);CHKERRQ(ierr); 219 PetscStackCallBLAS("LAPACKpotrs",LAPACKpotrs_("L",&B_N,&B_Nrhs,mpardiso->schur,&B_N,schur_sol,&B_N,&B_ierr)); 220 ierr = PetscFPTrapPop();CHKERRQ(ierr); 221 if (B_ierr) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_LIB,"Error in POTRS Lapack routine %d",(int)B_ierr); 222 } 223 break; 224 default: /* general */ 225 switch (mpardiso->iparm[12-1]) { 226 case 1: 227 sprintf(type,"C"); 228 break; 229 case 2: 230 sprintf(type,"T"); 231 break; 232 default: 233 sprintf(type,"N"); 234 break; 235 } 236 if (mpardiso->schur_inverted) { 237 PetscStackCallBLAS("BLASgemm",BLASgemm_(type,"N",&B_N,&B_Nrhs,&B_N,&one,mpardiso->schur,&B_N,schur_rhs,&B_N,&zero,schur_sol,&B_N)); 238 } else { 239 ierr = PetscFPTrapPush(PETSC_FP_TRAP_OFF);CHKERRQ(ierr); 240 PetscStackCallBLAS("LAPACKgetrs",LAPACKgetrs_(type,&B_N,&B_Nrhs,mpardiso->schur,&B_N,mpardiso->schur_pivots,schur_sol,&B_N,&B_ierr)); 241 ierr = PetscFPTrapPop();CHKERRQ(ierr); 242 if (B_ierr) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_LIB,"Error in GETRS Lapack routine %d",(int)B_ierr); 243 } 244 break; 245 } 246 PetscFunctionReturn(0); 247 } 248 249 250 #undef __FUNCT__ 251 #define __FUNCT__ "MatFactorSetSchurIS_MKL_PARDISO" 252 PetscErrorCode MatFactorSetSchurIS_MKL_PARDISO(Mat F, IS is) 253 { 254 Mat_MKL_PARDISO *mpardiso =(Mat_MKL_PARDISO*)F->spptr; 255 const PetscInt *idxs; 256 PetscInt size,i; 257 PetscMPIInt csize; 258 PetscErrorCode ierr; 259 260 PetscFunctionBegin; 261 ierr = MPI_Comm_size(PetscObjectComm((PetscObject)F),&csize);CHKERRQ(ierr); 262 if (csize > 1) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"MKL_PARDISO parallel Schur complements not yet supported from PETSc\n"); 263 ierr = ISGetLocalSize(is,&size);CHKERRQ(ierr); 264 if (mpardiso->schur_size != size) { 265 mpardiso->schur_size = size; 266 ierr = PetscFree2(mpardiso->schur,mpardiso->schur_work);CHKERRQ(ierr); 267 ierr = PetscFree(mpardiso->schur_idxs);CHKERRQ(ierr); 268 ierr = PetscFree(mpardiso->schur_pivots);CHKERRQ(ierr); 269 ierr = PetscBLASIntCast(PetscMax(mpardiso->n,2*size),&mpardiso->schur_work_size);CHKERRQ(ierr); 270 ierr = PetscMalloc2(size*size,&mpardiso->schur,mpardiso->schur_work_size,&mpardiso->schur_work);CHKERRQ(ierr); 271 ierr = PetscMalloc1(size,&mpardiso->schur_idxs);CHKERRQ(ierr); 272 } 273 ierr = PetscMemzero(mpardiso->perm,mpardiso->n*sizeof(INT_TYPE));CHKERRQ(ierr); 274 ierr = ISGetIndices(is,&idxs);CHKERRQ(ierr); 275 ierr = PetscMemcpy(mpardiso->schur_idxs,idxs,size*sizeof(PetscInt));CHKERRQ(ierr); 276 for (i=0;i<size;i++) mpardiso->perm[idxs[i]] = 1; 277 ierr = ISRestoreIndices(is,&idxs);CHKERRQ(ierr); 278 if (size) { /* turn on Schur switch if we the set of indices is not empty */ 279 mpardiso->iparm[36-1] = 2; 280 } 281 mpardiso->schur_factored = PETSC_FALSE; 282 mpardiso->schur_inverted = PETSC_FALSE; 283 PetscFunctionReturn(0); 284 } 285 286 #undef __FUNCT__ 287 #define __FUNCT__ "MatFactorCreateSchurComplement_MKL_PARDISO" 288 PetscErrorCode MatFactorCreateSchurComplement_MKL_PARDISO(Mat F,Mat* S) 289 { 290 Mat St; 291 Mat_MKL_PARDISO *mpardiso =(Mat_MKL_PARDISO*)F->spptr; 292 PetscScalar *array; 293 PetscErrorCode ierr; 294 295 PetscFunctionBegin; 296 if (!mpardiso->iparm[36-1]) SETERRQ(PetscObjectComm((PetscObject)F),PETSC_ERR_ORDER,"Schur complement mode not selected! You should call MatFactorSetSchurIS to enable it"); 297 else if (!mpardiso->schur_size) SETERRQ(PetscObjectComm((PetscObject)F),PETSC_ERR_ORDER,"Schur indices not set! You should call MatFactorSetSchurIS before"); 298 299 ierr = MatCreate(PetscObjectComm((PetscObject)F),&St);CHKERRQ(ierr); 300 ierr = MatSetSizes(St,PETSC_DECIDE,PETSC_DECIDE,mpardiso->schur_size,mpardiso->schur_size);CHKERRQ(ierr); 301 ierr = MatSetType(St,MATDENSE);CHKERRQ(ierr); 302 ierr = MatSetUp(St);CHKERRQ(ierr); 303 ierr = MatDenseGetArray(St,&array);CHKERRQ(ierr); 304 ierr = PetscMemcpy(array,mpardiso->schur,mpardiso->schur_size*mpardiso->schur_size*sizeof(PetscScalar));CHKERRQ(ierr); 305 ierr = MatDenseRestoreArray(St,&array);CHKERRQ(ierr); 306 *S = St; 307 PetscFunctionReturn(0); 308 } 309 310 #undef __FUNCT__ 311 #define __FUNCT__ "MatFactorGetSchurComplement_MKL_PARDISO" 312 PetscErrorCode MatFactorGetSchurComplement_MKL_PARDISO(Mat F,Mat* S) 313 { 314 Mat St; 315 Mat_MKL_PARDISO *mpardiso =(Mat_MKL_PARDISO*)F->spptr; 316 PetscErrorCode ierr; 317 318 PetscFunctionBegin; 319 if (!mpardiso->iparm[36-1]) SETERRQ(PetscObjectComm((PetscObject)F),PETSC_ERR_ORDER,"Schur complement mode not selected! You should call MatFactorSetSchurIS to enable it"); 320 else if (!mpardiso->schur_size) SETERRQ(PetscObjectComm((PetscObject)F),PETSC_ERR_ORDER,"Schur indices not set! You should call MatFactorSetSchurIS before"); 321 322 ierr = MatCreateSeqDense(PetscObjectComm((PetscObject)F),mpardiso->schur_size,mpardiso->schur_size,mpardiso->schur,&St);CHKERRQ(ierr); 323 *S = St; 324 PetscFunctionReturn(0); 325 } 326 327 #undef __FUNCT__ 328 #define __FUNCT__ "MatFactorInvertSchurComplement_MKL_PARDISO" 329 PetscErrorCode MatFactorInvertSchurComplement_MKL_PARDISO(Mat F) 330 { 331 Mat_MKL_PARDISO *mpardiso =(Mat_MKL_PARDISO*)F->spptr; 332 PetscErrorCode ierr; 333 334 PetscFunctionBegin; 335 if (!mpardiso->iparm[36-1]) { /* do nothing */ 336 PetscFunctionReturn(0); 337 } 338 if (!mpardiso->schur_size) SETERRQ(PetscObjectComm((PetscObject)F),PETSC_ERR_ORDER,"Schur indices not set! You should call MatFactorSetSchurIS before"); 339 ierr = MatMKLPardisoInvertSchur_Private(mpardiso);CHKERRQ(ierr); 340 PetscFunctionReturn(0); 341 } 342 343 #undef __FUNCT__ 344 #define __FUNCT__ "MatFactorSolveSchurComplement_MKL_PARDISO" 345 PetscErrorCode MatFactorSolveSchurComplement_MKL_PARDISO(Mat F, Vec rhs, Vec sol) 346 { 347 Mat_MKL_PARDISO *mpardiso =(Mat_MKL_PARDISO*)F->spptr; 348 PetscScalar *asol,*arhs; 349 PetscErrorCode ierr; 350 351 PetscFunctionBegin; 352 if (!mpardiso->iparm[36-1]) SETERRQ(PetscObjectComm((PetscObject)F),PETSC_ERR_ORDER,"Schur complement mode not selected! You should call MatFactorSetSchurIS to enable it"); 353 else if (!mpardiso->schur_size) SETERRQ(PetscObjectComm((PetscObject)F),PETSC_ERR_ORDER,"Schur indices not set! You should call MatFactorSetSchurIS before"); 354 355 mpardiso->nrhs = 1; 356 ierr = VecGetArrayRead(rhs,(const PetscScalar**)&arhs);CHKERRQ(ierr); 357 ierr = VecGetArray(sol,&asol);CHKERRQ(ierr); 358 ierr = MatMKLPardisoSolveSchur_Private(mpardiso,arhs,asol);CHKERRQ(ierr); 359 ierr = VecRestoreArrayRead(rhs,(const PetscScalar**)&arhs);CHKERRQ(ierr); 360 ierr = VecRestoreArray(sol,&asol);CHKERRQ(ierr); 361 PetscFunctionReturn(0); 362 } 363 364 #undef __FUNCT__ 365 #define __FUNCT__ "MatFactorSolveSchurComplementTranspose_MKL_PARDISO" 366 PetscErrorCode MatFactorSolveSchurComplementTranspose_MKL_PARDISO(Mat F, Vec rhs, Vec sol) 367 { 368 Mat_MKL_PARDISO *mpardiso =(Mat_MKL_PARDISO*)F->spptr; 369 PetscScalar *asol,*arhs; 370 PetscErrorCode ierr; 371 372 PetscFunctionBegin; 373 if (!mpardiso->iparm[36-1]) SETERRQ(PetscObjectComm((PetscObject)F),PETSC_ERR_ORDER,"Schur complement mode not selected! You should call MatFactorSetSchurIS to enable it"); 374 else if (!mpardiso->schur_size) SETERRQ(PetscObjectComm((PetscObject)F),PETSC_ERR_ORDER,"Schur indices not set! You should call MatFactorSetSchurIS before"); 375 376 mpardiso->nrhs = 1; 377 ierr = VecGetArrayRead(rhs,(const PetscScalar**)&arhs);CHKERRQ(ierr); 378 ierr = VecGetArray(sol,&asol);CHKERRQ(ierr); 379 mpardiso->iparm[12 - 1] = 2; 380 ierr = MatMKLPardisoSolveSchur_Private(mpardiso,arhs,asol);CHKERRQ(ierr); 381 mpardiso->iparm[12 - 1] = 0; 382 ierr = VecRestoreArrayRead(rhs,(const PetscScalar**)&arhs);CHKERRQ(ierr); 383 ierr = VecRestoreArray(sol,&asol);CHKERRQ(ierr); 384 PetscFunctionReturn(0); 385 } 386 /* 387 * Copy the elements of matrix A. 388 * Input: 389 * - Mat A: MATSEQAIJ matrix 390 * - int shift: matrix index. 391 * - 0 for c representation 392 * - 1 for fortran representation 393 * - MatReuse reuse: 394 * - MAT_INITIAL_MATRIX: Create a new aij representation 395 * - MAT_REUSE_MATRIX: Reuse all aij representation and just change values 396 * Output: 397 * - int *nnz: Number of nonzero-elements. 398 * - int **r pointer to i index 399 * - int **c pointer to j elements 400 * - MATRIXTYPE **v: Non-zero elements 401 */ 402 #undef __FUNCT__ 403 #define __FUNCT__ "MatCopy_MKL_PARDISO" 404 PetscErrorCode MatCopy_MKL_PARDISO(Mat A, MatReuse reuse, INT_TYPE *nnz, INT_TYPE **r, INT_TYPE **c, void **v) 405 { 406 Mat_SeqAIJ *aa=(Mat_SeqAIJ*)A->data; 407 408 PetscFunctionBegin; 409 *v=aa->a; 410 if (reuse == MAT_INITIAL_MATRIX) { 411 *r = (INT_TYPE*)aa->i; 412 *c = (INT_TYPE*)aa->j; 413 *nnz = aa->nz; 414 } 415 PetscFunctionReturn(0); 416 } 417 418 /* 419 * Free memory for Mat_MKL_PARDISO structure and pointers to objects. 420 */ 421 #undef __FUNCT__ 422 #define __FUNCT__ "MatDestroy_MKL_PARDISO" 423 PetscErrorCode MatDestroy_MKL_PARDISO(Mat A) 424 { 425 Mat_MKL_PARDISO *mat_mkl_pardiso=(Mat_MKL_PARDISO*)A->spptr; 426 PetscErrorCode ierr; 427 428 PetscFunctionBegin; 429 /* Terminate instance, deallocate memories */ 430 if (mat_mkl_pardiso->CleanUp) { 431 mat_mkl_pardiso->phase = JOB_RELEASE_OF_ALL_MEMORY; 432 433 MKL_PARDISO (mat_mkl_pardiso->pt, 434 &mat_mkl_pardiso->maxfct, 435 &mat_mkl_pardiso->mnum, 436 &mat_mkl_pardiso->mtype, 437 &mat_mkl_pardiso->phase, 438 &mat_mkl_pardiso->n, 439 NULL, 440 NULL, 441 NULL, 442 mat_mkl_pardiso->perm, 443 &mat_mkl_pardiso->nrhs, 444 mat_mkl_pardiso->iparm, 445 &mat_mkl_pardiso->msglvl, 446 NULL, 447 NULL, 448 &mat_mkl_pardiso->err); 449 } 450 ierr = PetscFree(mat_mkl_pardiso->perm);CHKERRQ(ierr); 451 ierr = PetscFree2(mat_mkl_pardiso->schur,mat_mkl_pardiso->schur_work);CHKERRQ(ierr); 452 ierr = PetscFree(mat_mkl_pardiso->schur_idxs);CHKERRQ(ierr); 453 ierr = PetscFree(mat_mkl_pardiso->schur_pivots);CHKERRQ(ierr); 454 ierr = PetscFree(A->spptr);CHKERRQ(ierr); 455 456 /* clear composed functions */ 457 ierr = PetscObjectComposeFunction((PetscObject)A,"MatFactorGetSolverPackage_C",NULL);CHKERRQ(ierr); 458 ierr = PetscObjectComposeFunction((PetscObject)A,"MatFactorSetSchurIS_C",NULL);CHKERRQ(ierr); 459 ierr = PetscObjectComposeFunction((PetscObject)A,"MatFactorCreateSchurComplement_C",NULL);CHKERRQ(ierr); 460 ierr = PetscObjectComposeFunction((PetscObject)A,"MatFactorGetSchurComplement_C",NULL);CHKERRQ(ierr); 461 ierr = PetscObjectComposeFunction((PetscObject)A,"MatFactorInvertSchurComplement_C",NULL);CHKERRQ(ierr); 462 ierr = PetscObjectComposeFunction((PetscObject)A,"MatFactorSolveSchurComplement_C",NULL);CHKERRQ(ierr); 463 ierr = PetscObjectComposeFunction((PetscObject)A,"MatFactorSolveSchurComplementTranspose_C",NULL);CHKERRQ(ierr); 464 ierr = PetscObjectComposeFunction((PetscObject)A,"MatMkl_PardisoSetCntl_C",NULL);CHKERRQ(ierr); 465 466 ierr = MatDestroy_SeqAIJ(A);CHKERRQ(ierr); 467 PetscFunctionReturn(0); 468 } 469 470 #undef __FUNCT__ 471 #define __FUNCT__ "MatMKLPardisoScatterSchur_Private" 472 static PetscErrorCode MatMKLPardisoScatterSchur_Private(Mat_MKL_PARDISO *mpardiso, PetscScalar *whole, PetscScalar *schur, PetscBool reduce) 473 { 474 PetscFunctionBegin; 475 476 if (reduce) { /* data given for the whole matrix */ 477 PetscInt i,m=0,p=0; 478 for (i=0;i<mpardiso->nrhs;i++) { 479 PetscInt j; 480 for (j=0;j<mpardiso->schur_size;j++) { 481 schur[p+j] = whole[m+mpardiso->schur_idxs[j]]; 482 } 483 m += mpardiso->n; 484 p += mpardiso->schur_size; 485 } 486 } else { /* from Schur to whole */ 487 PetscInt i,m=0,p=0; 488 for (i=0;i<mpardiso->nrhs;i++) { 489 PetscInt j; 490 for (j=0;j<mpardiso->schur_size;j++) { 491 whole[m+mpardiso->schur_idxs[j]] = schur[p+j]; 492 } 493 m += mpardiso->n; 494 p += mpardiso->schur_size; 495 } 496 } 497 PetscFunctionReturn(0); 498 } 499 500 /* 501 * Computes Ax = b 502 */ 503 #undef __FUNCT__ 504 #define __FUNCT__ "MatSolve_MKL_PARDISO" 505 PetscErrorCode MatSolve_MKL_PARDISO(Mat A,Vec b,Vec x) 506 { 507 Mat_MKL_PARDISO *mat_mkl_pardiso=(Mat_MKL_PARDISO*)(A)->spptr; 508 PetscErrorCode ierr; 509 PetscScalar *xarray; 510 const PetscScalar *barray; 511 512 PetscFunctionBegin; 513 mat_mkl_pardiso->nrhs = 1; 514 ierr = VecGetArray(x,&xarray);CHKERRQ(ierr); 515 ierr = VecGetArrayRead(b,&barray);CHKERRQ(ierr); 516 517 /* solve phase */ 518 /*-------------*/ 519 if (!mat_mkl_pardiso->schur) { 520 mat_mkl_pardiso->phase = JOB_SOLVE_ITERATIVE_REFINEMENT; 521 } else { 522 mat_mkl_pardiso->phase = JOB_SOLVE_FORWARD_SUBSTITUTION; 523 } 524 mat_mkl_pardiso->iparm[6-1] = 0; 525 MKL_PARDISO (mat_mkl_pardiso->pt, 526 &mat_mkl_pardiso->maxfct, 527 &mat_mkl_pardiso->mnum, 528 &mat_mkl_pardiso->mtype, 529 &mat_mkl_pardiso->phase, 530 &mat_mkl_pardiso->n, 531 mat_mkl_pardiso->a, 532 mat_mkl_pardiso->ia, 533 mat_mkl_pardiso->ja, 534 mat_mkl_pardiso->perm, 535 &mat_mkl_pardiso->nrhs, 536 mat_mkl_pardiso->iparm, 537 &mat_mkl_pardiso->msglvl, 538 (void*)barray, 539 (void*)xarray, 540 &mat_mkl_pardiso->err); 541 542 if (mat_mkl_pardiso->err < 0) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_LIB,"Error reported by MKL_PARDISO: err=%d. Please check manual\n",mat_mkl_pardiso->err); 543 544 if (mat_mkl_pardiso->schur) { /* solve Schur complement and expand solution */ 545 PetscInt shift = mat_mkl_pardiso->schur_size; 546 547 /* if inverted, uses BLAS *MM subroutines, otherwise LAPACK *TRS */ 548 if (!mat_mkl_pardiso->schur_inverted) { 549 shift = 0; 550 } 551 552 /* solve Schur complement */ 553 ierr = MatMKLPardisoScatterSchur_Private(mat_mkl_pardiso,xarray,mat_mkl_pardiso->schur_work,PETSC_TRUE);CHKERRQ(ierr); 554 ierr = MatMKLPardisoSolveSchur_Private(mat_mkl_pardiso,mat_mkl_pardiso->schur_work,mat_mkl_pardiso->schur_work+shift);CHKERRQ(ierr); 555 ierr = MatMKLPardisoScatterSchur_Private(mat_mkl_pardiso,xarray,mat_mkl_pardiso->schur_work+shift,PETSC_FALSE);CHKERRQ(ierr); 556 557 /* expansion phase */ 558 mat_mkl_pardiso->iparm[6-1] = 1; 559 mat_mkl_pardiso->phase = JOB_SOLVE_BACKWARD_SUBSTITUTION; 560 MKL_PARDISO (mat_mkl_pardiso->pt, 561 &mat_mkl_pardiso->maxfct, 562 &mat_mkl_pardiso->mnum, 563 &mat_mkl_pardiso->mtype, 564 &mat_mkl_pardiso->phase, 565 &mat_mkl_pardiso->n, 566 mat_mkl_pardiso->a, 567 mat_mkl_pardiso->ia, 568 mat_mkl_pardiso->ja, 569 mat_mkl_pardiso->perm, 570 &mat_mkl_pardiso->nrhs, 571 mat_mkl_pardiso->iparm, 572 &mat_mkl_pardiso->msglvl, 573 (void*)xarray, 574 (void*)mat_mkl_pardiso->schur_work, /* according to the specs, the solution vector is always used */ 575 &mat_mkl_pardiso->err); 576 577 if (mat_mkl_pardiso->err < 0) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_LIB,"Error reported by MKL_PARDISO: err=%d. Please check manual\n",mat_mkl_pardiso->err); 578 mat_mkl_pardiso->iparm[6-1] = 0; 579 } 580 ierr = VecRestoreArray(x,&xarray);CHKERRQ(ierr); 581 ierr = VecRestoreArrayRead(b,&barray);CHKERRQ(ierr); 582 mat_mkl_pardiso->CleanUp = PETSC_TRUE; 583 PetscFunctionReturn(0); 584 } 585 586 587 #undef __FUNCT__ 588 #define __FUNCT__ "MatSolveTranspose_MKL_PARDISO" 589 PetscErrorCode MatSolveTranspose_MKL_PARDISO(Mat A,Vec b,Vec x) 590 { 591 Mat_MKL_PARDISO *mat_mkl_pardiso=(Mat_MKL_PARDISO*)A->spptr; 592 PetscErrorCode ierr; 593 594 PetscFunctionBegin; 595 #if defined(PETSC_USE_COMPLEX) 596 mat_mkl_pardiso->iparm[12 - 1] = 1; 597 #else 598 mat_mkl_pardiso->iparm[12 - 1] = 2; 599 #endif 600 ierr = MatSolve_MKL_PARDISO(A,b,x);CHKERRQ(ierr); 601 mat_mkl_pardiso->iparm[12 - 1] = 0; 602 PetscFunctionReturn(0); 603 } 604 605 606 #undef __FUNCT__ 607 #define __FUNCT__ "MatMatSolve_MKL_PARDISO" 608 PetscErrorCode MatMatSolve_MKL_PARDISO(Mat A,Mat B,Mat X) 609 { 610 Mat_MKL_PARDISO *mat_mkl_pardiso=(Mat_MKL_PARDISO*)(A)->spptr; 611 PetscErrorCode ierr; 612 PetscScalar *barray, *xarray; 613 PetscBool flg; 614 615 PetscFunctionBegin; 616 ierr = PetscObjectTypeCompare((PetscObject)B,MATSEQDENSE,&flg);CHKERRQ(ierr); 617 if (!flg) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_ARG_WRONG,"Matrix B must be MATSEQDENSE matrix"); 618 ierr = PetscObjectTypeCompare((PetscObject)X,MATSEQDENSE,&flg);CHKERRQ(ierr); 619 if (!flg) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_ARG_WRONG,"Matrix X must be MATSEQDENSE matrix"); 620 621 ierr = MatGetSize(B,NULL,(PetscInt*)&mat_mkl_pardiso->nrhs);CHKERRQ(ierr); 622 623 if(mat_mkl_pardiso->nrhs > 0){ 624 ierr = MatDenseGetArray(B,&barray); 625 ierr = MatDenseGetArray(X,&xarray); 626 627 /* solve phase */ 628 /*-------------*/ 629 if (!mat_mkl_pardiso->schur) { 630 mat_mkl_pardiso->phase = JOB_SOLVE_ITERATIVE_REFINEMENT; 631 } else { 632 mat_mkl_pardiso->phase = JOB_SOLVE_FORWARD_SUBSTITUTION; 633 } 634 mat_mkl_pardiso->iparm[6-1] = 0; 635 MKL_PARDISO (mat_mkl_pardiso->pt, 636 &mat_mkl_pardiso->maxfct, 637 &mat_mkl_pardiso->mnum, 638 &mat_mkl_pardiso->mtype, 639 &mat_mkl_pardiso->phase, 640 &mat_mkl_pardiso->n, 641 mat_mkl_pardiso->a, 642 mat_mkl_pardiso->ia, 643 mat_mkl_pardiso->ja, 644 mat_mkl_pardiso->perm, 645 &mat_mkl_pardiso->nrhs, 646 mat_mkl_pardiso->iparm, 647 &mat_mkl_pardiso->msglvl, 648 (void*)barray, 649 (void*)xarray, 650 &mat_mkl_pardiso->err); 651 if (mat_mkl_pardiso->err < 0) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_LIB,"Error reported by MKL_PARDISO: err=%d. Please check manual\n",mat_mkl_pardiso->err); 652 653 if (mat_mkl_pardiso->schur) { /* solve Schur complement and expand solution */ 654 PetscScalar *o_schur_work = NULL; 655 PetscInt shift = mat_mkl_pardiso->schur_size*mat_mkl_pardiso->nrhs,scale; 656 PetscInt mem = mat_mkl_pardiso->n*mat_mkl_pardiso->nrhs; 657 658 /* allocate extra memory if it is needed */ 659 scale = 1; 660 if (mat_mkl_pardiso->schur_inverted) { 661 scale = 2; 662 } 663 mem *= scale; 664 if (mem > mat_mkl_pardiso->schur_work_size) { 665 o_schur_work = mat_mkl_pardiso->schur_work; 666 ierr = PetscMalloc1(mem,&mat_mkl_pardiso->schur_work);CHKERRQ(ierr); 667 } 668 669 /* if inverted, uses BLAS *MM subroutines, otherwise LAPACK *TRS */ 670 if (!mat_mkl_pardiso->schur_inverted) { 671 shift = 0; 672 } 673 674 /* solve Schur complement */ 675 ierr = MatMKLPardisoScatterSchur_Private(mat_mkl_pardiso,xarray,mat_mkl_pardiso->schur_work,PETSC_TRUE);CHKERRQ(ierr); 676 ierr = MatMKLPardisoSolveSchur_Private(mat_mkl_pardiso,mat_mkl_pardiso->schur_work,mat_mkl_pardiso->schur_work+shift);CHKERRQ(ierr); 677 ierr = MatMKLPardisoScatterSchur_Private(mat_mkl_pardiso,xarray,mat_mkl_pardiso->schur_work+shift,PETSC_FALSE);CHKERRQ(ierr); 678 679 /* expansion phase */ 680 mat_mkl_pardiso->iparm[6-1] = 1; 681 mat_mkl_pardiso->phase = JOB_SOLVE_BACKWARD_SUBSTITUTION; 682 MKL_PARDISO (mat_mkl_pardiso->pt, 683 &mat_mkl_pardiso->maxfct, 684 &mat_mkl_pardiso->mnum, 685 &mat_mkl_pardiso->mtype, 686 &mat_mkl_pardiso->phase, 687 &mat_mkl_pardiso->n, 688 mat_mkl_pardiso->a, 689 mat_mkl_pardiso->ia, 690 mat_mkl_pardiso->ja, 691 mat_mkl_pardiso->perm, 692 &mat_mkl_pardiso->nrhs, 693 mat_mkl_pardiso->iparm, 694 &mat_mkl_pardiso->msglvl, 695 (void*)xarray, 696 (void*)mat_mkl_pardiso->schur_work, /* according to the specs, the solution vector is always used */ 697 &mat_mkl_pardiso->err); 698 if (o_schur_work) { /* restore original schur_work (minimal size) */ 699 ierr = PetscFree(mat_mkl_pardiso->schur_work);CHKERRQ(ierr); 700 mat_mkl_pardiso->schur_work = o_schur_work; 701 } 702 if (mat_mkl_pardiso->err < 0) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_LIB,"Error reported by MKL_PARDISO: err=%d. Please check manual\n",mat_mkl_pardiso->err); 703 mat_mkl_pardiso->iparm[6-1] = 0; 704 } 705 } 706 mat_mkl_pardiso->CleanUp = PETSC_TRUE; 707 PetscFunctionReturn(0); 708 } 709 710 /* 711 * LU Decomposition 712 */ 713 #undef __FUNCT__ 714 #define __FUNCT__ "MatFactorNumeric_MKL_PARDISO" 715 PetscErrorCode MatFactorNumeric_MKL_PARDISO(Mat F,Mat A,const MatFactorInfo *info) 716 { 717 Mat_MKL_PARDISO *mat_mkl_pardiso=(Mat_MKL_PARDISO*)(F)->spptr; 718 PetscErrorCode ierr; 719 720 /* numerical factorization phase */ 721 /*-------------------------------*/ 722 PetscFunctionBegin; 723 mat_mkl_pardiso->matstruc = SAME_NONZERO_PATTERN; 724 ierr = MatCopy_MKL_PARDISO(A, MAT_REUSE_MATRIX, &mat_mkl_pardiso->nz, &mat_mkl_pardiso->ia, &mat_mkl_pardiso->ja, &mat_mkl_pardiso->a);CHKERRQ(ierr); 725 726 /* numerical factorization phase */ 727 /*-------------------------------*/ 728 mat_mkl_pardiso->phase = JOB_NUMERICAL_FACTORIZATION; 729 MKL_PARDISO (mat_mkl_pardiso->pt, 730 &mat_mkl_pardiso->maxfct, 731 &mat_mkl_pardiso->mnum, 732 &mat_mkl_pardiso->mtype, 733 &mat_mkl_pardiso->phase, 734 &mat_mkl_pardiso->n, 735 mat_mkl_pardiso->a, 736 mat_mkl_pardiso->ia, 737 mat_mkl_pardiso->ja, 738 mat_mkl_pardiso->perm, 739 &mat_mkl_pardiso->nrhs, 740 mat_mkl_pardiso->iparm, 741 &mat_mkl_pardiso->msglvl, 742 NULL, 743 (void*)mat_mkl_pardiso->schur, 744 &mat_mkl_pardiso->err); 745 if (mat_mkl_pardiso->err < 0) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_LIB,"Error reported by MKL_PARDISO: err=%d. Please check manual\n",mat_mkl_pardiso->err); 746 747 if (mat_mkl_pardiso->schur) { /* schur output from pardiso is in row major format */ 748 PetscInt j,k,n=mat_mkl_pardiso->schur_size; 749 for (j=0; j<n; j++) { 750 for (k=0; k<j; k++) { 751 PetscScalar tmp = mat_mkl_pardiso->schur[j + k*n]; 752 mat_mkl_pardiso->schur[j + k*n] = mat_mkl_pardiso->schur[k + j*n]; 753 mat_mkl_pardiso->schur[k + j*n] = tmp; 754 } 755 } 756 } 757 mat_mkl_pardiso->matstruc = SAME_NONZERO_PATTERN; 758 mat_mkl_pardiso->CleanUp = PETSC_TRUE; 759 mat_mkl_pardiso->schur_factored = PETSC_FALSE; 760 mat_mkl_pardiso->schur_inverted = PETSC_FALSE; 761 mat_mkl_pardiso->schur_solver_type = 0; 762 PetscFunctionReturn(0); 763 } 764 765 /* Sets mkl_pardiso options from the options database */ 766 #undef __FUNCT__ 767 #define __FUNCT__ "PetscSetMKL_PARDISOFromOptions" 768 PetscErrorCode PetscSetMKL_PARDISOFromOptions(Mat F, Mat A) 769 { 770 Mat_MKL_PARDISO *mat_mkl_pardiso = (Mat_MKL_PARDISO*)F->spptr; 771 PetscErrorCode ierr; 772 PetscInt icntl; 773 PetscBool flg; 774 int pt[IPARM_SIZE], threads = 1; 775 776 PetscFunctionBegin; 777 ierr = PetscOptionsBegin(PetscObjectComm((PetscObject)A),((PetscObject)A)->prefix,"MKL_PARDISO Options","Mat");CHKERRQ(ierr); 778 ierr = PetscOptionsInt("-mat_mkl_pardiso_65","Number of threads to use","None",threads,&threads,&flg);CHKERRQ(ierr); 779 if (flg) mkl_set_num_threads(threads); 780 781 ierr = PetscOptionsInt("-mat_mkl_pardiso_66","Maximum number of factors with identical sparsity structure that must be kept in memory at the same time","None",mat_mkl_pardiso->maxfct,&icntl,&flg);CHKERRQ(ierr); 782 if (flg) mat_mkl_pardiso->maxfct = icntl; 783 784 ierr = PetscOptionsInt("-mat_mkl_pardiso_67","Indicates the actual matrix for the solution phase","None",mat_mkl_pardiso->mnum,&icntl,&flg);CHKERRQ(ierr); 785 if (flg) mat_mkl_pardiso->mnum = icntl; 786 787 ierr = PetscOptionsInt("-mat_mkl_pardiso_68","Message level information","None",mat_mkl_pardiso->msglvl,&icntl,&flg);CHKERRQ(ierr); 788 if (flg) mat_mkl_pardiso->msglvl = icntl; 789 790 ierr = PetscOptionsInt("-mat_mkl_pardiso_69","Defines the matrix type","None",mat_mkl_pardiso->mtype,&icntl,&flg);CHKERRQ(ierr); 791 if(flg){ 792 mat_mkl_pardiso->mtype = icntl; 793 MKL_PARDISO_INIT(&pt, &mat_mkl_pardiso->mtype, mat_mkl_pardiso->iparm); 794 #if defined(PETSC_USE_REAL_SINGLE) 795 mat_mkl_pardiso->iparm[27] = 1; 796 #else 797 mat_mkl_pardiso->iparm[27] = 0; 798 #endif 799 mat_mkl_pardiso->iparm[34] = 1; 800 } 801 ierr = PetscOptionsInt("-mat_mkl_pardiso_1","Use default values","None",mat_mkl_pardiso->iparm[0],&icntl,&flg);CHKERRQ(ierr); 802 803 if(flg && icntl != 0){ 804 ierr = PetscOptionsInt("-mat_mkl_pardiso_2","Fill-in reducing ordering for the input matrix","None",mat_mkl_pardiso->iparm[1],&icntl,&flg);CHKERRQ(ierr); 805 if (flg) mat_mkl_pardiso->iparm[1] = icntl; 806 807 ierr = PetscOptionsInt("-mat_mkl_pardiso_4","Preconditioned CGS/CG","None",mat_mkl_pardiso->iparm[3],&icntl,&flg);CHKERRQ(ierr); 808 if (flg) mat_mkl_pardiso->iparm[3] = icntl; 809 810 ierr = PetscOptionsInt("-mat_mkl_pardiso_5","User permutation","None",mat_mkl_pardiso->iparm[4],&icntl,&flg);CHKERRQ(ierr); 811 if (flg) mat_mkl_pardiso->iparm[4] = icntl; 812 813 ierr = PetscOptionsInt("-mat_mkl_pardiso_6","Write solution on x","None",mat_mkl_pardiso->iparm[5],&icntl,&flg);CHKERRQ(ierr); 814 if (flg) mat_mkl_pardiso->iparm[5] = icntl; 815 816 ierr = PetscOptionsInt("-mat_mkl_pardiso_8","Iterative refinement step","None",mat_mkl_pardiso->iparm[7],&icntl,&flg);CHKERRQ(ierr); 817 if (flg) mat_mkl_pardiso->iparm[7] = icntl; 818 819 ierr = PetscOptionsInt("-mat_mkl_pardiso_10","Pivoting perturbation","None",mat_mkl_pardiso->iparm[9],&icntl,&flg);CHKERRQ(ierr); 820 if (flg) mat_mkl_pardiso->iparm[9] = icntl; 821 822 ierr = PetscOptionsInt("-mat_mkl_pardiso_11","Scaling vectors","None",mat_mkl_pardiso->iparm[10],&icntl,&flg);CHKERRQ(ierr); 823 if (flg) mat_mkl_pardiso->iparm[10] = icntl; 824 825 ierr = PetscOptionsInt("-mat_mkl_pardiso_12","Solve with transposed or conjugate transposed matrix A","None",mat_mkl_pardiso->iparm[11],&icntl,&flg);CHKERRQ(ierr); 826 if (flg) mat_mkl_pardiso->iparm[11] = icntl; 827 828 ierr = PetscOptionsInt("-mat_mkl_pardiso_13","Improved accuracy using (non-) symmetric weighted matching","None",mat_mkl_pardiso->iparm[12],&icntl,&flg);CHKERRQ(ierr); 829 if (flg) mat_mkl_pardiso->iparm[12] = icntl; 830 831 ierr = PetscOptionsInt("-mat_mkl_pardiso_18","Numbers of non-zero elements","None",mat_mkl_pardiso->iparm[17],&icntl,&flg);CHKERRQ(ierr); 832 if (flg) mat_mkl_pardiso->iparm[17] = icntl; 833 834 ierr = PetscOptionsInt("-mat_mkl_pardiso_19","Report number of floating point operations","None",mat_mkl_pardiso->iparm[18],&icntl,&flg);CHKERRQ(ierr); 835 if (flg) mat_mkl_pardiso->iparm[18] = icntl; 836 837 ierr = PetscOptionsInt("-mat_mkl_pardiso_21","Pivoting for symmetric indefinite matrices","None",mat_mkl_pardiso->iparm[20],&icntl,&flg);CHKERRQ(ierr); 838 if (flg) mat_mkl_pardiso->iparm[20] = icntl; 839 840 ierr = PetscOptionsInt("-mat_mkl_pardiso_24","Parallel factorization control","None",mat_mkl_pardiso->iparm[23],&icntl,&flg);CHKERRQ(ierr); 841 if (flg) mat_mkl_pardiso->iparm[23] = icntl; 842 843 ierr = PetscOptionsInt("-mat_mkl_pardiso_25","Parallel forward/backward solve control","None",mat_mkl_pardiso->iparm[24],&icntl,&flg);CHKERRQ(ierr); 844 if (flg) mat_mkl_pardiso->iparm[24] = icntl; 845 846 ierr = PetscOptionsInt("-mat_mkl_pardiso_27","Matrix checker","None",mat_mkl_pardiso->iparm[26],&icntl,&flg);CHKERRQ(ierr); 847 if (flg) mat_mkl_pardiso->iparm[26] = icntl; 848 849 ierr = PetscOptionsInt("-mat_mkl_pardiso_31","Partial solve and computing selected components of the solution vectors","None",mat_mkl_pardiso->iparm[30],&icntl,&flg);CHKERRQ(ierr); 850 if (flg) mat_mkl_pardiso->iparm[30] = icntl; 851 852 ierr = PetscOptionsInt("-mat_mkl_pardiso_34","Optimal number of threads for conditional numerical reproducibility (CNR) mode","None",mat_mkl_pardiso->iparm[33],&icntl,&flg);CHKERRQ(ierr); 853 if (flg) mat_mkl_pardiso->iparm[33] = icntl; 854 855 ierr = PetscOptionsInt("-mat_mkl_pardiso_60","Intel MKL_PARDISO mode","None",mat_mkl_pardiso->iparm[59],&icntl,&flg);CHKERRQ(ierr); 856 if (flg) mat_mkl_pardiso->iparm[59] = icntl; 857 } 858 PetscOptionsEnd(); 859 PetscFunctionReturn(0); 860 } 861 862 #undef __FUNCT__ 863 #define __FUNCT__ "MatFactorMKL_PARDISOInitialize_Private" 864 PetscErrorCode MatFactorMKL_PARDISOInitialize_Private(Mat A, MatFactorType ftype, Mat_MKL_PARDISO *mat_mkl_pardiso) 865 { 866 PetscErrorCode ierr; 867 PetscInt i; 868 869 PetscFunctionBegin; 870 for ( i = 0; i < IPARM_SIZE; i++ ){ 871 mat_mkl_pardiso->iparm[i] = 0; 872 } 873 874 for ( i = 0; i < IPARM_SIZE; i++ ){ 875 mat_mkl_pardiso->pt[i] = 0; 876 } 877 878 /* Default options for both sym and unsym */ 879 mat_mkl_pardiso->iparm[ 0] = 1; /* Solver default parameters overriden with provided by iparm */ 880 mat_mkl_pardiso->iparm[ 1] = 2; /* Metis reordering */ 881 mat_mkl_pardiso->iparm[ 5] = 0; /* Write solution into x */ 882 mat_mkl_pardiso->iparm[ 7] = 2; /* Max number of iterative refinement steps */ 883 mat_mkl_pardiso->iparm[17] = -1; /* Output: Number of nonzeros in the factor LU */ 884 mat_mkl_pardiso->iparm[18] = -1; /* Output: Mflops for LU factorization */ 885 #if 0 886 mat_mkl_pardiso->iparm[23] = 1; /* Parallel factorization control*/ 887 #endif 888 mat_mkl_pardiso->iparm[34] = 1; /* Cluster Sparse Solver use C-style indexing for ia and ja arrays */ 889 mat_mkl_pardiso->iparm[39] = 0; /* Input: matrix/rhs/solution stored on master */ 890 891 mat_mkl_pardiso->CleanUp = PETSC_FALSE; 892 mat_mkl_pardiso->maxfct = 1; /* Maximum number of numerical factorizations. */ 893 mat_mkl_pardiso->mnum = 1; /* Which factorization to use. */ 894 mat_mkl_pardiso->msglvl = 0; /* 0: do not print 1: Print statistical information in file */ 895 mat_mkl_pardiso->phase = -1; 896 mat_mkl_pardiso->err = 0; 897 898 mat_mkl_pardiso->n = A->rmap->N; 899 mat_mkl_pardiso->nrhs = 1; 900 mat_mkl_pardiso->err = 0; 901 mat_mkl_pardiso->phase = -1; 902 903 if(ftype == MAT_FACTOR_LU){ 904 /* Default type for non-sym */ 905 #if defined(PETSC_USE_COMPLEX) 906 mat_mkl_pardiso->mtype = 13; 907 #else 908 mat_mkl_pardiso->mtype = 11; 909 #endif 910 911 mat_mkl_pardiso->iparm[ 9] = 13; /* Perturb the pivot elements with 1E-13 */ 912 mat_mkl_pardiso->iparm[10] = 1; /* Use nonsymmetric permutation and scaling MPS */ 913 mat_mkl_pardiso->iparm[12] = 1; /* Switch on Maximum Weighted Matching algorithm (default for non-symmetric) */ 914 915 } else { 916 /* Default type for sym */ 917 #if defined(PETSC_USE_COMPLEX) 918 mat_mkl_pardiso ->mtype = 3; 919 #else 920 mat_mkl_pardiso ->mtype = -2; 921 #endif 922 mat_mkl_pardiso->iparm[ 9] = 13; /* Perturb the pivot elements with 1E-13 */ 923 mat_mkl_pardiso->iparm[10] = 0; /* Use nonsymmetric permutation and scaling MPS */ 924 mat_mkl_pardiso->iparm[12] = 1; /* Switch on Maximum Weighted Matching algorithm (default for non-symmetric) */ 925 /* mat_mkl_pardiso->iparm[20] = 1; */ /* Apply 1x1 and 2x2 Bunch-Kaufman pivoting during the factorization process */ 926 #if defined(PETSC_USE_DEBUG) 927 mat_mkl_pardiso->iparm[26] = 1; /* Matrix checker */ 928 #endif 929 } 930 ierr = PetscMalloc1(A->rmap->N*sizeof(INT_TYPE), &mat_mkl_pardiso->perm);CHKERRQ(ierr); 931 for(i = 0; i < A->rmap->N; i++){ 932 mat_mkl_pardiso->perm[i] = 0; 933 } 934 mat_mkl_pardiso->schur_size = 0; 935 PetscFunctionReturn(0); 936 } 937 938 /* 939 * Symbolic decomposition. Mkl_Pardiso analysis phase. 940 */ 941 #undef __FUNCT__ 942 #define __FUNCT__ "MatFactorSymbolic_AIJMKL_PARDISO_Private" 943 PetscErrorCode MatFactorSymbolic_AIJMKL_PARDISO_Private(Mat F,Mat A,const MatFactorInfo *info) 944 { 945 Mat_MKL_PARDISO *mat_mkl_pardiso = (Mat_MKL_PARDISO*)F->spptr; 946 PetscErrorCode ierr; 947 948 PetscFunctionBegin; 949 mat_mkl_pardiso->matstruc = DIFFERENT_NONZERO_PATTERN; 950 951 /* Set MKL_PARDISO options from the options database */ 952 ierr = PetscSetMKL_PARDISOFromOptions(F,A);CHKERRQ(ierr); 953 954 ierr = MatCopy_MKL_PARDISO(A, MAT_INITIAL_MATRIX, &mat_mkl_pardiso->nz, &mat_mkl_pardiso->ia, &mat_mkl_pardiso->ja, &mat_mkl_pardiso->a);CHKERRQ(ierr); 955 mat_mkl_pardiso->n = A->rmap->N; 956 957 /* analysis phase */ 958 /*----------------*/ 959 mat_mkl_pardiso->phase = JOB_ANALYSIS; 960 961 MKL_PARDISO (mat_mkl_pardiso->pt, 962 &mat_mkl_pardiso->maxfct, 963 &mat_mkl_pardiso->mnum, 964 &mat_mkl_pardiso->mtype, 965 &mat_mkl_pardiso->phase, 966 &mat_mkl_pardiso->n, 967 mat_mkl_pardiso->a, 968 mat_mkl_pardiso->ia, 969 mat_mkl_pardiso->ja, 970 mat_mkl_pardiso->perm, 971 &mat_mkl_pardiso->nrhs, 972 mat_mkl_pardiso->iparm, 973 &mat_mkl_pardiso->msglvl, 974 NULL, 975 NULL, 976 &mat_mkl_pardiso->err); 977 if (mat_mkl_pardiso->err < 0) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_LIB,"Error reported by MKL_PARDISO: err=%d\n. Please check manual",mat_mkl_pardiso->err); 978 979 mat_mkl_pardiso->CleanUp = PETSC_TRUE; 980 981 if (F->factortype == MAT_FACTOR_LU) { 982 F->ops->lufactornumeric = MatFactorNumeric_MKL_PARDISO; 983 } else { 984 F->ops->choleskyfactornumeric = MatFactorNumeric_MKL_PARDISO; 985 } 986 F->ops->solve = MatSolve_MKL_PARDISO; 987 F->ops->solvetranspose = MatSolveTranspose_MKL_PARDISO; 988 F->ops->matsolve = MatMatSolve_MKL_PARDISO; 989 PetscFunctionReturn(0); 990 } 991 992 #undef __FUNCT__ 993 #define __FUNCT__ "MatLUFactorSymbolic_AIJMKL_PARDISO" 994 PetscErrorCode MatLUFactorSymbolic_AIJMKL_PARDISO(Mat F,Mat A,IS r,IS c,const MatFactorInfo *info) 995 { 996 PetscErrorCode ierr; 997 998 PetscFunctionBegin; 999 ierr = MatFactorSymbolic_AIJMKL_PARDISO_Private(F, A, info);CHKERRQ(ierr); 1000 PetscFunctionReturn(0); 1001 } 1002 1003 #undef __FUNCT__ 1004 #define __FUNCT__ "MatCholeskyFactorSymbolic_AIJMKL_PARDISO" 1005 PetscErrorCode MatCholeskyFactorSymbolic_AIJMKL_PARDISO(Mat F,Mat A,IS r,const MatFactorInfo *info) 1006 { 1007 PetscErrorCode ierr; 1008 1009 PetscFunctionBegin; 1010 ierr = MatFactorSymbolic_AIJMKL_PARDISO_Private(F, A, info);CHKERRQ(ierr); 1011 PetscFunctionReturn(0); 1012 } 1013 1014 #undef __FUNCT__ 1015 #define __FUNCT__ "MatView_MKL_PARDISO" 1016 PetscErrorCode MatView_MKL_PARDISO(Mat A, PetscViewer viewer) 1017 { 1018 PetscErrorCode ierr; 1019 PetscBool iascii; 1020 PetscViewerFormat format; 1021 Mat_MKL_PARDISO *mat_mkl_pardiso=(Mat_MKL_PARDISO*)A->spptr; 1022 PetscInt i; 1023 1024 PetscFunctionBegin; 1025 /* check if matrix is mkl_pardiso type */ 1026 if (A->ops->solve != MatSolve_MKL_PARDISO) PetscFunctionReturn(0); 1027 1028 ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERASCII,&iascii);CHKERRQ(ierr); 1029 if (iascii) { 1030 ierr = PetscViewerGetFormat(viewer,&format);CHKERRQ(ierr); 1031 if (format == PETSC_VIEWER_ASCII_INFO) { 1032 ierr = PetscViewerASCIIPrintf(viewer,"MKL_PARDISO run parameters:\n");CHKERRQ(ierr); 1033 ierr = PetscViewerASCIIPrintf(viewer,"MKL_PARDISO phase: %d \n",mat_mkl_pardiso->phase);CHKERRQ(ierr); 1034 for(i = 1; i <= 64; i++){ 1035 ierr = PetscViewerASCIIPrintf(viewer,"MKL_PARDISO iparm[%d]: %d \n",i, mat_mkl_pardiso->iparm[i - 1]);CHKERRQ(ierr); 1036 } 1037 ierr = PetscViewerASCIIPrintf(viewer,"MKL_PARDISO maxfct: %d \n", mat_mkl_pardiso->maxfct);CHKERRQ(ierr); 1038 ierr = PetscViewerASCIIPrintf(viewer,"MKL_PARDISO mnum: %d \n", mat_mkl_pardiso->mnum);CHKERRQ(ierr); 1039 ierr = PetscViewerASCIIPrintf(viewer,"MKL_PARDISO mtype: %d \n", mat_mkl_pardiso->mtype);CHKERRQ(ierr); 1040 ierr = PetscViewerASCIIPrintf(viewer,"MKL_PARDISO n: %d \n", mat_mkl_pardiso->n);CHKERRQ(ierr); 1041 ierr = PetscViewerASCIIPrintf(viewer,"MKL_PARDISO nrhs: %d \n", mat_mkl_pardiso->nrhs);CHKERRQ(ierr); 1042 ierr = PetscViewerASCIIPrintf(viewer,"MKL_PARDISO msglvl: %d \n", mat_mkl_pardiso->msglvl);CHKERRQ(ierr); 1043 } 1044 } 1045 PetscFunctionReturn(0); 1046 } 1047 1048 1049 #undef __FUNCT__ 1050 #define __FUNCT__ "MatGetInfo_MKL_PARDISO" 1051 PetscErrorCode MatGetInfo_MKL_PARDISO(Mat A, MatInfoType flag, MatInfo *info) 1052 { 1053 Mat_MKL_PARDISO *mat_mkl_pardiso =(Mat_MKL_PARDISO*)A->spptr; 1054 1055 PetscFunctionBegin; 1056 info->block_size = 1.0; 1057 info->nz_allocated = mat_mkl_pardiso->nz + 0.0; 1058 info->nz_unneeded = 0.0; 1059 info->assemblies = 0.0; 1060 info->mallocs = 0.0; 1061 info->memory = 0.0; 1062 info->fill_ratio_given = 0; 1063 info->fill_ratio_needed = 0; 1064 info->factor_mallocs = 0; 1065 PetscFunctionReturn(0); 1066 } 1067 1068 #undef __FUNCT__ 1069 #define __FUNCT__ "MatMkl_PardisoSetCntl_MKL_PARDISO" 1070 PetscErrorCode MatMkl_PardisoSetCntl_MKL_PARDISO(Mat F,PetscInt icntl,PetscInt ival) 1071 { 1072 Mat_MKL_PARDISO *mat_mkl_pardiso =(Mat_MKL_PARDISO*)F->spptr; 1073 1074 PetscFunctionBegin; 1075 if(icntl <= 64){ 1076 mat_mkl_pardiso->iparm[icntl - 1] = ival; 1077 } else { 1078 if(icntl == 65) 1079 mkl_set_num_threads((int)ival); 1080 else if(icntl == 66) 1081 mat_mkl_pardiso->maxfct = ival; 1082 else if(icntl == 67) 1083 mat_mkl_pardiso->mnum = ival; 1084 else if(icntl == 68) 1085 mat_mkl_pardiso->msglvl = ival; 1086 else if(icntl == 69){ 1087 int pt[IPARM_SIZE]; 1088 mat_mkl_pardiso->mtype = ival; 1089 MKL_PARDISO_INIT(&pt, &mat_mkl_pardiso->mtype, mat_mkl_pardiso->iparm); 1090 #if defined(PETSC_USE_REAL_SINGLE) 1091 mat_mkl_pardiso->iparm[27] = 1; 1092 #else 1093 mat_mkl_pardiso->iparm[27] = 0; 1094 #endif 1095 mat_mkl_pardiso->iparm[34] = 1; 1096 } 1097 } 1098 PetscFunctionReturn(0); 1099 } 1100 1101 #undef __FUNCT__ 1102 #define __FUNCT__ "MatMkl_PardisoSetCntl" 1103 /*@ 1104 MatMkl_PardisoSetCntl - Set Mkl_Pardiso parameters 1105 1106 Logically Collective on Mat 1107 1108 Input Parameters: 1109 + F - the factored matrix obtained by calling MatGetFactor() 1110 . icntl - index of Mkl_Pardiso parameter 1111 - ival - value of Mkl_Pardiso parameter 1112 1113 Options Database: 1114 . -mat_mkl_pardiso_<icntl> <ival> 1115 1116 Level: beginner 1117 1118 References: Mkl_Pardiso Users' Guide 1119 1120 .seealso: MatGetFactor() 1121 @*/ 1122 PetscErrorCode MatMkl_PardisoSetCntl(Mat F,PetscInt icntl,PetscInt ival) 1123 { 1124 PetscErrorCode ierr; 1125 1126 PetscFunctionBegin; 1127 ierr = PetscTryMethod(F,"MatMkl_PardisoSetCntl_C",(Mat,PetscInt,PetscInt),(F,icntl,ival));CHKERRQ(ierr); 1128 PetscFunctionReturn(0); 1129 } 1130 1131 /*MC 1132 MATSOLVERMKL_PARDISO - A matrix type providing direct solvers (LU) for 1133 sequential matrices via the external package MKL_PARDISO. 1134 1135 Works with MATSEQAIJ matrices 1136 1137 Use -pc_type lu -pc_factor_mat_solver_package mkl_pardiso to us this direct solver 1138 1139 Options Database Keys: 1140 + -mat_mkl_pardiso_65 - Number of threads to use 1141 . -mat_mkl_pardiso_66 - Maximum number of factors with identical sparsity structure that must be kept in memory at the same time 1142 . -mat_mkl_pardiso_67 - Indicates the actual matrix for the solution phase 1143 . -mat_mkl_pardiso_68 - Message level information 1144 . -mat_mkl_pardiso_69 - Defines the matrix type. IMPORTANT: When you set this flag, iparm parameters are going to be set to the default ones for the matrix type 1145 . -mat_mkl_pardiso_1 - Use default values 1146 . -mat_mkl_pardiso_2 - Fill-in reducing ordering for the input matrix 1147 . -mat_mkl_pardiso_4 - Preconditioned CGS/CG 1148 . -mat_mkl_pardiso_5 - User permutation 1149 . -mat_mkl_pardiso_6 - Write solution on x 1150 . -mat_mkl_pardiso_8 - Iterative refinement step 1151 . -mat_mkl_pardiso_10 - Pivoting perturbation 1152 . -mat_mkl_pardiso_11 - Scaling vectors 1153 . -mat_mkl_pardiso_12 - Solve with transposed or conjugate transposed matrix A 1154 . -mat_mkl_pardiso_13 - Improved accuracy using (non-) symmetric weighted matching 1155 . -mat_mkl_pardiso_18 - Numbers of non-zero elements 1156 . -mat_mkl_pardiso_19 - Report number of floating point operations 1157 . -mat_mkl_pardiso_21 - Pivoting for symmetric indefinite matrices 1158 . -mat_mkl_pardiso_24 - Parallel factorization control 1159 . -mat_mkl_pardiso_25 - Parallel forward/backward solve control 1160 . -mat_mkl_pardiso_27 - Matrix checker 1161 . -mat_mkl_pardiso_31 - Partial solve and computing selected components of the solution vectors 1162 . -mat_mkl_pardiso_34 - Optimal number of threads for conditional numerical reproducibility (CNR) mode 1163 - -mat_mkl_pardiso_60 - Intel MKL_PARDISO mode 1164 1165 Level: beginner 1166 1167 For more information please check mkl_pardiso manual 1168 1169 .seealso: PCFactorSetMatSolverPackage(), MatSolverPackage 1170 1171 M*/ 1172 #undef __FUNCT__ 1173 #define __FUNCT__ "MatFactorGetSolverPackage_mkl_pardiso" 1174 static PetscErrorCode MatFactorGetSolverPackage_mkl_pardiso(Mat A, const MatSolverPackage *type) 1175 { 1176 PetscFunctionBegin; 1177 *type = MATSOLVERMKL_PARDISO; 1178 PetscFunctionReturn(0); 1179 } 1180 1181 /* MatGetFactor for Seq sbAIJ matrices */ 1182 #undef __FUNCT__ 1183 #define __FUNCT__ "MatGetFactor_sbaij_mkl_pardiso" 1184 PETSC_EXTERN PetscErrorCode MatGetFactor_sbaij_mkl_pardiso(Mat A,MatFactorType ftype,Mat *F) 1185 { 1186 Mat B; 1187 PetscErrorCode ierr; 1188 Mat_MKL_PARDISO *mat_mkl_pardiso; 1189 PetscBool isSeqSBAIJ; 1190 PetscInt bs; 1191 1192 PetscFunctionBegin; 1193 /* Create the factorization matrix */ 1194 ierr = PetscObjectTypeCompare((PetscObject)A,MATSEQSBAIJ,&isSeqSBAIJ);CHKERRQ(ierr); 1195 ierr = MatCreate(PetscObjectComm((PetscObject)A),&B);CHKERRQ(ierr); 1196 ierr = MatSetSizes(B,A->rmap->n,A->cmap->n,A->rmap->N,A->cmap->N);CHKERRQ(ierr); 1197 ierr = MatSetType(B,((PetscObject)A)->type_name);CHKERRQ(ierr); 1198 ierr = MatSeqSBAIJSetPreallocation(B,1,0,NULL);CHKERRQ(ierr); 1199 ierr = MatGetBlockSize(A,&bs); CHKERRQ(ierr); 1200 1201 if(bs != 1) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Matrice MATSEQSBAIJ with block size other than 1 is not supported by Pardiso"); 1202 if(ftype != MAT_FACTOR_CHOLESKY) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Matrice MATSEQAIJ should be used only with MAT_FACTOR_CHOLESKY."); 1203 1204 B->ops->choleskyfactorsymbolic = MatCholeskyFactorSymbolic_AIJMKL_PARDISO; 1205 B->factortype = MAT_FACTOR_CHOLESKY; 1206 B->ops->destroy = MatDestroy_MKL_PARDISO; 1207 B->ops->view = MatView_MKL_PARDISO; 1208 B->factortype = ftype; 1209 B->ops->getinfo = MatGetInfo_MKL_PARDISO; 1210 B->assembled = PETSC_TRUE; /* required by -ksp_view */ 1211 1212 ierr = PetscNewLog(B,&mat_mkl_pardiso);CHKERRQ(ierr); 1213 B->spptr = mat_mkl_pardiso; 1214 ierr = PetscObjectComposeFunction((PetscObject)B,"MatFactorGetSolverPackage_C",MatFactorGetSolverPackage_mkl_pardiso);CHKERRQ(ierr); 1215 ierr = PetscObjectComposeFunction((PetscObject)B,"MatFactorSetSchurIS_C",MatFactorSetSchurIS_MKL_PARDISO);CHKERRQ(ierr); 1216 ierr = PetscObjectComposeFunction((PetscObject)B,"MatFactorCreateSchurComplement_C",MatFactorCreateSchurComplement_MKL_PARDISO);CHKERRQ(ierr); 1217 ierr = PetscObjectComposeFunction((PetscObject)B,"MatFactorGetSchurComplement_C",MatFactorGetSchurComplement_MKL_PARDISO);CHKERRQ(ierr); 1218 ierr = PetscObjectComposeFunction((PetscObject)B,"MatFactorInvertSchurComplement_C",MatFactorInvertSchurComplement_MKL_PARDISO);CHKERRQ(ierr); 1219 ierr = PetscObjectComposeFunction((PetscObject)B,"MatFactorSolveSchurComplement_C",MatFactorSolveSchurComplement_MKL_PARDISO);CHKERRQ(ierr); 1220 ierr = PetscObjectComposeFunction((PetscObject)B,"MatFactorSolveSchurComplementTranspose_C",MatFactorSolveSchurComplementTranspose_MKL_PARDISO);CHKERRQ(ierr); 1221 ierr = PetscObjectComposeFunction((PetscObject)B,"MatMkl_PardisoSetCntl_C",MatMkl_PardisoSetCntl_MKL_PARDISO);CHKERRQ(ierr); 1222 ierr = MatFactorMKL_PARDISOInitialize_Private(A, ftype, mat_mkl_pardiso);CHKERRQ(ierr); 1223 *F = B; 1224 PetscFunctionReturn(0); 1225 } 1226 1227 /* MatGetFactor for Seq AIJ matrices */ 1228 #undef __FUNCT__ 1229 #define __FUNCT__ "MatGetFactor_aij_mkl_pardiso" 1230 PETSC_EXTERN PetscErrorCode MatGetFactor_aij_mkl_pardiso(Mat A,MatFactorType ftype,Mat *F) 1231 { 1232 Mat B; 1233 PetscErrorCode ierr; 1234 Mat_MKL_PARDISO *mat_mkl_pardiso; 1235 PetscBool isSeqAIJ; 1236 1237 PetscFunctionBegin; 1238 /* Create the factorization matrix */ 1239 ierr = PetscObjectTypeCompare((PetscObject)A,MATSEQAIJ,&isSeqAIJ);CHKERRQ(ierr); 1240 ierr = MatCreate(PetscObjectComm((PetscObject)A),&B);CHKERRQ(ierr); 1241 ierr = MatSetSizes(B,A->rmap->n,A->cmap->n,A->rmap->N,A->cmap->N);CHKERRQ(ierr); 1242 ierr = MatSetType(B,((PetscObject)A)->type_name);CHKERRQ(ierr); 1243 ierr = MatSeqAIJSetPreallocation(B,0,NULL);CHKERRQ(ierr); 1244 1245 if (ftype == MAT_FACTOR_LU) { 1246 B->ops->lufactorsymbolic = MatLUFactorSymbolic_AIJMKL_PARDISO; 1247 B->factortype = MAT_FACTOR_LU; 1248 } else { /* this is still broken */ 1249 B->ops->choleskyfactorsymbolic = MatCholeskyFactorSymbolic_AIJMKL_PARDISO; 1250 B->factortype = MAT_FACTOR_CHOLESKY; 1251 } 1252 B->ops->destroy = MatDestroy_MKL_PARDISO; 1253 B->ops->view = MatView_MKL_PARDISO; 1254 B->factortype = ftype; 1255 B->ops->getinfo = MatGetInfo_MKL_PARDISO; 1256 B->assembled = PETSC_TRUE; /* required by -ksp_view */ 1257 1258 ierr = PetscNewLog(B,&mat_mkl_pardiso);CHKERRQ(ierr); 1259 B->spptr = mat_mkl_pardiso; 1260 ierr = PetscObjectComposeFunction((PetscObject)B,"MatFactorGetSolverPackage_C",MatFactorGetSolverPackage_mkl_pardiso);CHKERRQ(ierr); 1261 ierr = PetscObjectComposeFunction((PetscObject)B,"MatFactorSetSchurIS_C",MatFactorSetSchurIS_MKL_PARDISO);CHKERRQ(ierr); 1262 ierr = PetscObjectComposeFunction((PetscObject)B,"MatFactorCreateSchurComplement_C",MatFactorCreateSchurComplement_MKL_PARDISO);CHKERRQ(ierr); 1263 ierr = PetscObjectComposeFunction((PetscObject)B,"MatFactorGetSchurComplement_C",MatFactorGetSchurComplement_MKL_PARDISO);CHKERRQ(ierr); 1264 ierr = PetscObjectComposeFunction((PetscObject)B,"MatFactorInvertSchurComplement_C",MatFactorInvertSchurComplement_MKL_PARDISO);CHKERRQ(ierr); 1265 ierr = PetscObjectComposeFunction((PetscObject)B,"MatFactorSolveSchurComplement_C",MatFactorSolveSchurComplement_MKL_PARDISO);CHKERRQ(ierr); 1266 ierr = PetscObjectComposeFunction((PetscObject)B,"MatFactorSolveSchurComplementTranspose_C",MatFactorSolveSchurComplementTranspose_MKL_PARDISO);CHKERRQ(ierr); 1267 ierr = PetscObjectComposeFunction((PetscObject)B,"MatMkl_PardisoSetCntl_C",MatMkl_PardisoSetCntl_MKL_PARDISO);CHKERRQ(ierr); 1268 ierr = MatFactorMKL_PARDISOInitialize_Private(A, ftype, mat_mkl_pardiso);CHKERRQ(ierr); 1269 1270 *F = B; 1271 PetscFunctionReturn(0); 1272 } 1273 1274 #undef __FUNCT__ 1275 #define __FUNCT__ "MatSolverPackageRegister_MKL_Pardiso" 1276 PETSC_EXTERN PetscErrorCode MatSolverPackageRegister_MKL_Pardiso(void) 1277 { 1278 PetscErrorCode ierr; 1279 1280 PetscFunctionBegin; 1281 ierr = MatSolverPackageRegister(MATSOLVERMKL_PARDISO,MATSEQAIJ,MAT_FACTOR_LU,MatGetFactor_aij_mkl_pardiso);CHKERRQ(ierr); 1282 ierr = MatSolverPackageRegister(MATSOLVERMKL_PARDISO,MATSEQSBAIJ,MAT_FACTOR_CHOLESKY,MatGetFactor_sbaij_mkl_pardiso);CHKERRQ(ierr); 1283 PetscFunctionReturn(0); 1284 } 1285 1286