From 05d6641b884cd6858a38c411cc6d7c1516732869 Mon Sep 17 00:00:00 2001
From: Kaushik Kulkarni <kaushikcfd@gmail.com>
Date: Wed, 25 Jan 2023 11:48:30 -0600
Subject: [PATCH] attach axes tags to some terms in dt_geometric_factors

---
 grudge/dt_utils.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/grudge/dt_utils.py b/grudge/dt_utils.py
index 64f22cd34..54a9343e8 100644
--- a/grudge/dt_utils.py
+++ b/grudge/dt_utils.py
@@ -338,7 +338,10 @@ def dt_geometric_factors(
             actx.tag(NameHint(f"dt_geometric_{dd.as_identifier()}"),
                 DOFArray(actx,
                     data=tuple(
-                        actx.einsum("e,ei->ei", 1/sae_i, cv_i,
+                        actx.einsum(
+                            "e,ei->ei",
+                            1/sae_i,
+                            actx.tag_axis(1, DiscretizationDOFAxisTag(), cv_i),
                             tagged=(FirstAxisIsElementsTag(),)) * dcoll.dim
                         for cv_i, sae_i in zip(cell_vols, surface_areas)))))