From e9cf2c94c521bf76b5f5b9ed7473b76904a065e2 Mon Sep 17 00:00:00 2001 From: schnellerhase <56360279+schnellerhase@users.noreply.github.com> Date: Mon, 19 Aug 2024 10:44:15 +0200 Subject: [PATCH] Adapt export --- python/dolfinx/wrappers/la.cpp | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/python/dolfinx/wrappers/la.cpp b/python/dolfinx/wrappers/la.cpp index e84590e31d1..41fd44fd386 100644 --- a/python/dolfinx/wrappers/la.cpp +++ b/python/dolfinx/wrappers/la.cpp @@ -135,10 +135,10 @@ void declare_objects(nb::module_& m, const std::string& type) { const std::array bs = self.block_size(); std::size_t nrows = self.num_all_rows() * bs[0]; - auto map_col = self.index_map(1); - std::size_t ncols - = (map_col->size_local() + map_col->num_ghosts()) * bs[1]; - return dolfinx_wrappers::as_nbarray(self.to_dense(), + std::size_t ncols = self.index_map(1)->size_global() * bs[1]; + auto dense = self.to_dense(); + assert(nrows*ncols == dense.size()); + return dolfinx_wrappers::as_nbarray(std::move(self.to_dense()), {nrows, ncols}); }) .def_prop_ro(