Lines Matching refs:tid

195 static KOKKOS_INLINE_FUNCTION PetscInt MapTidToIndex(const PetscInt *opt, PetscInt tid)  in MapTidToIndex()  argument
208 if (tid < offset[r + 1]) break; in MapTidToIndex()
210 m = (tid - offset[r]); in MapTidToIndex()
242 Kokkos::RangePolicy<DeviceExecutionSpace>(exec, 0, count), KOKKOS_LAMBDA(PetscInt tid) { in Pack()
246 PetscInt t = (iopt ? MapTidToIndex(iopt, tid) : (idx ? idx[tid] : start + tid)) * MBS; in Pack()
247 PetscInt s = tid * MBS; in Pack()
265 Kokkos::RangePolicy<DeviceExecutionSpace>(exec, 0, count), KOKKOS_LAMBDA(PetscInt tid) { in UnpackAndOp()
266 PetscInt t = (iopt ? MapTidToIndex(iopt, tid) : (idx ? idx[tid] : start + tid)) * MBS; in UnpackAndOp()
267 PetscInt s = tid * MBS; in UnpackAndOp()
284 Kokkos::RangePolicy<DeviceExecutionSpace>(exec, 0, count), KOKKOS_LAMBDA(PetscInt tid) { in FetchAndOp()
285 PetscInt r = (ropt ? MapTidToIndex(ropt, tid) : (idx ? idx[tid] : start + tid)) * MBS; in FetchAndOp()
286 PetscInt l = tid * MBS; in FetchAndOp()
328 Kokkos::RangePolicy<DeviceExecutionSpace>(exec, 0, count), KOKKOS_LAMBDA(PetscInt tid) { in ScatterAndOp()
332 k = tid / (srcx * srcy); in ScatterAndOp()
333 j = (tid - k * srcx * srcy) / srcx; in ScatterAndOp()
334 i = tid - k * srcx * srcy - j * srcx; in ScatterAndOp()
337 s = srcIdx[tid]; in ScatterAndOp()
341 k = tid / (dstx * dsty); in ScatterAndOp()
342 j = (tid - k * dstx * dsty) / dstx; in ScatterAndOp()
343 i = tid - k * dstx * dsty - j * dstx; in ScatterAndOp()
346 t = dstIdx[tid]; in ScatterAndOp()
391 Kokkos::RangePolicy<DeviceExecutionSpace>(exec, 0, count), KOKKOS_LAMBDA(PetscInt tid) { in FetchAndOpLocal()
392 … PetscInt r = (ropt ? MapTidToIndex(ropt, tid) : (rootidx ? rootidx[tid] : rootstart + tid)) * MBS; in FetchAndOpLocal()
393 … PetscInt l = (lopt ? MapTidToIndex(lopt, tid) : (leafidx ? leafidx[tid] : leafstart + tid)) * MBS; in FetchAndOpLocal()