Searched refs:row_in_slice (Results 1 – 3 of 3) sorted by relevance
| /petsc/src/mat/impls/sell/seq/seqhip/ |
| H A D | sellhip.hip.cxx | 80 PetscInt i, row, slice_id, row_in_slice; in matmult_seqsell_basic_kernel() local
86 row_in_slice = row % sliceheight; in matmult_seqsell_basic_kernel()
88 …for (i = sliidx[slice_id] + row_in_slice; i < sliidx[slice_id + 1]; i += sliceheight) sum += aval[… in matmult_seqsell_basic_kernel()
95 PetscInt i, row, slice_id, row_in_slice; in matmultadd_seqsell_basic_kernel() local
101 row_in_slice = row % sliceheight; in matmultadd_seqsell_basic_kernel()
103 …for (i = sliidx[slice_id] + row_in_slice; i < sliidx[slice_id + 1]; i += sliceheight) sum += aval[… in matmultadd_seqsell_basic_kernel()
334 PetscInt i, row, slice_id, row_in_slice; in PETSC_PRAGMA_DIAGNOSTIC_IGNORED_END() local
339 row_in_slice = row % sliceheight; in PETSC_PRAGMA_DIAGNOSTIC_IGNORED_END()
342 …for (i = sliidx[slice_id] + row_in_slice + sliceheight * threadIdx.y; i < sliidx[slice_id + 1]; i … in PETSC_PRAGMA_DIAGNOSTIC_IGNORED_END()
362 PetscInt i, row, slice_id, row_in_slice; in matmult_seqsell_tiled_kernel5() local
[all …]
|
| /petsc/src/mat/impls/sell/seq/seqcuda/ |
| H A D | sellcuda.cu | 77 PetscInt i, row, slice_id, row_in_slice; in matmult_seqsell_basic_kernel() local
83 row_in_slice = row % sliceheight; in matmult_seqsell_basic_kernel()
85 …for (i = sliidx[slice_id] + row_in_slice; i < sliidx[slice_id + 1]; i += sliceheight) sum += aval[… in matmult_seqsell_basic_kernel()
92 PetscInt i, row, slice_id, row_in_slice; in matmultadd_seqsell_basic_kernel() local
98 row_in_slice = row % sliceheight; in matmultadd_seqsell_basic_kernel()
100 …for (i = sliidx[slice_id] + row_in_slice; i < sliidx[slice_id + 1]; i += sliceheight) sum += aval[… in matmultadd_seqsell_basic_kernel()
329 PetscInt i, row, slice_id, row_in_slice; in matmult_seqsell_tiled_kernel6() local
334 row_in_slice = row % SLICE_HEIGHT; in matmult_seqsell_tiled_kernel6()
337 …for (i = sliidx[slice_id] + row_in_slice + SLICE_HEIGHT * threadIdx.y; i < sliidx[slice_id + 1]; i… in matmult_seqsell_tiled_kernel6()
357 PetscInt i, row, slice_id, row_in_slice; in matmult_seqsell_tiled_kernel5() local
[all …]
|
| /petsc/src/mat/impls/sell/seq/ |
| H A D | sell.c | 1439 PetscInt i, shift, row_in_slice, row, nrow, *cp, lastcol, j, k; in MatAssemblyEnd_SeqSELL() local
1457 …for (row_in_slice = 0; row_in_slice < a->sliceheight; ++row_in_slice) { /* loop over rows in the s… in MatAssemblyEnd_SeqSELL()
1458 row = a->sliceheight * i + row_in_slice; in MatAssemblyEnd_SeqSELL()
1467 …lastcol = cp[a->sliceheight * (nrow - 1) + row_in_slice]; /* use the index from the last nonzero a… in MatAssemblyEnd_SeqSELL()
1468 …} else if (!row_in_slice) { /* first row of the correct slice is e… in MatAssemblyEnd_SeqSELL()
1476 …if (a->sliidx[i + 1] != shift) lastcol = cp[row_in_slice - 1]; /* use the index from the previous … in MatAssemblyEnd_SeqSELL()
1480 cp[a->sliceheight * k + row_in_slice] = lastcol; in MatAssemblyEnd_SeqSELL()
1481 vp[a->sliceheight * k + row_in_slice] = (MatScalar)0; in MatAssemblyEnd_SeqSELL()
|