| /petsc/src/vec/vec/impls/seq/cupm/ |
| H A D | vecseqcupm_impl.hpp | 40 inline Vec_Seq *VecSeq_CUPM<T>::VecIMPLCast_(Vec v) noexcept in VecIMPLCast_()
46 inline constexpr VecType VecSeq_CUPM<T>::VECIMPLCUPM_() noexcept in VECIMPLCUPM_()
52 inline constexpr VecType VecSeq_CUPM<T>::VECIMPL_() noexcept in VECIMPL_()
58 inline PetscErrorCode VecSeq_CUPM<T>::ClearAsyncFunctions(Vec v) noexcept in ClearAsyncFunctions()
87 inline PetscErrorCode VecSeq_CUPM<T>::InitializeAsyncFunctions(Vec v) noexcept in InitializeAsyncFunctions()
90 …PetscCall(PetscObjectComposeFunction(PetscObjectCast(v), VecAsyncFnName(Abs), VecSeq_CUPM<T>::AbsA… in InitializeAsyncFunctions()
91 …PetscCall(PetscObjectComposeFunction(PetscObjectCast(v), VecAsyncFnName(AXPBY), VecSeq_CUPM<T>::AX… in InitializeAsyncFunctions()
92 …PetscCall(PetscObjectComposeFunction(PetscObjectCast(v), VecAsyncFnName(AXPBYPCZ), VecSeq_CUPM<T>:… in InitializeAsyncFunctions()
93 …PetscCall(PetscObjectComposeFunction(PetscObjectCast(v), VecAsyncFnName(AXPY), VecSeq_CUPM<T>::AXP… in InitializeAsyncFunctions()
94 …PetscCall(PetscObjectComposeFunction(PetscObjectCast(v), VecAsyncFnName(AYPX), VecSeq_CUPM<T>::AYP… in InitializeAsyncFunctions()
[all …]
|
| H A D | vecseqcupm.hpp | 26 class VecSeq_CUPM : Vec_CUPMBase<T, VecSeq_CUPM<T>> { class
28 PETSC_VEC_CUPM_BASE_CLASS_HEADER(base_type, T, VecSeq_CUPM<T>);
185 PetscCall(impl::VecSeq_CUPM<T>::CreateSeqCUPM(comm, 0, n, v, PETSC_TRUE)); in VecCreateSeqCUPMAsync()
195 PetscCall(impl::VecSeq_CUPM<T>::CreateSeqCUPMWithBothArrays(comm, bs, n, cpuarray, gpuarray, v)); in VecCreateSeqCUPMWithArraysAsync()
206 PetscCall(impl::VecSeq_CUPM<T>::template GetArray<PETSC_MEMTYPE_DEVICE, mode>(v, a, dctx)); in VecCUPMGetArrayAsync_Private()
216 PetscCall(impl::VecSeq_CUPM<T>::template RestoreArray<PETSC_MEMTYPE_DEVICE, mode>(v, a, dctx)); in VecCUPMRestoreArrayAsync_Private()
273 PetscCall(impl::VecSeq_CUPM<T>::template PlaceArray<PETSC_MEMTYPE_DEVICE>(vin, a)); in VecCUPMPlaceArrayAsync()
282 PetscCall(impl::VecSeq_CUPM<T>::template ReplaceArray<PETSC_MEMTYPE_DEVICE>(vin, a)); in VecCUPMReplaceArrayAsync()
291 PetscCall(impl::VecSeq_CUPM<T>::template ResetArray<PETSC_MEMTYPE_DEVICE>(vin)); in VecCUPMResetArrayAsync()
302 …ass PETSC_SINGLE_LIBRARY_VISIBILITY_INTERNAL ::Petsc::vec::cupm::impl::VecSeq_CUPM<::Petsc::device…
[all …]
|
| /petsc/src/vec/vec/impls/seq/cupm/cuda/ |
| H A D | vecseqcupm.cu | 7 template class impl::VecSeq_CUPM<DeviceType::CUDA>; member in impl
9 static constexpr auto VecSeq_CUDA = impl::VecSeq_CUPM<DeviceType::CUDA>{};
|
| /petsc/src/vec/vec/impls/seq/cupm/hip/ |
| H A D | vecseqcupm.hip.cxx | 7 template class impl::VecSeq_CUPM<DeviceType::HIP>; member in impl
9 static constexpr auto VecSeq_HIP = impl::VecSeq_CUPM<DeviceType::HIP>{};
|
| /petsc/src/mat/impls/dense/seq/cupm/ |
| H A D | matseqdensecupm.hpp | 656 PetscCall(PetscObjectTypeCompare(PetscObjectCast(x), VecSeq_CUPM::VECSEQCUPM(), &xiscupm)); in MatSolve_Factored_Dispatch_()
657 PetscCall(PetscObjectTypeCompare(PetscObjectCast(y), VecSeq_CUPM::VECSEQCUPM(), &yiscupm)); in MatSolve_Factored_Dispatch_()
816 …peCompareAny(PetscObjectCast(xx), &xiscupm, VecSeq_CUPM::VECSEQCUPM(), VecSeq_CUPM::VECMPICUPM(), … in MatMultAddColumnRange_Dispatch_()
820 PetscCall(VecSetType(x, VecSeq_CUPM::VECCUPM())); in MatMultAddColumnRange_Dispatch_()
825 …peCompareAny(PetscObjectCast(yy), &yiscupm, VecSeq_CUPM::VECSEQCUPM(), VecSeq_CUPM::VECMPICUPM(), … in MatMultAddColumnRange_Dispatch_()
829 PetscCall(VecSetType(y, VecSeq_CUPM::VECCUPM())); in MatMultAddColumnRange_Dispatch_()
835 …peCompareAny(PetscObjectCast(zz), &ziscupm, VecSeq_CUPM::VECSEQCUPM(), VecSeq_CUPM::VECMPICUPM(), … in MatMultAddColumnRange_Dispatch_()
839 PetscCall(VecSetType(z, VecSeq_CUPM::VECCUPM())); in MatMultAddColumnRange_Dispatch_()
845 if (y && y != z) PetscCall(VecSeq_CUPM::Copy(y, z)); // mult add in MatMultAddColumnRange_Dispatch_()
848 if (!y) PetscCall(VecSeq_CUPM::Set(z, 0.0)); in MatMultAddColumnRange_Dispatch_()
[all …]
|
| /petsc/src/vec/vec/impls/mpi/cupm/ |
| H A D | vecmpicupm.hpp | 23 using VecSeq_T = VecSeq_CUPM<T>;
|
| /petsc/include/petsc/private/ |
| H A D | matdensecupmimpl.h | 35 class VecSeq_CUPM; variable
123 using VecSeq_CUPM = ::Petsc::vec::cupm::impl::VecSeq_CUPM<T>; \
460 auto dv = VecSeq_CUPM::DeviceArrayWrite(dctx, v); in GetDiagonal()
|
| H A D | veccupmimpl.h | 71 class VecSeq_CUPM; variable
1012 using VecSeq_T = VecSeq_CUPM<T>; in BindToCPU_CUPMBase()
|
| /petsc/src/mat/impls/aij/seq/seqhipsparse/ |
| H A D | hipsparsematimpl.h | 298 using VecSeq_HIP = Petsc::vec::cupm::impl::VecSeq_CUPM<Petsc::device::cupm::DeviceType::HIP>;
|
| /petsc/src/mat/impls/aij/seq/seqcusparse/ |
| H A D | cusparsematimpl.h | 335 using VecSeq_CUDA = Petsc::vec::cupm::impl::VecSeq_CUPM<Petsc::device::cupm::DeviceType::CUDA>;
|