Searched refs:col_offset (Results 1 – 1 of 1) sorted by relevance
| /petsc/src/mat/impls/dense/seq/cupm/ |
| H A D | matseqdensecupm.hpp | 1664 …const auto col_offset = [&](const PetscScalar *ptr) { return ptr + col * MatIMPLCast(A)->… in GetColumnVector() local
1677 …PetscCall(PetscCUPMMemcpyAsync(x.data(), col_offset(DeviceArrayRead(dctx, A)), n, cupmMemcpyDevice… in GetColumnVector()
1679 …PetscCall(PetscCUPMMemcpyAsync(x.data(), col_offset(HostArrayRead(dctx, A)), n, cupmMemcpyHostToDe… in GetColumnVector()
1687 PetscCall(PetscArraycpy(x, col_offset(HostArrayRead(dctx, A)), n)); in GetColumnVector()
1689 …PetscCall(PetscCUPMMemcpyAsync(x, col_offset(DeviceArrayRead(dctx, A)), n, cupmMemcpyDeviceToHost,… in GetColumnVector()
|