From b762de80920dddca76aaaecf26823a7b7607b3a3 Mon Sep 17 00:00:00 2001
From: Thomas Gibson <gibsonthomas1120@hotmail.com>
Date: Tue, 11 May 2021 17:57:37 -0500
Subject: [PATCH] Use memoized meshmode functions to assemble the stiffness
matrix
---
grudge/op.py | 7 ++++---
1 file changed, 4 insertions(+), 3 deletions(-)
diff --git a/grudge/op.py b/grudge/op.py
index 7028d1b0..377065e5 100644
--- a/grudge/op.py
+++ b/grudge/op.py
@@ -277,13 +277,14 @@ def reference_stiffness_transpose_matrix(actx, out_element_group, in_element_gro
in_grp.discretization_key()))
def get_ref_stiffness_transpose_mat(out_grp, in_grp):
if in_grp == out_grp:
- mmat = reference_mass_matrix(actx, out_grp, in_grp)
+ from meshmode.discretization.poly_element import \
+ mass_matrix, diff_matrices
+ mmat = mass_matrix(out_grp)
return actx.freeze(
actx.from_numpy(
np.asarray(
- [actx.to_numpy(dmat.T).dot(actx.to_numpy(mmat.T))
- for dmat in reference_derivative_matrices(actx, in_grp)]
+ [dmat.T @ mmat.T for dmat in diff_matrices(out_grp)]
)
)
)
--
GitLab