Actual source code: sfallgatherv.h

  1: #pragma once

  3: #include <petsc/private/sfimpl.h>
  4: #include <../src/vec/is/sf/impls/basic/sfpack.h>
  5: #include <../src/vec/is/sf/impls/basic/sfbasic.h>

  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;

 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);