Searched refs:srcStart (Results 1 – 3 of 3) sorted by relevance
| /petsc/src/vec/is/sf/impls/basic/cupm/ |
| H A D | sfcupm_impl.hpp | 96 …t, PetscInt srcx, PetscInt srcy, PetscInt srcX, PetscInt srcY, PetscInt srcStart, const PetscInt *… in d_ScatterAndOp() argument
109 s = srcStart + k * srcX * srcY + j * srcX + i; in d_ScatterAndOp()
320 …SfInterface<T>::ScatterAndOp(PetscSFLink link, PetscInt count, PetscInt srcStart, PetscSFPackOpt s… in ScatterAndOp() argument
336 srcStart = srcOpt->start[0]; in ScatterAndOp()
355 …catterAndOp<Type, Op, BS, EQ>, link->bs, count, srcx, srcy, srcX, srcY, srcStart, srcIdx, (const T… in ScatterAndOp()
362 …terface<T>::ScatterAndInsert(PetscSFLink link, PetscInt count, PetscInt srcStart, PetscSFPackOpt s… in ScatterAndInsert() argument
368 …PetscCallCUPM(cupmMemcpyAsync((Type *)dst + dstStart * link->bs, (const Type *)src + srcStart * li… 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/kokkos/ |
| H A D | sfkok.kokkos.cxx | 293 static PetscErrorCode ScatterAndOp(PetscSFLink link, PetscInt count, PetscInt srcStart, PetscSFPack… in ScatterAndOp() argument
308 srcStart = srcOpt->start[0]; in ScatterAndOp()
335 s = srcStart + k * srcX * srcY + j * srcX + i; in ScatterAndOp()
358 static PetscErrorCode ScatterAndInsert(PetscSFLink link, PetscInt count, PetscInt srcStart, PetscSF… in ScatterAndInsert() argument
370 deviceConstBuffer_t sbuf(reinterpret_cast<const char *>(src + srcStart * link->bs), sz); 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 …erAnd##Opname, Type, BS, EQ)(PetscSFLink link, PetscInt count, PetscInt srcStart, PetscSFPackOpt s…
200 u += srcStart * MBS; \
214 s = (!srcIdx ? srcStart + i : srcIdx[i]) * MBS; \
882 …thMPIReduceLocal(PetscSF sf, PetscSFLink link, PetscInt count, PetscInt srcStart, const PetscInt *… in PetscSFLinkScatterDataWithMPIReduceLocal() argument
889 …ReduceLocal(sf, link, count, dstStart, dstIdx, dst, (const char *)src + srcStart * link->unitbytes… in PetscSFLinkScatterDataWithMPIReduceLocal()
|