File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change @@ -386,9 +386,7 @@ def _compute_integral_ir(
386386 # Pre-compute the dimension number of dofs in each local element.
387387 # Used to compute the shape of the locally assembled tensor, as well the
388388 # coefficient offset
389- element_dimensions = {
390- element : element .dim + element .num_global_support_dofs for element in unique_elements
391- }
389+ element_dimensions = {element : element .dim for element in unique_elements }
392390
393391 # Create dimensions of primary indices, needed to reset the argument
394392 # 'A' given to tabulate_tensor() by the assembler.
@@ -603,10 +601,7 @@ def _compute_expression_ir(
603601
604602 # Prepare dimensions of all unique element in expression, including
605603 # elements for arguments, coefficients and coordinate mappings
606- element_dimensions = {
607- element : element .dim + element .num_global_support_dofs
608- for element in analysis .unique_elements
609- }
604+ element_dimensions = {element : element .dim for element in analysis .unique_elements }
610605
611606 # Extract dimensions for elements of arguments only
612607 arguments = ufl .algorithms .extract_arguments (expr )
You can’t perform that action at this time.
0 commit comments