Searched refs:leafbufdisp (Results 1 – 2 of 2) sorted by relevance
| /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()
138 …for (i = 0; i < nRemoteLeafRanks; i++) PetscCallMPI(MPIU_Irecv(&bas->leafbufdisp[i], 1, MPIU_INT, … in PetscSFSetUp_Basic_NVSHMEM()
150 …PetscCallCUDA(cudaMemcpyAsync(bas->leafbufdisp_d, bas->leafbufdisp, nRemoteLeafRanks * sizeof(Pets… in PetscSFSetUp_Basic_NVSHMEM()
378 …srcdisp_h = bas->leafbufdisp; /* for my i-th remote root rank, I will access its buf at offset ro… in PetscSFLinkGetDataBegin_NVSHMEM()
537 …dstdisp_h = bas->leafbufdisp; /* for my i-th remote leaf rank, I will access its leaf buf at offs… in PetscSFLinkPutDataBegin_NVSHMEM()
|
| /petsc/src/vec/is/sf/impls/basic/ |
| H A D | sfbasic.h | 30 …PetscInt *leafbufdisp; /* [nRemoteLeafRanks]. For my i-th remote leaf rank, I will put to its leaf… member
|