@@ -488,9 +488,7 @@ <h1>Source code for ffcx.ir.representation</h1><div class="highlight"><pre>
488488 < span class ="c1 "> # Pre-compute the dimension number of dofs in each local element.</ span >
489489 < span class ="c1 "> # Used to compute the shape of the locally assembled tensor, as well the</ span >
490490 < span class ="c1 "> # coefficient offset</ span >
491- < span class ="n "> element_dimensions</ span > < span class ="o "> =</ span > < span class ="p "> {</ span >
492- < span class ="n "> element</ span > < span class ="p "> :</ span > < span class ="n "> element</ span > < span class ="o "> .</ span > < span class ="n "> dim</ span > < span class ="o "> +</ span > < span class ="n "> element</ span > < span class ="o "> .</ span > < span class ="n "> num_global_support_dofs</ span > < span class ="k "> for</ span > < span class ="n "> element</ span > < span class ="ow "> in</ span > < span class ="n "> unique_elements</ span >
493- < span class ="p "> }</ span >
491+ < span class ="n "> element_dimensions</ span > < span class ="o "> =</ span > < span class ="p "> {</ span > < span class ="n "> element</ span > < span class ="p "> :</ span > < span class ="n "> element</ span > < span class ="o "> .</ span > < span class ="n "> dim</ span > < span class ="k "> for</ span > < span class ="n "> element</ span > < span class ="ow "> in</ span > < span class ="n "> unique_elements</ span > < span class ="p "> }</ span >
494492
495493 < span class ="c1 "> # Create dimensions of primary indices, needed to reset the argument</ span >
496494 < span class ="c1 "> # 'A' given to tabulate_tensor() by the assembler.</ span >
@@ -705,10 +703,7 @@ <h1>Source code for ffcx.ir.representation</h1><div class="highlight"><pre>
705703
706704 < span class ="c1 "> # Prepare dimensions of all unique element in expression, including</ span >
707705 < span class ="c1 "> # elements for arguments, coefficients and coordinate mappings</ span >
708- < span class ="n "> element_dimensions</ span > < span class ="o "> =</ span > < span class ="p "> {</ span >
709- < span class ="n "> element</ span > < span class ="p "> :</ span > < span class ="n "> element</ span > < span class ="o "> .</ span > < span class ="n "> dim</ span > < span class ="o "> +</ span > < span class ="n "> element</ span > < span class ="o "> .</ span > < span class ="n "> num_global_support_dofs</ span >
710- < span class ="k "> for</ span > < span class ="n "> element</ span > < span class ="ow "> in</ span > < span class ="n "> analysis</ span > < span class ="o "> .</ span > < span class ="n "> unique_elements</ span >
711- < span class ="p "> }</ span >
706+ < span class ="n "> element_dimensions</ span > < span class ="o "> =</ span > < span class ="p "> {</ span > < span class ="n "> element</ span > < span class ="p "> :</ span > < span class ="n "> element</ span > < span class ="o "> .</ span > < span class ="n "> dim</ span > < span class ="k "> for</ span > < span class ="n "> element</ span > < span class ="ow "> in</ span > < span class ="n "> analysis</ span > < span class ="o "> .</ span > < span class ="n "> unique_elements</ span > < span class ="p "> }</ span >
712707
713708 < span class ="c1 "> # Extract dimensions for elements of arguments only</ span >
714709 < span class ="n "> arguments</ span > < span class ="o "> =</ span > < span class ="n "> ufl</ span > < span class ="o "> .</ span > < span class ="n "> algorithms</ span > < span class ="o "> .</ span > < span class ="n "> extract_arguments</ span > < span class ="p "> (</ span > < span class ="n "> expr</ span > < span class ="p "> )</ span >
0 commit comments