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