Lines Matching refs:sf
8 static PetscErrorCode PetscSFLinkStartCommunication_Gatherv(PetscSF sf, PetscSFLink link, PetscSFDi… in PetscSFLinkStartCommunication_Gatherv() argument
12 PetscSF_Gatherv *dat = (PetscSF_Gatherv *)sf->data; in PetscSFLinkStartCommunication_Gatherv()
19 …PetscCall(PetscSFLinkCopyRootBufferInCaseNotUseGpuAwareMPI(sf, link, PETSC_TRUE /* device2host bef… in PetscSFLinkStartCommunication_Gatherv()
21 …PetscCall(PetscSFLinkCopyLeafBufferInCaseNotUseGpuAwareMPI(sf, link, PETSC_TRUE /* device2host */)… in PetscSFLinkStartCommunication_Gatherv()
23 PetscCall(PetscObjectGetComm((PetscObject)sf, &comm)); in PetscSFLinkStartCommunication_Gatherv()
24 PetscCall(PetscMPIIntCast(sf->nroots, &count)); in PetscSFLinkStartCommunication_Gatherv()
25 …PetscCall(PetscSFLinkGetMPIBuffersAndRequests(sf, link, direction, &rootbuf, &leafbuf, &req, NULL)… in PetscSFLinkStartCommunication_Gatherv()
26 PetscCall(PetscSFLinkSyncStreamBeforeCallMPI(sf, link)); in PetscSFLinkStartCommunication_Gatherv()
36 static PetscErrorCode PetscSFSetCommunicationOps_Gatherv(PetscSF sf, PetscSFLink link) in PetscSFSetCommunicationOps_Gatherv() argument
43 PETSC_INTERN PetscErrorCode PetscSFFetchAndOpBegin_Gatherv(PetscSF sf, MPI_Datatype unit, PetscMemT… in PetscSFFetchAndOpBegin_Gatherv() argument
47 PetscCall(PetscSFBcastBegin(sf, unit, rootdata, leafupdate, MPI_REPLACE)); in PetscSFFetchAndOpBegin_Gatherv()
48 PetscCall(PetscSFBcastEnd(sf, unit, rootdata, leafupdate, MPI_REPLACE)); in PetscSFFetchAndOpBegin_Gatherv()
49 PetscCall(PetscSFReduceBegin(sf, unit, leafdata, rootdata, op)); in PetscSFFetchAndOpBegin_Gatherv()
53 PETSC_INTERN PetscErrorCode PetscSFCreate_Gatherv(PetscSF sf) in PetscSFCreate_Gatherv() argument
55 PetscSF_Gatherv *dat = (PetscSF_Gatherv *)sf->data; in PetscSFCreate_Gatherv()
58 sf->ops->BcastBegin = PetscSFBcastBegin_Basic; in PetscSFCreate_Gatherv()
59 sf->ops->BcastEnd = PetscSFBcastEnd_Basic; in PetscSFCreate_Gatherv()
60 sf->ops->ReduceBegin = PetscSFReduceBegin_Basic; in PetscSFCreate_Gatherv()
61 sf->ops->ReduceEnd = PetscSFReduceEnd_Basic; in PetscSFCreate_Gatherv()
64 sf->ops->SetUp = PetscSFSetUp_Allgatherv; in PetscSFCreate_Gatherv()
65 sf->ops->Reset = PetscSFReset_Allgatherv; in PetscSFCreate_Gatherv()
66 sf->ops->Destroy = PetscSFDestroy_Allgatherv; in PetscSFCreate_Gatherv()
67 sf->ops->GetGraph = PetscSFGetGraph_Allgatherv; in PetscSFCreate_Gatherv()
68 sf->ops->GetLeafRanks = PetscSFGetLeafRanks_Allgatherv; in PetscSFCreate_Gatherv()
69 sf->ops->GetRootRanks = PetscSFGetRootRanks_Allgatherv; in PetscSFCreate_Gatherv()
70 sf->ops->FetchAndOpEnd = PetscSFFetchAndOpEnd_Allgatherv; in PetscSFCreate_Gatherv()
71 sf->ops->CreateLocalSF = PetscSFCreateLocalSF_Allgatherv; in PetscSFCreate_Gatherv()
74 sf->ops->FetchAndOpBegin = PetscSFFetchAndOpBegin_Gatherv; in PetscSFCreate_Gatherv()
76 sf->ops->SetCommunicationOps = PetscSFSetCommunicationOps_Gatherv; in PetscSFCreate_Gatherv()
78 sf->collective = PETSC_TRUE; in PetscSFCreate_Gatherv()
81 sf->data = (void *)dat; in PetscSFCreate_Gatherv()