| /petsc/src/vec/is/sf/impls/basic/ |
| H A D | sfpack.c | 224 …Opname##Local, Type, BS, EQ)(PetscSFLink link, PetscInt count, PetscInt rootstart, PetscSFPackOpt …
233 r = (rootidx ? rootidx[i] : rootstart + i) * MBS; \
1085 PetscInt count, rootstart, leafstart; in PetscSFLinkScatterLocal() local
1123 …nkGetRootPackOptAndIndices(sf, link, rootmtype, PETSCSF_LOCAL, &count, &rootstart, &rootopt, &root… in PetscSFLinkScatterLocal()
1126 …PetscCall((*ScatterAndOp)(link, count, rootstart, rootopt, rootindices, rootdata, leafstart, leafo… in PetscSFLinkScatterLocal()
1128 …PetscCall((*ScatterAndOp)(link, count, leafstart, leafopt, leafindices, leafdata, rootstart, rooto… in PetscSFLinkScatterLocal()
1131 …PackOptAndIndices(sf, link, PETSC_MEMTYPE_HOST, PETSCSF_LOCAL, &count, &rootstart, &rootopt, &root… in PetscSFLinkScatterLocal()
1134 …PetscCall(PetscSFLinkScatterDataWithMPIReduceLocal(sf, link, count, rootstart, rootindices, rootda… in PetscSFLinkScatterLocal()
1136 …taWithMPIReduceLocal(sf, link, count, leafstart, leafindices, leafdata, rootstart, rootindices, ro… in PetscSFLinkScatterLocal()
1147 PetscInt count, rootstart, leafstart; in PetscSFLinkFetchAndOpLocal() local
[all …]
|
| H A D | sfbasic.h | 15 …PetscInt rootstart[2]; /* ... and start from rootstart[0] and rootstart[1] respectively …
|
| H A D | sfmpi.c | 190 link->rootbuf[i][rootmtype] = (char *)rootdata + bas->rootstart[i] * link->unitbytes; in PetscSFLinkCreate_MPI()
|
| H A D | sfpack.h | 376 *start = bas->rootstart[scope]; in PetscSFLinkGetRootPackOptAndIndices()
|
| /petsc/src/vec/is/sf/impls/basic/cupm/ |
| H A D | sfcupm_impl.hpp | 83 PETSC_KERNEL_DECL static void d_FetchAndOp(PetscInt bs, PetscInt count, PetscInt rootstart, const P… in d_FetchAndOp() argument
89 …PetscInt r = (rootopt ? MapTidToIndex(rootopt, tid) : (rootidx ? rootidx[tid] : rootstart + tid)) … in d_FetchAndOp()
131 PETSC_KERNEL_DECL static void d_FetchAndOpLocal(PetscInt bs, PetscInt count, PetscInt rootstart, co… in d_FetchAndOpLocal() argument
137 …PetscInt r = (rootopt ? MapTidToIndex(rootopt, tid) : (rootidx ? rootidx[tid] : rootstart + tid)) … in d_FetchAndOpLocal()
377 …nterface<T>::FetchAndOpLocal(PetscSFLink link, PetscInt count, PetscInt rootstart, PetscSFPackOpt … in FetchAndOpLocal() argument
384 …>stream, kernels::d_FetchAndOpLocal<Type, Op, BS, EQ>, link->bs, count, rootstart, rarray, rootidx… in FetchAndOpLocal()
|
| /petsc/src/vec/is/sf/impls/basic/allgather/ |
| H A D | sfallgather.c | 18 dat->rootstart[i] = 0; in PetscSFSetUp_Allgather()
|
| /petsc/src/vec/is/sf/impls/basic/kokkos/ |
| H A D | sfkok.kokkos.cxx | 379 static PetscErrorCode FetchAndOpLocal(PetscSFLink link, PetscInt count, PetscInt rootstart, PetscSF… in FetchAndOpLocal() argument
392 … PetscInt r = (ropt ? MapTidToIndex(ropt, tid) : (rootidx ? rootidx[tid] : rootstart + tid)) * MBS; in FetchAndOpLocal()
|
| /petsc/src/vec/is/sf/impls/basic/nvshmem/ |
| H A D | sfnvshmem.cu | 769 …link->rootbuf[PETSCSF_REMOTE][PETSC_MEMTYPE_DEVICE] = (char *)rootdata + bas->rootstart[PETSCSF_RE… in PetscSFLinkCreate_NVSHMEM()
|