From 9604734564d85ebc174841bd7c2fc143eb47ddcd Mon Sep 17 00:00:00 2001 From: Martin Kronbichler Date: Fri, 27 Sep 2019 16:11:39 +0200 Subject: [PATCH] Use proper partitioner in step-64 --- examples/step-64/step-64.cu | 16 ++++++++++++++-- 1 file changed, 14 insertions(+), 2 deletions(-) diff --git a/examples/step-64/step-64.cu b/examples/step-64/step-64.cu index abc67f1696..1ec73ef854 100644 --- a/examples/step-64/step-64.cu +++ b/examples/step-64/step-64.cu @@ -236,6 +236,9 @@ namespace Step64 const LinearAlgebra::distributed::Vector &src) const; + void initialize_dof_vector( + LinearAlgebra::distributed::Vector &vec) const; + private: CUDAWrappers::MatrixFree mf_data; LinearAlgebra::CUDAWrappers::Vector coef; @@ -304,6 +307,15 @@ namespace Step64 } + + template + void HelmholtzOperator::initialize_dof_vector( + LinearAlgebra::distributed::Vector &vec) const + { + mf_data.initialize_dof_vector(vec); + } + + // @sect3{Class HelmholtzProblem} // This is the main class of this program. It defines the usual @@ -401,8 +413,8 @@ namespace Step64 ghost_solution_host.reinit(locally_owned_dofs, locally_relevant_dofs, mpi_communicator); - solution_dev.reinit(locally_owned_dofs, mpi_communicator); - system_rhs_dev.reinit(locally_owned_dofs, mpi_communicator); + system_matrix_dev->initialize_dof_vector(solution_dev); + system_rhs_dev.reinit(solution_dev); } -- 2.39.5