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