Lines Matching refs:nloc
357 PetscInt arrsz, kk, ii, jj, nloc, ndatarows, ndf; in PCSetCoordinates_AGG() local
363 nloc = a_nloc; in PCSetCoordinates_AGG()
375 arrsz = nloc * pc_gamg->data_cell_rows * pc_gamg->data_cell_cols; in PCSetCoordinates_AGG()
382 for (kk = 0; kk < nloc; kk++) { in PCSetCoordinates_AGG()
383 const PetscInt M = nloc * pc_gamg->data_cell_rows; /* stride into data */ in PCSetCoordinates_AGG()
513 …PetscInt Istart, my0, Iend, nloc, clid, flid = 0, aggID, kk, jj, ii, mm, nSelected, minsz, ng… in formProl0() local
522 nloc = (Iend - Istart) / bs; in formProl0()
526 nghosts = data_stride / bs - nloc; in formProl0()
530 …for (kk = 0; kk < nghosts; kk++) PetscCall(PetscHMapISet(fgid_flid, flid_fgid[nloc + kk], nloc + k… in formProl0()
533 for (nSelected = mm = 0; mm < nloc; mm++) { in formProl0()
552 for (mm = clid = 0; mm < nloc; mm++) { in formProl0()
723 PetscInt Istart, Iend, ncols, nnz0, nnz1, NN, MM, nloc; in PCGAMGCreateGraph_AGG() local
747 nloc = Iend - Istart; in PCGAMGCreateGraph_AGG()
748 PetscCall(PetscMalloc2(nloc, &d_nnz, nloc, &o_nnz)); in PCGAMGCreateGraph_AGG()
760 for (PetscInt row = 0; row < nloc; row++) { in PCGAMGCreateGraph_AGG()
767 for (PetscInt row = 0; row < nloc; row++) { in PCGAMGCreateGraph_AGG()
774 PetscCall(MatSetSizes(tGmat, nloc, nloc, MM, MM)); in PCGAMGCreateGraph_AGG()
783 for (PetscInt row = 0, grow = Istart, ncol_row, jj; row < nloc; row++, grow++) { in PCGAMGCreateGraph_AGG()
805 … 100. * (double)nnz1 / (double)nnz0, (double)vfilter, (!nloc) ? 1. : (double)nnz0 / (double)nloc, … in PCGAMGCreateGraph_AGG()
841 const PetscInt nloc = Gmat_2->rmap->n; in fixAggregatesWithSquare() local
856 PetscCall(PetscMalloc3(nloc, &lid_state, nloc, &lid_parent_gid, nloc, &lid_cprowID_1)); in fixAggregatesWithSquare()
857 for (lid = 0; lid < nloc; lid++) lid_cprowID_1[lid] = -1; in fixAggregatesWithSquare()
870 …scCheck(lid <= nloc && lid >= -1, PETSC_COMM_SELF, PETSC_ERR_USER, "lid %" PetscInt_FMT " out of r… in fixAggregatesWithSquare()
880 …if (nloc > 0) PetscCheck(!matB_1 || matB_1->compressedrow.use, PETSC_COMM_SELF, PETSC_ERR_PLIB, "m… in fixAggregatesWithSquare()
882 for (lid = 0; lid < nloc; lid++) { in fixAggregatesWithSquare()
888 for (lid = 0; lid < nloc; lid++) { in fixAggregatesWithSquare()
902 for (lid = 0; lid < nloc; lid++) { in fixAggregatesWithSquare()
924 for (kk = 0, j = my0; kk < nloc; kk++, j++) { in fixAggregatesWithSquare()
939 for (kk = 0, j = my0; kk < nloc; kk++, j++) { in fixAggregatesWithSquare()
953 for (lid = 0; lid < nloc; lid++) { in fixAggregatesWithSquare()
1054 …for (kk = 0, j = my0; kk < nloc; kk++, j++) PetscCall(VecSetValues(tempVec, 1, &j, &lid_parent_gid… in fixAggregatesWithSquare()
1063 for (kk = 0, j = my0; kk < nloc; kk++, j++) { in fixAggregatesWithSquare()
1094 for (lid = 0; lid < nloc; lid++) { in fixAggregatesWithSquare()
1182 PetscInt Istart, Iend, Ii, nloc, bs, nn; in PCGAMGCoarsen_AGG() local
1196 nloc = nn / bs; in PCGAMGCoarsen_AGG()
1198 PetscCall(PetscMalloc2(nloc, &permute, nloc, °ree)); in PCGAMGCoarsen_AGG()
1199 PetscCall(PetscCalloc1(nloc, &bIndexSet)); in PCGAMGCoarsen_AGG()
1200 for (Ii = 0; Ii < nloc; Ii++) permute[Ii] = Ii; in PCGAMGCoarsen_AGG()
1203 for (Ii = 0; Ii < nloc; Ii++) { in PCGAMGCoarsen_AGG()
1210 for (Ii = 0; Ii < nloc; Ii++) { in PCGAMGCoarsen_AGG()
1212 iSwapIndex = (PetscInt)(hashfact * nloc) % nloc; in PCGAMGCoarsen_AGG()
1225 …if (pc_gamg_agg->use_minimum_degree_ordering) PetscCall(PetscSortIntWithArray(nloc, degree, permut… in PCGAMGCoarsen_AGG()
1228 PetscCall(ISCreateGeneral(PETSC_COMM_SELF, nloc, permute, PETSC_USE_POINTER, &perm)); in PCGAMGCoarsen_AGG()
1282 PetscInt Istart, Iend, nloc, ii, jj, kk, my0, nLocalSelected, bs; in PCGAMGConstructProlongator_AGG() local
1297 nloc = (Iend - Istart) / bs; in PCGAMGConstructProlongator_AGG()
1303 for (ii = 0, nLocalSelected = 0; ii < nloc; ii++) { in PCGAMGConstructProlongator_AGG()
1314 …PetscCall(MatSetSizes(Prol, nloc * bs, nLocalSelected * col_bs, PETSC_DETERMINE, PETSC_DETERMINE)); in PCGAMGConstructProlongator_AGG()
1347 PetscCall(PetscMalloc1(nloc, &tmp_ldata)); in PCGAMGConstructProlongator_AGG()
1351 const PetscReal *tp = PetscSafePointerPlusOffset(pc_gamg->data, jj * bs * nloc + kk); in PCGAMGConstructProlongator_AGG()
1353 for (ii = 0; ii < nloc; ii++, tp += bs) tmp_ldata[ii] = *tp; in PCGAMGConstructProlongator_AGG()
1368 nbnodes = bs * nloc; in PCGAMGConstructProlongator_AGG()
1377 PetscCall(PetscMalloc1(nloc, &fid_glid_loc)); in PCGAMGConstructProlongator_AGG()
1378 for (kk = 0; kk < nloc; kk++) fid_glid_loc[kk] = (PetscReal)(my0 + kk); in PCGAMGConstructProlongator_AGG()
1387 PetscCall(PetscMalloc1(nloc, &flid_fgid)); in PCGAMGConstructProlongator_AGG()
1388 for (kk = 0; kk < nloc; kk++) flid_fgid[kk] = my0 + kk; in PCGAMGConstructProlongator_AGG()