Lines Matching refs:tid
20 … static PETSC_DEVICE_INLINE_DECL PetscInt MapTidToIndex(const PetscInt *opt, PetscInt tid) noexcept in MapTidToIndex() argument
33 if (tid < offset[r + 1]) break; in MapTidToIndex()
35 m = (tid - offset[r]); in MapTidToIndex()
62 ::Petsc::device::cupm::kernels::util::grid_stride_1D(count, [&](PetscInt tid) { in d_Pack() argument
63 PetscInt t = (opt ? MapTidToIndex(opt, tid) : (idx ? idx[tid] : start + tid)) * MBS; in d_Pack()
64 PetscInt s = tid * MBS; in d_Pack()
75 ::Petsc::device::cupm::kernels::util::grid_stride_1D(count, [&](PetscInt tid) { in d_UnpackAndOp() argument
76 PetscInt t = (opt ? MapTidToIndex(opt, tid) : (idx ? idx[tid] : start + tid)) * MBS; in d_UnpackAndOp()
77 PetscInt s = tid * MBS; in d_UnpackAndOp()
88 ::Petsc::device::cupm::kernels::util::grid_stride_1D(count, [&](PetscInt tid) { in d_FetchAndOp() argument
89 …PetscInt r = (rootopt ? MapTidToIndex(rootopt, tid) : (rootidx ? rootidx[tid] : rootstart + tid)) … in d_FetchAndOp()
90 PetscInt l = tid * MBS; in d_FetchAndOp()
101 ::Petsc::device::cupm::kernels::util::grid_stride_1D(count, [&](PetscInt tid) { in d_ScatterAndOp() argument
105 PetscInt k = tid / (srcx * srcy); in d_ScatterAndOp()
106 PetscInt j = (tid - k * srcx * srcy) / srcx; in d_ScatterAndOp()
107 PetscInt i = tid - k * srcx * srcy - j * srcx; in d_ScatterAndOp()
111 s = srcIdx[tid]; in d_ScatterAndOp()
115 PetscInt k = tid / (dstx * dsty); in d_ScatterAndOp()
116 PetscInt j = (tid - k * dstx * dsty) / dstx; in d_ScatterAndOp()
117 PetscInt i = tid - k * dstx * dsty - j * dstx; in d_ScatterAndOp()
121 t = dstIdx[tid]; in d_ScatterAndOp()
136 ::Petsc::device::cupm::kernels::util::grid_stride_1D(count, [&](PetscInt tid) { in d_FetchAndOpLocal() argument
137 …PetscInt r = (rootopt ? MapTidToIndex(rootopt, tid) : (rootidx ? rootidx[tid] : rootstart + tid)) … in d_FetchAndOpLocal()
138 …PetscInt l = (leafopt ? MapTidToIndex(leafopt, tid) : (leafidx ? leafidx[tid] : leafstart + tid)) … in d_FetchAndOpLocal()