Searched refs:srcOpt (Results 1 – 3 of 3) sorted by relevance
| /petsc/src/vec/is/sf/impls/basic/kokkos/ ! |
| H A D | sfkok.kokkos.cxx | 293 …ndOp(PetscSFLink link, PetscInt count, PetscInt srcStart, PetscSFPackOpt srcOpt, const PetscInt *s… in ScatterAndOp() argument
303 if (srcOpt) { in ScatterAndOp()
304 srcx = srcOpt->dx[0]; in ScatterAndOp()
305 srcy = srcOpt->dy[0]; in ScatterAndOp()
306 srcX = srcOpt->X[0]; in ScatterAndOp()
307 srcY = srcOpt->Y[0]; in ScatterAndOp()
308 srcStart = srcOpt->start[0]; in ScatterAndOp()
358 …sert(PetscSFLink link, PetscInt count, PetscInt srcStart, PetscSFPackOpt srcOpt, const PetscInt *s… 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/cupm/ ! |
| H A D | sfcupm_impl.hpp | 320 …ndOp(PetscSFLink link, PetscInt count, PetscInt srcStart, PetscSFPackOpt srcOpt, const PetscInt *s… in ScatterAndOp() argument
331 if (srcOpt) { in ScatterAndOp()
332 srcx = srcOpt->dx[0]; in ScatterAndOp()
333 srcy = srcOpt->dy[0]; in ScatterAndOp()
334 srcX = srcOpt->X[0]; in ScatterAndOp()
335 srcY = srcOpt->Y[0]; in ScatterAndOp()
336 srcStart = srcOpt->start[0]; in ScatterAndOp()
362 …sert(PetscSFLink link, PetscInt count, PetscInt srcStart, PetscSFPackOpt srcOpt, const PetscInt *s… in ScatterAndInsert() argument
367 if ((!srcOpt && !srcIdx) && (!dstOpt && !dstIdx) && src != dst) { in ScatterAndInsert()
370 …PetscCall(ScatterAndOp<Type, kernels::Insert<Type>, BS, EQ>(link, count, srcStart, srcOpt, srcIdx,… in ScatterAndInsert()
|
| /petsc/src/vec/is/sf/impls/basic/ ! |
| H A D | sfpack.c | 191 … EQ)(PetscSFLink link, PetscInt count, PetscInt srcStart, PetscSFPackOpt srcOpt, const PetscInt *s…
202 } else if (srcOpt && !dstIdx) { /* src is 3D, dst is contiguous */ \
203 u += srcOpt->start[0] * MBS; \
205 X = srcOpt->X[0]; \
206 Y = srcOpt->Y[0]; \
207 for (k = 0; k < srcOpt->dz[0]; k++) \
208 for (j = 0; j < srcOpt->dy[0]; j++) { \
209 … for (i = 0; i < srcOpt->dx[0] * MBS; i++) OpApply(Op, v[i], u[(X * Y * k + X * j) * MBS + i]); \
210 v += srcOpt->dx[0] * MBS; \
|