| /petsc/include/petsc/mpiuni/ |
| H A D | mpi.h | 457 …#define MPI_Sendrecv(sendbuf, sendcount, sendtype, dest, sendtag, recvbuf, recvcount, recvtype, so… argument
458 …MPIUNI_ARG(recvtag), MPIUNI_ARG(comm), MPIUNI_ARG(status), MPIUNI_Memcpy(recvbuf, sendbuf, (sendco…
493 …pe, recvbuf, recvcount, recvtype, root, comm) (MPIUNI_ARG(recvcount), MPIUNI_ARG(root), MPIUNI_ARG… argument
494 …#define MPI_Gatherv(sendbuf, sendcount, sendtype, recvbuf, recvcounts, displs, recvtype, root, com… argument
495 … MPIUNI_ARG(recvtype), MPIUNI_ARG(root), MPIUNI_ARG(comm), MPIUNI_Memcpy(recvbuf, sendbuf, (sendco…
496 …#define MPI_Gatherv_c(sendbuf, sendcount, sendtype, recvbuf, recvcounts, displs, recvtype, root, c… argument
497 … MPIUNI_ARG(recvtype), MPIUNI_ARG(root), MPIUNI_ARG(comm), MPIUNI_Memcpy(recvbuf, sendbuf, (sendco…
498 #define MPI_Scatter(sendbuf, sendcount, sendtype, recvbuf, recvcount, recvtype, root, comm) \ argument
499 …PIUNI_ARG(sendtype), MPIUNI_ARG(recvbuf), MPIUNI_ARG(recvtype), MPIUNI_ARG(root), MPIUNI_ARG(comm)…
500 …#define MPI_Scatterv(sendbuf, sendcounts, displs, sendtype, recvbuf, recvcount, recvtype, root, co… argument
[all …]
|
| /petsc/include/ |
| H A D | petsclog.h | 530 …recvbuf, count, datatype, op, comm) (PetscAddLogDouble(&petsc_allreduce_ct, &petsc_allreduce_ct_th… argument
534 #define MPI_Reduce_scatter_block(sendbuf, recvbuf, recvcount, datatype, op, comm) \ argument
535 …_th, PetscMPIParallelComm(comm)) || MPI_Reduce_scatter_block((sendbuf), (recvbuf), (recvcount), (d…
537 #define MPI_Alltoall(sendbuf, sendcount, sendtype, recvbuf, recvcount, recvtype, comm) \ argument
538 …petsc_send_len_th)) || MPI_Alltoall((sendbuf), (sendcount), (sendtype), (recvbuf), (recvcount), (r…
540 …#define MPI_Alltoallv(sendbuf, sendcnts, sdispls, sendtype, recvbuf, recvcnts, rdispls, recvtype, … argument
541 …len_th)) || MPI_Alltoallv((sendbuf), (sendcnts), (sdispls), (sendtype), (recvbuf), (recvcnts), (rd…
543 #define MPI_Allgather(sendbuf, sendcount, sendtype, recvbuf, recvcount, recvtype, comm) \ argument
544 …arallelComm(comm)) || MPI_Allgather((sendbuf), (sendcount), (sendtype), (recvbuf), (recvcount), (r…
546 … #define MPI_Allgatherv(sendbuf, sendcount, sendtype, recvbuf, recvcount, displs, recvtype, comm) \ argument
[all …]
|
| H A D | petscsys.h | 2810 …counts, displs, sendtype, recvbuf, recvcount, recvtype, root, comm) MPI_Scatterv(sendbuf, sendcoun… argument
2811 …buf, sendcount, sendtype, recvbuf, recvcounts, displs, recvtype, root, comm) MPI_Gatherv(sendbuf,… argument
2813 …#define MPIU_Scatterv(sendbuf, sendcount, displs, sendtype, recvbuf, recvcount, recvtype, root, co… argument
2815 …#define MPIU_Gatherv(sendbuf, sendcount, sendtype, recvbuf, recvcounts, displs, recvtype, root, co… argument
2885 …counts, displs, sendtype, recvbuf, recvcount, recvtype, root, comm) MPI_Scatterv(sendbuf, sendcoun… argument
2886 …buf, sendcount, sendtype, recvbuf, recvcounts, displs, recvtype, root, comm) MPI_Gatherv(sendbuf,… argument
2888 …pe, recvbuf, recvcount, recvtype, root, comm) MPI_Scatterv_c(sendbuf, (const MPI_Count *)(sendcoun… argument
2889 …uf, sendcount, sendtype, recvbuf, recvcounts, displs, recvtype, root, comm) MPI_Gatherv_c(sendbuf… argument
|
| /petsc/src/sys/utils/ |
| H A D | mpiu.c | 150 PetscInt sendbuf[3], recvbuf[3]; in PetscGlobalMinMaxInt() local
157 PetscCallMPI(MPIU_Allreduce(sendbuf, recvbuf, 3, MPIU_INT, MPI_MAX, comm)); in PetscGlobalMinMaxInt()
158 minMaxValGlobal[0] = recvbuf[2] ? PETSC_INT_MIN : -recvbuf[0]; in PetscGlobalMinMaxInt()
159 minMaxValGlobal[1] = recvbuf[1]; in PetscGlobalMinMaxInt()
|
| /petsc/src/sys/mpiuni/ |
| H A D | mpi.c | 644 PETSC_EXTERN void petsc_mpi_reduce_(void *sendbuf, void *recvbuf, int *count, int *datatype, int *o… in petsc_mpi_reduce_() argument
646 *ierr = MPI_Reduce(sendbuf, recvbuf, *count, *datatype, *op, *root, *comm); in petsc_mpi_reduce_()
649 PETSC_EXTERN void petsc_mpi_allreduce_(void *sendbuf, void *recvbuf, int *count, int *datatype, int… in petsc_mpi_allreduce_() argument
651 *ierr = MPI_Allreduce(sendbuf, recvbuf, *count, *datatype, *op, *comm); in petsc_mpi_allreduce_()
664 PETSC_EXTERN void petsc_mpi_gather_(void *sendbuf, int *scount, int *sdatatype, void *recvbuf, int … in petsc_mpi_gather_() argument
666 *ierr = MPI_Gather(sendbuf, *scount, *sdatatype, recvbuf, rcount, rdatatype, *root, *comm); in petsc_mpi_gather_()
669 PETSC_EXTERN void petsc_mpi_allgather_(void *sendbuf, int *scount, int *sdatatype, void *recvbuf, i… in petsc_mpi_allgather_() argument
671 *ierr = MPI_Allgather(sendbuf, *scount, *sdatatype, recvbuf, rcount, rdatatype, *comm); in petsc_mpi_allgather_()
674 PETSC_EXTERN void petsc_mpi_scan_(void *sendbuf, void *recvbuf, int *count, int *datatype, int *op,… in petsc_mpi_scan_() argument
676 *ierr = MPIUNI_Memcpy(recvbuf, sendbuf, (*count) * MPI_sizeof(*datatype)); in petsc_mpi_scan_()
[all …]
|
| /petsc/src/sys/objects/ |
| H A D | subcomm.c | 198 PetscMPIInt size, icolor, duprank, *recvbuf, sendbuf[3], mysubsize, rank, *subsize; in PetscSubcommSetTypeGeneral() local
210 PetscCall(PetscMalloc1(2 * size, &recvbuf)); in PetscSubcommSetTypeGeneral()
217 PetscCallMPI(MPI_Allgather(sendbuf, 2, MPI_INT, recvbuf, 2, MPI_INT, comm)); in PetscSubcommSetTypeGeneral()
220 for (i = 0; i < 2 * size; i += 2) subsize[recvbuf[i]] = recvbuf[i + 1]; in PetscSubcommSetTypeGeneral()
221 PetscCall(PetscFree(recvbuf)); in PetscSubcommSetTypeGeneral()
|
| /petsc/src/vec/vec/impls/mpi/ |
| H A D | pdvec.c | 19 PetscCall(PetscFree4(vmpi->imap2, vmpi->jmap2, vmpi->sendbuf, vmpi->recvbuf)); in VecResetPreallocationCOO_MPI()
1059 PetscScalar *sendbuf, *recvbuf; in VecSetPreallocationCOO_MPI() local
1073 PetscCall(PetscMalloc4(nnz2, &imap2, nnz2 + 1, &jmap2, sendlen, &sendbuf, recvlen, &recvbuf)); in VecSetPreallocationCOO_MPI()
1104 vmpi->recvbuf = recvbuf; in VecSetPreallocationCOO_MPI()
1115 PetscScalar *a, *sendbuf = vmpi->sendbuf, *recvbuf = vmpi->recvbuf; in VecSetValuesCOO_MPI() local
1132 …vmpi->coo_sf, MPIU_SCALAR, PETSC_MEMTYPE_HOST, sendbuf, PETSC_MEMTYPE_HOST, recvbuf, MPI_REPLACE)); in VecSetValuesCOO_MPI()
1139 PetscCall(PetscSFReduceEnd(vmpi->coo_sf, MPIU_SCALAR, sendbuf, recvbuf, MPI_REPLACE)); in VecSetValuesCOO_MPI()
1143 for (PetscCount k = jmap2[i]; k < jmap2[i + 1]; k++) a[imap2[i]] += recvbuf[perm2[k]]; in VecSetValuesCOO_MPI()
|
| H A D | pvecimpl.h | 59 PetscScalar *sendbuf, *recvbuf; /* Buffers for remote values in VecSetValuesCOO() */ member
|
| /petsc/src/vec/vec/impls/mpi/kokkos/ |
| H A D | mpikok.kokkos.cxx | 161 PetscScalarKokkosView &recvbuf = veckok->recvbuf_d; in VecSetValuesCOO_MPIKokkos() local
178 …MPIU_SCALAR, PETSC_MEMTYPE_KOKKOS, sendbuf.data(), PETSC_MEMTYPE_KOKKOS, recvbuf.data(), MPI_REPLA… in VecSetValuesCOO_MPIKokkos()
190 …PetscCall(PetscSFReduceEnd(vecmpi->coo_sf, MPIU_SCALAR, sendbuf.data(), recvbuf.data(), MPI_REPLAC… in VecSetValuesCOO_MPIKokkos()
195 for (PetscCount k = jmap2(i); k < jmap2(i + 1); k++) xv(imap2(i)) += recvbuf(perm2(k)); in VecSetValuesCOO_MPIKokkos()
|
| /petsc/src/vec/vec/utils/ |
| H A D | comb.c | 23 static PetscMPIInt MPIU_Iallreduce(void *sendbuf, void *recvbuf, PetscMPIInt count, MPI_Datatype da… in MPIU_Iallreduce() argument
28 err = MPI_Iallreduce(sendbuf, recvbuf, count, datatype, op, comm, request); in MPIU_Iallreduce()
30 err = MPIU_Allreduce(sendbuf, recvbuf, count, datatype, op, comm); in MPIU_Iallreduce()
|
| /petsc/src/ksp/ksp/impls/cg/pipelcg/ |
| H A D | pipelcg.c | 102 static PetscMPIInt MPIU_Iallreduce(void *sendbuf, void *recvbuf, PetscMPIInt count, MPI_Datatype da… in MPIU_Iallreduce() argument
106 err = MPI_Iallreduce(sendbuf, recvbuf, count, datatype, op, comm, request); in MPIU_Iallreduce()
108 err = MPI_Allreduce(sendbuf, recvbuf, count, datatype, op, comm); in MPIU_Iallreduce()
|
| /petsc/src/tao/constrained/impls/ipm/ |
| H A D | pdipm.c | 154 PetscInt sendbuf[5], recvbuf[5]; in TaoPDIPMSetUpBounds() local
190 PetscCallMPI(MPIU_Allreduce(sendbuf, recvbuf, 5, MPIU_INT, MPI_SUM, comm)); in TaoPDIPMSetUpBounds()
191 pdipm->Nxlb = recvbuf[0]; in TaoPDIPMSetUpBounds()
192 pdipm->Nxub = recvbuf[1]; in TaoPDIPMSetUpBounds()
193 pdipm->Nxfixed = recvbuf[2]; in TaoPDIPMSetUpBounds()
194 pdipm->Nxbox = recvbuf[3]; in TaoPDIPMSetUpBounds()
195 pdipm->Nxfree = recvbuf[4]; in TaoPDIPMSetUpBounds()
|
| /petsc/src/mat/impls/aij/mpi/mpihipsparse/ |
| H A D | mpiaijhipsparse.hip.cxx | 39 PetscCallHIP(hipFree(coo->recvbuf)); in MatCOOStructDestroy_MPIAIJHIPSPARSE()
101 PetscCallHIP(hipMalloc((void **)&coo_d->recvbuf, coo_h->recvlen * sizeof(PetscScalar))); in MatSetPreallocationCOO_MPIAIJHIPSPARSE()
178 const auto &v2 = coo->recvbuf; in MatSetValuesCOO_MPIAIJHIPSPARSE()
|
| /petsc/src/mat/impls/aij/mpi/mpicusparse/ |
| H A D | mpiaijcusparse.cu | 39 PetscCallCUDA(cudaFree(coo->recvbuf)); in MatCOOStructDestroy_MPIAIJCUSPARSE()
101 PetscCallCUDA(cudaMalloc((void **)&coo_d->recvbuf, coo_h->recvlen * sizeof(PetscScalar))); in MatSetPreallocationCOO_MPIAIJCUSPARSE()
178 const auto &v2 = coo->recvbuf; in MatSetValuesCOO_MPIAIJCUSPARSE()
|
| /petsc/src/vec/vec/impls/seq/kokkos/ |
| H A D | veckokkosimpl.hpp | 118 …or_view_and_copy(DefaultMemorySpace(), PetscScalarKokkosViewHost(vecmpi->recvbuf, vecmpi->recvlen)… in SetUpCOO()
|
| /petsc/src/mat/impls/aij/mpi/ |
| H A D | mpiaij.h | 87 PetscScalar *sendbuf, *recvbuf; /* Buffers for remote values in MatSetValuesCOO() */ member
|
| H A D | mpiaij.c | 6334 PetscCall(PetscFree2(coo->sendbuf, coo->recvbuf)); in MatCOOStructDestroy_MPIAIJ()
6696 PetscCall(PetscMalloc2(coo->sendlen, &coo->sendbuf, coo->recvlen, &coo->recvbuf)); in MatSetPreallocationCOO_MPIAIJ()
6710 PetscScalar *sendbuf, *recvbuf; in MatSetValuesCOO_MPIAIJ() local
6723 recvbuf = coo->recvbuf; in MatSetValuesCOO_MPIAIJ()
6743 …egin(coo->sf, MPIU_SCALAR, PETSC_MEMTYPE_HOST, sendbuf, PETSC_MEMTYPE_HOST, recvbuf, MPI_REPLACE)); in MatSetValuesCOO_MPIAIJ()
6755 PetscCall(PetscSFReduceEnd(coo->sf, MPIU_SCALAR, sendbuf, recvbuf, MPI_REPLACE)); in MatSetValuesCOO_MPIAIJ()
6759 for (PetscCount k = Ajmap2[i]; k < Ajmap2[i + 1]; k++) Aa[Aimap2[i]] += recvbuf[Aperm2[k]]; in MatSetValuesCOO_MPIAIJ()
6762 for (PetscCount k = Bjmap2[i]; k < Bjmap2[i + 1]; k++) Ba[Bimap2[i]] += recvbuf[Bperm2[k]]; in MatSetValuesCOO_MPIAIJ()
|
| /petsc/src/mat/impls/dense/mpi/ |
| H A D | mpidense.c | 2269 PetscScalar *sendbuf, *recvbuf = NULL, *cv; in MatMatTransposeMultNumeric_MPIDense_MPIDense_Cyclic() local
2321 recvbuf = (i & 1) ? abt->buf[0] : abt->buf[1]; in MatMatTransposeMultNumeric_MPIDense_MPIDense_Cyclic()
2323 PetscCallMPI(MPIU_Irecv(recvbuf, recvsiz, MPIU_SCALAR, recvfrom, abt->tag, comm, &reqs[1])); in MatMatTransposeMultNumeric_MPIDense_MPIDense_Cyclic()
2336 sendbuf = recvbuf; in MatMatTransposeMultNumeric_MPIDense_MPIDense_Cyclic()
2353 PetscScalar *cv, *sendbuf, *recvbuf; in MatMatTransposeMultNumeric_MPIDense_MPIDense_Allgatherv() local
2376 recvbuf = abt->buf[1]; in MatMatTransposeMultNumeric_MPIDense_MPIDense_Allgatherv()
2382 …PetscCallMPI(MPI_Allgatherv(sendbuf, ibn, MPIU_SCALAR, recvbuf, abt->recvcounts, abt->recvdispls, … in MatMatTransposeMultNumeric_MPIDense_MPIDense_Allgatherv()
2386 …llBLAS("BLASgemm", BLASgemm_("N", "N", &cm, &cn, &ck, &_DOne, av, &alda, recvbuf, &ck, &_DZero, cv… in MatMatTransposeMultNumeric_MPIDense_MPIDense_Allgatherv()
|
| /petsc/src/vec/vec/impls/mpi/cupm/ |
| H A D | vecmpicupm_impl.hpp | 261 make_coo_pair(vcu->recvbuf_d, vmpi->recvbuf, vmpi->recvlen) in SetPreallocationCOO()
|
| /petsc/src/mat/impls/aij/mpi/kokkos/ |
| H A D | mpiaijkok.kokkos.cxx | 1512 MatScalarKokkosView sendbuf, recvbuf; member
1536 …recvbuf = Kokkos::create_mirror_view(Kokkos::WithoutInitializing, exec, MatScalarKokkosViewHost(co… in MatCOOStruct_MPIAIJKokkos()
1598 const auto &v2 = coo->recvbuf; in MatSetValuesCOO_MPIAIJKokkos()
|
| /petsc/src/mat/impls/aij/mpi/mumps/ |
| H A D | mumps.c | 1848 PetscScalar *recvbuf = NULL; in MatMumpsSetUpDistRHSInfo() local
1891 …recvbuf = (nrhs == 1) ? mumps->rhs_loc : mumps->rhs_recvbuf; /* Directly use rhs_loc[] as recvbuf.… in MatMumpsSetUpDistRHSInfo()
1895 …PetscCallMPI(MPI_Gatherv(array, sendcount, MPIU_SCALAR, recvbuf, mumps->rhs_recvcounts, mumps->rhs… in MatMumpsSetUpDistRHSInfo()
|