Searched refs:mimpl (Results 1 – 2 of 2) sorted by relevance
| /petsc/src/mat/impls/dense/mpi/cupm/ |
| H A D | matmpidensecupm.hpp | 243 const auto mimpl = MatIMPLCast(A); in BindToCPU() local
247 …PetscCheck(!mimpl->vecinuse, PetscObjectComm(pobj), PETSC_ERR_ORDER, "Need to call MatDenseRestore… in BindToCPU()
248 …PetscCheck(!mimpl->matinuse, PetscObjectComm(pobj), PETSC_ERR_ORDER, "Need to call MatDenseRestore… in BindToCPU()
249 if (const auto mimpl_A = mimpl->A) PetscCall(MatBindToCPU(mimpl_A, usehost)); in BindToCPU()
255 …PetscCall(PetscObjectTypeCompare(PetscObjectCast(mimpl->cvec), VecMPI_CUPM::VECMPICUPM(), &iscupm)… in BindToCPU()
256 if (!iscupm) PetscCall(VecDestroy(&mimpl->cvec)); in BindToCPU()
257 PetscCall(PetscObjectTypeCompare(PetscObjectCast(mimpl->cmat), MATMPIDENSECUPM(), &iscupm)); in BindToCPU()
258 if (!iscupm) PetscCall(MatDestroy(&mimpl->cmat)); in BindToCPU()
270 if (const auto mimpl_cmat = mimpl->cmat) PetscCall(MatBindToCPU(mimpl_cmat, usehost)); in BindToCPU()
321 const auto mimpl = MatIMPLCast(A); in GetColumnVec() local
[all …]
|
| /petsc/src/mat/impls/dense/seq/cupm/ |
| H A D | matseqdensecupm.hpp | 253 const auto mimpl = MatIMPLCast(m); in HostToDevice_() local
254 const auto lda = mimpl->lda; in HostToDevice_()
255 const auto src = mimpl->v; in HostToDevice_()
282 const auto mimpl = MatIMPLCast(m); in DeviceToHost_() local
286 if (!mimpl->v) PetscCall(MatSeqDenseSetPreallocation(m, nullptr)); in DeviceToHost_()
290 const auto lda = mimpl->lda; in DeviceToHost_()
291 const auto dest = mimpl->v; in DeviceToHost_()
562 const auto mimpl = MatIMPLCast(A); in Factor() local
571 mimpl->rank = min; in Factor()
575 const auto lda = static_cast<cupmBlasInt_t>(mimpl->lda); in Factor()
[all …]
|