Searched refs:vcu (Results 1 – 3 of 3) sorted by relevance
| /petsc/src/vec/vec/impls/mpi/cupm/ |
| H A D | vecmpicupm_impl.hpp | 246 const auto vcu = VecCUPMCast(x); in SetPreallocationCOO() local
254 make_coo_pair(vcu->imap2_d, vmpi->imap2, vmpi->nnz2), in SetPreallocationCOO()
255 make_coo_pair(vcu->jmap2_d, vmpi->jmap2, vmpi->nnz2 + 1), in SetPreallocationCOO()
256 make_coo_pair(vcu->perm2_d, vmpi->perm2, vmpi->recvlen), in SetPreallocationCOO()
257 make_coo_pair(vcu->Cperm_d, vmpi->Cperm, vmpi->sendlen) in SetPreallocationCOO()
260 make_coo_pair(vcu->sendbuf_d, vmpi->sendbuf, vmpi->sendlen), in SetPreallocationCOO()
261 make_coo_pair(vcu->recvbuf_d, vmpi->recvbuf, vmpi->recvlen) in SetPreallocationCOO()
329 const auto vcu = VecCUPMCast(x); in SetValuesCOO() local
331 const auto sendbuf_d = vcu->sendbuf_d; in SetValuesCOO()
332 const auto recvbuf_d = vcu->recvbuf_d; in SetValuesCOO()
[all …]
|
| /petsc/include/petsc/private/ |
| H A D | veccupmimpl.h | 101 const auto vcu = VecCUPMCast(v); variable
107 PetscAssertPointer(vcu, 1);
115 PetscCall(PetscPrintf(comm, "Device ptr: %p\n", vcu->array_d));
116 PetscCall(PetscPrintf(comm, "Device alloced ptr: %p\n", vcu->array_allocated_d));
117 PetscCall(PetscCUPMGetMemType(vcu->array_d, &mtype));
451 if (const auto vcu = VecCUPMCast(v)) { in DestroyDevice_() local
779 const auto vcu = VecCUPMCast(v); in ReplaceArray() local
783 vcu->array_d = vcu->array_allocated_d /* = aptr */; in ReplaceArray()
813 const auto vcu = VecCUPMCast(v); in ResetArray() local
826 vcu->array_d = host_array; in ResetArray()
[all …]
|
| /petsc/src/vec/vec/impls/seq/cupm/ |
| H A D | vecseqcupm_impl.hpp | 2266 const auto vcu = VecCUPMCast(x); in SetValuesCOO() local
2268 …tscCUPMLaunchKernel1D(n, 0, stream, kernels::add_coo_values, vv, n, vcu->jmap1_d, vcu->perm1_d, im… in SetValuesCOO()
|