| /petsc/src/vec/is/sf/tests/ |
| H A D | ex3.c | 12 PetscScalar *rootdata, *leafdata, *leafupdate; in main() local
64 PetscCall(VecGetArray(y2, &leafupdate)); in main()
74 PetscCall(PetscSFFetchAndOpBegin(gathersf, MPIU_SCALAR, rootdata, leafdata, leafupdate, op)); in main()
75 PetscCall(PetscSFFetchAndOpEnd(gathersf, MPIU_SCALAR, rootdata, leafdata, leafupdate, op)); in main()
78 PetscCall(VecCreateMPIWithArray(PETSC_COMM_WORLD, 1, nleaves, PETSC_DECIDE, leafupdate, &gy2)); in main()
86 PetscCall(VecRestoreArray(y2, &leafupdate)); in main()
119 PetscCall(VecGetArray(y2, &leafupdate)); in main()
129 PetscCall(PetscSFFetchAndOpBegin(allgathersf, MPIU_SCALAR, rootdata, leafdata, leafupdate, op)); in main()
130 PetscCall(PetscSFFetchAndOpEnd(allgathersf, MPIU_SCALAR, rootdata, leafdata, leafupdate, op)); in main()
138 PetscCall(VecCreateMPIWithArray(PETSC_COMM_WORLD, 1, nleaves, PETSC_DECIDE, leafupdate, &gy2)); in main()
[all …]
|
| /petsc/src/vec/is/sf/tests/output/ |
| H A D | ex3_2.out | 17 Vec Object: leafupdate 2 MPI processes
47 Vec Object: leafupdate 2 MPI processes
81 Vec Object: leafupdate 2 MPI processes
|
| H A D | ex3_1.out | 17 Vec Object: leafupdate 2 MPI processes
47 Vec Object: leafupdate 2 MPI processes
81 Vec Object: leafupdate 2 MPI processes
|
| H A D | ex3_4.out | 18 Vec Object: leafupdate 3 MPI processes
50 Vec Object: leafupdate 3 MPI processes
101 Vec Object: leafupdate 3 MPI processes
|
| H A D | ex3_3.out | 18 Vec Object: leafupdate 3 MPI processes
50 Vec Object: leafupdate 3 MPI processes
101 Vec Object: leafupdate 3 MPI processes
|
| /petsc/src/vec/is/sf/impls/basic/gatherv/ |
| H A D | sfgatherv.c | 43 …otmtype, void *rootdata, PetscMemType leafmtype, const void *leafdata, void *leafupdate, MPI_Op op) in PetscSFFetchAndOpBegin_Gatherv() argument
47 PetscCall(PetscSFBcastBegin(sf, unit, rootdata, leafupdate, MPI_REPLACE)); in PetscSFFetchAndOpBegin_Gatherv()
48 PetscCall(PetscSFBcastEnd(sf, unit, rootdata, leafupdate, MPI_REPLACE)); in PetscSFFetchAndOpBegin_Gatherv()
|
| /petsc/src/vec/is/sf/impls/basic/allgatherv/ |
| H A D | sfallgatherv.c | 281 …otmtype, void *rootdata, PetscMemType leafmtype, const void *leafdata, void *leafupdate, MPI_Op op) in PetscSFFetchAndOpBegin_Allgatherv() argument
293 …PetscCall((*link->Memcpy)(link, leafmtype, leafupdate, leafmtype, leafdata, sf->nleaves * link->un… in PetscSFFetchAndOpBegin_Allgatherv()
304 …PetscCallMPI(MPI_Sendrecv_replace(leafupdate, count, unit, next, link->tag, prev, link->tag, comm,… in PetscSFFetchAndOpBegin_Allgatherv()
307 PetscCallMPI(MPI_Exscan(MPI_IN_PLACE, leafupdate, count, link->basicunit, op, comm)); in PetscSFFetchAndOpBegin_Allgatherv()
310 PetscCall(PetscSFBcastBegin(sf, unit, rootdata, leafupdate, op)); in PetscSFFetchAndOpBegin_Allgatherv()
311 PetscCall(PetscSFBcastEnd(sf, unit, rootdata, leafupdate, op)); in PetscSFFetchAndOpBegin_Allgatherv()
314 …PetscCall(PetscSFBcastToZero_Private(sf, unit, rootdata, leafupdate)); /* Using this line makes Al… in PetscSFFetchAndOpBegin_Allgatherv()
321 …v(PetscSF sf, MPI_Datatype unit, void *rootdata, const void *leafdata, void *leafupdate, MPI_Op op) in PetscSFFetchAndOpEnd_Allgatherv() argument
|
| /petsc/src/vec/is/sf/tutorials/ |
| H A D | ex1.c | 397 PetscInt *leafdata, *leafupdate, *rootdata; in main() local
398 …PetscCall(PetscMalloc3(nleavesalloc, &leafdata, nleavesalloc, &leafupdate, nrootsalloc, &rootdata)… in main()
403 PetscCall(PetscSFFetchAndOpBegin(sf, MPIU_INT, rootdata, leafdata, leafupdate, mop)); in main()
404 PetscCall(PetscSFFetchAndOpEnd(sf, MPIU_INT, rootdata, leafdata, leafupdate, mop)); in main()
408 PetscCall(PetscIntView(nleavesalloc, leafupdate, PETSC_VIEWER_STDOUT_WORLD)); in main()
409 PetscCall(PetscFree3(leafdata, leafupdate, rootdata)); in main()
|
| /petsc/src/binding/petsc4py/src/petsc4py/PETSc/ |
| H A D | SF.pyx | 642 …def fetchAndOpBegin(self, unit: Datatype, rootdata: ndarray, leafdata: ndarray, leafupdate: ndarra…
661 leafupdate
676 <void*>PyArray_DATA(leafupdate), cop))
678 …def fetchAndOpEnd(self, unit: Datatype, rootdata: ndarray, leafdata: ndarray, leafupdate: ndarray,…
692 leafupdate
707 <void*>PyArray_DATA(leafupdate), cop))
|
| /petsc/src/vec/is/sf/impls/basic/ |
| H A D | sfbasic.c | 420 …otmtype, void *rootdata, PetscMemType leafmtype, const void *leafdata, void *leafupdate, MPI_Op op) in PetscSFFetchAndOpBegin_Basic() argument
426 PetscCall(PetscSFLinkFetchAndOpLocal(sf, link, rootdata, leafdata, leafupdate, op)); in PetscSFFetchAndOpBegin_Basic()
430 …c(PetscSF sf, MPI_Datatype unit, void *rootdata, const void *leafdata, void *leafupdate, MPI_Op op) in PetscSFFetchAndOpEnd_Basic() argument
444 PetscCall(PetscSFLinkUnpackLeafData(sf, link, PETSCSF_REMOTE, leafupdate, MPI_REPLACE)); in PetscSFFetchAndOpEnd_Basic()
|
| H A D | sfpack.c | 224 …afstart, PetscSFPackOpt leafopt, const PetscInt *leafidx, const void *leafdata, void *leafupdate) \
226 Type *rdata = (Type *)rootdata, *lupdate = (Type *)leafupdate; \
1144 …al(PetscSF sf, PetscSFLink link, void *rootdata, const void *leafdata, void *leafupdate, MPI_Op op) in PetscSFLinkFetchAndOpLocal() argument
1162 …rootstart, rootopt, rootindices, rootdata, leafstart, leafopt, leafindices, leafdata, leafupdate)); in PetscSFLinkFetchAndOpLocal()
|
| /petsc/src/vec/is/sf/interface/ |
| H A D | sf.c | 1685 …n(PetscSF sf, MPI_Datatype unit, void *rootdata, const void *leafdata, void *leafupdate, MPI_Op op) in PetscSFFetchAndOpBegin() argument
1695 PetscCall(PetscGetMemType(leafupdate, &leafupdatemtype)); in PetscSFFetchAndOpBegin()
1697 …UseTypeMethod(sf, FetchAndOpBegin, unit, rootmtype, rootdata, leafmtype, leafdata, leafupdate, op); in PetscSFFetchAndOpBegin()
1728 …MemType leafmtype, const void *leafdata, PetscMemType leafupdatemtype, void *leafupdate, MPI_Op op) in PetscSFFetchAndOpWithMemTypeBegin() argument
1735 …UseTypeMethod(sf, FetchAndOpBegin, unit, rootmtype, rootdata, leafmtype, leafdata, leafupdate, op); in PetscSFFetchAndOpWithMemTypeBegin()
1760 …d(PetscSF sf, MPI_Datatype unit, void *rootdata, const void *leafdata, void *leafupdate, MPI_Op op) in PetscSFFetchAndOpEnd() argument
1765 PetscUseTypeMethod(sf, FetchAndOpEnd, unit, rootdata, leafdata, leafupdate, op); in PetscSFFetchAndOpEnd()
|
| /petsc/src/vec/is/sf/impls/basic/cupm/ |
| H A D | sfcupm_impl.hpp | 131 …eafstart, const PetscInt *leafopt, const PetscInt *leafidx, const Type *leafdata, Type *leafupdate) in d_FetchAndOpLocal() argument
139 for (PetscInt i = 0; i < MBS; i++) leafupdate[l + i] = op(rootdata[r + i], leafdata[l + i]); in d_FetchAndOpLocal()
377 …, PetscSFPackOpt leafopt, const PetscInt *leafidx, const void *leafdata, void *leafupdate) noexcept in FetchAndOpLocal() argument
384 …ootidx, (Type *)rootdata, leafstart, larray, leafidx, (const Type *)leafdata, (Type *)leafupdate)); in FetchAndOpLocal()
|
| /petsc/src/vec/is/sf/impls/window/ |
| H A D | sfwindow.c | 1094 …otmtype, void *rootdata, PetscMemType leafmtype, const void *leafdata, void *leafupdate, MPI_Op op) in PetscSFFetchAndOpBegin_Window() argument
1125 PetscCallMPI(MPI_Get(leafupdate, 1, mine[i], ranks[i], tdp, 1, remote[i], win)); in PetscSFFetchAndOpBegin_Window()
1130 …PetscCallMPI(MPI_Get_accumulate((void *)leafdata, 1, mine[i], leafupdate, 1, mine[i], ranks[i], td… in PetscSFFetchAndOpBegin_Window()
1140 …w(PetscSF sf, MPI_Datatype unit, void *rootdata, const void *leafdata, void *leafupdate, MPI_Op op) in PetscSFFetchAndOpEnd_Window() argument
|
| /petsc/src/vec/is/sf/impls/basic/kokkos/ |
| H A D | sfkok.kokkos.cxx | 385 …Type *rootdata = static_cast<Type *>(rootdata_), *leafupdate = static_cast<Type *>(… in FetchAndOpLocal()
394 for (int i = 0; i < MBS; i++) leafupdate[l + i] = op(rootdata[r + i], leafdata[l + i]); in FetchAndOpLocal()
|