Home
last modified time | relevance | path

Searched refs:tid (Results 1 – 11 of 11) sorted by relevance

/petsc/src/dm/impls/swarm/tests/
H A Dex7.c249 for (int tid = 0; tid < numthreads; tid++) { in go() local
253 PetscCall(DMCreate(PETSC_COMM_SELF, &dm_t[tid])); in go()
254 PetscCall(DMSetType(dm_t[tid], DMPLEX)); in go()
255 PetscCall(DMSetFromOptions(dm_t[tid])); in go()
259 PetscCall(DMSetField(dm_t[tid], field, NULL, (PetscObject)fe)); in go()
260 PetscCall(DMCreateDS(dm_t[tid])); in go()
263 PetscCall(DMCreateGlobalVector(dm_t[tid], &rho_t[tid])); in go()
264 PetscCall(DMCreateGlobalVector(dm_t[tid], &rhs_t[tid])); in go()
266 PetscCall(DMGetBoundingBox(dm_t[tid], lo, hi)); in go()
267 if (tid == target) { in go()
[all …]
/petsc/src/ts/tests/
H A Dex30.c287 for (PetscInt tid = 0; tid < numthreads; tid++) { in gridToParticles_private() local
288 const PetscInt v_id = v_id_0 + tid, glb_v_id = global_vertex_id + v_id; in gridToParticles_private()
293 …PACK_IDX(v_id, grid)], globXArray[LAND_PACK_IDX(v_id, grid)], t_fhat[grid][tid], globMpArray[LAND_… in gridToParticles_private()
301 for (PetscInt tid = 0; tid < numthreads; tid++) { in gridToParticles_private() local
302 const PetscInt v_id = v_id_0 + tid, glb_v_id = global_vertex_id + v_id; in gridToParticles_private()
518 for (PetscInt tid = 0; tid < numthreads; tid++) { in go() local
520 PetscCall(VecDuplicate(subX, &t_fhat[grid][tid])); in go()
521 PetscCall(KSPCreate(PETSC_COMM_SELF, &t_ksp[grid][tid])); in go()
522 PetscCall(KSPSetType(t_ksp[grid][tid], KSPCG)); in go()
523 PetscCall(KSPGetPC(t_ksp[grid][tid], &pc)); in go()
[all …]
/petsc/src/ts/trajectory/utils/
H A Dreconstruct.c84 PetscInt tid = LagrangeGetId(t, tj->lag.order + 1, tj->lag.T, tj->lag.TT); in TSTrajectoryReconstruct_Private() local
85 if (tid < 0) continue; in TSTrajectoryReconstruct_Private()
86 …, "Reusing snapshot %" PetscInt_FMT ", step %" PetscInt_FMT ", time %g\n", tid, tshhist_id[s], (do… in TSTrajectoryReconstruct_Private()
87 tj->lag.TT[tid] = PETSC_TRUE; in TSTrajectoryReconstruct_Private()
88 tj->lag.WW[cnt] = tj->lag.W[tid]; in TSTrajectoryReconstruct_Private()
97 PetscInt tid; in TSTrajectoryReconstruct_Private() local
100 tid = LagrangeGetId(t, tj->lag.order + 1, tj->lag.T, tj->lag.TT); in TSTrajectoryReconstruct_Private()
101 … PetscCheck(tid < 0, PetscObjectComm((PetscObject)tj), PETSC_ERR_PLIB, "This should not happen"); in TSTrajectoryReconstruct_Private()
102 tid = -tid - 1; in TSTrajectoryReconstruct_Private()
104 if (tj->lag.T[tid] < PETSC_MAX_REAL) { in TSTrajectoryReconstruct_Private()
[all …]
/petsc/src/vec/is/sf/impls/basic/kokkos/
H A Dsfkok.kokkos.cxx195 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()
[all …]
/petsc/src/vec/is/sf/impls/basic/cupm/
H A Dsfcupm_impl.hpp20 … 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
[all …]
/petsc/src/ksp/pc/impls/vpbjacobi/cuda/
H A Dvpbjacobi_cuda.cu84 PetscInt tid = blockIdx.x * blockDim.x + threadIdx.x; in MatMultBatched() local
88 for (; tid < n; tid += gridSize) { in MatMultBatched()
89 k = matIdx[tid]; /* k-th block */ in MatMultBatched()
91 i = tid - bs[k]; /* i-th row of the block */ in MatMultBatched()
/petsc/src/ksp/pc/impls/gamg/
H A Dgeo.c129 PetscInt jj, tid, tt, idx, nselected_2; in triangulateAndFormProl() local
272 for (tid = 0, kk = 0; tid < mid.numberoftriangles; tid++) { in triangulateAndFormProl()
275 if (nTri[cid] == 0) node_tri[cid] = tid; in triangulateAndFormProl()
307 for (tid = node_tri[clid], jj = 0; jj < 5 && !haveit && tid != -1; jj++) { in triangulateAndFormProl()
309 PetscInt cid2 = mid.trianglelist[3 * tid + tt]; in triangulateAndFormProl()
333 tid = mid.neighborlist[3 * tid + idx]; in triangulateAndFormProl()
338 for (tid = 0; tid < mid.numberoftriangles && !haveit; tid++) { in triangulateAndFormProl()
340 PetscInt cid2 = mid.trianglelist[3 * tid + tt]; in triangulateAndFormProl()
359 bestTID = tid; in triangulateAndFormProl()
/petsc/src/mat/impls/sell/seq/seqhip/
H A Dsellhip.hip.cxx116 int tid = threadIdx.x + threadIdx.y * WARP_SIZE; in matmult_seqsell_tiled_kernel9() local
118 int tidx = tid % BLOCKY; in matmult_seqsell_tiled_kernel9()
119 int tidy = tid / BLOCKY; in matmult_seqsell_tiled_kernel9()
145 int tid = threadIdx.x + threadIdx.y * WARP_SIZE; in matmultadd_seqsell_tiled_kernel9() local
147 int tidx = tid % BLOCKY; in matmultadd_seqsell_tiled_kernel9()
148 int tidy = tid / BLOCKY; in matmultadd_seqsell_tiled_kernel9()
222 int tid = threadIdx.x + threadIdx.y * WARP_SIZE, tidx = tid % BLOCKY, tidy = tid / BLOCKY; in matmult_seqsell_tiled_kernel8() local
277 int tid = threadIdx.x + threadIdx.y * WARP_SIZE, tidx = tid % BLOCKY, tidy = tid / BLOCKY; in matmultadd_seqsell_tiled_kernel8() local
/petsc/src/mat/impls/sell/seq/seqcuda/
H A Dsellcuda.cu112 int tid = threadIdx.x + threadIdx.y * 32; in matmult_seqsell_tiled_kernel9() local
114 int tidx = tid % BLOCKY; in matmult_seqsell_tiled_kernel9()
115 int tidy = tid / BLOCKY; in matmult_seqsell_tiled_kernel9()
141 int tid = threadIdx.x + threadIdx.y * 32; in matmultadd_seqsell_tiled_kernel9() local
143 int tidx = tid % BLOCKY; in matmultadd_seqsell_tiled_kernel9()
144 int tidy = tid / BLOCKY; in matmultadd_seqsell_tiled_kernel9()
218 int tid = threadIdx.x + threadIdx.y * 32, tidx = tid % BLOCKY, tidy = tid / BLOCKY; in matmult_seqsell_tiled_kernel8() local
273 int tid = threadIdx.x + threadIdx.y * 32, tidx = tid % BLOCKY, tidy = tid / BLOCKY; in matmultadd_seqsell_tiled_kernel8() local
/petsc/src/ts/utils/dmplexlandau/
H A Dplexland.c115 for (PetscInt tid = 0; tid < ctx->batch_sz; tid++) { in LandauFormJacobian_Internal() local
116 …tx->num_grids; grid++) PetscCall(DMCreateMatrix(ctx->plex[grid], &subJ[LAND_PACK_IDX(tid, grid)])); in LandauFormJacobian_Internal()
243 …for (PetscInt tid = 0; tid < ctx->batch_sz * elem_offset[num_grids]; tid++) { … in LandauFormJacobian_Internal() local
244 …const PetscInt b_Nelem = elem_offset[num_grids], b_elem_idx = tid % b_Nelem, b_id = tid / b_Nelem;… in LandauFormJacobian_Internal()
313 … for (PetscInt tid = 0; tid < ctx->batch_sz * elem_offset[num_grids]; tid++) { // for each element in LandauFormJacobian_Internal() local
315 const PetscInt glb_elem_idx = tid % b_Nelem, b_id = tid / b_Nelem; in LandauFormJacobian_Internal()
/petsc/doc/
H A Dpetsc.bib7249 url = {http://www.llnl.gov/tid/lof/documents/pdf/237860.pdf},