solveVselfInBinsDevice.h

Go to the documentation of this file.

// ---------------------------------------------------------------------
//
// Copyright (c) 2017-2025 The Regents of the University of Michigan and DFT-FE
// authors.
//
// This file is part of the DFT-FE code.
//
// The DFT-FE code is free software; you can use it, redistribute
// it, and/or modify it under the terms of the GNU Lesser General
// Public License as published by the Free Software Foundation; either
// version 2.1 of the License, or (at your option) any later version.
// The full text of the license can be found in the file LICENSE at
// the top level of the DFT-FE distribution.
//
// ---------------------------------------------------------------------
//
 
#if defined(DFTFE_WITH_DEVICE)
#  ifndef solveVselfInBinsDevice_H_
#    define solveVselfInBinsDevice_H_
 
#    include <constraintMatrixInfo.h>
#    include <headers.h>
#    include <BLASWrapper.h>
namespace dftfe
{
  namespace poissonDevice
  {
    void
    solveVselfInBins(
      const dftfe::utils::MemoryStorage<double,
                                        dftfe::utils::MemorySpace::DEVICE>
        &cellGradNIGradNJIntergralDevice,
      const std::shared_ptr<
        dftfe::linearAlgebra::BLASWrapper<dftfe::utils::MemorySpace::DEVICE>>
        &                                      BLASWrapperPtr,
      const dealii::MatrixFree<3, double> &    matrixFreeData,
      const unsigned int                       mfDofHandlerIndex,
      const dealii::AffineConstraints<double> &hangingPeriodicConstraintMatrix,
      const double *                           rhsFlattenedH,
      const double *                           diagonalAH,
      const double *                           inhomoIdsColoredVecFlattenedH,
      const unsigned int                       localSize,
      const unsigned int                       ghostSize,
      const unsigned int                       numberBins,
      const MPI_Comm &                         mpiCommParent,
      const MPI_Comm &                         mpiCommDomain,
      double *                                 xH,
      const int                                verbosity,
      const unsigned int                       maxLinearSolverIterations,
      const double                             absLinearSolverTolerance,
      const bool isElectroFEOrderDifferentFromFEOrder = false);
 
    void
    cgSolver(
      const std::shared_ptr<
        dftfe::linearAlgebra::BLASWrapper<dftfe::utils::MemorySpace::DEVICE>>
        &BLASWrapperPtr,
      dftUtils::constraintMatrixInfo<dftfe::utils::MemorySpace::DEVICE>
        &           constraintsMatrixDataInfoDevice,
      const double *bD,
      const double *diagonalAD,
      const dftfe::utils::MemoryStorage<double,
                                        dftfe::utils::MemorySpace::DEVICE>
        &poissonCellStiffnessMatricesD,
      const dftfe::utils::MemoryStorage<double,
                                        dftfe::utils::MemorySpace::DEVICE>
        &inhomoIdsColoredVecFlattenedD,
      const dftfe::utils::MemoryStorage<dealii::types::global_dof_index,
                                        dftfe::utils::MemorySpace::DEVICE>
        &                           cellLocalProcIndexIdMapD,
      const unsigned int            localSize,
      const unsigned int            ghostSize,
      const unsigned int            numberBins,
      const unsigned int            totalLocallyOwnedCells,
      const unsigned int            numberNodesPerElement,
      const int                     debugLevel,
      const unsigned int            maxIter,
      const double                  absTol,
      const MPI_Comm &              mpiCommParent,
      const MPI_Comm &              mpiCommDomain,
      distributedDeviceVec<double> &x);
  } // namespace poissonDevice
} // namespace dftfe
#  endif
#endif