Home
last modified time | relevance | path

Searched refs:localsize (Results 1 – 9 of 9) sorted by relevance

/petsc/src/mat/utils/
H A Doverlapsplit.c16 PetscInt nindx, *indices_sc, *indices_ov, localsize, *localsizes_sc, localsize_tmp; in MatIncreaseOverlapSplit_Single() local
78 PetscCall(ISGetLocalSize(allis_sc, &localsize)); in MatIncreaseOverlapSplit_Single()
79 PetscCall(PetscCalloc2(localsize, &indices_ov, localsize, &sources_sc)); in MatIncreaseOverlapSplit_Single()
80 PetscCall(PetscCalloc2(localsize, &indices_ov_rd, localsize, &sources_sc_rd)); in MatIncreaseOverlapSplit_Single()
82 PetscCall(PetscArraycpy(indices_ov, indices, localsize)); in MatIncreaseOverlapSplit_Single()
91 PetscCall(PetscSortIntWithMPIIntArray(localsize, indices_ov, sources_sc)); in MatIncreaseOverlapSplit_Single()
95 if (localsize) { in MatIncreaseOverlapSplit_Single()
102 for (PetscInt i = 1; i < localsize; i++) { in MatIncreaseOverlapSplit_Single()
/petsc/src/dm/tests/
H A Dex3.c10 PetscInt M = 60, time_steps = 100, localsize, j, i, mybase, myend, *localnodes = NULL; in main() local
61 PetscCall(VecGetLocalSize(local, &localsize)); in main()
64 for (i = 1; i < localsize - 1; i++) { in main()
87 …for (i = 1; i < localsize - 1; i++) globalptr[i - 1] = .5 * (localptr[i + 1] + localptr[i - 1]) - … in main()
H A Dex12.c14 PetscInt M = 14, time_steps = 20, w = 1, s = 1, localsize, j, i, mybase, myend, globalsize; in main() local
54 PetscCall(VecGetLocalSize(local, &localsize)); in main()
67 …for (i = 1; i < localsize - 1; i++) globalptr[i - 1] = localptr[i] + (k / (h * h)) * (localptr[i +… in main()
/petsc/src/ts/tutorials/
H A Dex21.c302 PetscInt localsize; in SetBounds() local
307 PetscCall(VecGetLocalSize(xl, &localsize)); in SetBounds()
318 l[localsize - 1] = -PETSC_INFINITY; in SetBounds()
319 u[localsize - 1] = PETSC_INFINITY; in SetBounds()
470 PetscInt i, localsize; in RHSFunction() local
503 PetscCall(VecGetLocalSize(local_in, &localsize)); in RHSFunction()
521 if (rank == size - 1) copyptr[localsize - 1] = (t < .5) ? 2.0 : 0.0; in RHSFunction()
527 …for (i = 1; i < localsize - 1; i++) copyptr[i] = localptr[i] * sc * (localptr[i + 1] + localptr[i … in RHSFunction()
H A Dex2.c413 PetscInt i, localsize; in RHSFunction() local
446 PetscCall(VecGetLocalSize(local_in, &localsize)); in RHSFunction()
464 if (rank == size - 1) copyptr[localsize - 1] = 2.0; in RHSFunction()
470 …for (i = 1; i < localsize - 1; i++) copyptr[i] = localptr[i] * sc * (localptr[i + 1] + localptr[i … in RHSFunction()
/petsc/src/mat/graphops/partition/impls/hierarchical/
H A Dhierarchical.c244 PetscInt *local_indices, *global_indices, *sfineparts_indices, localsize, i; in MatPartitioningHierarchical_ReassembleFineparts() local
257 PetscCall(ISGetLocalSize(fineparts, &localsize)); in MatPartitioningHierarchical_ReassembleFineparts()
258 PetscCall(PetscMalloc2(localsize, &global_indices, localsize, &local_indices)); in MatPartitioningHierarchical_ReassembleFineparts()
259 for (i = 0; i < localsize; i++) local_indices[i] = i; in MatPartitioningHierarchical_ReassembleFineparts()
261 PetscCall(ISLocalToGlobalMappingApply(mapping, localsize, local_indices, global_indices)); in MatPartitioningHierarchical_ReassembleFineparts()
262 PetscCall(PetscCalloc1(localsize, &owners)); in MatPartitioningHierarchical_ReassembleFineparts()
264 …for (i = 0; i < localsize; i++) PetscCall(PetscLayoutFindOwner(rmap, global_indices[i], &owners[i]… in MatPartitioningHierarchical_ReassembleFineparts()
272 PetscCall(PetscMalloc1(localsize, &remote)); in MatPartitioningHierarchical_ReassembleFineparts()
273 for (i = 0; i < localsize; i++) { in MatPartitioningHierarchical_ReassembleFineparts()
280 …PetscCall(PetscSFSetGraph(sf, localsize, localsize, NULL, PETSC_OWN_POINTER, remote, PETSC_OWN_POI… in MatPartitioningHierarchical_ReassembleFineparts()
/petsc/src/ksp/ksp/tutorials/
H A Dex59.c180 PetscInt localsize, i, j, k, *indices; in ComputeSpecialBoundaryIndices() local
184 localsize = dd.xm_l * dd.ym_l * dd.zm_l; in ComputeSpecialBoundaryIndices()
186 PetscCall(PetscMalloc1(localsize, &indices)); in ComputeSpecialBoundaryIndices()
187 PetscCall(PetscMalloc1(localsize, &touched)); in ComputeSpecialBoundaryIndices()
188 for (i = 0; i < localsize; i++) touched[i] = PETSC_FALSE; in ComputeSpecialBoundaryIndices()
294 PetscInt i, j, k, ig, jg, kg, lindex, gindex, localsize; in ComputeMapping() local
300 localsize = dd.xm_l * dd.ym_l * dd.zm_l; in ComputeMapping()
301 PetscCall(PetscMalloc1(localsize, &global_indices)); in ComputeMapping()
326 …PetscCall(ISLocalToGlobalMappingCreate(dd.gcomm, 1, localsize, global_indices, PETSC_OWN_POINTER, … in ComputeMapping()
333 PetscInt localsize, zloc, yloc, xloc, auxnex, auxney, auxnez; in ComputeSubdomainMatrix() local
[all …]
/petsc/src/ts/tests/
H A Dex12.c324 PetscInt i, localsize; in RHSFunction() local
357 PetscCall(VecGetLocalSize(local_in, &localsize)); in RHSFunction()
375 if (rank == size - 1) copyptr[localsize - 1] = 2.0; in RHSFunction()
381 …for (i = 1; i < localsize - 1; i++) copyptr[i] = localptr[i] * sc * (localptr[i + 1] + localptr[i … in RHSFunction()
/petsc/src/ksp/pc/impls/fieldsplit/
H A Dfieldsplit.c2334 PetscInt localsize, size, sizez, i; in PCFieldSplitRestrictIS_FieldSplit() local
2340 PetscCall(ISGetLocalSize(isy, &localsize)); in PCFieldSplitRestrictIS_FieldSplit()
2341 …PetscCallMPI(MPI_Scan(&localsize, &size, 1, MPIU_INT, MPI_SUM, PetscObjectComm((PetscObject)isy))); in PCFieldSplitRestrictIS_FieldSplit()
2342 size -= localsize; in PCFieldSplitRestrictIS_FieldSplit()
2347 PetscCall(ISGetLocalSize(isrl, &localsize)); in PCFieldSplitRestrictIS_FieldSplit()
2348 PetscCall(PetscMalloc1(localsize, &indc)); in PCFieldSplitRestrictIS_FieldSplit()
2350 PetscCall(PetscArraycpy(indc, ind, localsize)); in PCFieldSplitRestrictIS_FieldSplit()
2353 for (i = 0; i < localsize; i++) *(indc + i) += size; in PCFieldSplitRestrictIS_FieldSplit()
2354 …PetscCall(ISCreateGeneral(PetscObjectComm((PetscObject)isy), localsize, indc, PETSC_OWN_POINTER, &… in PCFieldSplitRestrictIS_FieldSplit()
2367 PetscCall(ISGetLocalSize(ilink->is, &localsize)); in PCFieldSplitRestrictIS_FieldSplit()
[all …]