From 8e0c041337a5149f1b16fd90c276deb0f48abd0f Mon Sep 17 00:00:00 2001 From: Matthew Smith Date: Wed, 31 May 2023 11:19:22 -0700 Subject: [PATCH] tag a few more axes --- meshmode/discretization/connection/direct.py | 12 +++++++----- 1 file changed, 7 insertions(+), 5 deletions(-) diff --git a/meshmode/discretization/connection/direct.py b/meshmode/discretization/connection/direct.py index 37acf0090..0d55a7c10 100644 --- a/meshmode/discretization/connection/direct.py +++ b/meshmode/discretization/connection/direct.py @@ -167,12 +167,14 @@ def _global_from_element_indices( np_full_from_element_indices[~np_from_el_present] = 0 from_el_present = actx.freeze( - actx.tag(NameHint("from_el_present"), - actx.from_numpy( - np_from_el_present.astype(np.int8)))) + actx.tag_axis(0, DiscretizationElementAxisTag(), + actx.tag(NameHint("from_el_present"), + actx.from_numpy( + np_from_el_present.astype(np.int8))))) full_from_element_indices = actx.freeze( - actx.tag(NameHint("from_el_indices"), - actx.from_numpy(np_full_from_element_indices))) + actx.tag_axis(0, DiscretizationElementAxisTag(), + actx.tag(NameHint("from_el_indices"), + actx.from_numpy(np_full_from_element_indices)))) self._global_from_element_indices_cache = ( from_el_present, full_from_element_indices)