Searched refs:start_slice (Results 1 – 2 of 2) sorted by relevance
| /petsc/src/mat/impls/sell/seq/seqhip/ |
| H A D | sellhip.hip.cxx | 192 PetscInt gid, row, start_slice, cid; in matmult_seqsell_tiled_kernel8() local
203 start_slice = chunk_slice_map[cid]; /* starting slice at each iteration */ in matmult_seqsell_tiled_kernel8()
205 …if ((cid + 1) * BLOCKY * WARP_SIZE > sliidx[start_slice + 1]) { /* this iteration covers more than… in matmult_seqsell_tiled_kernel8()
208 …PetscInt slice_id = start_slice, totalslices = PetscCeilIntMacro(nrows, sliceheight), t… in matmult_seqsell_tiled_kernel8()
219 row = start_slice * sliceheight + threadIdx.x % sliceheight; in matmult_seqsell_tiled_kernel8()
221 …if (iter == chunksperblock - 1 || (cid + 2) * BLOCKY * WARP_SIZE > sliidx[start_slice + 1]) { /* l… in matmult_seqsell_tiled_kernel8()
247 PetscInt gid, row, start_slice, cid; in matmultadd_seqsell_tiled_kernel8() local
258 start_slice = chunk_slice_map[cid]; /* starting slice at each iteration */ in matmultadd_seqsell_tiled_kernel8()
260 …if ((cid + 1) * BLOCKY * WARP_SIZE > sliidx[start_slice + 1]) { /* this iteration covers more than… in matmultadd_seqsell_tiled_kernel8()
263 …PetscInt slice_id = start_slice, totalslices = PetscCeilIntMacro(nrows, sliceheight), t… in matmultadd_seqsell_tiled_kernel8()
[all …]
|
| /petsc/src/mat/impls/sell/seq/seqcuda/ |
| H A D | sellcuda.cu | 188 PetscInt gid, row, start_slice, cid; in matmult_seqsell_tiled_kernel8() local
199 start_slice = chunk_slice_map[cid]; /* starting slice at each iteration */ in matmult_seqsell_tiled_kernel8()
201 …if ((cid + 1) * BLOCKY * 32 > sliidx[start_slice + 1]) { /* this iteration covers more than one sl… in matmult_seqsell_tiled_kernel8()
204 …PetscInt slice_id = start_slice, totalslices = PetscCeilIntMacro(nrows, sliceheight), t… in matmult_seqsell_tiled_kernel8()
215 row = start_slice * sliceheight + threadIdx.x % sliceheight; in matmult_seqsell_tiled_kernel8()
217 …if (iter == chunksperblock - 1 || (cid + 2) * BLOCKY * 32 > sliidx[start_slice + 1]) { /* last ite… in matmult_seqsell_tiled_kernel8()
243 PetscInt gid, row, start_slice, cid; in matmultadd_seqsell_tiled_kernel8() local
254 start_slice = chunk_slice_map[cid]; /* starting slice at each iteration */ in matmultadd_seqsell_tiled_kernel8()
256 …if ((cid + 1) * BLOCKY * 32 > sliidx[start_slice + 1]) { /* this iteration covers more than one sl… in matmultadd_seqsell_tiled_kernel8()
259 …PetscInt slice_id = start_slice, totalslices = PetscCeilIntMacro(nrows, sliceheight), t… in matmultadd_seqsell_tiled_kernel8()
[all …]
|