Home
last modified time | relevance | path

Searched refs:array_d (Results 1 – 5 of 5) sorted by relevance

/petsc/src/vec/vec/impls/seq/kokkos/
H A Dveckokkosimpl.hpp46 Vec_Kokkos(PetscInt n, PetscScalar *array_h, PetscScalar *array_d = NULL) in Vec_Kokkos()
51 if (array_d) { in Vec_Kokkos()
52 v_d = PetscScalarKokkosView(array_d, n); /* Use the given device array */ in Vec_Kokkos()
57 if (!array_d) v_dual.modify_host(); in Vec_Kokkos()
H A Dveckok.kokkos.cxx1491 const PetscScalar *array_d = xkok->v_dual.view_device().data() + start; in VecGetSubVector_Kokkos_Private() local
1494 …if (xIsMPI) PetscCall(VecCreateMPIKokkosWithArrays_Private(comm, bs, n, N, array_h, array_d, &z));… in VecGetSubVector_Kokkos_Private()
1495 else PetscCall(VecCreateSeqKokkosWithArrays_Private(comm, bs, n, array_h, array_d, &z)); in VecGetSubVector_Kokkos_Private()
1873 PetscScalar *array_h, *array_d; in VecDuplicateVecs_SeqKokkos_GEMV() local
1887 array_d = w_dual.view_device().data(); in VecDuplicateVecs_SeqKokkos_GEMV()
1890 …PetscCall(VecCreateSeqKokkosWithLayoutAndArrays_Private(map, &array_h[i * lda], &array_d[i * lda],… in VecDuplicateVecs_SeqKokkos_GEMV()
/petsc/include/petsc/private/
H A Dveccupmimpl.h115 PetscCall(PetscPrintf(comm, "Device ptr: %p\n", vcu->array_d));
117 PetscCall(PetscCUPMGetMemType(vcu->array_d, &mtype));
145 PetscScalar *array_d; // gpu data member
396 if (auto &alloc = VecCUPMCast(v)->array_d) PetscFunctionReturn(PETSC_SUCCESS); in DeviceAllocateCheck_()
424 …PetscCall(PetscCUPMMemcpyAsync(VecCUPMCast(v)->array_d, VecIMPLCast(v)->array, v->map->n, cupmMemc… in CopyToDevice_()
441 …PetscCall(PetscCUPMMemcpyAsync(VecIMPLCast(v)->array, VecCUPMCast(v)->array_d, v->map->n, cupmMemc… in CopyToHost_()
593 *a = hostmem ? VecIMPLCast(v)->array : VecCUPMCast(v)->array_d; in GetArray()
736 backup_array = util::exchange(VecCUPMCast(v)->array_d, const_cast<PetscScalar *>(a)); in PlaceArray()
783 vcu->array_d = vcu->array_allocated_d /* = aptr */; in ReplaceArray()
826 vcu->array_d = host_array; in ResetArray()
[all …]
/petsc/src/vec/vec/impls/mpi/kokkos/
H A Dmpikok.kokkos.cxx286 PetscScalar *array_h, *array_d; in VecDuplicateVecs_MPIKokkos_GEMV() local
306 array_d = w_dual.view_device().data(); in VecDuplicateVecs_MPIKokkos_GEMV()
309 …PetscCall(VecCreateMPIKokkosWithLayoutAndArrays_Private(map, &array_h[i * lda], &array_d[i * lda],… in VecDuplicateVecs_MPIKokkos_GEMV()
/petsc/src/vec/vec/impls/seq/cupm/
H A Dvecseqcupm_impl.hpp369 if (auto &device_array = wcu->array_d) { in GetLocalVector()
433 PetscCallCUPM(cupmFreeAsync(VecCUPMCast(w)->array_d, stream)); in RestoreLocalVector()