Skip to content

Commit ac40fde

Browse files
author
FEniCS GitHub Actions
committed
1 parent 86d39d0 commit ac40fde

1 file changed

Lines changed: 2 additions & 7 deletions

File tree

ffcx/main/_modules/ffcx/ir/representation.html

Lines changed: 2 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -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"># &#39;A&#39; 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

Comments
 (0)