Searched refs:srcIdx (Results 1 – 3 of 3) sorted by relevance
| /petsc/src/vec/is/sf/impls/basic/cupm/ |
| H A D | sfcupm_impl.hpp | 96 …t srcy, PetscInt srcX, PetscInt srcY, PetscInt srcStart, const PetscInt *srcIdx, const Type *src, … in d_ScatterAndOp() argument
104 if (!srcIdx) { /* src is either contiguous or 3D */ in d_ScatterAndOp()
111 s = srcIdx[tid]; in d_ScatterAndOp()
320 …etscInt count, PetscInt srcStart, PetscSFPackOpt srcOpt, const PetscInt *srcIdx, const void *src, … in ScatterAndOp() argument
337 srcIdx = NULL; in ScatterAndOp()
338 } else if (!srcIdx) { in ScatterAndOp()
355 …Op<Type, Op, BS, EQ>, link->bs, count, srcx, srcy, srcX, srcY, srcStart, srcIdx, (const Type *)src… in ScatterAndOp()
362 …etscInt count, PetscInt srcStart, PetscSFPackOpt srcOpt, const PetscInt *srcIdx, const void *src, … in ScatterAndInsert() argument
367 if ((!srcOpt && !srcIdx) && (!dstOpt && !dstIdx) && src != dst) { in ScatterAndInsert()
370 …ndOp<Type, kernels::Insert<Type>, BS, EQ>(link, count, srcStart, srcOpt, srcIdx, src, dstStart, ds… in ScatterAndInsert()
|
| /petsc/src/vec/is/sf/impls/basic/kokkos/ |
| H A D | sfkok.kokkos.cxx | 293 …etscInt count, PetscInt srcStart, PetscSFPackOpt srcOpt, const PetscInt *srcIdx, const void *src_,… in ScatterAndOp() argument
309 srcIdx = NULL; in ScatterAndOp()
310 } else if (!srcIdx) { in ScatterAndOp()
331 if (!srcIdx) { /* src is in 3D */ in ScatterAndOp()
337 s = srcIdx[tid]; in ScatterAndOp()
358 …etscInt count, PetscInt srcStart, PetscSFPackOpt srcOpt, const PetscInt *srcIdx, const void *src_,… in ScatterAndInsert() argument
367 if ((!srcOpt && !srcIdx) && (!dstOpt && !dstIdx) && src != dst) { in ScatterAndInsert()
373 …PetscCall(ScatterAndOp<Type, Insert<Type>, BS, EQ>(link, count, srcStart, srcOpt, srcIdx, src, dst… in ScatterAndInsert()
|
| /petsc/src/vec/is/sf/impls/basic/ |
| H A D | sfpack.c | 191 …etscInt count, PetscInt srcStart, PetscSFPackOpt srcOpt, const PetscInt *srcIdx, const void *src, …
199 if (!srcIdx) { /* src is contiguous */ \
214 s = (!srcIdx ? srcStart + i : srcIdx[i]) * MBS; \
882 …sf, PetscSFLink link, PetscInt count, PetscInt srcStart, const PetscInt *srcIdx, const void *src, … in PetscSFLinkScatterDataWithMPIReduceLocal() argument
888 if (!srcIdx) { in PetscSFLinkScatterDataWithMPIReduceLocal()
894 …PetscCallMPI(MPIU_Reduce_local((const char *)src + srcIdx[i] * link->unitbytes, (char *)dst + disp… in PetscSFLinkScatterDataWithMPIReduceLocal()
|