Home
last modified time | relevance | path

Searched refs:cupmMemcpyDeviceToDevice (Results 1 – 7 of 7) sorted by relevance

/petsc/src/ksp/ksp/utils/lmvm/blas_cyclic/cupm/
H A Dblas_cyclic_cupm_impl.hpp150 if (x != y) PetscCall(PetscCUPMMemcpyAsync(y, x, m, cupmMemcpyDeviceToDevice, stream)); in dsv()
155 …etscCall(PetscCUPMMemcpyAsync(&y[i_oldest], &x[i_oldest], diff, cupmMemcpyDeviceToDevice, stream)); in dsv()
161 if (x != y) PetscCall(PetscCUPMMemcpyAsync(y, x, i_next, cupmMemcpyDeviceToDevice, stream)); in dsv()
165 …etscCall(PetscCUPMMemcpyAsync(&y[i_oldest], &x[i_oldest], diff, cupmMemcpyDeviceToDevice, stream)); in dsv()
201 if (x != y) PetscCall(PetscCUPMMemcpyAsync(y, x, n, cupmMemcpyDeviceToDevice, stream)); in trsv()
/petsc/src/sys/objects/device/impls/cupm/
H A Dcupminterface.cxx66 PETSC_CUPM_DEFINE_STATIC_VARIABLE_VIA_CLASS_TYPENAME(cupmMemcpyKind_t, cupmMemcpyDeviceToDevice);
H A Dcupmallocator.hpp134 …PetscCall(PetscCUPMMemcpyAsync(dest, src, n, cupmMemcpyDeviceToDevice, stream->get_stream(), true)… in uninitialized_copy()
/petsc/include/petsc/private/
H A Dcupminterface.hpp181 static const auto cupmMemcpyDeviceToDevice = cudaMemcpyDeviceToDevice; member
324 static const auto cupmMemcpyDeviceToDevice = hipMemcpyDeviceToDevice; member
469 using ::Petsc::device::cupm::impl::InterfaceImpl<T>::cupmMemcpyDeviceToDevice; \
618 return cupmMemcpyDeviceToDevice; in PetscDeviceCopyModeToCUPMMemcpyKind()
/petsc/src/vec/vec/impls/seq/cupm/
H A Dvecseqcupm_impl.hpp463 …yWrite(dctx, yin).data(), DeviceArrayRead(dctx, xin).data(), n, cupmMemcpyDeviceToDevice, stream)); in AYPXAsync()
834 …MMemcpyAsync(wptr.data(), DeviceArrayRead(dctx, yin).data(), n, cupmMemcpyDeviceToDevice, stream, … in WAXPYAsync()
1426 auto mode = cupmMemcpyDeviceToDevice; in CopyAsync()
1437 mode = PetscOffloadDevice(xmask) ? cupmMemcpyDeviceToDevice : cupmMemcpyHostToHost; in CopyAsync()
1449 mode = PetscOffloadHost(xmask) ? cupmMemcpyHostToDevice : cupmMemcpyDeviceToDevice; in CopyAsync()
1457 mode = PetscOffloadDevice(xmask) ? cupmMemcpyDeviceToDevice : cupmMemcpyHostToDevice; in CopyAsync()
1465 case cupmMemcpyDeviceToDevice: // the best case in CopyAsync()
1468 …const auto xptr = mode == cupmMemcpyDeviceToDevice ? DeviceArrayRead(dctx, xin).data() : HostArray… in CopyAsync()
1582 … PetscCall(PetscCUPMMemcpyAsync(yptr.data(), xptr.data(), n, cupmMemcpyDeviceToDevice, stream)); in AXPBYAsync()
/petsc/src/vec/is/sf/impls/basic/cupm/
H A Dsfcupm_impl.hpp283 …char *)data + start * link->unitbytes, count * link->unitbytes, cupmMemcpyDeviceToDevice, link->st… in Pack()
299 …*)data + start * link->unitbytes, buf, count * link->unitbytes, cupmMemcpyDeviceToDevice, link->st… in UnpackAndOp()
368 …onst Type *)src + srcStart * link->bs, count * link->unitbytes, cupmMemcpyDeviceToDevice, link->st… in ScatterAndInsert()
587 {cupmMemcpyDeviceToHost, cupmMemcpyDeviceToDevice} in LinkMemcpy()
/petsc/src/mat/impls/dense/seq/cupm/
H A Dmatseqdensecupm.hpp665 const auto copy_mode = xisdevice ? cupmMemcpyDeviceToDevice : cupmMemcpyHostToDevice; in MatSolve_Factored_Dispatch_()
695 const auto copy_mode = yiscupm ? cupmMemcpyDeviceToDevice : cupmMemcpyDeviceToHost; in MatSolve_Factored_Dispatch_()
753 const auto copy_mode = bisdevice ? cupmMemcpyDeviceToDevice : cupmMemcpyHostToDevice; in MatMatSolve_Factored_Dispatch_()
785 const auto copy_mode = xiscupm ? cupmMemcpyDeviceToDevice : cupmMemcpyDeviceToHost; in MatMatSolve_Factored_Dispatch_()
1365 …etscCUPMMemcpy2DAsync(vb.data(), lda_b, va.data(), lda_a, m, n, cupmMemcpyDeviceToDevice, stream)); in Copy()
1367 … PetscCall(PetscCUPMMemcpyAsync(vb.data(), va.data(), m * n, cupmMemcpyDeviceToDevice, stream)); in Copy()
1677 …MMemcpyAsync(x.data(), col_offset(DeviceArrayRead(dctx, A)), n, cupmMemcpyDeviceToDevice, stream)); in GetColumnVector()