From 0d2a281da2c52bb3f9d5d328f30e924a2373fee4 Mon Sep 17 00:00:00 2001 From: Thomas Gibson Date: Thu, 22 Apr 2021 13:36:58 -0500 Subject: [PATCH] Move actx.freeze/from_numpy into memoized function --- grudge/op.py | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/grudge/op.py b/grudge/op.py index 66ac1183..25a9d913 100644 --- a/grudge/op.py +++ b/grudge/op.py @@ -343,13 +343,13 @@ def reference_mass_matrix(actx, out_element_group, in_element_group): vand_inv_t = np.linalg.inv(vand).T weights = in_grp.quadrature_rule().weights - # actx.np.einsum? - return np.einsum("j,ik,jk->ij", weights, vand_inv_t, o_vand) + return actx.freeze( + actx.from_numpy( + np.einsum("j,ik,jk->ij", weights, vand_inv_t, o_vand) + ) + ) - return actx.freeze( - actx.from_numpy(get_ref_mass_mat(out_element_group, - in_element_group)) - ) + return get_ref_mass_mat(out_element_group, in_element_group) def _apply_mass_operator(dcoll, dd_out, dd_in, vec): -- GitLab