Lines Matching refs:nloc
38 PetscInt arrsz, bs, my0, kk, ii, nloc, Iend, aloc; in PCSetCoordinates_GEO() local
46 nloc = (Iend - my0) / bs; in PCSetCoordinates_GEO()
48 …nloc == a_nloc || aloc == a_nloc, PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Number of local blocks %"… in PCSetCoordinates_GEO()
51 …PetscCheck(coords || (nloc <= 0), PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Need coordinates for pc_g… in PCSetCoordinates_GEO()
54 arrsz = nloc * pc_gamg->data_cell_rows * pc_gamg->data_cell_cols; in PCSetCoordinates_GEO()
64 if (nloc == a_nloc) { in PCSetCoordinates_GEO()
65 for (kk = 0; kk < nloc; kk++) { in PCSetCoordinates_GEO()
66 for (ii = 0; ii < ndm; ii++) pc_gamg->data[ii * nloc + kk] = coords[kk * ndm + ii]; in PCSetCoordinates_GEO()
69 for (kk = 0; kk < nloc; kk++) { in PCSetCoordinates_GEO()
70 for (ii = 0; ii < ndm; ii++) pc_gamg->data[ii * nloc + kk] = coords[bs * kk * ndm + ii]; in PCSetCoordinates_GEO()
431 PetscInt *crsGID, kk, my0, Iend, nloc; in getGIDsOnSquareGraph() local
438 nloc = Iend - my0; /* this does not change */ in getGIDsOnSquareGraph()
488 selected_set[idx] = nloc + kk; in getGIDsOnSquareGraph()
496 for (kk = 0, idx = 0; kk < nloc; kk++) { in getGIDsOnSquareGraph()
531 PetscInt Istart, Iend, nloc, kk, Ii, ncols; in PCGAMGCoarsen_GEO() local
543 nloc = (Iend - Istart); in PCGAMGCoarsen_GEO()
546 PetscCall(PetscMalloc1(nloc, &gnodes)); in PCGAMGCoarsen_GEO()
547 PetscCall(PetscMalloc1(nloc, &permute)); in PCGAMGCoarsen_GEO()
565 PetscCall(PetscCalloc1(nloc, &bIndexSet)); in PCGAMGCoarsen_GEO()
566 for (Ii = 0; Ii < nloc; Ii++) { in PCGAMGCoarsen_GEO()
568 iSwapIndex = (PetscInt)(rr * nloc); in PCGAMGCoarsen_GEO()
581 if (gnodes) qsort(gnodes, nloc, sizeof(GAMGNode), petsc_geo_mg_compare); in PCGAMGCoarsen_GEO()
583 for (kk = 0; kk < nloc; kk++) permute[kk] = gnodes[kk].lid; /* locals only */ in PCGAMGCoarsen_GEO()
585 PetscCall(ISCreateGeneral(PETSC_COMM_SELF, nloc, permute, PETSC_OWN_POINTER, &perm)); in PCGAMGCoarsen_GEO()
607 PetscInt Istart, Iend, nloc, my0, jj, kk, ncols, nLocalSelected, bs, *clid_flid; in PCGAMGProlongator_GEO() local
622 nloc = (Iend - Istart) / bs; in PCGAMGProlongator_GEO()
634 if (lid < nloc) { in PCGAMGProlongator_GEO()
646 PetscCall(MatSetSizes(Prol, nloc * bs, nLocalSelected * bs, PETSC_DETERMINE, PETSC_DETERMINE)); in PCGAMGProlongator_GEO()
705 … for (jj = 0; jj < dim; jj++) crs_crds[jj * nLocalSelected + kk] = pc_gamg->data[jj * nloc + lid]; in PCGAMGProlongator_GEO()