Skip to content

Commit

Permalink
Update docs from 29758a5
Browse files Browse the repository at this point in the history
  • Loading branch information
olivedevteam committed Feb 4, 2025
1 parent b9bcaae commit 8ed55d0
Show file tree
Hide file tree
Showing 3 changed files with 82 additions and 1 deletion.
41 changes: 41 additions & 0 deletions _sources/how-to/configure-workflows/onnx-graph-surgeon.md.txt
Original file line number Diff line number Diff line change
Expand Up @@ -656,3 +656,44 @@ graph {
output: "zero_point"
}
```


### RMSNormToL2Norm

#### Description

Replace RMSNorm subgraph with L2Norm subgraph.

#### Example

Initial model graph:

```
RMSNorm pattern:
+-----------------------------------------------+
| |
| v
[Root] --> Pow --> ReduceMean --> Add --> Sqrt --> Div --> Mul
(y=2) (axis=-1) (B=E-6)
```

After applying:

```json
{
"type": "GraphSurgeries",
"surgeries": [
{
"surgeon": "RMSNormToL2Norm"
}
]
}
```


Transformed model graph:

```
[Root] --> LpNormalization --> Mul
(p=2, axis=-1)
```
40 changes: 40 additions & 0 deletions how-to/configure-workflows/onnx-graph-surgeon.html
Original file line number Diff line number Diff line change
Expand Up @@ -1056,6 +1056,41 @@ <h4>Example<a class="headerlink" href="#id27" title="Link to this heading">#</a>
</div>
</section>
</section>
<section id="rmsnormtol2norm">
<h3>RMSNormToL2Norm<a class="headerlink" href="#rmsnormtol2norm" title="Link to this heading">#</a></h3>
<section id="id28">
<h4>Description<a class="headerlink" href="#id28" title="Link to this heading">#</a></h4>
<p>Replace RMSNorm subgraph with L2Norm subgraph.</p>
</section>
<section id="id29">
<h4>Example<a class="headerlink" href="#id29" title="Link to this heading">#</a></h4>
<p>Initial model graph:</p>
<div class="highlight-default notranslate"><div class="highlight"><pre><span></span><span class="n">RMSNorm</span> <span class="n">pattern</span><span class="p">:</span>
<span class="o">+-----------------------------------------------+</span>
<span class="o">|</span> <span class="o">|</span>
<span class="o">|</span> <span class="n">v</span>
<span class="p">[</span><span class="n">Root</span><span class="p">]</span> <span class="o">--&gt;</span> <span class="n">Pow</span> <span class="o">--&gt;</span> <span class="n">ReduceMean</span> <span class="o">--&gt;</span> <span class="n">Add</span> <span class="o">--&gt;</span> <span class="n">Sqrt</span> <span class="o">--&gt;</span> <span class="n">Div</span> <span class="o">--&gt;</span> <span class="n">Mul</span>
<span class="p">(</span><span class="n">y</span><span class="o">=</span><span class="mi">2</span><span class="p">)</span> <span class="p">(</span><span class="n">axis</span><span class="o">=-</span><span class="mi">1</span><span class="p">)</span> <span class="p">(</span><span class="n">B</span><span class="o">=</span><span class="n">E</span><span class="o">-</span><span class="mi">6</span><span class="p">)</span>
</pre></div>
</div>
<p>After applying:</p>
<div class="highlight-json notranslate"><div class="highlight"><pre><span></span><span class="p">{</span>
<span class="w"> </span><span class="nt">&quot;type&quot;</span><span class="p">:</span><span class="w"> </span><span class="s2">&quot;GraphSurgeries&quot;</span><span class="p">,</span>
<span class="w"> </span><span class="nt">&quot;surgeries&quot;</span><span class="p">:</span><span class="w"> </span><span class="p">[</span>
<span class="w"> </span><span class="p">{</span>
<span class="w"> </span><span class="nt">&quot;surgeon&quot;</span><span class="p">:</span><span class="w"> </span><span class="s2">&quot;RMSNormToL2Norm&quot;</span>
<span class="w"> </span><span class="p">}</span>
<span class="w"> </span><span class="p">]</span>
<span class="p">}</span>
</pre></div>
</div>
<p>Transformed model graph:</p>
<div class="highlight-default notranslate"><div class="highlight"><pre><span></span><span class="p">[</span><span class="n">Root</span><span class="p">]</span> <span class="o">--&gt;</span> <span class="n">LpNormalization</span> <span class="o">--&gt;</span> <span class="n">Mul</span>
<span class="p">(</span><span class="n">p</span><span class="o">=</span><span class="mi">2</span><span class="p">,</span> <span class="n">axis</span><span class="o">=-</span><span class="mi">1</span><span class="p">)</span>
</pre></div>
</div>
</section>
</section>
</section>
</section>

Expand Down Expand Up @@ -1173,6 +1208,11 @@ <h4>Example<a class="headerlink" href="#id27" title="Link to this heading">#</a>
<li class="toc-h4 nav-item toc-entry"><a class="reference internal nav-link" href="#id27">Example</a></li>
</ul>
</li>
<li class="toc-h3 nav-item toc-entry"><a class="reference internal nav-link" href="#rmsnormtol2norm">RMSNormToL2Norm</a><ul class="nav section-nav flex-column">
<li class="toc-h4 nav-item toc-entry"><a class="reference internal nav-link" href="#id28">Description</a></li>
<li class="toc-h4 nav-item toc-entry"><a class="reference internal nav-link" href="#id29">Example</a></li>
</ul>
</li>
</ul>
</li>
</ul>
Expand Down
2 changes: 1 addition & 1 deletion searchindex.js

Large diffs are not rendered by default.

0 comments on commit 8ed55d0

Please sign in to comment.