Searched refs:bkok (Results 1 – 2 of 2) sorted by relevance
| /petsc/src/mat/impls/aij/seq/kokkos/ |
| H A D | aijkok.kokkos.cxx | 602 Mat_SeqAIJKokkos *akok = static_cast<Mat_SeqAIJKokkos *>(A->spptr), *bkok; in MatDuplicate_SeqAIJKokkos()
611 …bkok = new Mat_SeqAIJKokkos(mat, mat->rmap->n, mat->cmap->n, bseq, mat->nonzerostate, PETSC_FALSE); in MatDuplicate_SeqAIJKokkos()
612 bkok->a_dual.clear_sync_state(); /* Clear B's sync state as it will be decided below */ in MatDuplicate_SeqAIJKokkos()
616 Kokkos::deep_copy(bkok->a_dual.view_host(), akok->a_dual.view_host()); in MatDuplicate_SeqAIJKokkos()
617 bkok->a_dual.modify_host(); in MatDuplicate_SeqAIJKokkos()
619 Kokkos::deep_copy(bkok->a_dual.view_device(), akok->a_dual.view_device()); in MatDuplicate_SeqAIJKokkos()
620 bkok->a_dual.modify_device(); in MatDuplicate_SeqAIJKokkos()
624 bkok->a_dual.modify_host(); in MatDuplicate_SeqAIJKokkos()
626 mat->spptr = bkok; in MatDuplicate_SeqAIJKokkos()
640 Mat_SeqAIJKokkos *atkok, *bkok; in MatTranspose_SeqAIJKokkos() local
[all …]
|
| /petsc/src/mat/impls/aij/mpi/kokkos/ |
| H A D | mpiaijkok.kokkos.cxx | 743 …Mat_SeqAIJKokkos *akok = static_cast<Mat_SeqAIJKokkos *>(A->spptr), *bkok = static_cast<Mat_SeqAIJ… in MatMPIAIJKokkosBcastBegin()
951 PetscCall(KokkosDualViewSyncDevice(bkok->a_dual, PetscGetKokkosExecutionSpace())); in MatMPIAIJKokkosBcastBegin()
955 const auto &Ba = bkok->a_dual.view_device(); in MatMPIAIJKokkosBcastBegin()
957 const auto &Bi = bkok->i_dual.view_device(); in MatMPIAIJKokkosBcastBegin()
|