Home
last modified time | relevance | path

Searched refs:a_dual (Results 1 – 3 of 3) sorted by relevance

/petsc/src/mat/impls/aij/seq/kokkos/
H A Daijkok.hpp106 MatScalarKokkosDualView a_dual; member
136 a_dual = MatScalarKokkosDualView(a_d, a_h); in Mat_SeqAIJKokkos()
141 a_dual.modify_host(); /* Since caller provided values on host */ in Mat_SeqAIJKokkos()
142 if (copyValues) (void)KokkosDualViewSyncDevice(a_dual, exec); in Mat_SeqAIJKokkos()
160 a_dual = MatScalarKokkosDualView(a_d, a_h); in Mat_SeqAIJKokkos()
161 … a_dual.modify_device(); /* since we did not copy a_d to a_h, we mark device has the latest data */ in Mat_SeqAIJKokkos()
171 MatScalarType *a_host_data() { return a_dual.view_host().data(); } in a_host_data()
175 MatScalarType *a_device_data() { return a_dual.view_device().data(); } in a_device_data()
184 …void SetColSize(MatColIdxType n) { csrmat = KokkosCsrMatrix("csrmat", n, a_dual.view_device(), csr… in SetColSize()
H A Daijkok.kokkos.cxx117 if (aijkok->a_dual.need_sync_device()) { in MatSeqAIJKokkosSyncDevice()
118 PetscCall(KokkosDualViewSyncDevice(aijkok->a_dual, PetscGetKokkosExecutionSpace())); in MatSeqAIJKokkosSyncDevice()
132 aijkok->a_dual.clear_sync_state(); in MatSeqAIJKokkosModifyDevice()
133 aijkok->a_dual.modify_device(); in MatSeqAIJKokkosModifyDevice()
150 PetscCall(KokkosDualViewSyncHost(aijkok->a_dual, exec)); in MatSeqAIJKokkosSyncHost()
165 PetscCall(KokkosDualViewSyncHost(aijkok->a_dual, PetscGetKokkosExecutionSpace())); in MatSeqAIJGetArray_SeqAIJKokkos()
166 *array = aijkok->a_dual.view_host().data(); in MatSeqAIJGetArray_SeqAIJKokkos()
178 if (aijkok && A->nonzerostate == aijkok->nonzerostate) aijkok->a_dual.modify_host(); in MatSeqAIJRestoreArray_SeqAIJKokkos()
188 PetscCall(KokkosDualViewSyncHost(aijkok->a_dual, PetscGetKokkosExecutionSpace())); in MatSeqAIJGetArrayRead_SeqAIJKokkos()
189 *array = aijkok->a_dual.view_host().data(); in MatSeqAIJGetArrayRead_SeqAIJKokkos()
[all …]
/petsc/src/mat/impls/aij/mpi/kokkos/
H A Dmpiaijkok.kokkos.cxx950 PetscCall(KokkosDualViewSyncDevice(akok->a_dual, PetscGetKokkosExecutionSpace())); in MatMPIAIJKokkosBcastBegin()
951 PetscCall(KokkosDualViewSyncDevice(bkok->a_dual, PetscGetKokkosExecutionSpace())); in MatMPIAIJKokkosBcastBegin()
954 const auto &Aa = akok->a_dual.view_device(); in MatMPIAIJKokkosBcastBegin()
955 const auto &Ba = bkok->a_dual.view_device(); in MatMPIAIJKokkosBcastBegin()