Skip to content

Commit adf1bff

Browse files
Matthew Smithmajosm
authored andcommitted
fix issues with handling of empty groups
1 parent 369a541 commit adf1bff

File tree

2 files changed

+7
-3
lines changed

2 files changed

+7
-3
lines changed

grudge/dt_utils.py

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -295,7 +295,9 @@ def dt_geometric_factors(
295295
actx.einsum(
296296
"fej->e",
297297
face_ae_i.reshape(
298-
vgrp.mesh_el_group.nfaces, vgrp.nelements, -1),
298+
vgrp.mesh_el_group.nfaces,
299+
vgrp.nelements,
300+
face_ae_i.shape[-1]),
299301
tagged=(FirstAxisIsElementsTag(),))
300302

301303
for vgrp, face_ae_i in zip(volm_discr.groups, face_areas)
@@ -314,7 +316,9 @@ def dt_geometric_factors(
314316
actx.einsum(
315317
"fej->e",
316318
face_ae_i.reshape(
317-
vgrp.mesh_el_group.nfaces, vgrp.nelements, -1
319+
vgrp.mesh_el_group.nfaces,
320+
vgrp.nelements,
321+
face_ae_i.shape[-1]
318322
) / afgrp.nunit_dofs,
319323
tagged=(FirstAxisIsElementsTag(),))
320324

grudge/op.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -928,7 +928,7 @@ def _apply_face_mass_operator(dcoll: DiscretizationCollection, dd_in, vec):
928928
surf_ae_i.reshape(
929929
vgrp.mesh_el_group.nfaces,
930930
vgrp.nelements,
931-
-1),
931+
surf_ae_i.shape[-1]),
932932
vec_i.reshape(
933933
vgrp.mesh_el_group.nfaces,
934934
vgrp.nelements,

0 commit comments

Comments
 (0)