Searched refs:leafsigdisp (Results 1 – 2 of 2) sorted by relevance
| /petsc/src/vec/is/sf/impls/basic/ |
| H A D | sfbasic.h | 31 …PetscInt *leafsigdisp; /* [nRemoteLeafRanks]. For my i-th remote leaf rank, I am its leafsigdisp[i… member
|
| /petsc/src/vec/is/sf/impls/basic/nvshmem/ |
| H A D | sfnvshmem.cu | 59 PetscCall(PetscFree2(bas->leafsigdisp, bas->leafbufdisp)); in PetscSFReset_Basic_NVSHMEM()
133 PetscCall(PetscMalloc2(nRemoteLeafRanks, &bas->leafsigdisp, nRemoteLeafRanks, &bas->leafbufdisp)); in PetscSFSetUp_Basic_NVSHMEM()
134 …for (i = 0; i < nRemoteLeafRanks; i++) PetscCallMPI(MPIU_Irecv(&bas->leafsigdisp[i], 1, MPIU_INT, … in PetscSFSetUp_Basic_NVSHMEM()
151 …PetscCallCUDA(cudaMemcpyAsync(bas->leafsigdisp_d, bas->leafsigdisp, nRemoteLeafRanks * sizeof(Pets… in PetscSFSetUp_Basic_NVSHMEM()
|