Home
last modified time | relevance | path

Searched refs:sendbuf (Results 1 – 22 of 22) sorted by relevance

/petsc/src/sys/utils/
H A Dmpiu.c150 PetscInt sendbuf[3], recvbuf[3]; in PetscGlobalMinMaxInt() local
154 …sendbuf[0] = hasminint ? PETSC_MIN_INT : -minMaxVal[0]; /* Note that -PETSC_INT_MIN = PETSC_INT_MI… in PetscGlobalMinMaxInt()
155 sendbuf[1] = minMaxVal[1]; in PetscGlobalMinMaxInt()
156 sendbuf[2] = hasminint ? 1 : 0; /* Are there PETSC_INT_MIN in minMaxVal[0]? */ in PetscGlobalMinMaxInt()
157 PetscCallMPI(MPIU_Allreduce(sendbuf, recvbuf, 3, MPIU_INT, MPI_MAX, comm)); in PetscGlobalMinMaxInt()
181 PetscReal sendbuf[2]; in PetscGlobalMinMaxReal() local
184 sendbuf[0] = -minMaxVal[0]; in PetscGlobalMinMaxReal()
185 sendbuf[1] = minMaxVal[1]; in PetscGlobalMinMaxReal()
186 PetscCallMPI(MPIU_Allreduce(sendbuf, minMaxValGlobal, 2, MPIU_REAL, MPIU_MAX, comm)); in PetscGlobalMinMaxReal()
/petsc/include/petsc/mpiuni/
H A Dmpi.h457 …#define MPI_Sendrecv(sendbuf, sendcount, sendtype, dest, sendtag, recvbuf, recvcount, recvtype, so… argument
458 …G(recvtag), MPIUNI_ARG(comm), MPIUNI_ARG(status), MPIUNI_Memcpy(recvbuf, sendbuf, (sendcount) * MP…
493 …sendbuf, sendcount, sendtype, recvbuf, recvcount, recvtype, root, comm) (MPIUNI_ARG(recvcount), MP… argument
494 …#define MPI_Gatherv(sendbuf, sendcount, sendtype, recvbuf, recvcounts, displs, recvtype, root, com… argument
495 …RG(recvtype), MPIUNI_ARG(root), MPIUNI_ARG(comm), MPIUNI_Memcpy(recvbuf, sendbuf, (sendcount) * MP…
496 …#define MPI_Gatherv_c(sendbuf, sendcount, sendtype, recvbuf, recvcounts, displs, recvtype, root, c… argument
497 …RG(recvtype), MPIUNI_ARG(root), MPIUNI_ARG(comm), MPIUNI_Memcpy(recvbuf, sendbuf, (sendcount) * MP…
498 #define MPI_Scatter(sendbuf, sendcount, sendtype, recvbuf, recvcount, recvtype, root, comm) \ argument
499 …RG(recvtype), MPIUNI_ARG(root), MPIUNI_ARG(comm), MPIUNI_Memcpy(recvbuf, sendbuf, (recvcount) * MP…
500 …#define MPI_Scatterv(sendbuf, sendcounts, displs, sendtype, recvbuf, recvcount, recvtype, root, co… argument
[all …]
/petsc/include/
H A Dpetsclog.h530 …sendbuf, recvbuf, count, datatype, op, comm) (PetscAddLogDouble(&petsc_allreduce_ct, &petsc_allred… argument
534 #define MPI_Reduce_scatter_block(sendbuf, recvbuf, recvcount, datatype, op, comm) \ argument
535 …llreduce_ct_th, PetscMPIParallelComm(comm)) || MPI_Reduce_scatter_block((sendbuf), (recvbuf), (rec…
537 #define MPI_Alltoall(sendbuf, sendcount, sendtype, recvbuf, recvcount, recvtype, comm) \ argument
538 …), (sendtype), (&petsc_send_len), (&petsc_send_len_th)) || MPI_Alltoall((sendbuf), (sendcount), (s…
540 …#define MPI_Alltoallv(sendbuf, sendcnts, sdispls, sendtype, recvbuf, recvcnts, rdispls, recvtype, … argument
541 …, (sendtype), (&petsc_send_len), (&petsc_send_len_th)) || MPI_Alltoallv((sendbuf), (sendcnts), (sd…
543 #define MPI_Allgather(sendbuf, sendcount, sendtype, recvbuf, recvcount, recvtype, comm) \ argument
544 …r_ct, &petsc_gather_ct_th, PetscMPIParallelComm(comm)) || MPI_Allgather((sendbuf), (sendcount), (s…
546 … #define MPI_Allgatherv(sendbuf, sendcount, sendtype, recvbuf, recvcount, displs, recvtype, comm) \ argument
[all …]
H A Dpetscsys.h2810 …#define MPIU_Scatterv(sendbuf, sendcounts, displs, sendtype, recvbuf, recvcount, recvtype, root, c… argument
2811 …#define MPIU_Gatherv(sendbuf, sendcount, sendtype, recvbuf, recvcounts, displs, recvtype, root, co… 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 …#define MPIU_Scatterv(sendbuf, sendcounts, displs, sendtype, recvbuf, recvcount, recvtype, root, c… argument
2886 …#define MPIU_Gatherv(sendbuf, sendcount, sendtype, recvbuf, recvcounts, displs, recvtype, root, co… argument
2888 …#define MPIU_Scatterv(sendbuf, sendcounts, displs, sendtype, recvbuf, recvcount, recvtype, root, c… argument
2889 …#define MPIU_Gatherv(sendbuf, sendcount, sendtype, recvbuf, recvcounts, displs, recvtype, root, co… argument
/petsc/src/sys/mpiuni/
H A Dmpi.c644 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 Dsubcomm.c198 PetscMPIInt size, icolor, duprank, *recvbuf, sendbuf[3], mysubsize, rank, *subsize; in PetscSubcommSetTypeGeneral() local
215 sendbuf[0] = color; in PetscSubcommSetTypeGeneral()
216 sendbuf[1] = mysubsize; in PetscSubcommSetTypeGeneral()
217 PetscCallMPI(MPI_Allgather(sendbuf, 2, MPI_INT, recvbuf, 2, MPI_INT, comm)); in PetscSubcommSetTypeGeneral()
/petsc/src/mat/impls/dense/mpi/
H A Dmpidense.h13 PetscScalar *sendbuf; member
H A Dmpidense.c2050 PetscCall(PetscFree2(atb->sendbuf, atb->recvcounts)); in MatProductCtxDestroy_MatTransMatMult_MPIDense_MPIDense()
2073 PetscScalar *carray, *sendbuf; in MatTransposeMatMultNumeric_MPIDense_MPIDense() local
2083 sendbuf = atb->sendbuf; in MatTransposeMatMultNumeric_MPIDense_MPIDense()
2157 for (i = ranges[proc]; i < ranges[proc + 1]; i++) sendbuf[k++] = atbarray[i + j * lda]; in MatTransposeMatMultNumeric_MPIDense_MPIDense()
2164 PetscCallMPI(MPI_Reduce_scatter(sendbuf, carray, recvcounts, MPIU_SCALAR, MPIU_SUM, comm)); in MatTransposeMatMultNumeric_MPIDense_MPIDense()
2203 PetscCall(PetscMalloc2(cM * cN, &atb->sendbuf, size, &atb->recvcounts)); in MatTransposeMatMultSymbolic_MPIDense_MPIDense()
2269 PetscScalar *sendbuf, *recvbuf = NULL, *cv; in MatMatTransposeMultNumeric_MPIDense_MPIDense_Cyclic() local
2296 sendbuf = (PetscScalar *)bv; in MatMatTransposeMultNumeric_MPIDense_MPIDense_Cyclic()
2298 sendbuf = abt->buf[0]; in MatMatTransposeMultNumeric_MPIDense_MPIDense_Cyclic()
2300 for (j = 0; j < bn; j++, k++) sendbuf[k] = bv[i * blda + j]; in MatMatTransposeMultNumeric_MPIDense_MPIDense_Cyclic()
[all …]
/petsc/src/vec/vec/impls/mpi/
H A Dpdvec.c19 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()
1103 vmpi->sendbuf = sendbuf; in VecSetPreallocationCOO_MPI()
1115 PetscScalar *a, *sendbuf = vmpi->sendbuf, *recvbuf = vmpi->recvbuf; in VecSetValuesCOO_MPI() local
1129 for (PetscInt i = 0; i < vmpi->sendlen; i++) sendbuf[i] = v[Cperm[i]]; in VecSetValuesCOO_MPI()
1132 …PetscCall(PetscSFReduceWithMemTypeBegin(vmpi->coo_sf, MPIU_SCALAR, PETSC_MEMTYPE_HOST, sendbuf, PE… in VecSetValuesCOO_MPI()
1139 PetscCall(PetscSFReduceEnd(vmpi->coo_sf, MPIU_SCALAR, sendbuf, recvbuf, MPI_REPLACE)); in VecSetValuesCOO_MPI()
H A Dpvecimpl.h59 PetscScalar *sendbuf, *recvbuf; /* Buffers for remote values in VecSetValuesCOO() */ member
/petsc/src/vec/vec/impls/mpi/kokkos/
H A Dmpikok.kokkos.cxx160 PetscScalarKokkosView &sendbuf = veckok->sendbuf_d; in VecSetValuesCOO_MPIKokkos() local
177 …ecutionSpace(), 0, vecmpi->sendlen), KOKKOS_LAMBDA(const PetscCount i) { sendbuf(i) = vv(Cperm(i))… in VecSetValuesCOO_MPIKokkos()
178 …educeWithMemTypeBegin(vecmpi->coo_sf, MPIU_SCALAR, PETSC_MEMTYPE_KOKKOS, sendbuf.data(), PETSC_MEM… in VecSetValuesCOO_MPIKokkos()
190 …PetscCall(PetscSFReduceEnd(vecmpi->coo_sf, MPIU_SCALAR, sendbuf.data(), recvbuf.data(), MPI_REPLAC… in VecSetValuesCOO_MPIKokkos()
/petsc/src/ksp/ksp/impls/cg/pipelcg/
H A Dpipelcg.c102 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/vec/vec/utils/
H A Dcomb.c23 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/tao/constrained/impls/ipm/
H A Dpdipm.c154 PetscInt sendbuf[5], recvbuf[5]; in TaoPDIPMSetUpBounds() local
184 sendbuf[0] = pdipm->nxlb; in TaoPDIPMSetUpBounds()
185 sendbuf[1] = pdipm->nxub; in TaoPDIPMSetUpBounds()
186 sendbuf[2] = pdipm->nxfixed; in TaoPDIPMSetUpBounds()
187 sendbuf[3] = pdipm->nxbox; in TaoPDIPMSetUpBounds()
188 sendbuf[4] = pdipm->nxfree; in TaoPDIPMSetUpBounds()
190 PetscCallMPI(MPIU_Allreduce(sendbuf, recvbuf, 5, MPIU_INT, MPI_SUM, comm)); in TaoPDIPMSetUpBounds()
/petsc/src/mat/impls/aij/mpi/mpihipsparse/
H A Dmpiaijhipsparse.hip.cxx38 PetscCallHIP(hipFree(coo->sendbuf)); in MatCOOStructDestroy_MPIAIJHIPSPARSE()
100 PetscCallHIP(hipMalloc((void **)&coo_d->sendbuf, coo_h->sendlen * sizeof(PetscScalar))); in MatSetPreallocationCOO_MPIAIJHIPSPARSE()
177 const auto &vsend = coo->sendbuf; in MatSetValuesCOO_MPIAIJHIPSPARSE()
/petsc/src/mat/impls/aij/mpi/mpicusparse/
H A Dmpiaijcusparse.cu38 PetscCallCUDA(cudaFree(coo->sendbuf)); in MatCOOStructDestroy_MPIAIJCUSPARSE()
100 PetscCallCUDA(cudaMalloc((void **)&coo_d->sendbuf, coo_h->sendlen * sizeof(PetscScalar))); in MatSetPreallocationCOO_MPIAIJCUSPARSE()
177 const auto &vsend = coo->sendbuf; in MatSetValuesCOO_MPIAIJCUSPARSE()
/petsc/src/vec/vec/impls/seq/kokkos/
H A Dveckokkosimpl.hpp117 …or_view_and_copy(DefaultMemorySpace(), PetscScalarKokkosViewHost(vecmpi->sendbuf, vecmpi->sendlen)… in SetUpCOO()
/petsc/src/mat/impls/aij/mpi/
H A Dmpiaij.h87 PetscScalar *sendbuf, *recvbuf; /* Buffers for remote values in MatSetValuesCOO() */ member
H A Dmpiov.c1166 MatScalar *sendbuf; in MatCreateSubMatrix_MPIAIJ_All() local
1174 sendbuf = PetscSafePointerPlusOffset(b->a, b->i[rstarts[rank]]); in MatCreateSubMatrix_MPIAIJ_All()
1186 sendbuf[cnt++] = *b_sendbuf++; in MatCreateSubMatrix_MPIAIJ_All()
1192 for (PetscInt j = ad->i[i]; j < ad->i[i + 1]; j++) sendbuf[cnt++] = *a_sendbuf++; in MatCreateSubMatrix_MPIAIJ_All()
1196 sendbuf[cnt++] = *b_sendbuf++; in MatCreateSubMatrix_MPIAIJ_All()
H A Dmpiaij.c6334 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
6722 sendbuf = coo->sendbuf; in MatSetValuesCOO_MPIAIJ()
6740 for (PetscCount i = 0; i < coo->sendlen; i++) sendbuf[i] = v[Cperm1[i]]; in MatSetValuesCOO_MPIAIJ()
6743 …PetscCall(PetscSFReduceWithMemTypeBegin(coo->sf, MPIU_SCALAR, PETSC_MEMTYPE_HOST, sendbuf, PETSC_M… in MatSetValuesCOO_MPIAIJ()
6755 PetscCall(PetscSFReduceEnd(coo->sf, MPIU_SCALAR, sendbuf, recvbuf, MPI_REPLACE)); in MatSetValuesCOO_MPIAIJ()
/petsc/src/vec/vec/impls/mpi/cupm/
H A Dvecmpicupm_impl.hpp260 make_coo_pair(vcu->sendbuf_d, vmpi->sendbuf, vmpi->sendlen), in SetPreallocationCOO()
/petsc/src/mat/impls/aij/mpi/kokkos/
H A Dmpiaijkok.kokkos.cxx1512 MatScalarKokkosView sendbuf, recvbuf; member
1535 …sendbuf = Kokkos::create_mirror_view(Kokkos::WithoutInitializing, exec, MatScalarKokkosViewHost(co… in MatCOOStruct_MPIAIJKokkos()
1597 const auto &vsend = coo->sendbuf; in MatSetValuesCOO_MPIAIJKokkos()