Searched refs:Agpu (Results 1 – 1 of 1) sorted by relevance
| /petsc/src/mat/impls/aij/seq/seqviennacl/ |
| H A D | aijviennacl.cxx | 98 PetscErrorCode MatViennaCLCopyFromGPU(Mat A, const ViennaCLAIJMatrix *Agpu) in MatViennaCLCopyFromGPU() argument
106 if (A->offloadmask == PETSC_OFFLOAD_UNALLOCATED && Agpu) { in MatViennaCLCopyFromGPU()
109 …etscCheck((PetscInt)Agpu->size1() == m, PETSC_COMM_SELF, PETSC_ERR_ARG_SIZ, "GPU matrix has %lu ro… in MatViennaCLCopyFromGPU()
110 a->nz = Agpu->nnz(); in MatViennaCLCopyFromGPU()
130 row_buffer.raw_resize(Agpu->handle1(), Agpu->size1() + 1); in MatViennaCLCopyFromGPU()
133 viennacl::backend::memory_read(Agpu->handle1(), 0, row_buffer.raw_size(), row_buffer.get()); in MatViennaCLCopyFromGPU()
135 for (PetscInt i = 0; i < (PetscInt)Agpu->size1(); ++i) { in MatViennaCLCopyFromGPU()
142 col_buffer.raw_resize(Agpu->handle2(), Agpu->nnz()); in MatViennaCLCopyFromGPU()
143 viennacl::backend::memory_read(Agpu->handle2(), 0, col_buffer.raw_size(), col_buffer.get()); in MatViennaCLCopyFromGPU()
144 for (PetscInt i = 0; i < (PetscInt)Agpu->nnz(); ++i) (a->j)[i] = col_buffer[i]; in MatViennaCLCopyFromGPU()
[all …]
|