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