Use within MGTransferMatrixFree::build()

dealii · Jul 6, 2021 · 011b9d3 · 011b9d3
1 parent 4d01b83
commit 011b9d3
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 7 deletions.
diff --git a/include/deal.II/multigrid/mg_transfer.h b/include/deal.II/multigrid/mg_transfer.h
@@ -619,20 +619,20 @@ class MGLevelGlobalTransfer<LinearAlgebra::distributed::Vector<Number>>
   mutable MGLevelObject<LinearAlgebra::distributed::Vector<Number>>
     solution_ghosted_level_vector;
 
+  /**
+   * Function to initialize internal level vectors.
+   */
+  std::function<void(const unsigned int,
+                     LinearAlgebra::distributed::Vector<Number> &)>
+    initialize_dof_vector;
+
 private:
   /**
    * This function is called to make sure that build() has been invoked.
    */
   template <int dim, int spacedim>
   void
   assert_built(const DoFHandler<dim, spacedim> &dof_handler) const;
-
-  /**
-   * Function to initialize internal level vectors.
-   */
-  const std::function<void(const unsigned int,
-                           LinearAlgebra::distributed::Vector<Number> &)>
-    initialize_dof_vector;
 };
 
 

diff --git a/source/multigrid/mg_transfer_matrix_free.cc b/source/multigrid/mg_transfer_matrix_free.cc
@@ -109,6 +109,14 @@ MGTransferMatrixFree<dim, Number>::build(
   const std::vector<std::shared_ptr<const Utilities::MPI::Partitioner>>
     &external_partitioners)
 {
+  if (external_partitioners.size() > 0)
+    this->initialize_dof_vector =
+      [&external_partitioners](
+        const unsigned int                          level,
+        LinearAlgebra::distributed::Vector<Number> &vec) {
+        vec.reinit(external_partitioners[level]);
+      };
+
   this->fill_and_communicate_copy_indices(dof_handler);
 
   vector_partitioners.resize(0,