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