Home
last modified time | relevance | path

Searched refs:tidx (Results 1 – 3 of 3) sorted by relevance

/petsc/src/mat/impls/sell/seq/seqhip/
H A Dsellhip.hip.cxx118 int tidx = tid % BLOCKY; in matmult_seqsell_tiled_kernel9() local
131 if (tidy < sliceheight) t = shared[tidy][tidx]; in matmult_seqsell_tiled_kernel9()
134 if (tidx == 0 && tidy < sliceheight) shared[0][tidy] = t; in matmult_seqsell_tiled_kernel9()
147 int tidx = tid % BLOCKY; in matmultadd_seqsell_tiled_kernel9() local
160 if (tidy < sliceheight) t = shared[tidy][tidx]; in matmultadd_seqsell_tiled_kernel9()
163 if (tidx == 0 && tidy < sliceheight) shared[0][tidy] = t; 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
229 if (tidy < sliceheight) t = shared[tidy * BLOCKY + tidx]; /* shared[tidy][tidx] */ in matmult_seqsell_tiled_kernel8()
232 if (tidx == 0 && tidy < sliceheight) shared[tidy] = t; /* shared[0][tidy] = t */ in matmult_seqsell_tiled_kernel8()
277 int tid = threadIdx.x + threadIdx.y * WARP_SIZE, tidx = tid % BLOCKY, tidy = tid / BLOCKY; in matmultadd_seqsell_tiled_kernel8() local
[all …]
/petsc/src/mat/impls/sell/seq/seqcuda/
H A Dsellcuda.cu114 int tidx = tid % BLOCKY; in matmult_seqsell_tiled_kernel9() local
127 if (tidy < sliceheight) t = shared[tidy][tidx]; in matmult_seqsell_tiled_kernel9()
130 if (tidx == 0 && tidy < sliceheight) shared[0][tidy] = t; in matmult_seqsell_tiled_kernel9()
143 int tidx = tid % BLOCKY; in matmultadd_seqsell_tiled_kernel9() local
156 if (tidy < sliceheight) t = shared[tidy][tidx]; in matmultadd_seqsell_tiled_kernel9()
159 if (tidx == 0 && tidy < sliceheight) shared[0][tidy] = t; 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
225 if (tidy < sliceheight) t = shared[tidy * BLOCKY + tidx]; /* shared[tidy][tidx] */ in matmult_seqsell_tiled_kernel8()
228 if (tidx == 0 && tidy < sliceheight) shared[tidy] = t; /* shared[0][tidy] = t */ in matmult_seqsell_tiled_kernel8()
273 int tid = threadIdx.x + threadIdx.y * 32, tidx = tid % BLOCKY, tidy = tid / BLOCKY; in matmultadd_seqsell_tiled_kernel8() local
[all …]
/petsc/src/ksp/pc/impls/gamg/
H A Dgamg.c156 …PetscInt *counts, *newproc_idx, ii, jj, kk, strideNew, *tidx, ncrs_new, ncrs_eq_new, nloc_old, exp… in PCGAMGCreateLevel_GAMG() local
351 PetscCall(PetscMalloc1(ncrs * node_data_sz, &tidx)); in PCGAMGCreateLevel_GAMG()
355 for (kk = 0; kk < node_data_sz; kk++, jj++) tidx[jj] = id * node_data_sz + kk; in PCGAMGCreateLevel_GAMG()
358 PetscCall(ISCreateGeneral(comm, node_data_sz * ncrs, tidx, PETSC_COPY_VALUES, &isscat)); in PCGAMGCreateLevel_GAMG()
359 PetscCall(PetscFree(tidx)); in PCGAMGCreateLevel_GAMG()