Lines Matching refs:am
103 PetscInt i, rstart, rend, cm, ncols, row, *api, *apj, am = A->rmap->n, apnz, nout; in MatPtAPNumeric_MPIAIJ_MPIAIJ_scalable() local
141 for (i = 0; i < am; i++) { in MatPtAPNumeric_MPIAIJ_MPIAIJ_scalable()
223 PetscInt am = A->rmap->n, pm = P->rmap->n, pN = P->cmap->N, pn = P->cmap->n; in MatPtAPSymbolic_MPIAIJ_MPIAIJ_scalable() local
290 …et(PetscRealIntMultTruncate(fill, PetscIntSumTruncate(ad->i[am], PetscIntSumTruncate(ao->i[am], p_… in MatPtAPSymbolic_MPIAIJ_MPIAIJ_scalable()
292 …PetscCall(PetscFreeSpaceGet(PetscRealIntMultTruncate(fill, PetscIntSumTruncate(ad->i[am], p_loc->i… in MatPtAPSymbolic_MPIAIJ_MPIAIJ_scalable()
297 PetscCall(PetscMalloc1(am + 1, &api)); in MatPtAPSymbolic_MPIAIJ_MPIAIJ_scalable()
299 for (i = 0; i < am; i++) { in MatPtAPSymbolic_MPIAIJ_MPIAIJ_scalable()
343 PetscCall(PetscCalloc2(api[am], &apj, api[am], &apv)); in MatPtAPSymbolic_MPIAIJ_MPIAIJ_scalable()
348 PetscCall(MatCreateSeqAIJWithArrays(PETSC_COMM_SELF, am, pN, api, apj, apv, &ptap->AP_loc)); in MatPtAPSymbolic_MPIAIJ_MPIAIJ_scalable()
353 apfill = (PetscReal)api[am] / (ad->i[am] + ao->i[am] + p_loc->i[pm] + 1); in MatPtAPSymbolic_MPIAIJ_MPIAIJ_scalable()
355 apfill = (PetscReal)api[am] / (ad->i[am] + p_loc->i[pm] + 1); in MatPtAPSymbolic_MPIAIJ_MPIAIJ_scalable()
361 if (api[am]) { in MatPtAPSymbolic_MPIAIJ_MPIAIJ_scalable()
701 …nzi, *c_rmtj, voff, *c_othj, pn, pon, pcstart, pcend, ccstart, ccend, row, am, *poj, *pdj, *apindi… in MatPtAPNumeric_MPIAIJ_MPIXAIJ_allatonce() local
725 PetscCall(MatGetLocalSize(A, &am, NULL)); in MatPtAPNumeric_MPIAIJ_MPIXAIJ_allatonce()
731 for (i = 0; i < am && pon; i++) { in MatPtAPNumeric_MPIAIJ_MPIXAIJ_allatonce()
797 for (i = 0; i < am && pn; i++) { in MatPtAPNumeric_MPIAIJ_MPIXAIJ_allatonce()
852 …, jj, kk, nzi, dnzi, *c_rmtj, voff, *c_othj, pn, pon, pcstart, pcend, row, am, *poj, *pdj, *apindi… in MatPtAPNumeric_MPIAIJ_MPIXAIJ_allatonce_merged() local
878 PetscCall(MatGetLocalSize(A, &am, NULL)); in MatPtAPNumeric_MPIAIJ_MPIXAIJ_allatonce_merged()
890 for (i = 0; i < am && (pon || pn); i++) { in MatPtAPNumeric_MPIAIJ_MPIXAIJ_allatonce_merged()
994 PetscInt lidx, *rdj, col, pcstart, pcend, *dnz, *onz, am, arstart, arend, *poj, *pdj; in MatPtAPSymbolic_MPIAIJ_MPIXAIJ_allatonce() local
1030 PetscCall(MatGetLocalSize(A, &am, NULL)); in MatPtAPSymbolic_MPIAIJ_MPIXAIJ_allatonce()
1038 for (i = 0; i < am && pon; i++) { in MatPtAPSymbolic_MPIAIJ_MPIXAIJ_allatonce()
1156 for (i = 0; i < am && pn; i++) { in MatPtAPSymbolic_MPIAIJ_MPIXAIJ_allatonce()
1260 PetscInt lidx, *rdj, col, pcstart, pcend, *dnz, *onz, am, arstart, arend, *poj, *pdj; in MatPtAPSymbolic_MPIAIJ_MPIXAIJ_allatonce_merged() local
1297 PetscCall(MatGetLocalSize(A, &am, NULL)); in MatPtAPSymbolic_MPIAIJ_MPIXAIJ_allatonce_merged()
1311 for (i = 0; i < am && (pon || pn); i++) { in MatPtAPSymbolic_MPIAIJ_MPIXAIJ_allatonce_merged()
1510 PetscInt am = A->rmap->n, pm = P->rmap->n, pN = P->cmap->N, pn = P->cmap->n; in MatPtAPSymbolic_MPIAIJ_MPIAIJ() local
1578 …et(PetscRealIntMultTruncate(fill, PetscIntSumTruncate(ad->i[am], PetscIntSumTruncate(ao->i[am], p_… in MatPtAPSymbolic_MPIAIJ_MPIAIJ()
1580 …PetscCall(PetscFreeSpaceGet(PetscRealIntMultTruncate(fill, PetscIntSumTruncate(ad->i[am], p_loc->i… in MatPtAPSymbolic_MPIAIJ_MPIAIJ()
1585 PetscCall(PetscMalloc1(am + 1, &api)); in MatPtAPSymbolic_MPIAIJ_MPIAIJ()
1587 for (i = 0; i < am; i++) { in MatPtAPSymbolic_MPIAIJ_MPIAIJ()
1632 PetscCall(PetscMalloc2(api[am], &apj, api[am], &apv)); in MatPtAPSymbolic_MPIAIJ_MPIAIJ()
1637 PetscCall(MatCreateSeqAIJWithArrays(PETSC_COMM_SELF, am, pN, api, apj, apv, &ptap->AP_loc)); in MatPtAPSymbolic_MPIAIJ_MPIAIJ()
1641 apfill = (PetscReal)api[am] / (ad->i[am] + ao->i[am] + p_loc->i[pm] + 1); in MatPtAPSymbolic_MPIAIJ_MPIAIJ()
1643 apfill = (PetscReal)api[am] / (ad->i[am] + p_loc->i[pm] + 1); in MatPtAPSymbolic_MPIAIJ_MPIAIJ()
1649 if (api[am]) { in MatPtAPSymbolic_MPIAIJ_MPIAIJ()
1865 PetscInt *api, *apj, am = A->rmap->n, j, col, apnz; in MatPtAPNumeric_MPIAIJ_MPIAIJ() local
1907 for (i = 0; i < am; i++) { in MatPtAPNumeric_MPIAIJ_MPIAIJ()