Searched refs:nBlk (Results 1 – 2 of 2) sorted by relevance
| /petsc/src/ksp/pc/impls/bjacobi/bjkokkos/ |
| H A D | bjkokkoskernels.kokkos.cxx | 159 const PetscInt maxit = jac->ksp->max_it, nBlk = jac->nBlocks; in PCApply_BJKOKKOSKERNELS() local
160 const int Nsolves = nBlk; in PCApply_BJKOKKOSKERNELS()
299 for (int blkID = 0; blkID < nBlk; blkID++) { in PCApply_BJKOKKOSKERNELS()
311 for (int blkID = 0; blkID < nBlk; blkID++) { in PCApply_BJKOKKOSKERNELS()
|
| H A D | bjkokkos.kokkos.cxx | 539 const PetscInt nwork = jac->nwork, nBlk = jac->nBlocks; in PCApply_BJKOKKOS() local
589 …ck(nBlk % batch_sz == 0, PetscObjectComm((PetscObject)pc), PETSC_ERR_ARG_WRONG, "batch_sz = %" Pet… in PCApply_BJKOKKOS()
600 …Kokkos::View<Batch_MetaData *, Kokkos::DefaultExecutionSpace> d_metadata("solver meta data", nBlk); in PCApply_BJKOKKOS()
628 …ors\n", (int)jac->n, scr_bytes_team_shared, global_buff_words, rtol, (int)nBlk, (int)team_size, PC… in PCApply_BJKOKKOS()
632 …"Solve", Kokkos::TeamPolicy<Kokkos::LaunchBounds<256, 4>>(nBlk, team_size, PCBJKOKKOS_VEC_SIZE).se… in PCApply_BJKOKKOS()
706 for (int blkID = 0; blkID < nBlk; blkID++) { in PCApply_BJKOKKOS()
726 for (int blkID = 0; blkID < nBlk; blkID++) { in PCApply_BJKOKKOS()
734 nBlk, in PCApply_BJKOKKOS()
741 for (int blkID = 0; blkID < nBlk; blkID++) { in PCApply_BJKOKKOS()
|