Home
last modified time | relevance | path

Searched refs:pon (Results 1 – 2 of 2) sorted by relevance

/petsc/src/mat/impls/aij/mpi/
H A Dmpiptap.c701 …PetscInt i, j, jj, kk, nzi, *c_rmtj, voff, *c_othj, pn, pon, pcstart, pcend, ccstart, … in MatPtAPNumeric_MPIAIJ_MPIXAIJ_allatonce() local
722 PetscCall(MatGetLocalSize(p->B, NULL, &pon)); in MatPtAPNumeric_MPIAIJ_MPIXAIJ_allatonce()
723 pon *= dof; in MatPtAPNumeric_MPIAIJ_MPIXAIJ_allatonce()
724 PetscCall(PetscCalloc2(ptap->c_rmti[pon], &c_rmtj, ptap->c_rmti[pon], &c_rmta)); in MatPtAPNumeric_MPIAIJ_MPIXAIJ_allatonce()
727 for (i = 0; i < pon; i++) cmaxr = PetscMax(cmaxr, ptap->c_rmti[i + 1] - ptap->c_rmti[i]); in MatPtAPNumeric_MPIAIJ_MPIXAIJ_allatonce()
728 PetscCall(PetscCalloc4(cmaxr, &apindices, cmaxr, &apvalues, cmaxr, &apvaluestmp, pon, &c_rmtc)); in MatPtAPNumeric_MPIAIJ_MPIXAIJ_allatonce()
731 for (i = 0; i < am && pon; i++) { in MatPtAPNumeric_MPIAIJ_MPIXAIJ_allatonce()
852 …PetscInt i, j, jj, kk, nzi, dnzi, *c_rmtj, voff, *c_othj, pn, pon, pcstart, pcend, row… in MatPtAPNumeric_MPIAIJ_MPIXAIJ_allatonce_merged() local
872 PetscCall(MatGetLocalSize(p->B, NULL, &pon)); in MatPtAPNumeric_MPIAIJ_MPIXAIJ_allatonce_merged()
873 pon *= dof; in MatPtAPNumeric_MPIAIJ_MPIXAIJ_allatonce_merged()
[all …]
H A Dmpimatmatmult.c1598 PetscInt am = A->rmap->n, cm = C->rmap->n, pon = (p->B)->cmap->n; in MatTransposeMatMultNumeric_MPIAIJ_MPIAIJ() local
1627 PetscCall(PetscCalloc1(coi[pon], &coa)); in MatTransposeMatMultNumeric_MPIAIJ_MPIAIJ()
1776 PetscInt *ai, *aj, *Jptr, anz, *prmap = p->garray, pon, nspacedouble = 0, j; in MatTransposeMatMultSymbolic_MPIAIJ_MPIAIJ() local
1806 pon = (p->B)->cmap->n; /* total num of rows to be sent to other processors in MatTransposeMatMultSymbolic_MPIAIJ_MPIAIJ()
1808 PetscCall(PetscMalloc1(pon + 1, &coi)); in MatTransposeMatMultSymbolic_MPIAIJ_MPIAIJ()
1812 nnz = PetscRealIntMultTruncate(fill, PetscIntSumTruncate(poti[pon], ai[am])); in MatTransposeMatMultSymbolic_MPIAIJ_MPIAIJ()
1823 for (i = 0; i < pon; i++) { in MatTransposeMatMultSymbolic_MPIAIJ_MPIAIJ()
1851 PetscCall(PetscMalloc1(coi[pon], &coj)); in MatTransposeMatMultSymbolic_MPIAIJ_MPIAIJ()
1855 afill_tmp = (PetscReal)coi[pon] / (poti[pon] + ai[am] + 1); in MatTransposeMatMultSymbolic_MPIAIJ_MPIAIJ()
1879 for (i = 0; i < pon; i++) { in MatTransposeMatMultSymbolic_MPIAIJ_MPIAIJ()
[all …]