Date: 2022-03-31T04:46:35+00:00

Actual source code: sfgather.c

      1: #include <../src/vec/is/sf/impls/basic/gatherv/sfgatherv.h>
      2: #include <../src/vec/is/sf/impls/basic/allgather/sfallgather.h>

      4: /* Reuse the type. The difference is some fields (i.e., displs, recvcounts) are not used in Gather, which is not a big deal */
      5: typedef PetscSF_Allgatherv PetscSF_Gather;

      7: PETSC_INTERN PetscErrorCode PetscSFBcastBegin_Gather(PetscSF sf,MPI_Datatype unit,PetscMemType rootmtype,const void *rootdata,PetscMemType leafmtype,void *leafdata,MPI_Op op)
      8: {
      9:   PetscSFLink          link;
     10:   PetscMPIInt          sendcount;
     11:   MPI_Comm             comm;
     12:   void                 *rootbuf = NULL,*leafbuf = NULL;
     13:   MPI_Request          *req;

     15:   PetscSFLinkCreate(sf,unit,rootmtype,rootdata,leafmtype,leafdata,op,PETSCSF_BCAST,&link);
     16:   PetscSFLinkPackRootData(sf,link,PETSCSF_REMOTE,rootdata);
     17:   PetscSFLinkCopyRootBufferInCaseNotUseGpuAwareMPI(sf,link,PETSC_TRUE/* device2host before sending */);
     18:   PetscObjectGetComm((PetscObject)sf,&comm);
     19:   PetscMPIIntCast(sf->nroots,&sendcount);
     20:   PetscSFLinkGetMPIBuffersAndRequests(sf,link,PETSCSF_../../../../../..2LEAF,&rootbuf,&leafbuf,&req,NULL);
     21:   PetscSFLinkSyncStreamBeforeCallMPI(sf,link,PETSCSF_../../../../../..2LEAF);
     22:   MPIU_Igather(rootbuf == leafbuf ? MPI_IN_PLACE : rootbuf,sendcount,unit,leafbuf,sendcount,unit,0/*rank 0*/,comm,req);
     23:   return 0;
     24: }

     26: static PetscErrorCode PetscSFReduceBegin_Gather(PetscSF sf,MPI_Datatype unit,PetscMemType leafmtype,const void *leafdata,PetscMemType rootmtype,void *rootdata,MPI_Op op)
     27: {
     28:   PetscSFLink          link;
     29:   PetscMPIInt          recvcount;
     30:   MPI_Comm             comm;
     31:   void                 *rootbuf = NULL,*leafbuf = NULL;
     32:   MPI_Request          *req;

     34:   PetscSFLinkCreate(sf,unit,rootmtype,rootdata,leafmtype,leafdata,op,PETSCSF_REDUCE,&link);
     35:   PetscSFLinkPackLeafData(sf,link,PETSCSF_REMOTE,leafdata);
     36:   PetscSFLinkCopyLeafBufferInCaseNotUseGpuAwareMPI(sf,link,PETSC_TRUE/* device2host before sending */);
     37:   PetscObjectGetComm((PetscObject)sf,&comm);
     38:   PetscMPIIntCast(sf->nroots,&recvcount);
     39:   PetscSFLinkGetMPIBuffersAndRequests(sf,link,PETSCSF_LEAF2../../../../../..,&rootbuf,&leafbuf,&req,NULL);
     40:   PetscSFLinkSyncStreamBeforeCallMPI(sf,link,PETSCSF_LEAF2../../../../../..);
     41:   MPIU_Iscatter(leafbuf,recvcount,unit,rootbuf == leafbuf ? MPI_IN_PLACE : rootbuf,recvcount,unit,0/*rank 0*/,comm,req);
     42:   return 0;
     43: }

     45: PETSC_INTERN PetscErrorCode PetscSFCreate_Gather(PetscSF sf)
     46: {
     47:   PetscSF_Gather  *dat = (PetscSF_Gather*)sf->data;

     49:   sf->ops->BcastEnd        = PetscSFBcastEnd_Basic;
     50:   sf->ops->ReduceEnd       = PetscSFReduceEnd_Basic;

     52:   /* Inherit from Allgatherv */
     53:   sf->ops->Reset           = PetscSFReset_Allgatherv;
     54:   sf->ops->Destroy         = PetscSFDestroy_Allgatherv;
     55:   sf->ops->GetGraph        = PetscSFGetGraph_Allgatherv;
     56:   sf->ops->GetRootRanks    = PetscSFGetRootRanks_Allgatherv;
     57:   sf->ops->GetLeafRanks    = PetscSFGetLeafRanks_Allgatherv;
     58:   sf->ops->FetchAndOpEnd   = PetscSFFetchAndOpEnd_Allgatherv;
     59:   sf->ops->CreateLocalSF   = PetscSFCreateLocalSF_Allgatherv;

     61:   /* Inherit from Allgather */
     62:   sf->ops->SetUp           = PetscSFSetUp_Allgather;

     64:   /* Inherit from Gatherv */
     65:   sf->ops->FetchAndOpBegin = PetscSFFetchAndOpBegin_Gatherv;

     67:   /* Gather stuff */
     68:   sf->ops->BcastBegin      = PetscSFBcastBegin_Gather;
     69:   sf->ops->ReduceBegin     = PetscSFReduceBegin_Gather;

     71:   PetscNewLog(sf,&dat);
     72:   sf->data = (void*)dat;
     73:   return 0;
     74: }