Home
last modified time | relevance | path

Searched refs:bid (Results 1 – 9 of 9) sorted by relevance

/petsc/src/ksp/ksp/tutorials/
H A Dex19.c89 for (PetscInt bid = 0; bid < nblocks; bid++) { in main() local
91 …PetscCall(MatCreateSubMatrix(Aseq, loc_blocks[bid], loc_blocks[bid], MAT_INITIAL_MATRIX, &matblock… in main()
93 array[bid * nblocks + bid] = matblock; in main()
102 for (PetscInt bid = 0; bid < nblocks; bid++) PetscCall(MatDestroy(&array[bid * nblocks + bid])); in main() local
152 for (PetscInt bid = 0, gid0 = Istart; bid < nblocks; bid++) { in main() local
155 PetscCall(ISGetSize(loc_blocks[bid], &nn)); // size only in main()
157 PetscCall(ISDestroy(&loc_blocks[bid])); in main()
158 loc_blocks[bid] = new_loc_blocks; in main()
184 for (PetscInt bid = 0; bid < nblocks; bid++) PetscCall(ISDestroy(&loc_blocks[bid])); in main() local
/petsc/src/ksp/pc/impls/vpbjacobi/kokkos/
H A Dvpbjacobi_kok.kokkos.cxx92 PetscInt bid = team.league_rank(); // block id in PCApplyOrTranspose_VPBJacobi_Kokkos()
93 PetscInt n = bs(bid + 1) - bs(bid); // size of this block in PCApplyOrTranspose_VPBJacobi_Kokkos()
94 const PetscScalar *bbuf = &diag(bs2(bid)); in PCApplyOrTranspose_VPBJacobi_Kokkos()
95 const PetscScalar *xbuf = &xv(bs(bid)); in PCApplyOrTranspose_VPBJacobi_Kokkos()
96 PetscScalar *ybuf = &yv(bs(bid)); in PCApplyOrTranspose_VPBJacobi_Kokkos()
/petsc/src/ksp/pc/impls/bjacobi/bjkokkos/
H A Dbjkokkoskernels.kokkos.cxx278 …for (int bid = 0; bid < batch_sz; bid++) PetscCall(PetscPrintf(PetscObjectComm((PetscObject)A), "%… in PCApply_BJKOKKOSKERNELS() local
282 for (int bid = 0; bid < batch_sz; bid++) { in PCApply_BJKOKKOSKERNELS()
283 if ((ii = handle.get_iteration_host(idx + bid * jac->dm_Nf[dmIdx])) > count) count = ii; in PCApply_BJKOKKOSKERNELS()
H A Dbjkokkos.kokkos.cxx677 for (int bid = 0; bid < batch_sz; bid++) { in PCApply_BJKOKKOS() local
679 …jac->max_nits += h_metadata[idx + bid * jac->dm_Nf[dmIdx]].its; // report total number of iteratio… in PCApply_BJKOKKOS()
680 if (h_metadata[idx + bid * jac->dm_Nf[dmIdx]].its > max_nnit) { in PCApply_BJKOKKOS()
681 max_nnit = h_metadata[idx + bid * jac->dm_Nf[dmIdx]].its; in PCApply_BJKOKKOS()
682 mbid = bid; in PCApply_BJKOKKOS()
685 if (h_metadata[idx + bid * jac->dm_Nf[dmIdx]].its > max_nnit) { in PCApply_BJKOKKOS()
686 jac->max_nits = max_nnit = h_metadata[idx + bid * jac->dm_Nf[dmIdx]].its; in PCApply_BJKOKKOS()
687 mbid = bid; in PCApply_BJKOKKOS()
693 …for (int bid = 0; bid < batch_sz; bid++) PetscCall(PetscPrintf(PetscObjectComm((PetscObject)A), "%… in PCApply_BJKOKKOS() local
/petsc/src/vec/is/sf/impls/basic/nvshmem/
H A Dsfnvshmem.cu329 int bid = blockIdx.x; in GetDataFromRemotelyAccessible() local
330 PetscMPIInt pe = srcranks[bid]; in GetDataFromRemotelyAccessible()
333 PetscInt nelems = (dstdisp[bid + 1] - dstdisp[bid]) * unitbytes; in GetDataFromRemotelyAccessible()
334 …nvshmem_getmem_nbi(dst + (dstdisp[bid] - dstdisp[0]) * unitbytes, src + srcdisp[bid] * unitbytes, … in GetDataFromRemotelyAccessible()
490 int bid = blockIdx.x; in WaitAndPutDataToRemotelyAccessible() local
491 PetscMPIInt pe = dstranks[bid]; in WaitAndPutDataToRemotelyAccessible()
494 PetscInt nelems = (srcdisp[bid + 1] - srcdisp[bid]) * unitbytes; in WaitAndPutDataToRemotelyAccessible()
495 nvshmem_uint64_wait_until(srcsig + bid, NVSHMEM_CMP_EQ, 0); /* Wait until the sig = 0 */ in WaitAndPutDataToRemotelyAccessible()
496 srcsig[bid] = 1; in WaitAndPutDataToRemotelyAccessible()
497 …nvshmem_putmem_nbi(dst + dstdisp[bid] * unitbytes, src + (srcdisp[bid] - srcdisp[0]) * unitbytes, … in WaitAndPutDataToRemotelyAccessible()
/petsc/src/sys/objects/device/interface/
H A Ddcontext.cxx522 PetscObjectId bid; in PetscDeviceContextWaitForContext() local
529 bid = PetscObjectCast(dctxb)->id; in PetscDeviceContextWaitForContext()
532 PetscCallCXX(CxxDataCast(dctxa)->upstream()[bid] = CxxDataCast(dctxb)->weak_snapshot()); in PetscDeviceContextWaitForContext()
534 …dctx %" PetscInt64_FMT " waiting on dctx %" PetscInt64_FMT "\n", PetscObjectCast(dctxa)->id, bid)); in PetscDeviceContextWaitForContext()
/petsc/src/mat/impls/aij/seq/kokkos/
H A Daijkok.kokkos.cxx1513 …const PetscInt bid = teamMember.league_rank(); … in MatInvertVariableBlockDiagonal_SeqAIJKokkos()
1514 …const PetscInt rstart = bs(bid); … in MatInvertVariableBlockDiagonal_SeqAIJKokkos()
1515 …const PetscInt m = bs(bid + 1) - bs(bid); … in MatInvertVariableBlockDiagonal_SeqAIJKokkos()
1516 …const auto &B = Kokkos::View<PetscScalar **, Kokkos::LayoutLeft>(&diagVal(bs2(bid)), m, m)… in MatInvertVariableBlockDiagonal_SeqAIJKokkos()
1517 const auto &W = PetscScalarKokkosView(&work(bs2(bid)), m * m); in MatInvertVariableBlockDiagonal_SeqAIJKokkos()
/petsc/src/sys/objects/
H A Daoptions.c328 PetscBool bid; in PetscOptionsGetFromTextInput() local
470 PetscCall(PetscOptionsStringToBool(str, &bid)); in PetscOptionsGetFromTextInput()
472 *((PetscBool *)next->data) = bid; in PetscOptionsGetFromTextInput()
/petsc/doc/faq/
H A Dindex.md331 pinch you can work with the optimized version, although we bid you good luck in