d9/d34/_anderson_integrator_8cuh_source.html

#ifndef PBAT_GPU_IMPL_VBD_ANDERSONINTEGRATOR_CUH

#define PBAT_GPU_IMPL_VBD_ANDERSONINTEGRATOR_CUH


#include "Integrator.cuh"

#include "pbat/gpu/impl/common/Buffer.cuh"

#include "pbat/gpu/impl/math/Blas.cuh"

#include "pbat/gpu/impl/math/LinearSolver.cuh"

#include "pbat/gpu/impl/math/Matrix.cuh"


namespace pbat::gpu::impl::vbd {


class AndersonIntegrator : public Integrator

{

  public:

    AndersonIntegrator(Data const& data);

    virtual void Solve(kernels::BackwardEulerMinimization& bdf, GpuIndex iterations) override;

    virtual ~AndersonIntegrator() = default;

    void UpdateAndersonWindow(GpuIndex k);

    void TakeAndersonAcceleratedStep(GpuIndex k);

    void RegularizeRFactor(GpuIndex mk);


  private:

    math::Vector<GpuScalar> Fk;

    math::Vector<GpuScalar> Fkm1;

    math::Vector<GpuScalar> Gk;

    math::Vector<GpuScalar> Gkm1;

    math::Vector<GpuScalar> xkm1;

    math::Matrix<GpuScalar>

        mDFK;

    math::Matrix<GpuScalar>

        mDGK;


    math::Matrix<GpuScalar> mQR;

    math::Vector<GpuScalar> mTau;

    math::Blas mBlas;

    math::LinearSolver mLinearSolver;

    common::Buffer<GpuScalar> mLinearSolverWorkspace;

};


} // namespace pbat::gpu::impl::vbd


#endif // PBAT_GPU_IMPL_VBD_ANDERSONINTEGRATOR_CUH

Blas.cuh
BLAS API wrapper over cuBLAS.

LinearSolver.cuh
Linear solver abstractions over cuSolver.

Matrix.cuh
Matrix and vector cuBLAS abstractions.

pbat::gpu::impl::common::Buffer
Definition Buffer.cuh:21

pbat::gpu::impl::math::Blas
Definition Blas.cuh:37

pbat::gpu::impl::math::LinearSolver
Definition LinearSolver.cuh:40

pbat::gpu::impl::vbd::AndersonIntegrator::AndersonIntegrator
AndersonIntegrator(Data const &data)
Construct a new AndersonIntegrator object.
Definition AndersonIntegrator.cu:15

pbat::gpu::impl::vbd::AndersonIntegrator::TakeAndersonAcceleratedStep
void TakeAndersonAcceleratedStep(GpuIndex k)
Take an Anderson accelerated step.
Definition AndersonIntegrator.cu:72

pbat::gpu::impl::vbd::AndersonIntegrator::~AndersonIntegrator
virtual ~AndersonIntegrator()=default
Destroy the Chebyshev Integrator object.

pbat::gpu::impl::vbd::AndersonIntegrator::UpdateAndersonWindow
void UpdateAndersonWindow(GpuIndex k)
Update the Anderson window with the current residual and iterate.
Definition AndersonIntegrator.cu:54

pbat::gpu::impl::vbd::AndersonIntegrator::Solve
virtual void Solve(kernels::BackwardEulerMinimization &bdf, GpuIndex iterations) override
Solve the optimization problem using the Anderson accelerated VBD method.
Definition AndersonIntegrator.cu:36

pbat::gpu::impl::vbd::AndersonIntegrator::RegularizeRFactor
void RegularizeRFactor(GpuIndex mk)
Add regularization to the diagonal of the QR factorization.
Definition AndersonIntegrator.cu:93

pbat::gpu::impl::vbd::Integrator::Integrator
Integrator(Data const &data)
Construct Integrator from data.
Definition Integrator.cu:22

pbat::gpu::impl::vbd::Integrator::Data
pbat::sim::vbd::Data Data
Data type for VBD.
Definition Integrator.cuh:43

pbat::gpu::impl::vbd
Vertex Block Descent (VBD) algorithms.
Definition AndersonIntegrator.cu:13

pbat::GpuIndex
std::int32_t GpuIndex
Index type for GPU code.
Definition Aliases.h:20

pbat::gpu::impl::math::Matrix
Definition Matrix.cuh:150

pbat::gpu::impl::math::Vector
Definition Matrix.cuh:201

pbat::gpu::impl::vbd::kernels::BackwardEulerMinimization
Device-side BFD1 minimization problem.
Definition Kernels.cuh:40

Integrator.cuh
VBD integrator implementation.