Lines Matching refs:pn
223 PetscInt am = A->rmap->n, pm = P->rmap->n, pN = P->cmap->N, pn = P->cmap->n; in MatPtAPSymbolic_MPIAIJ_MPIAIJ_scalable() local
281 PetscCall(PetscHMapICreateWithSize(pn, &ta)); /* for compute AP_loc and Cmpi */ in MatPtAPSymbolic_MPIAIJ_MPIAIJ_scalable()
384 PetscCall(PetscLayoutSetLocalSize(rowmap, pn)); in MatPtAPSymbolic_MPIAIJ_MPIAIJ_scalable()
510 MatPreallocateBegin(comm, pn, pn, dnz, onz); in MatPtAPSymbolic_MPIAIJ_MPIAIJ_scalable()
512 for (i = 0; i < pn; i++) { in MatPtAPSymbolic_MPIAIJ_MPIAIJ_scalable()
539 PetscCall(MatSetSizes(Cmpi, pn, pn, PETSC_DETERMINE, PETSC_DETERMINE)); in MatPtAPSymbolic_MPIAIJ_MPIAIJ_scalable()
701 …PetscInt i, j, jj, kk, nzi, *c_rmtj, voff, *c_othj, pn, pon, pcstart, pcend, ccstart, … in MatPtAPNumeric_MPIAIJ_MPIXAIJ_allatonce() local
781 PetscCall(MatGetLocalSize(P, NULL, &pn)); in MatPtAPNumeric_MPIAIJ_MPIXAIJ_allatonce()
782 pn *= dof; in MatPtAPNumeric_MPIAIJ_MPIXAIJ_allatonce()
783 PetscCall(PetscCalloc2(ptap->c_othi[pn], &c_othj, ptap->c_othi[pn], &c_otha)); in MatPtAPNumeric_MPIAIJ_MPIXAIJ_allatonce()
794 …for (i = 0; i < pn; i++) cmaxr = PetscMax(cmaxr, (cd->i[i + 1] - cd->i[i]) + (co->i[i + 1] - co->i… in MatPtAPNumeric_MPIAIJ_MPIXAIJ_allatonce()
795 …PetscCall(PetscCalloc5(cmaxr, &apindices, cmaxr, &apvalues, cmaxr, &apvaluestmp, pn, &dcc, pn, &oc… in MatPtAPNumeric_MPIAIJ_MPIXAIJ_allatonce()
797 for (i = 0; i < am && pn; i++) { in MatPtAPNumeric_MPIAIJ_MPIXAIJ_allatonce()
826 for (i = 0; i < pn; 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
874 PetscCall(MatGetLocalSize(P, NULL, &pn)); in MatPtAPNumeric_MPIAIJ_MPIXAIJ_allatonce_merged()
875 pn *= dof; in MatPtAPNumeric_MPIAIJ_MPIXAIJ_allatonce_merged()
886 …for (i = 0; i < pn; i++) cmaxr = PetscMax(cmaxr, (cd->i[i + 1] - cd->i[i]) + (co->i[i + 1] - co->i… in MatPtAPNumeric_MPIAIJ_MPIXAIJ_allatonce_merged()
890 for (i = 0; i < am && (pon || pn); i++) { in MatPtAPNumeric_MPIAIJ_MPIXAIJ_allatonce_merged()
951 PetscCall(PetscCalloc2(ptap->c_othi[pn], &c_othj, ptap->c_othi[pn], &c_otha)); in MatPtAPNumeric_MPIAIJ_MPIXAIJ_allatonce_merged()
960 for (i = 0; i < pn; i++) { in MatPtAPNumeric_MPIAIJ_MPIXAIJ_allatonce_merged()
993 …PetscInt pn, pon, *c_rmtc, i, j, nzi, htsize, htosize, *c_rmtj, off, *c_othj, rcvncols… in MatPtAPSymbolic_MPIAIJ_MPIXAIJ_allatonce() local
1008 PetscCall(MatGetLocalSize(P, NULL, &pn)); in MatPtAPSymbolic_MPIAIJ_MPIXAIJ_allatonce()
1009 pn *= dof; in MatPtAPSymbolic_MPIAIJ_MPIXAIJ_allatonce()
1012 PetscCall(MatSetSizes(Cmpi, pn, pn, PETSC_DETERMINE, PETSC_DETERMINE)); in MatPtAPSymbolic_MPIAIJ_MPIXAIJ_allatonce()
1083 PetscCall(PetscSFSetGraph(sf, pn, pon, NULL, PETSC_OWN_POINTER, iremote, PETSC_OWN_POINTER)); in MatPtAPSymbolic_MPIAIJ_MPIXAIJ_allatonce()
1092 for (i = 0; i < pn; i++) rootspacesize += rootdegrees[i]; in MatPtAPSymbolic_MPIAIJ_MPIXAIJ_allatonce()
1101 PetscCall(PetscCalloc1(pn + 1, &ptap->c_othi)); in MatPtAPSymbolic_MPIAIJ_MPIXAIJ_allatonce()
1106 for (i = 0; i < pn; i++) { in MatPtAPSymbolic_MPIAIJ_MPIXAIJ_allatonce()
1136 PetscCall(PetscCalloc1(ptap->c_othi[pn], &c_othj)); in MatPtAPSymbolic_MPIAIJ_MPIXAIJ_allatonce()
1139 …PetscCall(PetscSFSetGraph(ptap->sf, ptap->c_othi[pn], nleaves, NULL, PETSC_OWN_POINTER, iremote, P… in MatPtAPSymbolic_MPIAIJ_MPIXAIJ_allatonce()
1144 PetscCall(PetscMalloc2(pn, &dnz, pn, &onz)); in MatPtAPSymbolic_MPIAIJ_MPIXAIJ_allatonce()
1149 PetscCall(PetscMalloc2(pn, &hta, pn, &hto)); in MatPtAPSymbolic_MPIAIJ_MPIXAIJ_allatonce()
1150 for (i = 0; i < pn; i++) { in MatPtAPSymbolic_MPIAIJ_MPIXAIJ_allatonce()
1156 for (i = 0; i < am && pn; i++) { in MatPtAPSymbolic_MPIAIJ_MPIXAIJ_allatonce()
1185 for (i = 0; i < pn; i++) { in MatPtAPSymbolic_MPIAIJ_MPIXAIJ_allatonce()
1209 PetscCall(MatSetSizes(Cmpi, pn, pn, PETSC_DETERMINE, PETSC_DETERMINE)); in MatPtAPSymbolic_MPIAIJ_MPIXAIJ_allatonce()
1259 …PetscInt pn, pon, *c_rmtc, i, j, nzi, dnzi, htsize, htosize, *c_rmtj, off, *c_othj, rc… in MatPtAPSymbolic_MPIAIJ_MPIXAIJ_allatonce_merged() local
1274 PetscCall(MatGetLocalSize(P, NULL, &pn)); in MatPtAPSymbolic_MPIAIJ_MPIXAIJ_allatonce_merged()
1275 pn *= dof; in MatPtAPSymbolic_MPIAIJ_MPIXAIJ_allatonce_merged()
1278 PetscCall(MatSetSizes(Cmpi, pn, pn, PETSC_DETERMINE, PETSC_DETERMINE)); in MatPtAPSymbolic_MPIAIJ_MPIXAIJ_allatonce_merged()
1302 PetscCall(PetscMalloc2(pn, &htd, pn, &hto)); in MatPtAPSymbolic_MPIAIJ_MPIXAIJ_allatonce_merged()
1303 for (i = 0; i < pn; i++) { in MatPtAPSymbolic_MPIAIJ_MPIXAIJ_allatonce_merged()
1311 for (i = 0; i < am && (pon || pn); i++) { in MatPtAPSymbolic_MPIAIJ_MPIXAIJ_allatonce_merged()
1375 PetscCall(PetscSFSetGraph(sf, pn, pon, NULL, PETSC_OWN_POINTER, iremote, PETSC_OWN_POINTER)); in MatPtAPSymbolic_MPIAIJ_MPIXAIJ_allatonce_merged()
1384 for (i = 0; i < pn; i++) rootspacesize += rootdegrees[i]; in MatPtAPSymbolic_MPIAIJ_MPIXAIJ_allatonce_merged()
1393 PetscCall(PetscMalloc1(pn + 1, &ptap->c_othi)); in MatPtAPSymbolic_MPIAIJ_MPIXAIJ_allatonce_merged()
1398 for (i = 0; i < pn; i++) { in MatPtAPSymbolic_MPIAIJ_MPIXAIJ_allatonce_merged()
1428 PetscCall(PetscCalloc1(ptap->c_othi[pn], &c_othj)); in MatPtAPSymbolic_MPIAIJ_MPIXAIJ_allatonce_merged()
1431 …PetscCall(PetscSFSetGraph(ptap->sf, ptap->c_othi[pn], nleaves, NULL, PETSC_OWN_POINTER, iremote, P… in MatPtAPSymbolic_MPIAIJ_MPIXAIJ_allatonce_merged()
1438 PetscCall(PetscMalloc2(pn, &dnz, pn, &onz)); in MatPtAPSymbolic_MPIAIJ_MPIXAIJ_allatonce_merged()
1442 for (i = 0; i < pn; i++) { in MatPtAPSymbolic_MPIAIJ_MPIXAIJ_allatonce_merged()
1466 PetscCall(MatSetSizes(Cmpi, pn, pn, PETSC_DETERMINE, PETSC_DETERMINE)); 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
1568 PetscCall(PetscHMapICreateWithSize(pn, &ta)); /* for compute AP_loc and Cmpi */ in MatPtAPSymbolic_MPIAIJ_MPIAIJ()
1674 rowmap->n = pn; in MatPtAPSymbolic_MPIAIJ_MPIAIJ()
1799 MatPreallocateBegin(comm, pn, pn, dnz, onz); in MatPtAPSymbolic_MPIAIJ_MPIAIJ()
1801 for (i = 0; i < pn; i++) { in MatPtAPSymbolic_MPIAIJ_MPIAIJ()
1828 PetscCall(MatSetSizes(Cmpi, pn, pn, PETSC_DETERMINE, PETSC_DETERMINE)); in MatPtAPSymbolic_MPIAIJ_MPIAIJ()