Lines Matching refs:am
203 PetscInt am = A->rmap->n, pN = P->cmap->N, pn = P->cmap->n, pm = P->rmap->n; in MatMatMultSymbolic_MPIAIJ_MPIAIJ_nonscalable() local
237 PetscCall(PetscMalloc1(am + 1, &api)); in MatMatMultSymbolic_MPIAIJ_MPIAIJ_nonscalable()
245 …ceGet(PetscRealIntMultTruncate(fill, PetscIntSumTruncate(adi[am], PetscIntSumTruncate(aoi[am], pi_… in MatMatMultSymbolic_MPIAIJ_MPIAIJ_nonscalable()
248 MatPreallocateBegin(comm, am, pn, dnz, onz); in MatMatMultSymbolic_MPIAIJ_MPIAIJ_nonscalable()
249 for (i = 0; i < am; i++) { in MatMatMultSymbolic_MPIAIJ_MPIAIJ_nonscalable()
293 PetscCall(PetscMalloc1(api[am], &ptap->apj)); in MatMatMultSymbolic_MPIAIJ_MPIAIJ_nonscalable()
302 PetscCall(MatSetSizes(C, am, pn, PETSC_DETERMINE, PETSC_DETERMINE)); in MatMatMultSymbolic_MPIAIJ_MPIAIJ_nonscalable()
324 afill = (PetscReal)api[am] / (adi[am] + aoi[am] + pi_loc[pm] + 1) + 1.e-5; in MatMatMultSymbolic_MPIAIJ_MPIAIJ_nonscalable()
331 if (api[am]) { in MatMatMultSymbolic_MPIAIJ_MPIAIJ_nonscalable()
810 PetscInt am = A->rmap->n, pn = P->cmap->n, pm = P->rmap->n, lsize = pn + 20; in MatMatMultSymbolic_MPIAIJ_MPIAIJ() local
843 PetscCall(PetscMalloc1(am + 1, &api)); in MatMatMultSymbolic_MPIAIJ_MPIAIJ()
850 …ceGet(PetscRealIntMultTruncate(fill, PetscIntSumTruncate(adi[am], PetscIntSumTruncate(aoi[am], pi_… in MatMatMultSymbolic_MPIAIJ_MPIAIJ()
852 MatPreallocateBegin(comm, am, pn, dnz, onz); in MatMatMultSymbolic_MPIAIJ_MPIAIJ()
853 for (i = 0; i < am; i++) { in MatMatMultSymbolic_MPIAIJ_MPIAIJ()
916 PetscCall(PetscMalloc1(api[am], &ptap->apj)); in MatMatMultSymbolic_MPIAIJ_MPIAIJ()
922 PetscCall(MatSetSizes(C, am, pn, PETSC_DETERMINE, PETSC_DETERMINE)); in MatMatMultSymbolic_MPIAIJ_MPIAIJ()
946 afill = (PetscReal)api[am] / (adi[am] + aoi[am] + pi_loc[pm] + 1) + 1.e-5; in MatMatMultSymbolic_MPIAIJ_MPIAIJ()
953 if (api[am]) { in MatMatMultSymbolic_MPIAIJ_MPIAIJ()
1052 …PetscInt am = A->rmap->n, pN = P->cmap->N, pn = P->cmap->n, pm = P->rmap->n, p_colstar… in MatMatMultSymbolic_MPIAIJ_MPIAIJ_seqMPI() local
1081 PetscCall(PetscMalloc1(am + 1, &api)); in MatMatMultSymbolic_MPIAIJ_MPIAIJ_seqMPI()
1082 PetscCall(PetscMalloc1(am + 1, &adpoi)); in MatMatMultSymbolic_MPIAIJ_MPIAIJ_seqMPI()
1090 MatPreallocateBegin(comm, am, pn, dnz, onz); in MatMatMultSymbolic_MPIAIJ_MPIAIJ_seqMPI()
1119 …ceGet(PetscRealIntMultTruncate(fill, PetscIntSumTruncate(adi[am], PetscIntSumTruncate(aoi[am], pi_… in MatMatMultSymbolic_MPIAIJ_MPIAIJ_seqMPI()
1122 for (i = 0; i < am; i++) { in MatMatMultSymbolic_MPIAIJ_MPIAIJ_seqMPI()
1163 PetscCall(PetscMalloc1(aopothi[am] + adpoi[am] + adpdi[am], &ptap->apj)); in MatMatMultSymbolic_MPIAIJ_MPIAIJ_seqMPI()
1164 PetscCall(PetscMalloc1(adpoi[am], &adpoj)); in MatMatMultSymbolic_MPIAIJ_MPIAIJ_seqMPI()
1178 for (i = 0; i < am; i++) { in MatMatMultSymbolic_MPIAIJ_MPIAIJ_seqMPI()
1200 PetscCall(MatSetSizes(C, am, pn, PETSC_DETERMINE, PETSC_DETERMINE)); in MatMatMultSymbolic_MPIAIJ_MPIAIJ_seqMPI()
1221 afill = (PetscReal)api[am] / (adi[am] + aoi[am] + pi_loc[pm] + 1) + 1.e-5; in MatMatMultSymbolic_MPIAIJ_MPIAIJ_seqMPI()
1228 if (api[am]) { in MatMatMultSymbolic_MPIAIJ_MPIAIJ_seqMPI()
1598 PetscInt am = A->rmap->n, cm = C->rmap->n, pon = (p->B)->cmap->n; in MatTransposeMatMultNumeric_MPIAIJ_MPIAIJ() local
1645 for (i = 0; i < am; i++) { in MatTransposeMatMultNumeric_MPIAIJ_MPIAIJ()
1766 PetscInt am = A->rmap->n, pn = P->cmap->n; in MatTransposeMatMultSymbolic_MPIAIJ_MPIAIJ() local
1812 nnz = PetscRealIntMultTruncate(fill, PetscIntSumTruncate(poti[pon], ai[am])); in MatTransposeMatMultSymbolic_MPIAIJ_MPIAIJ()
1818 MatRowMergeMax_SeqAIJ(a_loc, am, ta); in MatTransposeMatMultSymbolic_MPIAIJ_MPIAIJ()
1855 afill_tmp = (PetscReal)coi[pon] / (poti[pon] + ai[am] + 1); in MatTransposeMatMultSymbolic_MPIAIJ_MPIAIJ()
1974 …scRealIntMultTruncate(fill, PetscIntSumTruncate(pdti[pn], PetscIntSumTruncate(poti[pon], ai[am]))); in MatTransposeMatMultSymbolic_MPIAIJ_MPIAIJ()
2040 afill_tmp = (PetscReal)bi[pn] / (pdti[pn] + poti[pon] + ai[am] + 1); in MatTransposeMatMultSymbolic_MPIAIJ_MPIAIJ()