Lines Matching refs:mdn
100 Mat_MPIDense *mdn = (Mat_MPIDense *)A->data; in MatGetDiagonalBlock_MPIDense() local
113 PetscCall(PetscObjectTypeCompare((PetscObject)mdn->A, MATSEQDENSECUDA, &flg)); in MatGetDiagonalBlock_MPIDense()
116 PetscCall(PetscObjectTypeCompare((PetscObject)mdn->A, MATSEQDENSEHIP, &flg)); in MatGetDiagonalBlock_MPIDense()
119 PetscCall(PetscObjectGetComm((PetscObject)mdn->A, &comm)); in MatGetDiagonalBlock_MPIDense()
122 PetscCall(MatSetType(B, ((PetscObject)mdn->A)->type_name)); in MatGetDiagonalBlock_MPIDense()
123 PetscCall(MatDenseGetArrayRead(mdn->A, (const PetscScalar **)&array)); in MatGetDiagonalBlock_MPIDense()
125 PetscCall(MatDenseRestoreArrayRead(mdn->A, (const PetscScalar **)&array)); in MatGetDiagonalBlock_MPIDense()
168 Mat_MPIDense *mdn = (Mat_MPIDense *)mat->data; in MatGetValues_MPIDense() local
180 PetscCall(MatGetValues(mdn->A, 1, &row, 1, &idxn[j], v + i * n + j)); in MatGetValues_MPIDense()
381 Mat_MPIDense *mdn = (Mat_MPIDense *)mat->data; in MatAssemblyBegin_MPIDense() local
385 if (mdn->donotstash || mat->nooffprocentries) PetscFunctionReturn(PETSC_SUCCESS); in MatAssemblyBegin_MPIDense()
389 …PetscCall(PetscInfo(mdn->A, "Stash has %" PetscInt_FMT " entries, uses %" PetscInt_FMT " mallocs.\… in MatAssemblyBegin_MPIDense()
395 Mat_MPIDense *mdn = (Mat_MPIDense *)mat->data; in MatAssemblyEnd_MPIDense() local
401 if (!mdn->donotstash && !mat->nooffprocentries) { in MatAssemblyEnd_MPIDense()
422 PetscCall(MatAssemblyBegin(mdn->A, mode)); in MatAssemblyEnd_MPIDense()
423 PetscCall(MatAssemblyEnd(mdn->A, mode)); in MatAssemblyEnd_MPIDense()
475 Mat_MPIDense *mdn = (Mat_MPIDense *)mat->data; in MatMultColumnRange_MPIDense() local
481 if (!mdn->Mvctx) PetscCall(MatSetUpMultiply_MPIDense(mat)); in MatMultColumnRange_MPIDense()
483 PetscCall(VecGetArrayWriteAndMemType(mdn->lvec, &ay, &aymtype)); in MatMultColumnRange_MPIDense()
484 …PetscCall(PetscSFBcastWithMemTypeBegin(mdn->Mvctx, MPIU_SCALAR, axmtype, ax, aymtype, ay, MPI_REPL… in MatMultColumnRange_MPIDense()
485 PetscCall(PetscSFBcastEnd(mdn->Mvctx, MPIU_SCALAR, ax, ay, MPI_REPLACE)); in MatMultColumnRange_MPIDense()
486 PetscCall(VecRestoreArrayWriteAndMemType(mdn->lvec, &ay)); in MatMultColumnRange_MPIDense()
488 …PetscUseMethod(mdn->A, "MatMultColumnRange_C", (Mat, Vec, Vec, PetscInt, PetscInt), (mdn->A, mdn->… in MatMultColumnRange_MPIDense()
494 Mat_MPIDense *mdn = (Mat_MPIDense *)mat->data; in MatMult_MPIDense() local
500 if (!mdn->Mvctx) PetscCall(MatSetUpMultiply_MPIDense(mat)); in MatMult_MPIDense()
502 PetscCall(VecGetArrayWriteAndMemType(mdn->lvec, &ay, &aymtype)); in MatMult_MPIDense()
503 …PetscCall(PetscSFBcastWithMemTypeBegin(mdn->Mvctx, MPIU_SCALAR, axmtype, ax, aymtype, ay, MPI_REPL… in MatMult_MPIDense()
504 PetscCall(PetscSFBcastEnd(mdn->Mvctx, MPIU_SCALAR, ax, ay, MPI_REPLACE)); in MatMult_MPIDense()
505 PetscCall(VecRestoreArrayWriteAndMemType(mdn->lvec, &ay)); in MatMult_MPIDense()
507 PetscCall((*mdn->A->ops->mult)(mdn->A, mdn->lvec, yy)); in MatMult_MPIDense()
513 Mat_MPIDense *mdn = (Mat_MPIDense *)mat->data; in MatMultAddColumnRange_MPIDense() local
519 if (!mdn->Mvctx) PetscCall(MatSetUpMultiply_MPIDense(mat)); in MatMultAddColumnRange_MPIDense()
521 PetscCall(VecGetArrayAndMemType(mdn->lvec, &ay, &aymtype)); in MatMultAddColumnRange_MPIDense()
522 …PetscCall(PetscSFBcastWithMemTypeBegin(mdn->Mvctx, MPIU_SCALAR, axmtype, ax, aymtype, ay, MPI_REPL… in MatMultAddColumnRange_MPIDense()
523 PetscCall(PetscSFBcastEnd(mdn->Mvctx, MPIU_SCALAR, ax, ay, MPI_REPLACE)); in MatMultAddColumnRange_MPIDense()
524 PetscCall(VecRestoreArrayAndMemType(mdn->lvec, &ay)); in MatMultAddColumnRange_MPIDense()
526 …PetscUseMethod(mdn->A, "MatMultAddColumnRange_C", (Mat, Vec, Vec, Vec, PetscInt, PetscInt), (mdn->… in MatMultAddColumnRange_MPIDense()
532 Mat_MPIDense *mdn = (Mat_MPIDense *)mat->data; in MatMultAdd_MPIDense() local
538 if (!mdn->Mvctx) PetscCall(MatSetUpMultiply_MPIDense(mat)); in MatMultAdd_MPIDense()
540 PetscCall(VecGetArrayAndMemType(mdn->lvec, &ay, &aymtype)); in MatMultAdd_MPIDense()
541 …PetscCall(PetscSFBcastWithMemTypeBegin(mdn->Mvctx, MPIU_SCALAR, axmtype, ax, aymtype, ay, MPI_REPL… in MatMultAdd_MPIDense()
542 PetscCall(PetscSFBcastEnd(mdn->Mvctx, MPIU_SCALAR, ax, ay, MPI_REPLACE)); in MatMultAdd_MPIDense()
543 PetscCall(VecRestoreArrayAndMemType(mdn->lvec, &ay)); in MatMultAdd_MPIDense()
545 PetscCall((*mdn->A->ops->multadd)(mdn->A, mdn->lvec, yy, zz)); in MatMultAdd_MPIDense()
698 Mat_MPIDense *mdn = (Mat_MPIDense *)mat->data; in MatDestroy_MPIDense() local
703 …PetscCheck(!mdn->vecinuse, PetscObjectComm((PetscObject)mat), PETSC_ERR_ORDER, "Need to call MatDe… in MatDestroy_MPIDense()
704 …PetscCheck(!mdn->matinuse, PetscObjectComm((PetscObject)mat), PETSC_ERR_ORDER, "Need to call MatDe… in MatDestroy_MPIDense()
705 PetscCall(MatDestroy(&mdn->A)); in MatDestroy_MPIDense()
706 PetscCall(VecDestroy(&mdn->lvec)); in MatDestroy_MPIDense()
707 PetscCall(PetscSFDestroy(&mdn->Mvctx)); in MatDestroy_MPIDense()
708 PetscCall(VecDestroy(&mdn->cvec)); in MatDestroy_MPIDense()
709 PetscCall(MatDestroy(&mdn->cmat)); in MatDestroy_MPIDense()
798 Mat_MPIDense *mdn = (Mat_MPIDense *)mat->data; in MatView_MPIDense_ASCIIorDraworSocket() local
820 if (mdn->Mvctx) PetscCall(PetscSFView(mdn->Mvctx, viewer)); in MatView_MPIDense_ASCIIorDraworSocket()
856 m = mdn->A->rmap->n; in MatView_MPIDense_ASCIIorDraworSocket()
895 Mat mdn = mat->A; in MatGetInfo_MPIDense() local
901 PetscCall(MatGetInfo(mdn, MAT_LOCAL, info)); in MatGetInfo_MPIDense()
975 Mat_MPIDense *mdn = (Mat_MPIDense *)A->data; in MatDiagonalScale_MPIDense() local
978 PetscInt i, j, s2a, s3a, s2, s3, m = mdn->A->rmap->n, n = mdn->A->cmap->n, lda; in MatDiagonalScale_MPIDense()
981 PetscCall(MatDenseGetArray(mdn->A, &vv)); in MatDiagonalScale_MPIDense()
982 PetscCall(MatDenseGetLDA(mdn->A, &lda)); in MatDiagonalScale_MPIDense()
1005 if (!mdn->Mvctx) PetscCall(MatSetUpMultiply_MPIDense(A)); in MatDiagonalScale_MPIDense()
1006 PetscCall(VecGetArray(mdn->lvec, &r)); in MatDiagonalScale_MPIDense()
1007 PetscCall(PetscSFBcastBegin(mdn->Mvctx, MPIU_SCALAR, ar, r, MPI_REPLACE)); in MatDiagonalScale_MPIDense()
1008 PetscCall(PetscSFBcastEnd(mdn->Mvctx, MPIU_SCALAR, ar, r, MPI_REPLACE)); in MatDiagonalScale_MPIDense()
1015 PetscCall(VecRestoreArray(mdn->lvec, &r)); in MatDiagonalScale_MPIDense()
1018 PetscCall(MatDenseRestoreArray(mdn->A, &vv)); in MatDiagonalScale_MPIDense()
1024 Mat_MPIDense *mdn = (Mat_MPIDense *)A->data; in MatNorm_MPIDense() local
1031 PetscCall(MatDenseGetArrayRead(mdn->A, &av)); in MatNorm_MPIDense()
1035 PetscCall(MatNorm(mdn->A, type, nrm)); in MatNorm_MPIDense()
1038 for (i = 0; i < mdn->A->cmap->n * mdn->A->rmap->n; i++) { in MatNorm_MPIDense()
1044 PetscCall(PetscLogFlops(2.0 * mdn->A->cmap->n * mdn->A->rmap->n)); in MatNorm_MPIDense()
1051 for (j = 0; j < mdn->A->cmap->n; j++) { in MatNorm_MPIDense()
1052 for (i = 0; i < mdn->A->rmap->n; i++) { in MatNorm_MPIDense()
1064 PetscCall(MatNorm(mdn->A, type, nrm)); in MatNorm_MPIDense()
1068 PetscCall(MatDenseRestoreArrayRead(mdn->A, &av)); in MatNorm_MPIDense()
2430 Mat_MPIDense *mdn = (Mat_MPIDense *)A->data; in MatMatMultNumeric_MPIDense_MPIDense() local
2454 …if (!mdn->Mvctx) PetscCall(MatSetUpMultiply_MPIDense(A)); /* cannot be done during the symbolic ph… in MatMatMultNumeric_MPIDense_MPIDense()
2496 …PetscCall(PetscSFBcastWithMemTypeBegin(mdn->Mvctx, vector_type, b_memtype, b_array, be_memtype, be… in MatMatMultNumeric_MPIDense_MPIDense()
2497 PetscCall(PetscSFBcastEnd(mdn->Mvctx, vector_type, b_array, be_array, MPI_REPLACE)); in MatMatMultNumeric_MPIDense_MPIDense()
2509 …PetscCall(PetscSFBcastBegin(mdn->Mvctx, MPIU_SCALAR, read + i * lda, write + i * ab->Be->rmap->n, … in MatMatMultNumeric_MPIDense_MPIDense()
2510 …PetscCall(PetscSFBcastEnd(mdn->Mvctx, MPIU_SCALAR, read + i * lda, write + i * ab->Be->rmap->n, MP… in MatMatMultNumeric_MPIDense_MPIDense()