Home
last modified time | relevance | path

Searched refs:PetscCUPMMallocAsync (Results 1 – 4 of 4) sorted by relevance

/petsc/src/sys/objects/device/impls/cupm/
H A Dcupmallocator.hpp107 PetscCall(PetscCUPMMallocAsync(ptr, n, stream->get_stream())); in allocate()
/petsc/src/mat/impls/dense/seq/cupm/
H A Dmatseqdensecupm.hpp226 PetscCall(PetscCUPMMallocAsync(&mcu->d_v, size, stream)); in SetPreallocation_()
343 PetscCall(PetscCUPMMallocAsync(&mcu->d_fact_work, lwork, stream)); in ResizeFactLwork()
361 if (!mcu->d_fact_info) PetscCall(PetscCUPMMallocAsync(&mcu->d_fact_info, 1, stream)); in FactorPrepare()
401 if (!mcu->d_fact_ipiv) PetscCall(PetscCUPMMallocAsync(&mcu->d_fact_ipiv, n, stream)); in Factor()
499 if (!mcu->d_fact_ipiv) PetscCall(PetscCUPMMallocAsync(&mcu->d_fact_ipiv, n, stream)); in Factor()
502 PetscCall(PetscCUPMMallocAsync(&mcu->d_fact_work, mcu->d_fact_lwork, stream)); in Factor()
504 if (mcu->d_fact_info) PetscCall(PetscCUPMMallocAsync(&mcu->d_fact_info, 1, stream)); in Factor()
578 if (!mcu->d_fact_tau) PetscCall(PetscCUPMMallocAsync(&mcu->d_fact_tau, min, stream)); in Factor()
768 PetscCall(PetscCUPMMallocAsync(&y, nrhs * m)); in MatMatSolve_Factored_Dispatch_()
1810 PetscCall(PetscCUPMMallocAsync(&mcu->d_fact_work, il, stream)); in InvertFactors()
/petsc/include/petsc/private/
H A Dcupminterface.hpp630 …static PetscErrorCode PetscCUPMMallocAsync(M **ptr, std::size_t n, cupmStream_t stream = nullptr) … in PetscCUPMMallocAsync() function
657 PetscCall(PetscCUPMMallocAsync(ptr, n)); in PetscCUPMMalloc()
937 using ::Petsc::device::cupm::impl::Interface<T>::PetscCUPMMallocAsync; \
H A Dveccupmimpl.h403 PetscCall(PetscCUPMMallocAsync(&array_allocated_d, n, stream)); in DeviceAllocateCheck_()
1117 for (auto &elem : cntptrs) PetscCall(PetscCUPMMallocAsync(&elem.device, elem.size, stream)); in SetPreallocationCOO_CUPMBase()
1118 for (auto &elem : bufptrs) PetscCall(PetscCUPMMallocAsync(&elem.device, elem.size, stream)); in SetPreallocationCOO_CUPMBase()