Lines Matching refs:sf
7 static PetscErrorCode PetscSFLinkStartCommunication_Gather(PetscSF sf, PetscSFLink link, PetscSFDir… in PetscSFLinkStartCommunication_Gather() argument
17 …PetscCall(PetscSFLinkCopyRootBufferInCaseNotUseGpuAwareMPI(sf, link, PETSC_TRUE /* device2host bef… in PetscSFLinkStartCommunication_Gather()
19 …PetscCall(PetscSFLinkCopyLeafBufferInCaseNotUseGpuAwareMPI(sf, link, PETSC_TRUE /* device2host */)… in PetscSFLinkStartCommunication_Gather()
21 PetscCall(PetscObjectGetComm((PetscObject)sf, &comm)); in PetscSFLinkStartCommunication_Gather()
22 PetscCall(PetscMPIIntCast(sf->nroots, &count)); in PetscSFLinkStartCommunication_Gather()
23 …PetscCall(PetscSFLinkGetMPIBuffersAndRequests(sf, link, direction, &rootbuf, &leafbuf, &req, NULL)… in PetscSFLinkStartCommunication_Gather()
24 PetscCall(PetscSFLinkSyncStreamBeforeCallMPI(sf, link)); in PetscSFLinkStartCommunication_Gather()
34 static PetscErrorCode PetscSFSetCommunicationOps_Gather(PetscSF sf, PetscSFLink link) in PetscSFSetCommunicationOps_Gather() argument
41 PETSC_INTERN PetscErrorCode PetscSFCreate_Gather(PetscSF sf) in PetscSFCreate_Gather() argument
43 PetscSF_Gather *dat = (PetscSF_Gather *)sf->data; in PetscSFCreate_Gather()
46 sf->ops->BcastBegin = PetscSFBcastBegin_Basic; in PetscSFCreate_Gather()
47 sf->ops->BcastEnd = PetscSFBcastEnd_Basic; in PetscSFCreate_Gather()
48 sf->ops->ReduceBegin = PetscSFReduceBegin_Basic; in PetscSFCreate_Gather()
49 sf->ops->ReduceEnd = PetscSFReduceEnd_Basic; in PetscSFCreate_Gather()
52 sf->ops->Reset = PetscSFReset_Allgatherv; in PetscSFCreate_Gather()
53 sf->ops->Destroy = PetscSFDestroy_Allgatherv; in PetscSFCreate_Gather()
54 sf->ops->GetGraph = PetscSFGetGraph_Allgatherv; in PetscSFCreate_Gather()
55 sf->ops->GetRootRanks = PetscSFGetRootRanks_Allgatherv; in PetscSFCreate_Gather()
56 sf->ops->GetLeafRanks = PetscSFGetLeafRanks_Allgatherv; in PetscSFCreate_Gather()
57 sf->ops->FetchAndOpEnd = PetscSFFetchAndOpEnd_Allgatherv; in PetscSFCreate_Gather()
58 sf->ops->CreateLocalSF = PetscSFCreateLocalSF_Allgatherv; in PetscSFCreate_Gather()
61 sf->ops->SetUp = PetscSFSetUp_Allgather; in PetscSFCreate_Gather()
64 sf->ops->FetchAndOpBegin = PetscSFFetchAndOpBegin_Gatherv; in PetscSFCreate_Gather()
66 sf->ops->SetCommunicationOps = PetscSFSetCommunicationOps_Gather; in PetscSFCreate_Gather()
68 sf->collective = PETSC_TRUE; in PetscSFCreate_Gather()
71 sf->data = (void *)dat; in PetscSFCreate_Gather()