Home
last modified time | relevance | path

Searched refs:cupmMemcpyDeviceToHost (Results 1 – 6 of 6) sorted by relevance

/petsc/src/sys/objects/device/impls/cupm/
H A Dcupminterface.cxx65 PETSC_CUPM_DEFINE_STATIC_VARIABLE_VIA_CLASS_TYPENAME(cupmMemcpyKind_t, cupmMemcpyDeviceToHost);
/petsc/include/petsc/private/
H A Dcupminterface.hpp180 static const auto cupmMemcpyDeviceToHost = cudaMemcpyDeviceToHost; member
323 static const auto cupmMemcpyDeviceToHost = hipMemcpyDeviceToHost; member
468 using ::Petsc::device::cupm::impl::InterfaceImpl<T>::cupmMemcpyDeviceToHost; \
620 return cupmMemcpyDeviceToHost; in PetscDeviceCopyModeToCUPMMemcpyKind()
699 if (use_async || stream || (kind != cupmMemcpyDeviceToHost)) { in PetscCUPMMemcpyAsync()
729 if (stream || (kind != cupmMemcpyDeviceToHost)) { in PetscCUPMMemcpy2DAsync()
917 …if (kind == cupmMemcpyDeviceToHost) PetscCall(PetscLogGpuToCpu(static_cast<PetscLogDouble>(size))); in PetscLogCUPMMemcpyTransfer()
H A Dveccupmimpl.h441 …Async(VecIMPLCast(v)->array, VecCUPMCast(v)->array_d, v->map->n, cupmMemcpyDeviceToHost, stream, f… in CopyToHost_()
/petsc/src/vec/vec/impls/seq/cupm/
H A Dvecseqcupm_impl.hpp1255 PetscCall(PetscCUPMMemcpyAsync(z, d_results, nv, cupmMemcpyDeviceToHost, stream)); in MDot_()
1294 PetscCall(PetscCUPMMemcpyAsync(z, d_z, nv, cupmMemcpyDeviceToHost, stream)); in MDot_()
1451 mode = PetscOffloadHost(xmask) ? cupmMemcpyHostToHost : cupmMemcpyDeviceToHost; in CopyAsync()
1474 case cupmMemcpyDeviceToHost: // not great in CopyAsync()
1476 …const auto xptr = mode == cupmMemcpyDeviceToHost ? DeviceArrayRead(dctx, xin).data() : HostArray… in CopyAsync()
1480 if (mode == cupmMemcpyDeviceToHost) PetscCall(PetscLogGpuTimeBegin()); in CopyAsync()
1482 if (mode == cupmMemcpyDeviceToHost) PetscCall(PetscLogGpuTimeEnd()); in CopyAsync()
1667 …PetscCall(PetscCUPMMemcpyAsync(&xv, xptr.data() + max_loc - 1, 1, cupmMemcpyDeviceToHost, stream)); in Norm()
/petsc/src/mat/impls/dense/seq/cupm/
H A Dmatseqdensecupm.hpp295 …PetscCall(PetscCUPMMemcpy2DAsync(dest, lda, src, lda, nrows, ncols, cupmMemcpyDeviceToHost, stream… in DeviceToHost_()
297 PetscCall(PetscCUPMMemcpyAsync(dest, src, lda * ncols, cupmMemcpyDeviceToHost, stream)); in DeviceToHost_()
314 PetscCall(PetscCUPMMemcpyAsync(&info, fact_info, 1, cupmMemcpyDeviceToHost, stream)); in CheckCUPMSolverInfo_()
695 const auto copy_mode = yiscupm ? cupmMemcpyDeviceToDevice : cupmMemcpyDeviceToHost; in MatSolve_Factored_Dispatch_()
785 const auto copy_mode = xiscupm ? cupmMemcpyDeviceToDevice : cupmMemcpyDeviceToHost; in MatMatSolve_Factored_Dispatch_()
1689 …(PetscCUPMMemcpyAsync(x, col_offset(DeviceArrayRead(dctx, A)), n, cupmMemcpyDeviceToHost, stream)); in GetColumnVector()
/petsc/src/vec/is/sf/impls/basic/cupm/
H A Dsfcupm_impl.hpp587 {cupmMemcpyDeviceToHost, cupmMemcpyDeviceToDevice} in LinkMemcpy()