diff --git a/src/solver/fe/pywrap/include/bindings_sem_solver.h b/src/solver/fe/pywrap/include/bindings_sem_solver.h index 90dd944c..5bd90364 100644 --- a/src/solver/fe/pywrap/include/bindings_sem_solver.h +++ b/src/solver/fe/pywrap/include/bindings_sem_solver.h @@ -61,6 +61,11 @@ void bind_sem_solver_base(py::module_ &m) py::arg("time_sample"), py::arg("data")) .def("update_solution", &Solver::updateSolution, py::arg("dt"), py::arg("data")) + .def("get_mass_matrix", + [](Solver& self) -> Kokkos::Experimental::python_view_type_t { + return self.getMassMatrix(); + }, + py::return_value_policy::reference_internal) .def("output_solution_values", &Solver::outputSolutionValues, py::arg("t"), py::arg("e"), py::arg("field_global"), py::arg("field_name"));