| /petsc/src/vec/is/ao/impls/basic/ |
| H A D | aobasic.c | 158 PetscInt napp, *allpetsc, *allapp, ip, ia, N, i, *petsc = NULL, start; in AOCreate_Basic() local
170 PetscCall(ISGetLocalSize(isapp, &napp)); in AOCreate_Basic()
173 PetscCall(PetscMPIIntCast(napp, &count)); in AOCreate_Basic()
190 if (napp) { in AOCreate_Basic()
193 PetscCall(PetscMalloc1(napp + 1, &petsc)); in AOCreate_Basic()
194 for (i = 0; i < napp; i++) petsc[i] = start + i; in AOCreate_Basic()
233 if (napp && !mypetsc) PetscCall(PetscFree(petsc)); in AOCreate_Basic()
242 if (napp) { in AOCreate_Basic()
275 PetscErrorCode AOCreateBasic(MPI_Comm comm, PetscInt napp, const PetscInt myapp[], const PetscInt m… in AOCreateBasic() argument
281 PetscCall(ISCreateGeneral(comm, napp, app, PETSC_USE_POINTER, &isapp)); in AOCreateBasic()
[all …]
|
| /petsc/src/binding/petsc4py/src/petsc4py/PETSc/ |
| H A D | AO.pyx | 84 cdef PetscInt napp = 0, *idxapp = NULL,
94 app = iarray_i(app, &napp, &idxapp)
97 assert napp == npetsc, "incompatible array sizes"
98 CHKERR(AOCreateBasic(ccomm, napp, idxapp, idxpetsc, &newao))
136 cdef PetscInt napp = 0, *idxapp = NULL,
146 app = iarray_i(app, &napp, &idxapp)
149 assert napp == npetsc, "incompatible array sizes"
150 CHKERR(AOCreateMemoryScalable(ccomm, napp, idxapp, idxpetsc, &newao))
182 cdef PetscInt napp = 0, *idxapp = NULL,
192 app = iarray_i(app, &napp, &idxapp)
[all …]
|
| /petsc/src/vec/is/ao/impls/memscalable/ |
| H A D | aomemscalable.c | 266 static PetscErrorCode AOCreateMemoryScalable_private(MPI_Comm comm, PetscInt napp, const PetscInt f… in AOCreateMemoryScalable_private() argument
279 PetscInt n = napp, ip, ia; in AOCreateMemoryScalable_private()
389 PetscInt napp, n_local, N, i, start, *petsc, *lens, *disp; in AOCreate_MemoryScalable() local
408 PetscCall(ISGetLocalSize(isapp, &napp)); in AOCreate_MemoryScalable()
409 PetscCallMPI(MPI_Allgather(&napp, 1, MPIU_INT, lens, 1, MPIU_INT, comm)); in AOCreate_MemoryScalable()
418 if (napp) { in AOCreate_MemoryScalable()
421 PetscCall(PetscMalloc1(napp + 1, &petsc)); in AOCreate_MemoryScalable()
422 for (i = 0; i < napp; i++) petsc[i] = start + i; in AOCreate_MemoryScalable()
446 PetscCall(AOCreateMemoryScalable_private(comm, napp, petsc, myapp, ao, aomems->app_loc)); in AOCreate_MemoryScalable()
447 PetscCall(AOCreateMemoryScalable_private(comm, napp, myapp, petsc, ao, aomems->petsc_loc)); in AOCreate_MemoryScalable()
[all …]
|
| /petsc/src/vec/is/ao/impls/mapping/ |
| H A D | aomapping.c | 238 PetscErrorCode AOCreateMapping(MPI_Comm comm, PetscInt napp, const PetscInt myapp[], const PetscInt… in AOCreateMapping() argument
263 PetscCall(PetscMPIIntCast(napp, &nnapp)); in AOCreateMapping()
277 PetscCall(PetscMalloc1(napp + 1, &petsc)); in AOCreateMapping()
278 for (PetscInt i = 0; i < napp; i++) petsc[i] = start + i; in AOCreateMapping()
351 PetscInt napp, npetsc; in AOCreateMappingIS() local
355 PetscCall(ISGetLocalSize(isapp, &napp)); in AOCreateMappingIS()
358 PetscCheck(napp == npetsc, PETSC_COMM_SELF, PETSC_ERR_ARG_SIZ, "Local IS lengths must match"); in AOCreateMappingIS()
365 PetscCall(AOCreateMapping(comm, napp, myapp, mypetsc, aoout)); in AOCreateMappingIS()
|
| /petsc/src/vec/is/ao/interface/ |
| H A D | ao.c | 460 PetscInt napp, npetsc; in AOSetIS() local
461 PetscCall(ISGetLocalSize(isapp, &napp)); in AOSetIS()
463 …cCheck(napp == npetsc, PETSC_COMM_SELF, PETSC_ERR_ARG_SIZ, "napp %" PetscInt_FMT " != npetsc %" Pe… in AOSetIS()
|