From 3de08d98fe7dd0b181b364191e52da79bbd38d65 Mon Sep 17 00:00:00 2001
From: Andreas Kloeckner <inform@tiker.net>
Date: Wed, 17 Jun 2020 10:56:08 -0500
Subject: [PATCH] Use group.ndofs in unflatten

---
 meshmode/dof_array.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/meshmode/dof_array.py b/meshmode/dof_array.py
index 016cf92e..07aa96d4 100644
--- a/meshmode/dof_array.py
+++ b/meshmode/dof_array.py
@@ -201,7 +201,7 @@ def unflatten(actx: ArrayContext, discr: "_Discretization", ary) -> np.ndarray:
             "result[iel, idof] = ary[grp_start + iel*nunit_dofs + idof]",
             name="unflatten")
 
-    group_sizes = [grp.nelements*grp.nunit_dofs for grp in discr.groups]
+    group_sizes = [grp.ndofs for grp in discr.groups]
     if ary.size != sum(group_sizes):
         raise ValueError("array has size %d, expected %d"
                 % (ary.size, sum(group_sizes)))
-- 
GitLab