Searched refs:VecCUPMCast (Results 1 – 3 of 3) sorted by relevance
101 const auto vcu = VecCUPMCast(v);167 PETSC_NODISCARD static Vec_CUPM *VecCUPMCast(Vec) noexcept;298 auto &device_array = VecCUPMCast(v)->array_allocated_d; in ResetAllocatedDevicePtr_()302 if (PetscDefined(HAVE_NVSHMEM) && VecCUPMCast(v)->nvshmem) { in ResetAllocatedDevicePtr_()366 PetscCall(VecAllocateCheck_(v, v->spptr, VecCUPMCast)); in VecCUPMAllocateCheck_()396 if (auto &alloc = VecCUPMCast(v)->array_d) PetscFunctionReturn(PETSC_SUCCESS); in DeviceAllocateCheck_()399 auto &array_allocated_d = VecCUPMCast(v)->array_allocated_d; in DeviceAllocateCheck_()424 …PetscCall(PetscCUPMMemcpyAsync(VecCUPMCast(v)->array_d, VecIMPLCast(v)->array, v->map->n, cupmMemc… in CopyToDevice_()441 …PetscCall(PetscCUPMMemcpyAsync(VecIMPLCast(v)->array, VecCUPMCast(v)->array_d, v->map->n, cupmMemc… in CopyToHost_()451 if (const auto vcu = VecCUPMCast(v)) { in DestroyDevice_()[all …]
246 const auto vcu = VecCUPMCast(x); in SetPreallocationCOO()329 const auto vcu = VecCUPMCast(x); in SetValuesCOO()
368 if (const auto wcu = VecCUPMCast(w)) { in GetLocalVector()433 PetscCallCUPM(cupmFreeAsync(VecCUPMCast(w)->array_d, stream)); in RestoreLocalVector()2266 const auto vcu = VecCUPMCast(x); in SetValuesCOO()