Lines Matching refs:pon
701 …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()
877 PetscCall(PetscCalloc2(ptap->c_rmti[pon], &c_rmtj, ptap->c_rmti[pon], &c_rmta)); in MatPtAPNumeric_MPIAIJ_MPIXAIJ_allatonce_merged()
883 for (i = 0; i < pon; i++) cmaxr = PetscMax(cmaxr, ptap->c_rmti[i + 1] - ptap->c_rmti[i]); in MatPtAPNumeric_MPIAIJ_MPIXAIJ_allatonce_merged()
887 PetscCall(PetscCalloc4(cmaxr, &apindices, cmaxr, &apvalues, cmaxr, &apvaluestmp, pon, &c_rmtc)); in MatPtAPNumeric_MPIAIJ_MPIXAIJ_allatonce_merged()
890 for (i = 0; i < am && (pon || 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
1022 PetscCall(MatGetLocalSize(p->B, NULL, &pon)); in MatPtAPSymbolic_MPIAIJ_MPIXAIJ_allatonce()
1023 pon *= dof; in MatPtAPSymbolic_MPIAIJ_MPIXAIJ_allatonce()
1025 PetscCall(PetscMalloc1(pon + 1, &ptap->c_rmti)); in MatPtAPSymbolic_MPIAIJ_MPIXAIJ_allatonce()
1027 PetscCall(PetscMalloc1(pon, &c_rmtc)); in MatPtAPSymbolic_MPIAIJ_MPIXAIJ_allatonce()
1028 PetscCall(PetscMalloc1(pon, &hta)); in MatPtAPSymbolic_MPIAIJ_MPIXAIJ_allatonce()
1029 for (i = 0; i < pon; i++) PetscCall(PetscHSetICreate(&hta[i])); in MatPtAPSymbolic_MPIAIJ_MPIXAIJ_allatonce()
1038 for (i = 0; i < am && pon; i++) { in MatPtAPSymbolic_MPIAIJ_MPIXAIJ_allatonce()
1056 for (i = 0; i < pon; i++) { in MatPtAPSymbolic_MPIAIJ_MPIXAIJ_allatonce()
1062 PetscCall(PetscMalloc1(ptap->c_rmti[pon], &c_rmtj)); in MatPtAPSymbolic_MPIAIJ_MPIXAIJ_allatonce()
1064 for (i = 0; i < pon; i++) { in MatPtAPSymbolic_MPIAIJ_MPIXAIJ_allatonce()
1071 PetscCall(PetscMalloc1(pon, &iremote)); in MatPtAPSymbolic_MPIAIJ_MPIXAIJ_allatonce()
1072 for (i = 0; i < pon; i++) { in MatPtAPSymbolic_MPIAIJ_MPIXAIJ_allatonce()
1083 PetscCall(PetscSFSetGraph(sf, pn, pon, NULL, PETSC_OWN_POINTER, iremote, PETSC_OWN_POINTER)); in MatPtAPSymbolic_MPIAIJ_MPIXAIJ_allatonce()
1117 PetscCall(PetscMalloc1(pon, &c_rmtoffsets)); in MatPtAPSymbolic_MPIAIJ_MPIXAIJ_allatonce()
1123 PetscCall(PetscCalloc1(ptap->c_rmti[pon], &iremote)); in MatPtAPSymbolic_MPIAIJ_MPIXAIJ_allatonce()
1125 for (i = 0; i < pon; i++) { 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
1289 PetscCall(MatGetLocalSize(p->B, NULL, &pon)); in MatPtAPSymbolic_MPIAIJ_MPIXAIJ_allatonce_merged()
1290 pon *= dof; in MatPtAPSymbolic_MPIAIJ_MPIXAIJ_allatonce_merged()
1292 PetscCall(PetscMalloc1(pon + 1, &ptap->c_rmti)); in MatPtAPSymbolic_MPIAIJ_MPIXAIJ_allatonce_merged()
1294 PetscCall(PetscMalloc1(pon, &c_rmtc)); in MatPtAPSymbolic_MPIAIJ_MPIXAIJ_allatonce_merged()
1295 PetscCall(PetscMalloc1(pon, &hta)); in MatPtAPSymbolic_MPIAIJ_MPIXAIJ_allatonce_merged()
1296 for (i = 0; i < pon; i++) PetscCall(PetscHSetICreate(&hta[i])); in MatPtAPSymbolic_MPIAIJ_MPIXAIJ_allatonce_merged()
1311 for (i = 0; i < am && (pon || pn); i++) { in MatPtAPSymbolic_MPIAIJ_MPIXAIJ_allatonce_merged()
1348 for (i = 0; i < pon; i++) { in MatPtAPSymbolic_MPIAIJ_MPIXAIJ_allatonce_merged()
1354 PetscCall(PetscMalloc1(ptap->c_rmti[pon], &c_rmtj)); in MatPtAPSymbolic_MPIAIJ_MPIXAIJ_allatonce_merged()
1356 for (i = 0; i < pon; i++) { in MatPtAPSymbolic_MPIAIJ_MPIXAIJ_allatonce_merged()
1363 PetscCall(PetscMalloc1(pon, &iremote)); in MatPtAPSymbolic_MPIAIJ_MPIXAIJ_allatonce_merged()
1364 for (i = 0; i < pon; 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()
1409 PetscCall(PetscMalloc1(pon, &c_rmtoffsets)); in MatPtAPSymbolic_MPIAIJ_MPIXAIJ_allatonce_merged()
1415 PetscCall(PetscCalloc1(ptap->c_rmti[pon], &iremote)); in MatPtAPSymbolic_MPIAIJ_MPIXAIJ_allatonce_merged()
1417 for (i = 0; i < pon; i++) { in MatPtAPSymbolic_MPIAIJ_MPIXAIJ_allatonce_merged()