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