From 542c99fa691736f493de05dbfe47baf59e4bdef7 Mon Sep 17 00:00:00 2001
From: Matt Wala <wala1@illinois.edu>
Date: Fri, 26 Jun 2020 16:09:56 -0500
Subject: [PATCH] Fix a loopy kerel error

---
 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 b26732c3..c77fdd0f 100644
--- a/meshmode/dof_array.py
+++ b/meshmode/dof_array.py
@@ -195,7 +195,7 @@ def flatten(ary: np.ndarray) -> np.ndarray:
     def prg():
         return make_loopy_program(
             "{[iel,idof]: 0<=iel<nelements and 0<=idof<ndofs_per_element}",
-            "result[grp_start + iel*nunit_dofs + idof] = grp_ary[iel, idof]",
+            "result[grp_start + iel*ndofs_per_element + idof] = grp_ary[iel, idof]",  # noqa
             name="flatten")
 
     result = actx.empty(group_starts[-1], dtype=ary.entry_dtype)
-- 
GitLab