Home
last modified time | relevance | path

Searched refs:ranges (Results 1 – 25 of 42) sorted by relevance

12

/petsc/src/binding/petsc4py/test/ !
H A Dtest_dmda.py46 ranges = self.da.getRanges()
48 self.assertEqual(dim, len(ranges))
52 s, e = ranges[i]
58 ranges = self.da.getGhostRanges()
60 self.assertEqual(dim, len(ranges))
64 s, e = ranges[i]
188 ranges = self.da.getRanges()
189 ranges = list(zip(*ranges))
190 dim = len(ranges[0])
192 _ = ag[ranges[0][0] : ranges[1][0]]
[all …]
/petsc/src/dm/impls/plex/tests/ !
H A Dex28.c13 const PetscInt *ranges = NULL; in main() local
39 PetscCall(MatGetOwnershipRanges(A, &ranges)); in main()
42 const PetscInt partsize = ranges[p + 1] - ranges[p]; in main()
51 PetscCall(PetscPrintf(comm, "%" PetscInt_FMT, ranges[p])); in main()
/petsc/src/binding/petsc4py/src/petsc4py/PETSc/ !
H A Dpetscdmstag.pxi191 cdef object ranges = list(ownership_ranges)
192 cdef PetscInt rdim = <PetscInt>len(ranges)
196 "number of dimensions %d and number ownership ranges %d" %
199 ranges[0] = iarray_i(ranges[0], &nlx, _x)
205 ranges[1] = iarray_i(ranges[1], &nly, _y)
211 ranges[2] = iarray_i(ranges[2], &nlz, _z)
216 return tuple(ranges)
224 # Returns tuple of arrays containing ownership ranges as Python arrays
225 ranges = [array_i(m, lx)]
227 ranges.append(array_i(n, ly))
[all …]
H A Dpetscdmda.pxi156 cdef object ranges = list(ownership_ranges)
157 cdef PetscInt rdim = <PetscInt>len(ranges)
161 "number of dimensions %d and number ownership ranges %d" %
164 ranges[0] = iarray_i(ranges[0], &nlx, _x)
170 ranges[1] = iarray_i(ranges[1], &nly, _y)
176 ranges[2] = iarray_i(ranges[2], &nlz, _z)
181 return tuple(ranges)
188 # Returns tuple of arrays containing ownership ranges as Python arrays
189 ranges = [array_i(m, lx)]
191 ranges.append(array_i(n, ly))
[all …]
H A DDMStag.pyx142 # ownership ranges
286 def setOwnershipRanges(self, ranges: tuple[Sequence[int], ...]) -> None:
293 ranges
305 asStagOwnershipRanges(ranges, dim, &m, &n, &p, &lx, &ly, &lz)
/petsc/src/vec/is/sf/tests/ !
H A Dex15.c17 const PetscInt *ranges; in main() local
40 PetscCall(VecGetOwnershipRanges(x, &ranges)); in main()
41 …for (i = ranges[rank]; i < ranges[rank + 1]; i++) PetscCall(VecSetValue(x, i, (PetscScalar)i, INSE… in main()
50 start = ranges[rank]; in main()
51 for (i = ranges[rank]; i < ranges[rank + 1]; i++) ix[i - start] = i; in main()
/petsc/src/vec/ftn-mod/ !
H A Dpetscis.h902 subroutine PetscLayoutRestoreRanges(map, ranges, ierr)
6 PetscInt, pointer :: ranges(:)
H A Dpetscvec.h9062 subroutine VecRestoreOwnershipRanges(x, ranges, ierr)
66 PetscInt, pointer :: ranges(:)
/petsc/src/mat/impls/scalapack/ !
H A Dmatscalapack.c163 const PetscInt *ranges; in MatMultXXXYYY_ScaLAPACK() local
169 PetscCall(PetscLayoutGetRanges(A->rmap, &ranges)); in MatMultXXXYYY_ScaLAPACK()
170 PetscCall(PetscBLASIntCast(ranges[1], &mb)); /* x block size */ in MatMultXXXYYY_ScaLAPACK()
174 PetscCall(PetscLayoutGetRanges(A->cmap, &ranges)); in MatMultXXXYYY_ScaLAPACK()
175 PetscCall(PetscBLASIntCast(ranges[1], &nb)); /* y block size */ in MatMultXXXYYY_ScaLAPACK()
208 PetscCall(PetscLayoutGetRanges(A->cmap, &ranges)); in MatMultXXXYYY_ScaLAPACK()
209 PetscCall(PetscBLASIntCast(ranges[1], &nb)); /* x block size */ in MatMultXXXYYY_ScaLAPACK()
213 PetscCall(PetscLayoutGetRanges(A->rmap, &ranges)); in MatMultXXXYYY_ScaLAPACK()
214 PetscCall(PetscBLASIntCast(ranges[1], &mb)); /* y block size */ in MatMultXXXYYY_ScaLAPACK()
434 const PetscInt *ranges; in MatGetDiagonal_ScaLAPACK() local
[all …]
/petsc/src/snes/interface/ !
H A Dsnesj.c60 const PetscInt *ranges; in SNESComputeJacobianDefault() local
136 PetscCall(VecGetOwnershipRanges(x1, &ranges)); in SNESComputeJacobianDefault()
140 if (i >= ranges[j]) break; in SNESComputeJacobianDefault()
/petsc/src/mat/graphops/partition/impls/hierarchical/ !
H A Dhierarchical.c245 const PetscInt *ranges, *fineparts_indices; in MatPartitioningHierarchical_ReassembleFineparts() local
265 PetscCall(PetscLayoutGetRanges(rmap, &ranges)); in MatPartitioningHierarchical_ReassembleFineparts()
266 PetscCall(PetscMalloc1(ranges[rank + 1] - ranges[rank], &sfineparts_indices)); in MatPartitioningHierarchical_ReassembleFineparts()
268 for (i = 0; i < ranges[rank + 1] - ranges[rank]; i++) sfineparts_indices[i] = -1; in MatPartitioningHierarchical_ReassembleFineparts()
275 remote[i].index = global_indices[i] - ranges[owners[i]]; in MatPartitioningHierarchical_ReassembleFineparts()
285 …PetscCall(ISCreateGeneral(comm, ranges[rank + 1] - ranges[rank], sfineparts_indices, PETSC_OWN_POI… in MatPartitioningHierarchical_ReassembleFineparts()
/petsc/src/tao/constrained/impls/ipm/ !
H A Dpdipm.c279 const PetscInt *aj, *ranges, *Jranges, *rranges, *cranges; in TaoSNESJacobian_PDIPM() local
388 PetscCall(MatGetOwnershipRanges(tao->jacobian_equality, &ranges)); in TaoSNESJacobian_PDIPM()
392 while (aj[j] >= ranges[proc + 1]) proc++; in TaoSNESJacobian_PDIPM()
394 cols[0] = aj[j] - ranges[proc] + Jranges[proc] + nx_all; in TaoSNESJacobian_PDIPM()
403 PetscCall(MatGetOwnershipRanges(tao->jacobian_inequality, &ranges)); in TaoSNESJacobian_PDIPM()
407 while (aj[j] >= ranges[proc + 1]) proc++; in TaoSNESJacobian_PDIPM()
409 cols[0] = aj[j] - ranges[proc] + Jranges[proc] + nx_all + nce_all[proc]; in TaoSNESJacobian_PDIPM()
845 const PetscInt *cols, *rranges, *cranges, *aj, *ranges; in TaoSetup_PDIPM() local
1084 PetscCall(MatGetOwnershipRanges(tao->jacobian_equality, &ranges)); in TaoSetup_PDIPM()
1088 while (aj[j] >= ranges[proc + 1]) proc++; in TaoSetup_PDIPM()
[all …]
/petsc/src/dm/impls/patch/ !
H A Dpatch.c46 const PetscInt *ranges, *indices; in DMPatchZoom() local
98 PetscCall(VecGetOwnershipRanges(X, &ranges)); in DMPatchZoom()
127 PetscCall(PetscFindInt(indices[q], size + 1, ranges, &ir)); in DMPatchZoom()
131 remotePoints[q].index = indices[q] - ranges[remotePoints[q].rank]; in DMPatchZoom()
163 PetscCall(PetscFindInt(indices[q], size + 1, ranges, &ir)); in DMPatchZoom()
166 remotePoints[q].index = indices[q] - ranges[remotePoints[q].rank]; in DMPatchZoom()
/petsc/src/mat/utils/ !
H A Daxpy.c513 const PetscInt *ranges; in MatFilter() local
518 PetscCall(MatGetOwnershipRanges(A, &ranges)); in MatFilter()
519 rStart = ranges[rank]; in MatFilter()
520 rEnd = ranges[rank + 1]; in MatFilter()
530 for (r = 0; r < size; ++r) maxRows = PetscMax(maxRows, ranges[r + 1] - ranges[r]); in MatFilter()
/petsc/src/vec/vec/tutorials/ !
H A Dex1f90.F9016 PetscInt, pointer :: ranges(:)
174 PetscCallA(VecGetOwnershipRanges(x, ranges, ierr))
/petsc/share/petsc/bin/ !
H A Ddmnetwork_view.py16 ranges = str(text).split(',')
17 for rangeStr in ranges:
446 ranges = parseIndexSet(args.draw_rank_range)
447 for rank in ranges:
/petsc/src/mat/tests/ !
H A Dex115.c85 const PetscInt *ranges; in main() local
89 PetscCall(MatGetOwnershipRanges(A, &ranges)); in main()
90 rows[0] = ranges[(rank + 1) % size]; in main()
91 rows[1] = ranges[(rank + 1) % size + 1]; in main()
/petsc/doc/changes/ !
H A D233.md17 set and determines the ranges etc.
24 set and determines the ranges etc.
/petsc/src/mat/impls/dense/mpi/ !
H A Dmpidense.c2076 const PetscInt *ranges; in MatTransposeMatMultNumeric_MPIDense_MPIDense() local
2091 PetscCall(MatGetOwnershipRanges(C, &ranges)); in MatTransposeMatMultNumeric_MPIDense_MPIDense()
2093 if (ranges[1] == C->rmap->N) { in MatTransposeMatMultNumeric_MPIDense_MPIDense()
2157 for (i = ranges[proc]; i < ranges[proc + 1]; i++) sendbuf[k++] = atbarray[i + j * lda]; in MatTransposeMatMultNumeric_MPIDense_MPIDense()
2179 const PetscInt *ranges; in MatTransposeMatMultSymbolic_MPIDense_MPIDense() local
2204 PetscCall(MatGetOwnershipRanges(C, &ranges)); in MatTransposeMatMultSymbolic_MPIDense_MPIDense()
2205 …for (PetscMPIInt i = 0; i < size; i++) PetscCall(PetscMPIIntCast((ranges[i + 1] - ranges[i]) * cN,… in MatTransposeMatMultSymbolic_MPIDense_MPIDense()
2275 const PetscInt *ranges; in MatMatTransposeMultNumeric_MPIDense_MPIDense_Cyclic() local
2293 PetscCall(MatGetOwnershipRanges(B, &ranges)); in MatMatTransposeMultNumeric_MPIDense_MPIDense_Cyclic()
2315 PetscInt nextbn = ranges[nextrecvisfrom + 1] - ranges[nextrecvisfrom]; in MatMatTransposeMultNumeric_MPIDense_MPIDense_Cyclic()
[all …]
/petsc/src/mat/impls/htool/ !
H A Dhtool.cxx411 const PetscInt *ranges; in MatAssemblyEnd_Htool() local
430 PetscCall(MatGetOwnershipRanges(A, &ranges)); in MatAssemblyEnd_Htool()
432 offset[2 * i] = ranges[i]; in MatAssemblyEnd_Htool()
433 offset[2 * i + 1] = ranges[i + 1] - ranges[i]; in MatAssemblyEnd_Htool()
451 PetscCall(MatGetOwnershipRangesColumn(A, &ranges)); in MatAssemblyEnd_Htool()
453 offset[2 * i] = ranges[i]; in MatAssemblyEnd_Htool()
454 offset[2 * i + 1] = ranges[i + 1] - ranges[i]; in MatAssemblyEnd_Htool()
/petsc/src/mat/impls/adj/mpi/ !
H A Dmpiadj.c752 const PetscInt *ranges; in MatMPIAdjCreateNonemptySubcommMat_MPIAdj() local
761 PetscCall(MatGetOwnershipRanges(A, &ranges)); in MatMPIAdjCreateNonemptySubcommMat_MPIAdj()
763 if (ranges[i + 1] - ranges[i] > 0) nranks++; in MatMPIAdjCreateNonemptySubcommMat_MPIAdj()
773 if (ranges[i + 1] - ranges[i] > 0) ranks[nranks++] = i; in MatMPIAdjCreateNonemptySubcommMat_MPIAdj()
/petsc/src/vec/is/sf/utils/ !
H A Dsfutils.c132 const PetscInt *ranges; in PetscSFSetGraphSection() local
152 PetscCall(PetscLayoutGetRanges(layout, &ranges)); in PetscSFSetGraphSection()
193 PetscCall(PetscFindInt(offset, size + 1, ranges, &ir)); in PetscSFSetGraphSection()
198 remote[l].index = offset - ranges[r]; in PetscSFSetGraphSection()
203 remote[l].index = goff + d - ranges[rank]; in PetscSFSetGraphSection()
/petsc/src/snes/utils/dm/ !
H A Ddminterpolatesnes.c184 const PetscInt *ranges; in DMInterpolationSetUp() local
206 PetscCall(PetscLayoutGetRanges(layout, &ranges)); in DMInterpolationSetUp()
208 PetscCall(PetscMPIIntCast((ranges[p + 1] - ranges[p]) * ctx->dim, &counts[p])); in DMInterpolationSetUp()
209 PetscCall(PetscMPIIntCast(ranges[p] * ctx->dim, &displs[p])); in DMInterpolationSetUp()
/petsc/src/dm/impls/plex/cgns/ !
H A Dplexcgns2.c759 const PetscInt *ranges; in PetscLayoutCreateFromSizesAndOffset() local
765 PetscCall(PetscLayoutGetRanges(init, &ranges)); in PetscLayoutCreateFromSizesAndOffset()
768 PetscCall(PetscArraycpy(new_ranges, ranges, num_ranks + 1)); in PetscLayoutCreateFromSizesAndOffset()
803 CGRange *ranges; in DMPlexCGNS_CreateCornersConnectivitySection() local
809 PetscCall(PetscMalloc2(num_sections, &ranges, num_sections, &sectionCellTypes)); in DMPlexCGNS_CreateCornersConnectivitySection()
815 …gid, base, zone, section_ids[s], buffer, &sectionCellTypes[s], &ranges[s].start, &ranges[s].end, &… in DMPlexCGNS_CreateCornersConnectivitySection()
817 PetscInt num_section_cells = ranges[s].end - ranges[s].start + 1; in DMPlexCGNS_CreateCornersConnectivitySection()
818 …PetscCall(PetscLayoutCreateFromSizesAndOffset(comm, PETSC_DECIDE, num_section_cells, 1, ranges[s].… in DMPlexCGNS_CreateCornersConnectivitySection()
901 PetscCall(PetscFree2(ranges, sectionCellTypes)); in DMPlexCGNS_CreateCornersConnectivitySection()
/petsc/include/petsc/mpiuni/ !
H A Dmpi.h532 …ine MPI_Group_range_incl(group, n, ranges, newgroup) (MPIUNI_ARG(group), MPIUNI_ARG(n)… argument
533 …ine MPI_Group_range_excl(group, n, ranges, newgroup) (MPIUNI_ARG(group), MPIUNI_ARG(n)… argument

12