Add DealIIMatrixFreeMeshEvaluator::build_range_vector()

ORNL-CEES · Dec 18, 2018 · 6cecfec · 6cecfec
1 parent ebe4438
commit 6cecfec
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 0 deletions.
diff --git a/include/mfmg/dealii/dealii_matrix_free_mesh_evaluator.hpp b/include/mfmg/dealii/dealii_matrix_free_mesh_evaluator.hpp
@@ -34,6 +34,9 @@ class DealIIMatrixFreeMeshEvaluator : public DealIIMeshEvaluator<dim>
   vmult(dealii::LinearAlgebra::distributed::Vector<double> &dst,
         dealii::LinearAlgebra::distributed::Vector<double> const &src) const;
 
+  std::shared_ptr<dealii::LinearAlgebra::distributed::Vector<double>>
+  build_range_vector() const;
+
 private:
   std::shared_ptr<dealii::TrilinosWrappers::SparseMatrix> _sparse_matrix;
 };

diff --git a/source/dealii/dealii_matrix_free_mesh_evaluator.cc b/source/dealii/dealii_matrix_free_mesh_evaluator.cc
@@ -44,6 +44,15 @@ void DealIIMatrixFreeMeshEvaluator<dim>::vmult(
 {
   _sparse_matrix->vmult(dst, src);
 }
+
+template <int dim>
+std::shared_ptr<dealii::LinearAlgebra::distributed::Vector<double>>
+DealIIMatrixFreeMeshEvaluator<dim>::build_range_vector() const
+{
+  return std::make_shared<dealii::LinearAlgebra::distributed::Vector<double>>(
+      _sparse_matrix->locally_owned_range_indices(),
+      _sparse_matrix->get_mpi_communicator());
+}
 } // namespace mfmg
 
 // Explicit Instantiation