Searched refs:totalentries (Results 1 – 2 of 2) sorted by relevance
| /petsc/src/mat/impls/sell/seq/seqhip/ |
| H A D | sellhip.hip.cxx | 12 PetscInt totalentries; member
56 hipstruct->totalentries = a->sliidx[a->totalslices]; in MatSeqSELLHIPCopyToGPU()
208 …d = start_slice, totalslices = PetscCeilIntMacro(nrows, sliceheight), totalentries = sliidx[totals… in matmult_seqsell_tiled_kernel8() local
210 while (gid < totalentries && gid >= sliidx[slice_id + 1]) slice_id++; in matmult_seqsell_tiled_kernel8()
213 if (row < nrows && gid < totalentries) t = aval[gid] * x[acolidx[gid]]; in matmult_seqsell_tiled_kernel8()
216 if (row < nrows && gid < totalentries && write) atomAdd(y[row], t); in matmult_seqsell_tiled_kernel8()
263 …d = start_slice, totalslices = PetscCeilIntMacro(nrows, sliceheight), totalentries = sliidx[totals… in matmultadd_seqsell_tiled_kernel8() local
265 while (gid < totalentries && gid >= sliidx[slice_id + 1]) slice_id++; in matmultadd_seqsell_tiled_kernel8()
268 if (row < nrows && gid < totalentries) t = aval[gid] * x[acolidx[gid]]; in matmultadd_seqsell_tiled_kernel8()
271 if (row < nrows && gid < totalentries && write) atomAdd(z[row], t); in matmultadd_seqsell_tiled_kernel8()
[all …]
|
| /petsc/src/mat/impls/sell/seq/seqcuda/ |
| H A D | sellcuda.cu | 9 PetscInt totalentries; member
53 cudastruct->totalentries = a->sliidx[a->totalslices]; in MatSeqSELLCUDACopyToGPU()
204 …d = start_slice, totalslices = PetscCeilIntMacro(nrows, sliceheight), totalentries = sliidx[totals… in matmult_seqsell_tiled_kernel8() local
206 while (gid < totalentries && gid >= sliidx[slice_id + 1]) slice_id++; in matmult_seqsell_tiled_kernel8()
209 if (row < nrows && gid < totalentries) t = aval[gid] * x[acolidx[gid]]; in matmult_seqsell_tiled_kernel8()
212 if (row < nrows && gid < totalentries && write) atomAdd(y[row], t); in matmult_seqsell_tiled_kernel8()
259 …d = start_slice, totalslices = PetscCeilIntMacro(nrows, sliceheight), totalentries = sliidx[totals… in matmultadd_seqsell_tiled_kernel8() local
261 while (gid < totalentries && gid >= sliidx[slice_id + 1]) slice_id++; in matmultadd_seqsell_tiled_kernel8()
264 if (row < nrows && gid < totalentries) t = aval[gid] * x[acolidx[gid]]; in matmultadd_seqsell_tiled_kernel8()
267 if (row < nrows && gid < totalentries && write) atomAdd(z[row], t); in matmultadd_seqsell_tiled_kernel8()
[all …]
|