From 4e0cfda9ae6ff463d6e8ca8f1eca07fd5abfe505 Mon Sep 17 00:00:00 2001
From: Andreas Kloeckner <inform@tiker.net>
Date: Wed, 1 Jul 2020 15:40:12 -0500
Subject: [PATCH] Don't freeze in unflatten
---
meshmode/dof_array.py | 21 ++++++++++-----------
1 file changed, 10 insertions(+), 11 deletions(-)
diff --git a/meshmode/dof_array.py b/meshmode/dof_array.py
index f4099b33..f74f2090 100644
--- a/meshmode/dof_array.py
+++ b/meshmode/dof_array.py
@@ -244,17 +244,16 @@ def unflatten(actx: ArrayContext, discr: "_Discretization", ary,
group_starts = np.cumsum([0] + group_sizes)
return DOFArray.from_list(actx, [
- actx.freeze(
- actx.call_loopy(
- prg(),
- grp_start=grp_start, ary=ary,
- nelements=grp.nelements,
- ndofs_per_element=ndofs_per_element,
- )["result"])
- for grp_start, grp, ndofs_per_element in zip(
- group_starts,
- discr.groups,
- ndofs_per_element_per_group)])
+ actx.call_loopy(
+ prg(),
+ grp_start=grp_start, ary=ary,
+ nelements=grp.nelements,
+ ndofs_per_element=ndofs_per_element,
+ )["result"]
+ for grp_start, grp, ndofs_per_element in zip(
+ group_starts,
+ discr.groups,
+ ndofs_per_element_per_group)])
def flat_norm(ary: DOFArray, ord=2):
--
GitLab