xref: /petsc/src/vec/is/sf/impls/basic/allgatherv/sfallgatherv.h (revision d5b43468fb8780a8feea140ccd6fa3e6a50411cc)
1 #ifndef _SFALLGATHERV_H
2 #define _SFALLGATHERV_H
3 
4 #include <petsc/private/sfimpl.h> /*I "petscsf.h" I*/
5 #include <../src/vec/is/sf/impls/basic/sfpack.h>
6 #include <../src/vec/is/sf/impls/basic/sfbasic.h>
7 
8 typedef struct {
9   SFBASICHEADER;
10   PetscMPIInt *displs, *recvcounts;
11   /* special treatment for one-to-all patterns detected at setup time */
12   PetscBool   bcast_pattern; /* bcast here means one-to-all; we might do MPI_Reduce with this pattern */
13   PetscMPIInt bcast_root;    /* the root rank in MPI_Bcast */
14 } PetscSF_Allgatherv;
15 
16 PETSC_INTERN PetscErrorCode PetscSFSetUp_Allgather(PetscSF);
17 PETSC_INTERN PetscErrorCode PetscSFSetUp_Allgatherv(PetscSF);
18 PETSC_INTERN PetscErrorCode PetscSFReset_Allgatherv(PetscSF);
19 PETSC_INTERN PetscErrorCode PetscSFDestroy_Allgatherv(PetscSF);
20 PETSC_INTERN PetscErrorCode PetscSFFetchAndOpBegin_Allgatherv(PetscSF sf, MPI_Datatype, PetscMemType, void *, PetscMemType, const void *, void *, MPI_Op);
21 PETSC_INTERN PetscErrorCode PetscSFFetchAndOpEnd_Allgatherv(PetscSF, MPI_Datatype, void *, const void *, void *, MPI_Op);
22 PETSC_INTERN PetscErrorCode PetscSFGetRootRanks_Allgatherv(PetscSF, PetscInt *, const PetscMPIInt **, const PetscInt **, const PetscInt **, const PetscInt **);
23 PETSC_INTERN PetscErrorCode PetscSFGetLeafRanks_Allgatherv(PetscSF, PetscInt *, const PetscMPIInt **, const PetscInt **, const PetscInt **);
24 PETSC_INTERN PetscErrorCode PetscSFCreateLocalSF_Allgatherv(PetscSF, PetscSF *);
25 PETSC_INTERN PetscErrorCode PetscSFGetGraph_Allgatherv(PetscSF, PetscInt *, PetscInt *, const PetscInt **, const PetscSFNode **);
26 PETSC_INTERN PetscErrorCode PetscSFReduceEnd_Allgatherv(PetscSF, MPI_Datatype, const void *, void *, MPI_Op);
27 #endif
28