1 #include <../src/mat/impls/elemental/matelemimpl.h> /*I "petscmat.h" I*/ 2 3 /* 4 The variable Petsc_Elemental_keyval is used to indicate an MPI attribute that 5 is attached to a communicator, in this case the attribute is a Mat_Elemental_Grid 6 */ 7 static PetscMPIInt Petsc_Elemental_keyval = MPI_KEYVAL_INVALID; 8 9 #undef __FUNCT__ 10 #define __FUNCT__ "PetscElementalInitializePackage" 11 /*@C 12 PetscElementalInitializePackage - Initialize Elemental package 13 14 Logically Collective 15 16 Input Arguments: 17 . path - the dynamic library path or PETSC_NULL 18 19 Level: developer 20 21 .seealso: MATELEMENTAL, PetscElementalFinalizePackage() 22 @*/ 23 PetscErrorCode PetscElementalInitializePackage(const char *path) 24 { 25 PetscErrorCode ierr; 26 27 PetscFunctionBegin; 28 if (elem::Initialized()) PetscFunctionReturn(0); 29 { /* We have already initialized MPI, so this song and dance is just to pass these variables (which won't be used by Elemental) through the interface that needs references */ 30 int zero = 0; 31 char **nothing = 0; 32 elem::Initialize(zero,nothing); 33 } 34 ierr = PetscRegisterFinalize(PetscElementalFinalizePackage);CHKERRQ(ierr); 35 PetscFunctionReturn(0); 36 } 37 38 #undef __FUNCT__ 39 #define __FUNCT__ "PetscElementalFinalizePackage" 40 /*@C 41 PetscElementalFinalizePackage - Finalize Elemental package 42 43 Logically Collective 44 45 Level: developer 46 47 .seealso: MATELEMENTAL, PetscElementalInitializePackage() 48 @*/ 49 PetscErrorCode PetscElementalFinalizePackage(void) 50 { 51 52 PetscFunctionBegin; 53 elem::Finalize(); 54 PetscFunctionReturn(0); 55 } 56 57 /* Sets Elemental options from the options database */ 58 #undef __FUNCT__ 59 #define __FUNCT__ "PetscSetElementalFromOptions" 60 PetscErrorCode PetscSetElementalFromOptions(Mat A) 61 { 62 //Mat_Elemental *a = (Mat_Elemental*)A->data; 63 PetscErrorCode ierr; 64 //PetscInt optv1,optv2; 65 //PetscBool flg; 66 67 PetscFunctionBegin; 68 ierr = PetscOptionsBegin(((PetscObject)A)->comm,((PetscObject)A)->prefix,"Elemental Options","Mat");CHKERRQ(ierr); 69 //ierr = PetscOptionsInt("-mat_elemental_grid_height","Gird Height","None",(*a->emat).Grid().Height(),&optv1,&flg);CHKERRQ(ierr); 70 //ierr = PetscOptionsInt("-mat_elemental_grid_width","Gird Width","None",(*a->emat).Grid().Width(),&optv2,&flg);CHKERRQ(ierr); 71 //if (flg) (*a->emat).Grid().ResizeTo(optv1,optv2); 72 // ierr = PetscOptionsInt("-mat_mumps_icntl_2","ICNTL(2): output stream for diagnostic printing, statistics, and warning","None",mumps->id.ICNTL(2),&icntl,&flg);CHKERRQ(ierr); 73 // if (flg) mumps->id.ICNTL(2) = icntl; 74 // ierr = PetscOptionsInt("-mat_mumps_icntl_3","ICNTL(3): output stream for global information, collected on the host","None",mumps->id.ICNTL(3),&icntl,&flg);CHKERRQ(ierr); 75 // if (flg) mumps->id.ICNTL(3) = icntl; 76 77 78 PetscOptionsEnd(); 79 PetscFunctionReturn(0); 80 } 81 82 #undef __FUNCT__ 83 #define __FUNCT__ "MatView_Elemental" 84 static PetscErrorCode MatView_Elemental(Mat A,PetscViewer viewer) 85 { 86 PetscErrorCode ierr; 87 Mat_Elemental *a = (Mat_Elemental*)A->data; 88 PetscBool iascii; 89 90 PetscFunctionBegin; 91 ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERASCII,&iascii);CHKERRQ(ierr); 92 if (iascii) { 93 PetscViewerFormat format; 94 ierr = PetscViewerGetFormat(viewer,&format);CHKERRQ(ierr); 95 if (format == PETSC_VIEWER_ASCII_INFO) { 96 /* call elemental viewing function */ 97 ierr = PetscViewerASCIIPrintf(viewer,"Elemental run parameters:\n");CHKERRQ(ierr); 98 ierr = PetscViewerASCIIPrintf(viewer," allocated entries=%d\n",(*a->emat).AllocatedMemory());CHKERRQ(ierr); 99 ierr = PetscViewerASCIIPrintf(viewer," grid height=%d, grid width=%d\n",(*a->emat).Grid().Height(),(*a->emat).Grid().Width());CHKERRQ(ierr); 100 if (format == PETSC_VIEWER_ASCII_FACTOR_INFO) { 101 /* call elemental viewing function */ 102 ierr = PetscPrintf(((PetscObject)viewer)->comm,"test matview_elemental 2\n");CHKERRQ(ierr); 103 } 104 105 } else if (format == PETSC_VIEWER_DEFAULT) { 106 ierr = PetscViewerASCIIUseTabs(viewer,PETSC_FALSE);CHKERRQ(ierr); 107 ierr = PetscObjectPrintClassNamePrefixType((PetscObject)A,viewer,"Matrix Object");CHKERRQ(ierr); 108 a->emat->Print("Elemental matrix (cyclic ordering)"); 109 ierr = PetscViewerASCIIUseTabs(viewer,PETSC_TRUE);CHKERRQ(ierr); 110 if (A->factortype == MAT_FACTOR_NONE){ 111 Mat Aaij; 112 ierr = PetscPrintf(((PetscObject)viewer)->comm,"Elemental matrix (explicit ordering)\n");CHKERRQ(ierr); 113 ierr = MatComputeExplicitOperator(A,&Aaij);CHKERRQ(ierr); 114 ierr = MatView(Aaij,viewer);CHKERRQ(ierr); 115 ierr = MatDestroy(&Aaij);CHKERRQ(ierr); 116 } 117 } else SETERRQ(((PetscObject)viewer)->comm,PETSC_ERR_SUP,"Format"); 118 } else { 119 /* convert to aij/mpidense format and call MatView() */ 120 Mat Aaij; 121 ierr = PetscPrintf(((PetscObject)viewer)->comm,"Elemental matrix (explicit ordering)\n");CHKERRQ(ierr); 122 ierr = MatComputeExplicitOperator(A,&Aaij);CHKERRQ(ierr); 123 ierr = MatView(Aaij,viewer);CHKERRQ(ierr); 124 ierr = MatDestroy(&Aaij);CHKERRQ(ierr); 125 } 126 PetscFunctionReturn(0); 127 } 128 129 #undef __FUNCT__ 130 #define __FUNCT__ "MatGetInfo_Elemental" 131 static PetscErrorCode MatGetInfo_Elemental(Mat A,MatInfoType flag,MatInfo *info) 132 { 133 Mat_Elemental *a = (Mat_Elemental*)A->data; 134 PetscMPIInt rank; 135 136 PetscFunctionBegin; 137 MPI_Comm_rank(((PetscObject)A)->comm,&rank); 138 139 /* if (!rank) printf(" .........MatGetInfo_Elemental ...\n"); */ 140 info->block_size = 1.0; /* ? */ 141 142 if (flag == MAT_LOCAL) { 143 info->nz_allocated = (double)(*a->emat).AllocatedMemory(); /* locally allocated */ 144 info->nz_used = info->nz_allocated; 145 } else if (flag == MAT_GLOBAL_MAX) { 146 //ierr = MPI_Allreduce(isend,irecv,5,MPIU_REAL,MPIU_MAX,((PetscObject)matin)->comm);CHKERRQ(ierr); 147 /* see MatGetInfo_MPIAIJ() for getting global info->nz_allocated! */ 148 //SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP," MAT_GLOBAL_MAX not written yet"); 149 } else if (flag == MAT_GLOBAL_SUM) { 150 //SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP," MAT_GLOBAL_SUM not written yet"); 151 info->nz_allocated = (double)(*a->emat).AllocatedMemory(); /* locally allocated */ 152 info->nz_used = info->nz_allocated; /* assume Elemental does accurate allocation */ 153 //ierr = MPI_Allreduce(isend,irecv,1,MPIU_REAL,MPIU_SUM,((PetscObject)A)->comm);CHKERRQ(ierr); 154 //PetscPrintf(PETSC_COMM_SELF," ... [%d] locally allocated %g\n",rank,info->nz_allocated); 155 } 156 157 info->nz_unneeded = 0.0; 158 info->assemblies = (double)A->num_ass; 159 info->mallocs = 0; 160 info->memory = ((PetscObject)A)->mem; 161 info->fill_ratio_given = 0; /* determined by Elemental */ 162 info->fill_ratio_needed = 0; 163 info->factor_mallocs = 0; 164 PetscFunctionReturn(0); 165 } 166 167 #undef __FUNCT__ 168 #define __FUNCT__ "MatSetValues_Elemental" 169 static PetscErrorCode MatSetValues_Elemental(Mat A,PetscInt nr,const PetscInt *rows,PetscInt nc,const PetscInt *cols,const PetscScalar *vals,InsertMode imode) 170 { 171 PetscErrorCode ierr; 172 Mat_Elemental *a = (Mat_Elemental*)A->data; 173 PetscMPIInt rank; 174 PetscInt i,j,rrank,ridx,crank,cidx; 175 176 PetscFunctionBegin; 177 ierr = MPI_Comm_rank(((PetscObject)A)->comm,&rank);CHKERRQ(ierr); 178 179 const elem::Grid &grid = a->emat->Grid(); 180 for (i=0; i<nr; i++) { 181 PetscInt erow,ecol,elrow,elcol; 182 if (rows[i] < 0) continue; 183 P2RO(A,0,rows[i],&rrank,&ridx); 184 RO2E(A,0,rrank,ridx,&erow); 185 if (rrank < 0 || ridx < 0 || erow < 0) SETERRQ(((PetscObject)A)->comm,PETSC_ERR_PLIB,"Incorrect row translation"); 186 for (j=0; j<nc; j++) { 187 if (cols[j] < 0) continue; 188 P2RO(A,1,cols[j],&crank,&cidx); 189 RO2E(A,1,crank,cidx,&ecol); 190 if (crank < 0 || cidx < 0 || ecol < 0) SETERRQ(((PetscObject)A)->comm,PETSC_ERR_PLIB,"Incorrect col translation"); 191 if (erow % grid.MCSize() != grid.MCRank() || ecol % grid.MRSize() != grid.MRRank()){ /* off-proc entry */ 192 if (imode != ADD_VALUES) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Only ADD_VALUES to off-processor entry is supported"); 193 /* PetscPrintf(PETSC_COMM_SELF,"[%D] add off-proc entry (%D,%D, %g) (%D %D)\n",rank,rows[i],cols[j],*(vals+i*nc),erow,ecol); */ 194 a->esubmat->Set(0,0, (PetscElemScalar)vals[i*nc+j]); 195 a->interface->Axpy(1.0,*(a->esubmat),erow,ecol); 196 continue; 197 } 198 elrow = erow / grid.MCSize(); 199 elcol = ecol / grid.MRSize(); 200 switch (imode) { 201 case INSERT_VALUES: a->emat->SetLocal(elrow,elcol,(PetscElemScalar)vals[i*nc+j]); break; 202 case ADD_VALUES: a->emat->UpdateLocal(elrow,elcol,(PetscElemScalar)vals[i*nc+j]); break; 203 default: SETERRQ1(((PetscObject)A)->comm,PETSC_ERR_SUP,"No support for InsertMode %d",(int)imode); 204 } 205 } 206 } 207 PetscFunctionReturn(0); 208 } 209 210 #undef __FUNCT__ 211 #define __FUNCT__ "MatMult_Elemental" 212 static PetscErrorCode MatMult_Elemental(Mat A,Vec X,Vec Y) 213 { 214 Mat_Elemental *a = (Mat_Elemental*)A->data; 215 PetscErrorCode ierr; 216 const PetscElemScalar *x; 217 PetscElemScalar *y; 218 PetscElemScalar one = 1,zero = 0; 219 220 PetscFunctionBegin; 221 ierr = VecGetArrayRead(X,(const PetscScalar **)&x);CHKERRQ(ierr); 222 ierr = VecGetArray(Y,(PetscScalar **)&y);CHKERRQ(ierr); 223 { /* Scoping so that constructor is called before pointer is returned */ 224 elem::DistMatrix<PetscElemScalar,elem::VC,elem::STAR> xe(A->cmap->N,1,0,x,A->cmap->n,*a->grid); 225 elem::DistMatrix<PetscElemScalar,elem::VC,elem::STAR> ye(A->rmap->N,1,0,y,A->rmap->n,*a->grid); 226 elem::Gemv(elem::NORMAL,one,*a->emat,xe,zero,ye); 227 } 228 ierr = VecRestoreArrayRead(X,(const PetscScalar **)&x);CHKERRQ(ierr); 229 ierr = VecRestoreArray(Y,(PetscScalar **)&y);CHKERRQ(ierr); 230 PetscFunctionReturn(0); 231 } 232 233 #undef __FUNCT__ 234 #define __FUNCT__ "MatMultTranspose_Elemental" 235 static PetscErrorCode MatMultTranspose_Elemental(Mat A,Vec X,Vec Y) 236 { 237 Mat_Elemental *a = (Mat_Elemental*)A->data; 238 PetscErrorCode ierr; 239 const PetscElemScalar *x; 240 PetscElemScalar *y; 241 PetscElemScalar one = 1,zero = 0; 242 243 PetscFunctionBegin; 244 ierr = VecGetArrayRead(X,(const PetscScalar **)&x);CHKERRQ(ierr); 245 ierr = VecGetArray(Y,(PetscScalar **)&y);CHKERRQ(ierr); 246 { /* Scoping so that constructor is called before pointer is returned */ 247 elem::DistMatrix<PetscElemScalar,elem::VC,elem::STAR> xe(A->rmap->N,1,0,x,A->rmap->n,*a->grid); 248 elem::DistMatrix<PetscElemScalar,elem::VC,elem::STAR> ye(A->cmap->N,1,0,y,A->cmap->n,*a->grid); 249 elem::Gemv(elem::TRANSPOSE,one,*a->emat,xe,zero,ye); 250 } 251 ierr = VecRestoreArrayRead(X,(const PetscScalar **)&x);CHKERRQ(ierr); 252 ierr = VecRestoreArray(Y,(PetscScalar **)&y);CHKERRQ(ierr); 253 PetscFunctionReturn(0); 254 } 255 256 #undef __FUNCT__ 257 #define __FUNCT__ "MatMultAdd_Elemental" 258 static PetscErrorCode MatMultAdd_Elemental(Mat A,Vec X,Vec Y,Vec Z) 259 { 260 Mat_Elemental *a = (Mat_Elemental*)A->data; 261 PetscErrorCode ierr; 262 const PetscElemScalar *x; 263 PetscElemScalar *z; 264 PetscElemScalar one = 1; 265 266 PetscFunctionBegin; 267 if (Y != Z) {ierr = VecCopy(Y,Z);CHKERRQ(ierr);} 268 ierr = VecGetArrayRead(X,(const PetscScalar **)&x);CHKERRQ(ierr); 269 ierr = VecGetArray(Z,(PetscScalar **)&z);CHKERRQ(ierr); 270 { /* Scoping so that constructor is called before pointer is returned */ 271 elem::DistMatrix<PetscElemScalar,elem::VC,elem::STAR> xe(A->cmap->N,1,0,x,A->cmap->n,*a->grid); 272 elem::DistMatrix<PetscElemScalar,elem::VC,elem::STAR> ze(A->rmap->N,1,0,z,A->rmap->n,*a->grid); 273 elem::Gemv(elem::NORMAL,one,*a->emat,xe,one,ze); 274 } 275 ierr = VecRestoreArrayRead(X,(const PetscScalar **)&x);CHKERRQ(ierr); 276 ierr = VecRestoreArray(Z,(PetscScalar **)&z);CHKERRQ(ierr); 277 PetscFunctionReturn(0); 278 } 279 280 #undef __FUNCT__ 281 #define __FUNCT__ "MatMultTransposeAdd_Elemental" 282 static PetscErrorCode MatMultTransposeAdd_Elemental(Mat A,Vec X,Vec Y,Vec Z) 283 { 284 Mat_Elemental *a = (Mat_Elemental*)A->data; 285 PetscErrorCode ierr; 286 const PetscElemScalar *x; 287 PetscElemScalar *z; 288 PetscElemScalar one = 1; 289 290 PetscFunctionBegin; 291 if (Y != Z) {ierr = VecCopy(Y,Z);CHKERRQ(ierr);} 292 ierr = VecGetArrayRead(X,(const PetscScalar **)&x);CHKERRQ(ierr); 293 ierr = VecGetArray(Z,(PetscScalar **)&z);CHKERRQ(ierr); 294 { /* Scoping so that constructor is called before pointer is returned */ 295 elem::DistMatrix<PetscElemScalar,elem::VC,elem::STAR> xe(A->rmap->N,1,0,x,A->rmap->n,*a->grid); 296 elem::DistMatrix<PetscElemScalar,elem::VC,elem::STAR> ze(A->cmap->N,1,0,z,A->cmap->n,*a->grid); 297 elem::Gemv(elem::TRANSPOSE,one,*a->emat,xe,one,ze); 298 } 299 ierr = VecRestoreArrayRead(X,(const PetscScalar **)&x);CHKERRQ(ierr); 300 ierr = VecRestoreArray(Z,(PetscScalar **)&z);CHKERRQ(ierr); 301 PetscFunctionReturn(0); 302 } 303 304 #undef __FUNCT__ 305 #define __FUNCT__ "MatMatMultNumeric_Elemental" 306 static PetscErrorCode MatMatMultNumeric_Elemental(Mat A,Mat B,Mat C) 307 { 308 Mat_Elemental *a = (Mat_Elemental*)A->data; 309 Mat_Elemental *b = (Mat_Elemental*)B->data; 310 Mat_Elemental *c = (Mat_Elemental*)C->data; 311 PetscElemScalar one = 1,zero = 0; 312 313 PetscFunctionBegin; 314 { /* Scoping so that constructor is called before pointer is returned */ 315 elem::Gemm(elem::NORMAL,elem::NORMAL,one,*a->emat,*b->emat,zero,*c->emat); 316 } 317 C->assembled = PETSC_TRUE; 318 PetscFunctionReturn(0); 319 } 320 321 #undef __FUNCT__ 322 #define __FUNCT__ "MatMatMultSymbolic_Elemental" 323 static PetscErrorCode MatMatMultSymbolic_Elemental(Mat A,Mat B,PetscReal fill,Mat *C) 324 { 325 PetscErrorCode ierr; 326 Mat Ce; 327 MPI_Comm comm=((PetscObject)A)->comm; 328 329 PetscFunctionBegin; 330 ierr = MatCreate(comm,&Ce);CHKERRQ(ierr); 331 ierr = MatSetSizes(Ce,A->rmap->n,B->cmap->n,PETSC_DECIDE,PETSC_DECIDE);CHKERRQ(ierr); 332 ierr = MatSetType(Ce,MATELEMENTAL);CHKERRQ(ierr); 333 ierr = MatSetUp(Ce);CHKERRQ(ierr); 334 *C = Ce; 335 PetscFunctionReturn(0); 336 } 337 338 #undef __FUNCT__ 339 #define __FUNCT__ "MatMatMult_Elemental" 340 static PetscErrorCode MatMatMult_Elemental(Mat A,Mat B,MatReuse scall,PetscReal fill,Mat *C) 341 { 342 PetscErrorCode ierr; 343 344 PetscFunctionBegin; 345 if (scall == MAT_INITIAL_MATRIX){ 346 ierr = PetscLogEventBegin(MAT_MatMultSymbolic,A,B,0,0);CHKERRQ(ierr); 347 ierr = MatMatMultSymbolic_Elemental(A,B,1.0,C);CHKERRQ(ierr); 348 ierr = PetscLogEventEnd(MAT_MatMultSymbolic,A,B,0,0);CHKERRQ(ierr); 349 } 350 ierr = PetscLogEventBegin(MAT_MatMultNumeric,A,B,0,0);CHKERRQ(ierr); 351 ierr = MatMatMultNumeric_Elemental(A,B,*C);CHKERRQ(ierr); 352 ierr = PetscLogEventEnd(MAT_MatMultNumeric,A,B,0,0);CHKERRQ(ierr); 353 PetscFunctionReturn(0); 354 } 355 356 #undef __FUNCT__ 357 #define __FUNCT__ "MatMatTransposeMultNumeric_Elemental" 358 static PetscErrorCode MatMatTransposeMultNumeric_Elemental(Mat A,Mat B,Mat C) 359 { 360 Mat_Elemental *a = (Mat_Elemental*)A->data; 361 Mat_Elemental *b = (Mat_Elemental*)B->data; 362 Mat_Elemental *c = (Mat_Elemental*)C->data; 363 PetscElemScalar one = 1,zero = 0; 364 365 PetscFunctionBegin; 366 { /* Scoping so that constructor is called before pointer is returned */ 367 elem::Gemm(elem::NORMAL,elem::TRANSPOSE,one,*a->emat,*b->emat,zero,*c->emat); 368 } 369 C->assembled = PETSC_TRUE; 370 PetscFunctionReturn(0); 371 } 372 373 #undef __FUNCT__ 374 #define __FUNCT__ "MatMatTransposeMultSymbolic_Elemental" 375 static PetscErrorCode MatMatTransposeMultSymbolic_Elemental(Mat A,Mat B,PetscReal fill,Mat *C) 376 { 377 PetscErrorCode ierr; 378 Mat Ce; 379 MPI_Comm comm=((PetscObject)A)->comm; 380 381 PetscFunctionBegin; 382 ierr = MatCreate(comm,&Ce);CHKERRQ(ierr); 383 ierr = MatSetSizes(Ce,A->rmap->n,B->rmap->n,PETSC_DECIDE,PETSC_DECIDE);CHKERRQ(ierr); 384 ierr = MatSetType(Ce,MATELEMENTAL);CHKERRQ(ierr); 385 ierr = MatSetUp(Ce);CHKERRQ(ierr); 386 *C = Ce; 387 PetscFunctionReturn(0); 388 } 389 390 #undef __FUNCT__ 391 #define __FUNCT__ "MatMatTransposeMult_Elemental" 392 static PetscErrorCode MatMatTransposeMult_Elemental(Mat A,Mat B,MatReuse scall,PetscReal fill,Mat *C) 393 { 394 PetscErrorCode ierr; 395 396 PetscFunctionBegin; 397 if (scall == MAT_INITIAL_MATRIX){ 398 ierr = PetscLogEventBegin(MAT_MatTransposeMultSymbolic,A,B,0,0);CHKERRQ(ierr); 399 ierr = MatMatMultSymbolic_Elemental(A,B,1.0,C);CHKERRQ(ierr); 400 ierr = PetscLogEventEnd(MAT_MatTransposeMultSymbolic,A,B,0,0);CHKERRQ(ierr); 401 } 402 ierr = PetscLogEventBegin(MAT_MatTransposeMultNumeric,A,B,0,0);CHKERRQ(ierr); 403 ierr = MatMatTransposeMultNumeric_Elemental(A,B,*C);CHKERRQ(ierr); 404 ierr = PetscLogEventEnd(MAT_MatTransposeMultNumeric,A,B,0,0);CHKERRQ(ierr); 405 PetscFunctionReturn(0); 406 } 407 408 #undef __FUNCT__ 409 #define __FUNCT__ "MatScale_Elemental" 410 static PetscErrorCode MatScale_Elemental(Mat X,PetscScalar a) 411 { 412 Mat_Elemental *x = (Mat_Elemental*)X->data; 413 414 PetscFunctionBegin; 415 elem::Scal((PetscElemScalar)a,*x->emat); 416 PetscFunctionReturn(0); 417 } 418 419 #undef __FUNCT__ 420 #define __FUNCT__ "MatAXPY_Elemental" 421 static PetscErrorCode MatAXPY_Elemental(Mat Y,PetscScalar a,Mat X,MatStructure str) 422 { 423 Mat_Elemental *x = (Mat_Elemental*)X->data; 424 Mat_Elemental *y = (Mat_Elemental*)Y->data; 425 426 PetscFunctionBegin; 427 elem::Axpy((PetscElemScalar)a,*x->emat,*y->emat); 428 PetscFunctionReturn(0); 429 } 430 431 #undef __FUNCT__ 432 #define __FUNCT__ "MatCopy_Elemental" 433 static PetscErrorCode MatCopy_Elemental(Mat A,Mat B,MatStructure str) 434 { 435 Mat_Elemental *a=(Mat_Elemental*)A->data; 436 Mat_Elemental *b=(Mat_Elemental*)B->data; 437 438 PetscFunctionBegin; 439 elem::Copy(*a->emat,*b->emat); 440 PetscFunctionReturn(0); 441 } 442 443 #undef __FUNCT__ 444 #define __FUNCT__ "MatDuplicate_Elemental" 445 static PetscErrorCode MatDuplicate_Elemental(Mat A,MatDuplicateOption op,Mat *B) 446 { 447 Mat Be; 448 MPI_Comm comm=((PetscObject)A)->comm; 449 Mat_Elemental *a=(Mat_Elemental*)A->data; 450 PetscErrorCode ierr; 451 452 PetscFunctionBegin; 453 ierr = MatCreate(comm,&Be);CHKERRQ(ierr); 454 ierr = MatSetSizes(Be,A->rmap->n,A->cmap->n,PETSC_DECIDE,PETSC_DECIDE);CHKERRQ(ierr); 455 ierr = MatSetType(Be,MATELEMENTAL);CHKERRQ(ierr); 456 ierr = MatSetUp(Be);CHKERRQ(ierr); 457 *B = Be; 458 if (op == MAT_COPY_VALUES) { 459 Mat_Elemental *b=(Mat_Elemental*)Be->data; 460 elem::Copy(*a->emat,*b->emat); 461 } 462 Be->assembled = PETSC_TRUE; 463 PetscFunctionReturn(0); 464 } 465 466 #undef __FUNCT__ 467 #define __FUNCT__ "MatTranspose_Elemental" 468 static PetscErrorCode MatTranspose_Elemental(Mat A,MatReuse reuse,Mat *B) 469 { 470 /* Only out-of-place supported */ 471 Mat Be; 472 PetscErrorCode ierr; 473 MPI_Comm comm=((PetscObject)A)->comm; 474 Mat_Elemental *a = (Mat_Elemental*)A->data, *b; 475 476 PetscFunctionBegin; 477 if (reuse == MAT_INITIAL_MATRIX){ 478 ierr = MatCreate(comm,&Be);CHKERRQ(ierr); 479 ierr = MatSetSizes(Be,A->cmap->n,A->rmap->n,PETSC_DECIDE,PETSC_DECIDE);CHKERRQ(ierr); 480 ierr = MatSetType(Be,MATELEMENTAL);CHKERRQ(ierr); 481 ierr = MatSetUp(Be);CHKERRQ(ierr); 482 *B = Be; 483 } 484 b = (Mat_Elemental*)Be->data; 485 elem::Transpose(*a->emat,*b->emat); 486 Be->assembled = PETSC_TRUE; 487 PetscFunctionReturn(0); 488 } 489 490 #undef __FUNCT__ 491 #define __FUNCT__ "MatSolve_Elemental" 492 static PetscErrorCode MatSolve_Elemental(Mat A,Vec B,Vec X) 493 { 494 Mat_Elemental *a = (Mat_Elemental*)A->data; 495 PetscErrorCode ierr; 496 PetscElemScalar *x; 497 498 PetscFunctionBegin; 499 ierr = VecCopy(B,X);CHKERRQ(ierr); 500 ierr = VecGetArray(X,(PetscScalar **)&x);CHKERRQ(ierr); 501 elem::DistMatrix<PetscElemScalar,elem::VC,elem::STAR> xe(A->rmap->N,1,0,x,A->rmap->n,*a->grid); 502 elem::DistMatrix<PetscElemScalar,elem::MC,elem::MR> xer = xe; 503 switch (A->factortype) { 504 case MAT_FACTOR_LU: 505 if ((*a->pivot).AllocatedMemory()) { 506 elem::SolveAfterLU(elem::NORMAL,*a->emat,*a->pivot,xer); 507 elem::Copy(xer,xe); 508 } else { 509 elem::SolveAfterLU(elem::NORMAL,*a->emat,xer); 510 elem::Copy(xer,xe); 511 } 512 break; 513 case MAT_FACTOR_CHOLESKY: 514 elem::SolveAfterCholesky(elem::UPPER,elem::NORMAL,*a->emat,xer); 515 elem::Copy(xer,xe); 516 break; 517 default: 518 SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Unfactored Matrix or Unsupported MatFactorType"); 519 break; 520 } 521 ierr = VecRestoreArray(X,(PetscScalar **)&x);CHKERRQ(ierr); 522 PetscFunctionReturn(0); 523 } 524 525 #undef __FUNCT__ 526 #define __FUNCT__ "MatSolveAdd_Elemental" 527 static PetscErrorCode MatSolveAdd_Elemental(Mat A,Vec B,Vec Y,Vec X) 528 { 529 PetscErrorCode ierr; 530 531 PetscFunctionBegin; 532 ierr = MatSolve_Elemental(A,B,X);CHKERRQ(ierr); 533 ierr = VecAXPY(X,1,Y);CHKERRQ(ierr); 534 PetscFunctionReturn(0); 535 } 536 537 #undef __FUNCT__ 538 #define __FUNCT__ "MatMatSolve_Elemental" 539 static PetscErrorCode MatMatSolve_Elemental(Mat A,Mat B,Mat X) 540 { 541 Mat_Elemental *a=(Mat_Elemental*)A->data; 542 Mat_Elemental *b=(Mat_Elemental*)B->data; 543 Mat_Elemental *x=(Mat_Elemental*)X->data; 544 545 PetscFunctionBegin; 546 elem::Copy(*b->emat,*x->emat); 547 switch (A->factortype) { 548 case MAT_FACTOR_LU: 549 if ((*a->pivot).AllocatedMemory()) { 550 elem::SolveAfterLU(elem::NORMAL,*a->emat,*a->pivot,*x->emat); 551 } else { 552 elem::SolveAfterLU(elem::NORMAL,*a->emat,*x->emat); 553 } 554 break; 555 case MAT_FACTOR_CHOLESKY: 556 elem::SolveAfterCholesky(elem::UPPER,elem::NORMAL,*a->emat,*x->emat); 557 break; 558 default: 559 SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Unfactored Matrix or Unsupported MatFactorType"); 560 break; 561 } 562 PetscFunctionReturn(0); 563 } 564 565 #undef __FUNCT__ 566 #define __FUNCT__ "MatLUFactor_Elemental" 567 static PetscErrorCode MatLUFactor_Elemental(Mat A,IS row,IS col,const MatFactorInfo *info) 568 { 569 Mat_Elemental *a = (Mat_Elemental*)A->data; 570 571 PetscFunctionBegin; 572 if (info->dtcol){ 573 elem::LU(*a->emat,*a->pivot); 574 } else { 575 elem::LU(*a->emat); 576 } 577 A->factortype = MAT_FACTOR_LU; 578 A->assembled = PETSC_TRUE; 579 PetscFunctionReturn(0); 580 } 581 582 #undef __FUNCT__ 583 #define __FUNCT__ "MatLUFactorNumeric_Elemental" 584 static PetscErrorCode MatLUFactorNumeric_Elemental(Mat F,Mat A,const MatFactorInfo *info) 585 { 586 PetscErrorCode ierr; 587 588 PetscFunctionBegin; 589 ierr = MatCopy(A,F,SAME_NONZERO_PATTERN);CHKERRQ(ierr); 590 ierr = MatLUFactor_Elemental(F,0,0,info);CHKERRQ(ierr); 591 PetscFunctionReturn(0); 592 } 593 594 #undef __FUNCT__ 595 #define __FUNCT__ "MatLUFactorSymbolic_Elemental" 596 static PetscErrorCode MatLUFactorSymbolic_Elemental(Mat F,Mat A,IS r,IS c,const MatFactorInfo *info) 597 { 598 PetscFunctionBegin; 599 /* F is create and allocated by MatGetFactor_elemental_petsc(), skip this routine. */ 600 PetscFunctionReturn(0); 601 } 602 603 #undef __FUNCT__ 604 #define __FUNCT__ "MatCholeskyFactor_Elemental" 605 static PetscErrorCode MatCholeskyFactor_Elemental(Mat A,IS perm,const MatFactorInfo *info) 606 { 607 Mat_Elemental *a = (Mat_Elemental*)A->data; 608 elem::DistMatrix<PetscElemScalar,elem::MC,elem::STAR> d; 609 610 PetscFunctionBegin; 611 elem::Cholesky(elem::UPPER,*a->emat); 612 A->factortype = MAT_FACTOR_CHOLESKY; 613 A->assembled = PETSC_TRUE; 614 PetscFunctionReturn(0); 615 } 616 617 #undef __FUNCT__ 618 #define __FUNCT__ "MatCholeskyFactorNumeric_Elemental" 619 static PetscErrorCode MatCholeskyFactorNumeric_Elemental(Mat F,Mat A,const MatFactorInfo *info) 620 { 621 PetscErrorCode ierr; 622 623 PetscFunctionBegin; 624 ierr = MatCopy(A,F,SAME_NONZERO_PATTERN);CHKERRQ(ierr); 625 ierr = MatCholeskyFactor_Elemental(F,0,info);CHKERRQ(ierr); 626 PetscFunctionReturn(0); 627 } 628 629 #undef __FUNCT__ 630 #define __FUNCT__ "MatCholeskyFactorSymbolic_Elemental" 631 static PetscErrorCode MatCholeskyFactorSymbolic_Elemental(Mat F,Mat A,IS perm,const MatFactorInfo *info) 632 { 633 PetscFunctionBegin; 634 /* F is create and allocated by MatGetFactor_elemental_petsc(), skip this routine. */ 635 PetscFunctionReturn(0); 636 } 637 638 EXTERN_C_BEGIN 639 #undef __FUNCT__ 640 #define __FUNCT__ "MatFactorGetSolverPackage_elemental_elemental" 641 PetscErrorCode MatFactorGetSolverPackage_elemental_elemental(Mat A,const MatSolverPackage *type) 642 { 643 PetscFunctionBegin; 644 *type = MATSOLVERELEMENTAL; 645 PetscFunctionReturn(0); 646 } 647 EXTERN_C_END 648 649 EXTERN_C_BEGIN 650 #undef __FUNCT__ 651 #define __FUNCT__ "MatGetFactor_elemental_elemental" 652 static PetscErrorCode MatGetFactor_elemental_elemental(Mat A,MatFactorType ftype,Mat *F) 653 { 654 Mat B; 655 PetscErrorCode ierr; 656 657 PetscFunctionBegin; 658 /* Create the factorization matrix */ 659 ierr = MatCreate(((PetscObject)A)->comm,&B);CHKERRQ(ierr); 660 ierr = MatSetSizes(B,A->rmap->n,A->cmap->n,PETSC_DECIDE,PETSC_DECIDE);CHKERRQ(ierr); 661 ierr = MatSetType(B,MATELEMENTAL);CHKERRQ(ierr); 662 ierr = MatSetUp(B);CHKERRQ(ierr); 663 B->factortype = ftype; 664 ierr = PetscObjectComposeFunctionDynamic((PetscObject)B,"MatFactorGetSolverPackage_C","MatFactorGetSolverPackage_elemental_elemental",MatFactorGetSolverPackage_elemental_elemental);CHKERRQ(ierr); 665 *F = B; 666 PetscFunctionReturn(0); 667 } 668 EXTERN_C_END 669 670 #undef __FUNCT__ 671 #define __FUNCT__ "MatNorm_Elemental" 672 static PetscErrorCode MatNorm_Elemental(Mat A,NormType type,PetscReal *nrm) 673 { 674 Mat_Elemental *a=(Mat_Elemental*)A->data; 675 676 PetscFunctionBegin; 677 switch (type){ 678 case NORM_1: 679 *nrm = elem::Norm(*a->emat,elem::ONE_NORM); 680 break; 681 case NORM_FROBENIUS: 682 *nrm = elem::Norm(*a->emat,elem::FROBENIUS_NORM); 683 break; 684 case NORM_INFINITY: 685 *nrm = elem::Norm(*a->emat,elem::INFINITY_NORM); 686 break; 687 default: 688 printf("Error: unsupported norm type!\n"); 689 } 690 PetscFunctionReturn(0); 691 } 692 693 #undef __FUNCT__ 694 #define __FUNCT__ "MatZeroEntries_Elemental" 695 static PetscErrorCode MatZeroEntries_Elemental(Mat A) 696 { 697 Mat_Elemental *a=(Mat_Elemental*)A->data; 698 699 PetscFunctionBegin; 700 elem::Zero(*a->emat); 701 PetscFunctionReturn(0); 702 } 703 704 EXTERN_C_BEGIN 705 #undef __FUNCT__ 706 #define __FUNCT__ "MatGetOwnershipIS_Elemental" 707 static PetscErrorCode MatGetOwnershipIS_Elemental(Mat A,IS *rows,IS *cols) 708 { 709 Mat_Elemental *a = (Mat_Elemental*)A->data; 710 PetscErrorCode ierr; 711 PetscInt i,m,shift,stride,*idx; 712 713 PetscFunctionBegin; 714 if (rows) { 715 m = a->emat->LocalHeight(); 716 shift = a->emat->ColShift(); 717 stride = a->emat->ColStride(); 718 ierr = PetscMalloc(m*sizeof(PetscInt),&idx);CHKERRQ(ierr); 719 for (i=0; i<m; i++) { 720 PetscInt rank,offset; 721 E2RO(A,0,shift+i*stride,&rank,&offset); 722 RO2P(A,0,rank,offset,&idx[i]); 723 } 724 ierr = ISCreateGeneral(PETSC_COMM_SELF,m,idx,PETSC_OWN_POINTER,rows);CHKERRQ(ierr); 725 } 726 if (cols) { 727 m = a->emat->LocalWidth(); 728 shift = a->emat->RowShift(); 729 stride = a->emat->RowStride(); 730 ierr = PetscMalloc(m*sizeof(PetscInt),&idx);CHKERRQ(ierr); 731 for (i=0; i<m; i++) { 732 PetscInt rank,offset; 733 E2RO(A,1,shift+i*stride,&rank,&offset); 734 RO2P(A,1,rank,offset,&idx[i]); 735 } 736 ierr = ISCreateGeneral(PETSC_COMM_SELF,m,idx,PETSC_OWN_POINTER,cols);CHKERRQ(ierr); 737 } 738 PetscFunctionReturn(0); 739 } 740 EXTERN_C_END 741 742 #undef __FUNCT__ 743 #define __FUNCT__ "MatConvert_Elemental_Dense" 744 static PetscErrorCode MatConvert_Elemental_Dense(Mat A,const MatType newtype,MatReuse reuse,Mat *B) 745 { 746 Mat Bmpi; 747 Mat_Elemental *a = (Mat_Elemental*)A->data; 748 MPI_Comm comm=((PetscObject)A)->comm; 749 PetscErrorCode ierr; 750 PetscInt rrank,ridx,crank,cidx,nrows,ncols,i,j; 751 PetscElemScalar v; 752 753 PetscFunctionBegin; 754 if (strcmp(newtype,MATDENSE) && strcmp(newtype,MATSEQDENSE) && strcmp(newtype,MATMPIDENSE)) { 755 SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Unsupported New MatType: must be MATDENSE, MATSEQDENSE or MATMPIDENSE"); 756 } 757 ierr = MatCreate(comm,&Bmpi);CHKERRQ(ierr); 758 ierr = MatSetSizes(Bmpi,A->rmap->n,A->cmap->n,PETSC_DECIDE,PETSC_DECIDE);CHKERRQ(ierr); 759 ierr = MatSetType(Bmpi,MATDENSE);CHKERRQ(ierr); 760 ierr = MatSetUp(Bmpi);CHKERRQ(ierr); 761 ierr = MatGetSize(A,&nrows,&ncols);CHKERRQ(ierr); 762 for (i=0; i<nrows; i++) { 763 PetscInt erow,ecol; 764 P2RO(A,0,i,&rrank,&ridx); 765 RO2E(A,0,rrank,ridx,&erow); 766 if (rrank < 0 || ridx < 0 || erow < 0) SETERRQ(comm,PETSC_ERR_PLIB,"Incorrect row translation"); 767 for (j=0; j<ncols; j++) { 768 P2RO(A,1,j,&crank,&cidx); 769 RO2E(A,1,crank,cidx,&ecol); 770 if (crank < 0 || cidx < 0 || ecol < 0) SETERRQ(comm,PETSC_ERR_PLIB,"Incorrect col translation"); 771 v = a->emat->Get(erow,ecol); 772 ierr = MatSetValues(Bmpi,1,&i,1,&j,(PetscScalar *)&v,INSERT_VALUES);CHKERRQ(ierr); 773 } 774 } 775 ierr = MatAssemblyBegin(Bmpi,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 776 ierr = MatAssemblyEnd(Bmpi,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 777 if (reuse == MAT_REUSE_MATRIX) { 778 ierr = MatHeaderReplace(A,Bmpi);CHKERRQ(ierr); 779 } else { 780 *B = Bmpi; 781 } 782 PetscFunctionReturn(0); 783 } 784 785 #undef __FUNCT__ 786 #define __FUNCT__ "MatDestroy_Elemental" 787 static PetscErrorCode MatDestroy_Elemental(Mat A) 788 { 789 Mat_Elemental *a = (Mat_Elemental*)A->data; 790 PetscErrorCode ierr; 791 Mat_Elemental_Grid *commgrid; 792 PetscBool flg; 793 MPI_Comm icomm; 794 795 PetscFunctionBegin; 796 a->interface->Detach(); 797 delete a->interface; 798 delete a->esubmat; 799 delete a->emat; 800 801 elem::mpi::Comm cxxcomm(((PetscObject)A)->comm); 802 ierr = PetscCommDuplicate(cxxcomm,&icomm,PETSC_NULL);CHKERRQ(ierr); 803 ierr = MPI_Attr_get(icomm,Petsc_Elemental_keyval,(void**)&commgrid,(int*)&flg);CHKERRQ(ierr); 804 if (--commgrid->grid_refct == 0) { 805 delete commgrid->grid; 806 ierr = PetscFree(commgrid);CHKERRQ(ierr); 807 } 808 ierr = PetscCommDestroy(&icomm);CHKERRQ(ierr); 809 ierr = PetscObjectComposeFunctionDynamic((PetscObject)A,"MatGetOwnershipIS_C","",PETSC_NULL);CHKERRQ(ierr); 810 ierr = PetscObjectComposeFunctionDynamic((PetscObject)A,"MatGetFactor_petsc_C","",PETSC_NULL);CHKERRQ(ierr); 811 ierr = PetscObjectComposeFunctionDynamic((PetscObject)A,"MatFactorGetSolverPackage_C","",PETSC_NULL);CHKERRQ(ierr); 812 ierr = PetscFree(A->data);CHKERRQ(ierr); 813 PetscFunctionReturn(0); 814 } 815 816 #undef __FUNCT__ 817 #define __FUNCT__ "MatSetUp_Elemental" 818 PetscErrorCode MatSetUp_Elemental(Mat A) 819 { 820 Mat_Elemental *a = (Mat_Elemental*)A->data; 821 PetscErrorCode ierr; 822 PetscMPIInt rsize,csize; 823 824 PetscFunctionBegin; 825 ierr = PetscLayoutSetUp(A->rmap);CHKERRQ(ierr); 826 ierr = PetscLayoutSetUp(A->cmap);CHKERRQ(ierr); 827 828 a->emat->ResizeTo(A->rmap->N,A->cmap->N);CHKERRQ(ierr); 829 elem::Zero(*a->emat); 830 831 ierr = MPI_Comm_size(A->rmap->comm,&rsize);CHKERRQ(ierr); 832 ierr = MPI_Comm_size(A->cmap->comm,&csize);CHKERRQ(ierr); 833 if (csize != rsize) SETERRQ(((PetscObject)A)->comm,PETSC_ERR_ARG_INCOMP,"Cannot use row and column communicators of different sizes"); 834 a->commsize = rsize; 835 a->mr[0] = A->rmap->N % rsize; if (!a->mr[0]) a->mr[0] = rsize; 836 a->mr[1] = A->cmap->N % csize; if (!a->mr[1]) a->mr[1] = csize; 837 a->m[0] = A->rmap->N / rsize + (a->mr[0] != rsize); 838 a->m[1] = A->cmap->N / csize + (a->mr[1] != csize); 839 PetscFunctionReturn(0); 840 } 841 842 #undef __FUNCT__ 843 #define __FUNCT__ "MatAssemblyBegin_Elemental" 844 PetscErrorCode MatAssemblyBegin_Elemental(Mat A, MatAssemblyType type) 845 { 846 Mat_Elemental *a = (Mat_Elemental*)A->data; 847 848 PetscFunctionBegin; 849 a->interface->Detach(); 850 a->interface->Attach(elem::LOCAL_TO_GLOBAL,*(a->emat)); 851 PetscFunctionReturn(0); 852 } 853 854 #undef __FUNCT__ 855 #define __FUNCT__ "MatAssemblyEnd_Elemental" 856 PetscErrorCode MatAssemblyEnd_Elemental(Mat A, MatAssemblyType type) 857 { 858 PetscFunctionBegin; 859 /* Currently does nothing */ 860 PetscFunctionReturn(0); 861 } 862 863 /* -------------------------------------------------------------------*/ 864 static struct _MatOps MatOps_Values = { 865 MatSetValues_Elemental, 866 0, 867 0, 868 MatMult_Elemental, 869 /* 4*/ MatMultAdd_Elemental, 870 MatMultTranspose_Elemental, 871 MatMultTransposeAdd_Elemental, 872 MatSolve_Elemental, 873 MatSolveAdd_Elemental, 874 0, //MatSolveTranspose_Elemental, 875 /*10*/ 0, //MatSolveTransposeAdd_Elemental, 876 MatLUFactor_Elemental, 877 MatCholeskyFactor_Elemental, 878 0, 879 MatTranspose_Elemental, 880 /*15*/ MatGetInfo_Elemental, 881 0, 882 0, 883 0, 884 MatNorm_Elemental, 885 /*20*/ MatAssemblyBegin_Elemental, 886 MatAssemblyEnd_Elemental, 887 0, //MatSetOption_Elemental, 888 MatZeroEntries_Elemental, 889 /*24*/ 0, 890 MatLUFactorSymbolic_Elemental, 891 MatLUFactorNumeric_Elemental, 892 MatCholeskyFactorSymbolic_Elemental, 893 MatCholeskyFactorNumeric_Elemental, 894 /*29*/ MatSetUp_Elemental, 895 0, 896 0, 897 0, 898 0, 899 /*34*/ MatDuplicate_Elemental, 900 0, 901 0, 902 0, 903 0, 904 /*39*/ MatAXPY_Elemental, 905 0, 906 0, 907 0, 908 MatCopy_Elemental, 909 /*44*/ 0, 910 MatScale_Elemental, 911 0, 912 0, 913 0, 914 /*49*/ 0, 915 0, 916 0, 917 0, 918 0, 919 /*54*/ 0, 920 0, 921 0, 922 0, 923 0, 924 /*59*/ 0, 925 MatDestroy_Elemental, 926 MatView_Elemental, 927 0, 928 0, 929 /*64*/ 0, 930 0, 931 0, 932 0, 933 0, 934 /*69*/ 0, 935 0, 936 MatConvert_Elemental_Dense, 937 0, 938 0, 939 /*74*/ 0, 940 0, 941 0, 942 0, 943 0, 944 /*79*/ 0, 945 0, 946 0, 947 0, 948 0, 949 /*84*/ 0, 950 0, 951 0, 952 0, 953 0, 954 /*89*/ MatMatMult_Elemental, 955 MatMatMultSymbolic_Elemental, 956 MatMatMultNumeric_Elemental, 957 0, 958 0, 959 /*94*/ 0, 960 MatMatTransposeMult_Elemental, 961 MatMatTransposeMultSymbolic_Elemental, 962 MatMatTransposeMultNumeric_Elemental, 963 0, 964 /*99*/ 0, 965 0, 966 0, 967 0, 968 0, 969 /*104*/0, 970 0, 971 0, 972 0, 973 0, 974 /*109*/MatMatSolve_Elemental, 975 0, 976 0, 977 0, 978 0, 979 /*114*/0, 980 0, 981 0, 982 0, 983 0, 984 /*119*/0, 985 0, 986 0, 987 0, 988 0, 989 /*124*/0, 990 0, 991 0, 992 0, 993 0, 994 /*129*/0, 995 0, 996 0, 997 0, 998 0, 999 /*134*/0, 1000 0, 1001 0, 1002 0, 1003 0 1004 }; 1005 1006 /*MC 1007 MATELEMENTAL = "elemental" - A matrix type for dense matrices using the Elemental package 1008 1009 Options Database Keys: 1010 . -mat_type elemental - sets the matrix type to "elemental" during a call to MatSetFromOptions() 1011 1012 Level: beginner 1013 1014 .seealso: MATDENSE,MatCreateElemental() 1015 M*/ 1016 EXTERN_C_BEGIN 1017 #undef __FUNCT__ 1018 #define __FUNCT__ "MatCreate_Elemental" 1019 PETSC_EXTERN_C PetscErrorCode MatCreate_Elemental(Mat A) 1020 { 1021 Mat_Elemental *a; 1022 PetscErrorCode ierr; 1023 PetscBool flg,flg1,flg2; 1024 Mat_Elemental_Grid *commgrid; 1025 MPI_Comm icomm; 1026 PetscInt optv1,optv2; 1027 1028 PetscFunctionBegin; 1029 ierr = PetscElementalInitializePackage(PETSC_NULL);CHKERRQ(ierr); 1030 ierr = PetscMemcpy(A->ops,&MatOps_Values,sizeof(struct _MatOps));CHKERRQ(ierr); 1031 A->insertmode = NOT_SET_VALUES; 1032 1033 ierr = PetscNewLog(A,Mat_Elemental,&a);CHKERRQ(ierr); 1034 A->data = (void*)a; 1035 1036 /* Set up the elemental matrix */ 1037 elem::mpi::Comm cxxcomm(((PetscObject)A)->comm); 1038 ierr = PetscOptionsBegin(((PetscObject)A)->comm,((PetscObject)A)->prefix,"Elemental Options","Mat");CHKERRQ(ierr); 1039 1040 /* Grid needs to be shared between multiple Mats on the same communicator, implement by attribute caching on the MPI_Comm */ 1041 if (Petsc_Elemental_keyval == MPI_KEYVAL_INVALID) { 1042 ierr = MPI_Keyval_create(MPI_NULL_COPY_FN,MPI_NULL_DELETE_FN,&Petsc_Elemental_keyval,(void*)0); 1043 } 1044 ierr = PetscCommDuplicate(cxxcomm,&icomm,PETSC_NULL);CHKERRQ(ierr); 1045 ierr = MPI_Attr_get(icomm,Petsc_Elemental_keyval,(void**)&commgrid,(int*)&flg);CHKERRQ(ierr); 1046 if (!flg) { 1047 ierr = PetscNewLog(A,Mat_Elemental_Grid,&commgrid);CHKERRQ(ierr); 1048 ierr = PetscOptionsInt("-mat_elemental_grid_height","Grid Height","None",elem::mpi::CommSize(cxxcomm),&optv1,&flg1);CHKERRQ(ierr); 1049 ierr = PetscOptionsInt("-mat_elemental_grid_width","Grid Width","None",1,&optv2,&flg2);CHKERRQ(ierr); 1050 if (flg1 || flg2) { 1051 if (optv1*optv2 != elem::mpi::CommSize(cxxcomm)) { 1052 SETERRQ(((PetscObject)A)->comm,PETSC_ERR_ARG_INCOMP,"Grid Height times Grid Width must equal CommSize"); 1053 } 1054 commgrid->grid = new elem::Grid(cxxcomm,optv1,optv2); 1055 } else { 1056 commgrid->grid = new elem::Grid(cxxcomm); 1057 } 1058 commgrid->grid_refct = 1; 1059 ierr = MPI_Attr_put(icomm,Petsc_Elemental_keyval,(void*)commgrid);CHKERRQ(ierr); 1060 } else { 1061 commgrid->grid_refct++; 1062 } 1063 ierr = PetscCommDestroy(&icomm);CHKERRQ(ierr); 1064 a->grid = commgrid->grid; 1065 a->emat = new elem::DistMatrix<PetscElemScalar>(*a->grid); 1066 a->esubmat = new elem::Matrix<PetscElemScalar>(1,1); 1067 a->interface = new elem::AxpyInterface<PetscElemScalar>; 1068 a->pivot = new elem::DistMatrix<PetscInt,elem::VC,elem::STAR>; 1069 1070 /* build cache for off array entries formed */ 1071 a->interface->Attach(elem::LOCAL_TO_GLOBAL,*(a->emat)); 1072 1073 ierr = PetscObjectComposeFunctionDynamic((PetscObject)A,"MatGetOwnershipIS_C","MatGetOwnershipIS_Elemental",MatGetOwnershipIS_Elemental);CHKERRQ(ierr); 1074 ierr = PetscObjectComposeFunctionDynamic((PetscObject)A,"MatGetFactor_elemental_C","MatGetFactor_elemental_elemental",MatGetFactor_elemental_elemental);CHKERRQ(ierr); 1075 1076 ierr = PetscObjectChangeTypeName((PetscObject)A,MATELEMENTAL);CHKERRQ(ierr); 1077 PetscOptionsEnd(); 1078 PetscFunctionReturn(0); 1079 } 1080 EXTERN_C_END 1081