[GH-PAGES] Updated website

This commit is contained in:
Philippe Tillet
2021-04-21 16:45:37 -04:00
parent 4f471eaf18
commit 817a174c88
32 changed files with 216 additions and 99 deletions

View File

@@ -3,4 +3,4 @@
.. currentmodule:: triton .. currentmodule:: triton
.. autodata:: maximum .. autofunction:: maximum

View File

@@ -3,4 +3,4 @@
.. currentmodule:: triton .. currentmodule:: triton
.. autodata:: minimum .. autofunction:: minimum

View File

@@ -3,4 +3,4 @@
.. currentmodule:: triton .. currentmodule:: triton
.. autodata:: ravel .. autofunction:: ravel

View File

@@ -3,4 +3,4 @@
.. currentmodule:: triton .. currentmodule:: triton
.. autodata:: sigmoid .. autofunction:: sigmoid

View File

@@ -3,4 +3,4 @@
.. currentmodule:: triton .. currentmodule:: triton
.. autodata:: softmax .. autofunction:: softmax

View File

@@ -238,13 +238,13 @@
<td style="width: 33%; vertical-align: top;"><ul> <td style="width: 33%; vertical-align: top;"><ul>
<li><a href="language-reference/python-api/generated/triton.max.html#triton.max">max() (in module triton)</a> <li><a href="language-reference/python-api/generated/triton.max.html#triton.max">max() (in module triton)</a>
</li> </li>
<li><a href="language-reference/python-api/generated/triton.maximum.html#triton.maximum">maximum (in module triton)</a> <li><a href="language-reference/python-api/generated/triton.maximum.html#triton.maximum">maximum() (in module triton)</a>
</li> </li>
</ul></td> </ul></td>
<td style="width: 33%; vertical-align: top;"><ul> <td style="width: 33%; vertical-align: top;"><ul>
<li><a href="language-reference/python-api/generated/triton.min.html#triton.min">min() (in module triton)</a> <li><a href="language-reference/python-api/generated/triton.min.html#triton.min">min() (in module triton)</a>
</li> </li>
<li><a href="language-reference/python-api/generated/triton.minimum.html#triton.minimum">minimum (in module triton)</a> <li><a href="language-reference/python-api/generated/triton.minimum.html#triton.minimum">minimum() (in module triton)</a>
</li> </li>
<li><a href="language-reference/python-api/generated/triton.multiple_of.html#triton.multiple_of">multiple_of() (in module triton)</a> <li><a href="language-reference/python-api/generated/triton.multiple_of.html#triton.multiple_of">multiple_of() (in module triton)</a>
</li> </li>
@@ -270,7 +270,7 @@
<h2 id="R">R</h2> <h2 id="R">R</h2>
<table style="width: 100%" class="indextable genindextable"><tr> <table style="width: 100%" class="indextable genindextable"><tr>
<td style="width: 33%; vertical-align: top;"><ul> <td style="width: 33%; vertical-align: top;"><ul>
<li><a href="language-reference/python-api/generated/triton.ravel.html#triton.ravel">ravel (in module triton)</a> <li><a href="language-reference/python-api/generated/triton.ravel.html#triton.ravel">ravel() (in module triton)</a>
</li> </li>
</ul></td> </ul></td>
<td style="width: 33%; vertical-align: top;"><ul> <td style="width: 33%; vertical-align: top;"><ul>
@@ -282,9 +282,9 @@
<h2 id="S">S</h2> <h2 id="S">S</h2>
<table style="width: 100%" class="indextable genindextable"><tr> <table style="width: 100%" class="indextable genindextable"><tr>
<td style="width: 33%; vertical-align: top;"><ul> <td style="width: 33%; vertical-align: top;"><ul>
<li><a href="language-reference/python-api/generated/triton.sigmoid.html#triton.sigmoid">sigmoid (in module triton)</a> <li><a href="language-reference/python-api/generated/triton.sigmoid.html#triton.sigmoid">sigmoid() (in module triton)</a>
</li> </li>
<li><a href="language-reference/python-api/generated/triton.softmax.html#triton.softmax">softmax (in module triton)</a> <li><a href="language-reference/python-api/generated/triton.softmax.html#triton.softmax">softmax() (in module triton)</a>
</li> </li>
</ul></td> </ul></td>
<td style="width: 33%; vertical-align: top;"><ul> <td style="width: 33%; vertical-align: top;"><ul>

View File

@@ -192,13 +192,13 @@
<dl class="py function"> <dl class="py function">
<dt id="triton.arange"> <dt id="triton.arange">
<code class="sig-prename descclassname"><span class="pre">triton.</span></code><code class="sig-name descname"><span class="pre">arange</span></code><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">start</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">end</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">builder</span></span><span class="o"><span class="pre">=</span></span><span class="default_value"><span class="pre">None</span></span></em><span class="sig-paren">)</span><a class="headerlink" href="#triton.arange" title="Permalink to this definition"></a></dt> <code class="sig-prename descclassname"><span class="pre">triton.</span></code><code class="sig-name descname"><span class="pre">arange</span></code><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">start</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">end</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">builder</span></span><span class="o"><span class="pre">=</span></span><span class="default_value"><span class="pre">None</span></span></em><span class="sig-paren">)</span><a class="headerlink" href="#triton.arange" title="Permalink to this definition"></a></dt>
<dd><p>Returns contiguous values within the open interval [start, end).</p> <dd><p>Returns contiguous values within the open interval [<code class="code docutils literal notranslate"><span class="pre">start</span></code>, <code class="code docutils literal notranslate"><span class="pre">end</span></code>).</p>
<dl class="field-list simple"> <dl class="field-list simple">
<dt class="field-odd">Parameters</dt> <dt class="field-odd">Parameters</dt>
<dd class="field-odd"><ul class="simple"> <dd class="field-odd"><ul class="simple">
<li><p><strong>start</strong> (<em>int</em>) Start of the interval.</p></li> <li><p><strong>start</strong> (<em>int</em>) Start of the interval. Must be a power of two.</p></li>
<li><p><strong>stop</strong> (<em>int</em>) End of the interval.</p></li> <li><p><strong>stop</strong> (<em>int</em>) End of the interval. Must be a power of two &gt;= start.</p></li>
<li><p><strong>builder</strong> (<em>triton.ir.builder</em>) IR builder to generate code into, optional from within &#64;triton.jit functions</p></li> <li><p><strong>builder</strong> (<em>triton.ir.builder</em><em>, </em><em>optional from within JIT'ed functions</em>) IR builder to generate code into</p></li>
</ul> </ul>
</dd> </dd>
</dl> </dl>

View File

@@ -193,8 +193,19 @@
<h1>triton.atomic_cas<a class="headerlink" href="#triton-atomic-cas" title="Permalink to this headline"></a></h1> <h1>triton.atomic_cas<a class="headerlink" href="#triton-atomic-cas" title="Permalink to this headline"></a></h1>
<dl class="py function"> <dl class="py function">
<dt id="triton.atomic_cas"> <dt id="triton.atomic_cas">
<code class="sig-prename descclassname"><span class="pre">triton.</span></code><code class="sig-name descname"><span class="pre">atomic_cas</span></code><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">ptr</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">cmp</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">val</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">builder</span></span><span class="o"><span class="pre">=</span></span><span class="default_value"><span class="pre">None</span></span></em><span class="sig-paren">)</span><a class="headerlink" href="#triton.atomic_cas" title="Permalink to this definition"></a></dt> <code class="sig-prename descclassname"><span class="pre">triton.</span></code><code class="sig-name descname"><span class="pre">atomic_cas</span></code><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">pointer</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">cmp</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">val</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">builder</span></span><span class="o"><span class="pre">=</span></span><span class="default_value"><span class="pre">None</span></span></em><span class="sig-paren">)</span><a class="headerlink" href="#triton.atomic_cas" title="Permalink to this definition"></a></dt>
<dd></dd></dl> <dd><p>Performs an atomic “compare-and-swap” and the memory locations specified by <code class="code docutils literal notranslate"><span class="pre">pointer</span></code>.</p>
<dl class="field-list simple">
<dt class="field-odd">Parameters</dt>
<dd class="field-odd"><ul class="simple">
<li><p><strong>pointer</strong> (<em>Block of dtype=triton.PointerDType</em>) The memory locations to compare-and-swap.</p></li>
<li><p><strong>cmp</strong> (<em>Block of dtype=`pointer.dtype.element_ty`</em>) The values expected to be found in the atomic object</p></li>
<li><p><strong>val</strong> (<em>Block of dtype=`pointer.dtype.element_ty`</em>) The values to copy in case the expected value matches the contained value.</p></li>
<li><p><strong>builder</strong> (<em>triton.ir.builder</em><em>, </em><em>optional from within JIT'ed functions</em>) IR builder to generate code into</p></li>
</ul>
</dd>
</dl>
</dd></dl>
</div> </div>

View File

@@ -193,8 +193,18 @@
<h1>triton.atomic_xchg<a class="headerlink" href="#triton-atomic-xchg" title="Permalink to this headline"></a></h1> <h1>triton.atomic_xchg<a class="headerlink" href="#triton-atomic-xchg" title="Permalink to this headline"></a></h1>
<dl class="py function"> <dl class="py function">
<dt id="triton.atomic_xchg"> <dt id="triton.atomic_xchg">
<code class="sig-prename descclassname"><span class="pre">triton.</span></code><code class="sig-name descname"><span class="pre">atomic_xchg</span></code><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">ptr</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">val</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">builder</span></span><span class="o"><span class="pre">=</span></span><span class="default_value"><span class="pre">None</span></span></em><span class="sig-paren">)</span><a class="headerlink" href="#triton.atomic_xchg" title="Permalink to this definition"></a></dt> <code class="sig-prename descclassname"><span class="pre">triton.</span></code><code class="sig-name descname"><span class="pre">atomic_xchg</span></code><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">pointer</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">val</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">builder</span></span><span class="o"><span class="pre">=</span></span><span class="default_value"><span class="pre">None</span></span></em><span class="sig-paren">)</span><a class="headerlink" href="#triton.atomic_xchg" title="Permalink to this definition"></a></dt>
<dd></dd></dl> <dd><p>Swaps the <em>old</em> values stored at location <code class="code docutils literal notranslate"><span class="pre">pointer</span></code> with the new values given by <code class="code docutils literal notranslate"><span class="pre">val</span></code>. Returns the old values.</p>
<dl class="field-list simple">
<dt class="field-odd">Parameters</dt>
<dd class="field-odd"><ul class="simple">
<li><p><strong>pointer</strong> (<em>Block of dtype=triton.PointerDType</em>) The memory locations which contain the old values</p></li>
<li><p><strong>val</strong> (<em>Block of dtype=`pointer.dtype.element_ty`</em>) The new values to store</p></li>
<li><p><strong>builder</strong> (<em>triton.ir.builder</em><em>, </em><em>optional from within JIT'ed functions</em>) IR builder to generate code into</p></li>
</ul>
</dd>
</dl>
</dd></dl>
</div> </div>

View File

@@ -193,13 +193,13 @@
<dl class="py function"> <dl class="py function">
<dt id="triton.broadcast_to"> <dt id="triton.broadcast_to">
<code class="sig-prename descclassname"><span class="pre">triton.</span></code><code class="sig-name descname"><span class="pre">broadcast_to</span></code><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">input</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">shape</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">builder</span></span><span class="o"><span class="pre">=</span></span><span class="default_value"><span class="pre">None</span></span></em><span class="sig-paren">)</span><a class="headerlink" href="#triton.broadcast_to" title="Permalink to this definition"></a></dt> <code class="sig-prename descclassname"><span class="pre">triton.</span></code><code class="sig-name descname"><span class="pre">broadcast_to</span></code><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">input</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">shape</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">builder</span></span><span class="o"><span class="pre">=</span></span><span class="default_value"><span class="pre">None</span></span></em><span class="sig-paren">)</span><a class="headerlink" href="#triton.broadcast_to" title="Permalink to this definition"></a></dt>
<dd><p>Tries to broadcast a block to a new shape.</p> <dd><p>Tries to broadcast the given block to a new <code class="code docutils literal notranslate"><span class="pre">shape</span></code>.</p>
<dl class="field-list simple"> <dl class="field-list simple">
<dt class="field-odd">Parameters</dt> <dt class="field-odd">Parameters</dt>
<dd class="field-odd"><ul class="simple"> <dd class="field-odd"><ul class="simple">
<li><p><strong>input</strong> (<em>triton.value</em>) The input block.</p></li> <li><p><strong>input</strong> (<em>Block</em>) The input block.</p></li>
<li><p><strong>shape</strong> (<em>tuple of int</em>) The new shape.</p></li> <li><p><strong>shape</strong> (<em>Tuple</em><em>[</em><em>int</em><em>]</em>) The desired shape.</p></li>
<li><p><strong>builder</strong> (<em>triton.ir.builder</em>) IR builder to generate code into, optional from within &#64;triton.jit functions</p></li> <li><p><strong>builder</strong> (<em>triton.ir.builder</em><em>, </em><em>optional from within JIT'ed functions</em>) IR builder to generate code into</p></li>
</ul> </ul>
</dd> </dd>
</dl> </dl>

View File

@@ -191,14 +191,14 @@
<dl class="py function"> <dl class="py function">
<dt id="triton.dot"> <dt id="triton.dot">
<code class="sig-prename descclassname"><span class="pre">triton.</span></code><code class="sig-name descname"><span class="pre">dot</span></code><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">input</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">other</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">builder</span></span><span class="o"><span class="pre">=</span></span><span class="default_value"><span class="pre">None</span></span></em><span class="sig-paren">)</span><a class="headerlink" href="#triton.dot" title="Permalink to this definition"></a></dt> <code class="sig-prename descclassname"><span class="pre">triton.</span></code><code class="sig-name descname"><span class="pre">dot</span></code><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">input</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">other</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">builder</span></span><span class="o"><span class="pre">=</span></span><span class="default_value"><span class="pre">None</span></span></em><span class="sig-paren">)</span><a class="headerlink" href="#triton.dot" title="Permalink to this definition"></a></dt>
<dd><p>Returns the matrix product of two blocks. <dd><p>Returns the matrix product of two blocks.</p>
The two blocks must be two dimensionals and have compatible inner dimensions.</p> <p>The two blocks must be two dimensionals and have compatible inner dimensions.</p>
<dl class="field-list simple"> <dl class="field-list simple">
<dt class="field-odd">Parameters</dt> <dt class="field-odd">Parameters</dt>
<dd class="field-odd"><ul class="simple"> <dd class="field-odd"><ul class="simple">
<li><p><strong>input</strong> (2D block of scalar-type in {<cite>float16</cite>, <cite>float32</cite>}) The first block to be multiplied.</p></li> <li><p><strong>input</strong> (2D block of scalar-type in {<code class="code docutils literal notranslate"><span class="pre">float16</span></code>, <code class="code docutils literal notranslate"><span class="pre">float32</span></code>}) The first block to be multiplied.</p></li>
<li><p><strong>other</strong> (2D block of scalar-type in {<cite>float16</cite>, <cite>float32</cite>}) The second block to be multiplied.</p></li> <li><p><strong>other</strong> (2D block of scalar-type in {<code class="code docutils literal notranslate"><span class="pre">float16</span></code>, <code class="code docutils literal notranslate"><span class="pre">float32</span></code>}) The second block to be multiplied.</p></li>
<li><p><strong>builder</strong> (<em>triton.ir.builder</em>) IR builder to generate code into, optional from within &#64;triton.jit functions</p></li> <li><p><strong>builder</strong> (<em>triton.ir.builder</em><em>, </em><em>optional from within JIT'ed functions</em>) IR builder to generate code into</p></li>
</ul> </ul>
</dd> </dd>
</dl> </dl>

View File

@@ -194,7 +194,16 @@
<dl class="py function"> <dl class="py function">
<dt id="triton.exp"> <dt id="triton.exp">
<code class="sig-prename descclassname"><span class="pre">triton.</span></code><code class="sig-name descname"><span class="pre">exp</span></code><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">x</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">builder</span></span><span class="o"><span class="pre">=</span></span><span class="default_value"><span class="pre">None</span></span></em><span class="sig-paren">)</span><a class="headerlink" href="#triton.exp" title="Permalink to this definition"></a></dt> <code class="sig-prename descclassname"><span class="pre">triton.</span></code><code class="sig-name descname"><span class="pre">exp</span></code><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">x</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">builder</span></span><span class="o"><span class="pre">=</span></span><span class="default_value"><span class="pre">None</span></span></em><span class="sig-paren">)</span><a class="headerlink" href="#triton.exp" title="Permalink to this definition"></a></dt>
<dd></dd></dl> <dd><p>Computes the element-wise exponential of <code class="code docutils literal notranslate"><span class="pre">x</span></code></p>
<dl class="field-list simple">
<dt class="field-odd">Parameters</dt>
<dd class="field-odd"><ul class="simple">
<li><p><strong>x</strong> (<em>Block</em>) the input values</p></li>
<li><p><strong>builder</strong> (<em>triton.ir.builder</em><em>, </em><em>optional from within JIT'ed functions</em>) IR builder to generate code into</p></li>
</ul>
</dd>
</dl>
</dd></dl>
</div> </div>

View File

@@ -194,14 +194,16 @@
<dl class="py function"> <dl class="py function">
<dt id="triton.load"> <dt id="triton.load">
<code class="sig-prename descclassname"><span class="pre">triton.</span></code><code class="sig-name descname"><span class="pre">load</span></code><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">pointer</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">mask</span></span><span class="o"><span class="pre">=</span></span><span class="default_value"><span class="pre">None</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">other</span></span><span class="o"><span class="pre">=</span></span><span class="default_value"><span class="pre">None</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">builder</span></span><span class="o"><span class="pre">=</span></span><span class="default_value"><span class="pre">None</span></span></em><span class="sig-paren">)</span><a class="headerlink" href="#triton.load" title="Permalink to this definition"></a></dt> <code class="sig-prename descclassname"><span class="pre">triton.</span></code><code class="sig-name descname"><span class="pre">load</span></code><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">pointer</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">mask</span></span><span class="o"><span class="pre">=</span></span><span class="default_value"><span class="pre">None</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">other</span></span><span class="o"><span class="pre">=</span></span><span class="default_value"><span class="pre">None</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">builder</span></span><span class="o"><span class="pre">=</span></span><span class="default_value"><span class="pre">None</span></span></em><span class="sig-paren">)</span><a class="headerlink" href="#triton.load" title="Permalink to this definition"></a></dt>
<dd><p>Return a block of data whose values are, elementwise, loaded from memory at location defined by <cite>pointer</cite>.</p> <dd><p>Return a block of data whose values are, elementwise, loaded from memory at location defined by <code class="code docutils literal notranslate"><span class="pre">pointer</span></code>.</p>
<p><code class="code docutils literal notranslate"><span class="pre">mask</span></code> and <code class="code docutils literal notranslate"><span class="pre">other</span></code> are implicitly broadcast to <code class="code docutils literal notranslate"><span class="pre">pointer.shape</span></code>.</p>
<p><code class="code docutils literal notranslate"><span class="pre">other</span></code> is implicitly typecast to <code class="code docutils literal notranslate"><span class="pre">pointer.dtype.element_ty</span></code>.</p>
<dl class="field-list simple"> <dl class="field-list simple">
<dt class="field-odd">Parameters</dt> <dt class="field-odd">Parameters</dt>
<dd class="field-odd"><ul class="simple"> <dd class="field-odd"><ul class="simple">
<li><p><strong>pointer</strong> (<em>Block of triton.pointer</em>) Pointer to the data to be loaded.</p></li> <li><p><strong>pointer</strong> (<em>Block of dtype=triton.PointerDType</em>) Pointers to the data to be loaded.</p></li>
<li><p><strong>mask</strong> (<em>Block of triton.bool</em><em>, </em><em>optional</em>) if mask[idx] is false, do not load the data at <cite>pointer[idx]</cite>.</p></li> <li><p><strong>mask</strong> (<em>Block of triton.int1</em><em>, </em><em>optional</em>) if mask[idx] is false, do not load the data at address <code class="code docutils literal notranslate"><span class="pre">pointer[idx]</span></code>.</p></li>
<li><p><strong>other</strong> (<em>Block of triton.value</em><em>, </em><em>optional</em>) if mask[idx] is false, return other[idx] instead of pointer[idx]`</p></li> <li><p><strong>other</strong> (<em>Block</em><em>, </em><em>optional</em>) if mask[idx] is false, return other[idx]</p></li>
<li><p><strong>builder</strong> (<em>triton.ir.builder</em>) IR builder to generate code into, optional from within &#64;triton.jit functions</p></li> <li><p><strong>builder</strong> (<em>triton.ir.builder</em><em>, </em><em>optional from within JIT'ed functions</em>) IR builder to generate code into</p></li>
</ul> </ul>
</dd> </dd>
</dl> </dl>

View File

@@ -194,7 +194,16 @@
<dl class="py function"> <dl class="py function">
<dt id="triton.log"> <dt id="triton.log">
<code class="sig-prename descclassname"><span class="pre">triton.</span></code><code class="sig-name descname"><span class="pre">log</span></code><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">x</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">builder</span></span><span class="o"><span class="pre">=</span></span><span class="default_value"><span class="pre">None</span></span></em><span class="sig-paren">)</span><a class="headerlink" href="#triton.log" title="Permalink to this definition"></a></dt> <code class="sig-prename descclassname"><span class="pre">triton.</span></code><code class="sig-name descname"><span class="pre">log</span></code><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">x</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">builder</span></span><span class="o"><span class="pre">=</span></span><span class="default_value"><span class="pre">None</span></span></em><span class="sig-paren">)</span><a class="headerlink" href="#triton.log" title="Permalink to this definition"></a></dt>
<dd></dd></dl> <dd><p>Computes the element-wise natural logarithm of <code class="code docutils literal notranslate"><span class="pre">x</span></code></p>
<dl class="field-list simple">
<dt class="field-odd">Parameters</dt>
<dd class="field-odd"><ul class="simple">
<li><p><strong>x</strong> (<em>Block</em>) the input values</p></li>
<li><p><strong>builder</strong> (<em>triton.ir.builder</em><em>, </em><em>optional from within JIT'ed functions</em>) IR builder to generate code into</p></li>
</ul>
</dd>
</dl>
</dd></dl>
</div> </div>

View File

@@ -193,7 +193,17 @@
<dl class="py function"> <dl class="py function">
<dt id="triton.max"> <dt id="triton.max">
<code class="sig-prename descclassname"><span class="pre">triton.</span></code><code class="sig-name descname"><span class="pre">max</span></code><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">input</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">axis</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">builder</span></span><span class="o"><span class="pre">=</span></span><span class="default_value"><span class="pre">None</span></span></em><span class="sig-paren">)</span><a class="headerlink" href="#triton.max" title="Permalink to this definition"></a></dt> <code class="sig-prename descclassname"><span class="pre">triton.</span></code><code class="sig-name descname"><span class="pre">max</span></code><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">input</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">axis</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">builder</span></span><span class="o"><span class="pre">=</span></span><span class="default_value"><span class="pre">None</span></span></em><span class="sig-paren">)</span><a class="headerlink" href="#triton.max" title="Permalink to this definition"></a></dt>
<dd></dd></dl> <dd><p>Returns the maximum value of all elements in the <code class="code docutils literal notranslate"><span class="pre">input</span></code> block along the provided <code class="code docutils literal notranslate"><span class="pre">axis</span></code></p>
<dl class="field-list simple">
<dt class="field-odd">Parameters</dt>
<dd class="field-odd"><ul class="simple">
<li><p><strong>input</strong> the input values</p></li>
<li><p><strong>axis</strong> the dimension along which the reduction should be done</p></li>
<li><p><strong>builder</strong> (<em>triton.ir.builder</em><em>, </em><em>optional from within JIT'ed functions</em>) IR builder to generate code into</p></li>
</ul>
</dd>
</dl>
</dd></dl>
</div> </div>

View File

@@ -189,10 +189,19 @@
<div class="section" id="triton-maximum"> <div class="section" id="triton-maximum">
<h1>triton.maximum<a class="headerlink" href="#triton-maximum" title="Permalink to this headline"></a></h1> <h1>triton.maximum<a class="headerlink" href="#triton-maximum" title="Permalink to this headline"></a></h1>
<dl class="py data"> <dl class="py function">
<dt id="triton.maximum"> <dt id="triton.maximum">
<code class="sig-prename descclassname"><span class="pre">triton.</span></code><code class="sig-name descname"><span class="pre">maximum</span></code><em class="property"> <span class="pre">=</span> <span class="pre">&lt;triton.code_gen.JITFunction</span> <span class="pre">object&gt;</span></em><a class="headerlink" href="#triton.maximum" title="Permalink to this definition"></a></dt> <code class="sig-prename descclassname"><span class="pre">triton.</span></code><code class="sig-name descname"><span class="pre">maximum</span></code><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">x</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">y</span></span></em><span class="sig-paren">)</span><a class="headerlink" href="#triton.maximum" title="Permalink to this definition"></a></dt>
<dd></dd></dl> <dd><p>Computes the element-wise maximum of <code class="code docutils literal notranslate"><span class="pre">x</span></code> and <code class="code docutils literal notranslate"><span class="pre">y</span></code>.</p>
<dl class="field-list simple">
<dt class="field-odd">Parameters</dt>
<dd class="field-odd"><ul class="simple">
<li><p><strong>input</strong> (<em>Block</em>) the first input block</p></li>
<li><p><strong>other</strong> (<em>Block</em>) the second input block</p></li>
</ul>
</dd>
</dl>
</dd></dl>
</div> </div>

View File

@@ -193,7 +193,17 @@
<dl class="py function"> <dl class="py function">
<dt id="triton.min"> <dt id="triton.min">
<code class="sig-prename descclassname"><span class="pre">triton.</span></code><code class="sig-name descname"><span class="pre">min</span></code><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">input</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">axis</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">builder</span></span><span class="o"><span class="pre">=</span></span><span class="default_value"><span class="pre">None</span></span></em><span class="sig-paren">)</span><a class="headerlink" href="#triton.min" title="Permalink to this definition"></a></dt> <code class="sig-prename descclassname"><span class="pre">triton.</span></code><code class="sig-name descname"><span class="pre">min</span></code><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">input</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">axis</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">builder</span></span><span class="o"><span class="pre">=</span></span><span class="default_value"><span class="pre">None</span></span></em><span class="sig-paren">)</span><a class="headerlink" href="#triton.min" title="Permalink to this definition"></a></dt>
<dd></dd></dl> <dd><p>Returns the minimum value of all elements in the <code class="code docutils literal notranslate"><span class="pre">input</span></code> block along the provided <code class="code docutils literal notranslate"><span class="pre">axis</span></code></p>
<dl class="field-list simple">
<dt class="field-odd">Parameters</dt>
<dd class="field-odd"><ul class="simple">
<li><p><strong>input</strong> the input values</p></li>
<li><p><strong>axis</strong> the dimension along which the reduction should be done</p></li>
<li><p><strong>builder</strong> (<em>triton.ir.builder</em><em>, </em><em>optional from within JIT'ed functions</em>) IR builder to generate code into</p></li>
</ul>
</dd>
</dl>
</dd></dl>
</div> </div>

View File

@@ -189,10 +189,19 @@
<div class="section" id="triton-minimum"> <div class="section" id="triton-minimum">
<h1>triton.minimum<a class="headerlink" href="#triton-minimum" title="Permalink to this headline"></a></h1> <h1>triton.minimum<a class="headerlink" href="#triton-minimum" title="Permalink to this headline"></a></h1>
<dl class="py data"> <dl class="py function">
<dt id="triton.minimum"> <dt id="triton.minimum">
<code class="sig-prename descclassname"><span class="pre">triton.</span></code><code class="sig-name descname"><span class="pre">minimum</span></code><em class="property"> <span class="pre">=</span> <span class="pre">&lt;triton.code_gen.JITFunction</span> <span class="pre">object&gt;</span></em><a class="headerlink" href="#triton.minimum" title="Permalink to this definition"></a></dt> <code class="sig-prename descclassname"><span class="pre">triton.</span></code><code class="sig-name descname"><span class="pre">minimum</span></code><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">x</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">y</span></span></em><span class="sig-paren">)</span><a class="headerlink" href="#triton.minimum" title="Permalink to this definition"></a></dt>
<dd></dd></dl> <dd><p>Computes the element-wise minimum of <code class="code docutils literal notranslate"><span class="pre">x</span></code> and <code class="code docutils literal notranslate"><span class="pre">y</span></code>.</p>
<dl class="field-list simple">
<dt class="field-odd">Parameters</dt>
<dd class="field-odd"><ul class="simple">
<li><p><strong>input</strong> (<em>Block</em>) the first input block</p></li>
<li><p><strong>other</strong> (<em>Block</em>) the second input block</p></li>
</ul>
</dd>
</dl>
</dd></dl>
</div> </div>

View File

@@ -190,8 +190,11 @@
<h1>triton.multiple_of<a class="headerlink" href="#triton-multiple-of" title="Permalink to this headline"></a></h1> <h1>triton.multiple_of<a class="headerlink" href="#triton-multiple-of" title="Permalink to this headline"></a></h1>
<dl class="py function"> <dl class="py function">
<dt id="triton.multiple_of"> <dt id="triton.multiple_of">
<code class="sig-prename descclassname"><span class="pre">triton.</span></code><code class="sig-name descname"><span class="pre">multiple_of</span></code><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">x</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">value</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">builder</span></span><span class="o"><span class="pre">=</span></span><span class="default_value"><span class="pre">None</span></span></em><span class="sig-paren">)</span><a class="headerlink" href="#triton.multiple_of" title="Permalink to this definition"></a></dt> <code class="sig-prename descclassname"><span class="pre">triton.</span></code><code class="sig-name descname"><span class="pre">multiple_of</span></code><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">input</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">value</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">builder</span></span><span class="o"><span class="pre">=</span></span><span class="default_value"><span class="pre">None</span></span></em><span class="sig-paren">)</span><a class="headerlink" href="#triton.multiple_of" title="Permalink to this definition"></a></dt>
<dd></dd></dl> <dd><p>Let the compiler knows that the values in <code class="code docutils literal notranslate"><span class="pre">input</span></code> are all multiples of <code class="code docutils literal notranslate"><span class="pre">value</span></code>.
:param builder: IR builder to generate code into
:type builder: triton.ir.builder, optional from within JITed functions</p>
</dd></dl>
</div> </div>

View File

@@ -192,12 +192,12 @@
<dl class="py function"> <dl class="py function">
<dt id="triton.num_programs"> <dt id="triton.num_programs">
<code class="sig-prename descclassname"><span class="pre">triton.</span></code><code class="sig-name descname"><span class="pre">num_programs</span></code><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">axis</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">builder</span></span><span class="o"><span class="pre">=</span></span><span class="default_value"><span class="pre">None</span></span></em><span class="sig-paren">)</span><a class="headerlink" href="#triton.num_programs" title="Permalink to this definition"></a></dt> <code class="sig-prename descclassname"><span class="pre">triton.</span></code><code class="sig-name descname"><span class="pre">num_programs</span></code><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">axis</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">builder</span></span><span class="o"><span class="pre">=</span></span><span class="default_value"><span class="pre">None</span></span></em><span class="sig-paren">)</span><a class="headerlink" href="#triton.num_programs" title="Permalink to this definition"></a></dt>
<dd><p>Returns the number of program instances launched along the given <cite>axis</cite>.</p> <dd><p>Returns the number of program instances launched along the given <code class="code docutils literal notranslate"><span class="pre">axis</span></code>.</p>
<dl class="field-list simple"> <dl class="field-list simple">
<dt class="field-odd">Parameters</dt> <dt class="field-odd">Parameters</dt>
<dd class="field-odd"><ul class="simple"> <dd class="field-odd"><ul class="simple">
<li><p><strong>axis</strong> (<em>int</em>) The axis of the 3D launch grid. Has to be either 0, 1 or 2.</p></li> <li><p><strong>axis</strong> (<em>int</em>) The axis of the 3D launch grid. Has to be either 0, 1 or 2.</p></li>
<li><p><strong>builder</strong> (<em>triton.ir.builder</em>) IR builder to generate code into, optional from within &#64;triton.jit functions</p></li> <li><p><strong>builder</strong> (<em>triton.ir.builder</em><em>, </em><em>optional from within JIT'ed functions</em>) IR builder to generate code into</p></li>
</ul> </ul>
</dd> </dd>
</dl> </dl>

View File

@@ -192,13 +192,12 @@
<dl class="py function"> <dl class="py function">
<dt id="triton.program_id"> <dt id="triton.program_id">
<code class="sig-prename descclassname"><span class="pre">triton.</span></code><code class="sig-name descname"><span class="pre">program_id</span></code><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">axis</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">builder</span></span><span class="o"><span class="pre">=</span></span><span class="default_value"><span class="pre">None</span></span></em><span class="sig-paren">)</span><a class="headerlink" href="#triton.program_id" title="Permalink to this definition"></a></dt> <code class="sig-prename descclassname"><span class="pre">triton.</span></code><code class="sig-name descname"><span class="pre">program_id</span></code><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">axis</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">builder</span></span><span class="o"><span class="pre">=</span></span><span class="default_value"><span class="pre">None</span></span></em><span class="sig-paren">)</span><a class="headerlink" href="#triton.program_id" title="Permalink to this definition"></a></dt>
<dd><p>Returns the id of the current program instance along the given <cite>axis</cite>. <dd><p>Returns the id of the current program instance along the given <code class="code docutils literal notranslate"><span class="pre">axis</span></code>.</p>
Triton uses an SPMD model in which different &#64;triton.jit functions run in parallel with different <a href="#id1"><span class="problematic" id="id2">`</span></a>program_id`s.</p>
<dl class="field-list simple"> <dl class="field-list simple">
<dt class="field-odd">Parameters</dt> <dt class="field-odd">Parameters</dt>
<dd class="field-odd"><ul class="simple"> <dd class="field-odd"><ul class="simple">
<li><p><strong>axis</strong> (<em>int</em>) The axis of the 3D launch grid. Has to be either 0, 1 or 2.</p></li> <li><p><strong>axis</strong> (<em>int</em>) The axis of the 3D launch grid. Has to be either 0, 1 or 2.</p></li>
<li><p><strong>builder</strong> (<em>triton.ir.builder</em>) IR builder to generate code into, optional from within &#64;triton.jit functions</p></li> <li><p><strong>builder</strong> (<em>triton.ir.builder</em><em>, </em><em>optional from within JIT'ed functions</em>) IR builder to generate code into</p></li>
</ul> </ul>
</dd> </dd>
</dl> </dl>

View File

@@ -190,10 +190,16 @@
<div class="section" id="triton-ravel"> <div class="section" id="triton-ravel">
<h1>triton.ravel<a class="headerlink" href="#triton-ravel" title="Permalink to this headline"></a></h1> <h1>triton.ravel<a class="headerlink" href="#triton-ravel" title="Permalink to this headline"></a></h1>
<dl class="py data"> <dl class="py function">
<dt id="triton.ravel"> <dt id="triton.ravel">
<code class="sig-prename descclassname"><span class="pre">triton.</span></code><code class="sig-name descname"><span class="pre">ravel</span></code><em class="property"> <span class="pre">=</span> <span class="pre">&lt;triton.code_gen.JITFunction</span> <span class="pre">object&gt;</span></em><a class="headerlink" href="#triton.ravel" title="Permalink to this definition"></a></dt> <code class="sig-prename descclassname"><span class="pre">triton.</span></code><code class="sig-name descname"><span class="pre">ravel</span></code><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">x</span></span></em><span class="sig-paren">)</span><a class="headerlink" href="#triton.ravel" title="Permalink to this definition"></a></dt>
<dd></dd></dl> <dd><p>Returns a contiguous flattened view of <code class="code docutils literal notranslate"><span class="pre">x</span></code></p>
<dl class="field-list simple">
<dt class="field-odd">Parameters</dt>
<dd class="field-odd"><p><strong>x</strong> (<em>Block</em>) the input block</p>
</dd>
</dl>
</dd></dl>
</div> </div>

View File

@@ -193,9 +193,16 @@
<dl class="py function"> <dl class="py function">
<dt id="triton.reshape"> <dt id="triton.reshape">
<code class="sig-prename descclassname"><span class="pre">triton.</span></code><code class="sig-name descname"><span class="pre">reshape</span></code><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">input</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">shape</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">builder</span></span><span class="o"><span class="pre">=</span></span><span class="default_value"><span class="pre">None</span></span></em><span class="sig-paren">)</span><a class="headerlink" href="#triton.reshape" title="Permalink to this definition"></a></dt> <code class="sig-prename descclassname"><span class="pre">triton.</span></code><code class="sig-name descname"><span class="pre">reshape</span></code><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">input</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">shape</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">builder</span></span><span class="o"><span class="pre">=</span></span><span class="default_value"><span class="pre">None</span></span></em><span class="sig-paren">)</span><a class="headerlink" href="#triton.reshape" title="Permalink to this definition"></a></dt>
<dd><p>Reshapes a block to a new shape. <dd><p>Tries to reshape the given block to a new shape.</p>
:param builder: IR builder to generate code into, optional from within &#64;triton.jit functions <dl class="field-list simple">
:type builder: triton.ir.builder</p> <dt class="field-odd">Parameters</dt>
<dd class="field-odd"><ul class="simple">
<li><p><strong>input</strong> The input block.</p></li>
<li><p><strong>shape</strong> (<em>Tuple</em><em>[</em><em>int</em><em>]</em>) The desired shape.</p></li>
<li><p><strong>builder</strong> (<em>triton.ir.builder</em><em>, </em><em>optional from within JIT'ed functions</em>) IR builder to generate code into</p></li>
</ul>
</dd>
</dl>
</dd></dl> </dd></dl>
</div> </div>

View File

@@ -191,10 +191,16 @@
<div class="section" id="triton-sigmoid"> <div class="section" id="triton-sigmoid">
<h1>triton.sigmoid<a class="headerlink" href="#triton-sigmoid" title="Permalink to this headline"></a></h1> <h1>triton.sigmoid<a class="headerlink" href="#triton-sigmoid" title="Permalink to this headline"></a></h1>
<dl class="py data"> <dl class="py function">
<dt id="triton.sigmoid"> <dt id="triton.sigmoid">
<code class="sig-prename descclassname"><span class="pre">triton.</span></code><code class="sig-name descname"><span class="pre">sigmoid</span></code><em class="property"> <span class="pre">=</span> <span class="pre">&lt;triton.code_gen.JITFunction</span> <span class="pre">object&gt;</span></em><a class="headerlink" href="#triton.sigmoid" title="Permalink to this definition"></a></dt> <code class="sig-prename descclassname"><span class="pre">triton.</span></code><code class="sig-name descname"><span class="pre">sigmoid</span></code><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">x</span></span></em><span class="sig-paren">)</span><a class="headerlink" href="#triton.sigmoid" title="Permalink to this definition"></a></dt>
<dd></dd></dl> <dd><p>Computes the element-wise sigmoid of <code class="code docutils literal notranslate"><span class="pre">x</span></code>.</p>
<dl class="field-list simple">
<dt class="field-odd">Parameters</dt>
<dd class="field-odd"><p><strong>x</strong> (<em>Block</em>) the input block</p>
</dd>
</dl>
</dd></dl>
</div> </div>

View File

@@ -191,10 +191,16 @@
<div class="section" id="triton-softmax"> <div class="section" id="triton-softmax">
<h1>triton.softmax<a class="headerlink" href="#triton-softmax" title="Permalink to this headline"></a></h1> <h1>triton.softmax<a class="headerlink" href="#triton-softmax" title="Permalink to this headline"></a></h1>
<dl class="py data"> <dl class="py function">
<dt id="triton.softmax"> <dt id="triton.softmax">
<code class="sig-prename descclassname"><span class="pre">triton.</span></code><code class="sig-name descname"><span class="pre">softmax</span></code><em class="property"> <span class="pre">=</span> <span class="pre">&lt;triton.code_gen.JITFunction</span> <span class="pre">object&gt;</span></em><a class="headerlink" href="#triton.softmax" title="Permalink to this definition"></a></dt> <code class="sig-prename descclassname"><span class="pre">triton.</span></code><code class="sig-name descname"><span class="pre">softmax</span></code><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">x</span></span></em><span class="sig-paren">)</span><a class="headerlink" href="#triton.softmax" title="Permalink to this definition"></a></dt>
<dd></dd></dl> <dd><p>Computes the element-wise softmax of <code class="code docutils literal notranslate"><span class="pre">x</span></code>.</p>
<dl class="field-list simple">
<dt class="field-odd">Parameters</dt>
<dd class="field-odd"><p><strong>x</strong> (<em>Block</em>) the input block</p>
</dd>
</dl>
</dd></dl>
</div> </div>

View File

@@ -194,14 +194,15 @@
<dl class="py function"> <dl class="py function">
<dt id="triton.store"> <dt id="triton.store">
<code class="sig-prename descclassname"><span class="pre">triton.</span></code><code class="sig-name descname"><span class="pre">store</span></code><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">pointer</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">value</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">mask</span></span><span class="o"><span class="pre">=</span></span><span class="default_value"><span class="pre">None</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">builder</span></span><span class="o"><span class="pre">=</span></span><span class="default_value"><span class="pre">None</span></span></em><span class="sig-paren">)</span><a class="headerlink" href="#triton.store" title="Permalink to this definition"></a></dt> <code class="sig-prename descclassname"><span class="pre">triton.</span></code><code class="sig-name descname"><span class="pre">store</span></code><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">pointer</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">value</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">mask</span></span><span class="o"><span class="pre">=</span></span><span class="default_value"><span class="pre">None</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">builder</span></span><span class="o"><span class="pre">=</span></span><span class="default_value"><span class="pre">None</span></span></em><span class="sig-paren">)</span><a class="headerlink" href="#triton.store" title="Permalink to this definition"></a></dt>
<dd><p>Stores <cite>value</cite> block of elements in memory, element-wise, at the memory locations specified by <cite>pointer</cite>.</p> <dd><p>Stores <code class="code docutils literal notranslate"><span class="pre">value</span></code> block of elements in memory, element-wise, at the memory locations specified by <code class="code docutils literal notranslate"><span class="pre">pointer</span></code>.</p>
<p><code class="code docutils literal notranslate"><span class="pre">value</span></code> is implicitly broadcast to <code class="code docutils literal notranslate"><span class="pre">pointer.shape</span></code> and typecast to <code class="code docutils literal notranslate"><span class="pre">pointer.dtype.element_ty</span></code>.</p>
<dl class="field-list simple"> <dl class="field-list simple">
<dt class="field-odd">Parameters</dt> <dt class="field-odd">Parameters</dt>
<dd class="field-odd"><ul class="simple"> <dd class="field-odd"><ul class="simple">
<li><p><strong>pointer</strong> (<em>Block of triton.pointer</em>) The memory locations where the elements of <cite>value</cite> are stored.</p></li> <li><p><strong>pointer</strong> (<em>Block of dtype=triton.PointerDType</em>) The memory locations where the elements of <code class="code docutils literal notranslate"><span class="pre">value</span></code> are stored.</p></li>
<li><p><strong>value</strong> (<em>Block of triton.value</em>) The block of elements to be stored.</p></li> <li><p><strong>value</strong> (<em>Block</em>) The block of elements to be stored.</p></li>
<li><p><strong>mask</strong> (<em>Block of triton.bool</em><em>, </em><em>optional</em>) If mask[idx] is false, do not store <cite>value[idx]</cite> at <cite>pointer[idx]</cite>.</p></li> <li><p><strong>mask</strong> (<em>Block of triton.int1</em><em>, </em><em>optional</em>) If mask[idx] is false, do not store <code class="code docutils literal notranslate"><span class="pre">value[idx]</span></code> at <code class="code docutils literal notranslate"><span class="pre">pointer[idx]</span></code>.</p></li>
<li><p><strong>builder</strong> (<em>triton.ir.builder</em>) IR builder to generate code into, optional from within &#64;triton.jit functions</p></li> <li><p><strong>builder</strong> (<em>triton.ir.builder</em><em>, </em><em>optional from within JIT'ed functions</em>) IR builder to generate code into</p></li>
</ul> </ul>
</dd> </dd>
</dl> </dl>

View File

@@ -193,7 +193,17 @@
<dl class="py function"> <dl class="py function">
<dt id="triton.sum"> <dt id="triton.sum">
<code class="sig-prename descclassname"><span class="pre">triton.</span></code><code class="sig-name descname"><span class="pre">sum</span></code><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">input</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">axis</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">builder</span></span><span class="o"><span class="pre">=</span></span><span class="default_value"><span class="pre">None</span></span></em><span class="sig-paren">)</span><a class="headerlink" href="#triton.sum" title="Permalink to this definition"></a></dt> <code class="sig-prename descclassname"><span class="pre">triton.</span></code><code class="sig-name descname"><span class="pre">sum</span></code><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">input</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">axis</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">builder</span></span><span class="o"><span class="pre">=</span></span><span class="default_value"><span class="pre">None</span></span></em><span class="sig-paren">)</span><a class="headerlink" href="#triton.sum" title="Permalink to this definition"></a></dt>
<dd></dd></dl> <dd><p>Returns the sum of all elements in the <code class="code docutils literal notranslate"><span class="pre">input</span></code> block along the provided <code class="code docutils literal notranslate"><span class="pre">axis</span></code></p>
<dl class="field-list simple">
<dt class="field-odd">Parameters</dt>
<dd class="field-odd"><ul class="simple">
<li><p><strong>input</strong> the input values</p></li>
<li><p><strong>axis</strong> the dimension along which the reduction should be done</p></li>
<li><p><strong>builder</strong> (<em>triton.ir.builder</em><em>, </em><em>optional from within JIT'ed functions</em>) IR builder to generate code into</p></li>
</ul>
</dd>
</dl>
</dd></dl>
</div> </div>

View File

@@ -191,18 +191,18 @@
<dl class="py function"> <dl class="py function">
<dt id="triton.where"> <dt id="triton.where">
<code class="sig-prename descclassname"><span class="pre">triton.</span></code><code class="sig-name descname"><span class="pre">where</span></code><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">condition</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">x</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">y</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">builder</span></span><span class="o"><span class="pre">=</span></span><span class="default_value"><span class="pre">None</span></span></em><span class="sig-paren">)</span><a class="headerlink" href="#triton.where" title="Permalink to this definition"></a></dt> <code class="sig-prename descclassname"><span class="pre">triton.</span></code><code class="sig-name descname"><span class="pre">where</span></code><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">condition</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">x</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">y</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">builder</span></span><span class="o"><span class="pre">=</span></span><span class="default_value"><span class="pre">None</span></span></em><span class="sig-paren">)</span><a class="headerlink" href="#triton.where" title="Permalink to this definition"></a></dt>
<dd><p>Returns a block of elements from either <cite>x</cite> or <cite>y</cite>, depending on <cite>condition</cite>. <dd><p>Returns a block of elements from either <code class="code docutils literal notranslate"><span class="pre">x</span></code> or <code class="code docutils literal notranslate"><span class="pre">y</span></code>, depending on <code class="code docutils literal notranslate"><span class="pre">condition</span></code>.</p>
Note that <cite>x</cite> and <cite>y</cite> are always evaluated regardless of the value of <cite>condition</cite>. <p>Note that <code class="code docutils literal notranslate"><span class="pre">x</span></code> and <code class="code docutils literal notranslate"><span class="pre">y</span></code> are always evaluated regardless of the value of <code class="code docutils literal notranslate"><span class="pre">condition</span></code>.</p>
If you want to avoid unintented memory operations, use the <cite>mask</cite> arguments in <cite>triton.load</cite> and <cite>triton.store</cite> instead. <p>If you want to avoid unintented memory operations, use the <code class="code docutils literal notranslate"><span class="pre">mask</span></code> arguments in <cite>triton.load</cite> and <cite>triton.store</cite> instead.</p>
The shape of <cite>x</cite> and <cite>y</cite> are both broadcast to the shape of <cite>condition</cite>. <p>The shape of <code class="code docutils literal notranslate"><span class="pre">x</span></code> and <code class="code docutils literal notranslate"><span class="pre">y</span></code> are both broadcast to the shape of <code class="code docutils literal notranslate"><span class="pre">condition</span></code>.
<cite>x</cite> and <cite>y</cite> must have the data type.</p> <code class="code docutils literal notranslate"><span class="pre">x</span></code> and <code class="code docutils literal notranslate"><span class="pre">y</span></code> must have the data type.</p>
<dl class="field-list simple"> <dl class="field-list simple">
<dt class="field-odd">Parameters</dt> <dt class="field-odd">Parameters</dt>
<dd class="field-odd"><ul class="simple"> <dd class="field-odd"><ul class="simple">
<li><p><strong>condition</strong> (<em>Block of triton.bool</em>) When True (nonzero), yield x, otherwise yield y.</p></li> <li><p><strong>condition</strong> (<em>Block of triton.bool</em>) When True (nonzero), yield x, otherwise yield y.</p></li>
<li><p><strong>x</strong> values selected at indices where condition is True.</p></li> <li><p><strong>x</strong> values selected at indices where condition is True.</p></li>
<li><p><strong>y</strong> values selected at indices where condition is False.</p></li> <li><p><strong>y</strong> values selected at indices where condition is False.</p></li>
<li><p><strong>builder</strong> (<em>triton.ir.builder</em>) IR builder to generate code into, optional from within &#64;triton.jit functions</p></li> <li><p><strong>builder</strong> (<em>triton.ir.builder</em><em>, </em><em>optional from within JIT'ed functions</em>) IR builder to generate code into</p></li>
</ul> </ul>
</dd> </dd>
</dl> </dl>

View File

@@ -192,13 +192,13 @@
<dl class="py function"> <dl class="py function">
<dt id="triton.zeros"> <dt id="triton.zeros">
<code class="sig-prename descclassname"><span class="pre">triton.</span></code><code class="sig-name descname"><span class="pre">zeros</span></code><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">shape</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">dtype</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">builder</span></span><span class="o"><span class="pre">=</span></span><span class="default_value"><span class="pre">None</span></span></em><span class="sig-paren">)</span><a class="headerlink" href="#triton.zeros" title="Permalink to this definition"></a></dt> <code class="sig-prename descclassname"><span class="pre">triton.</span></code><code class="sig-name descname"><span class="pre">zeros</span></code><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">shape</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">dtype</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">builder</span></span><span class="o"><span class="pre">=</span></span><span class="default_value"><span class="pre">None</span></span></em><span class="sig-paren">)</span><a class="headerlink" href="#triton.zeros" title="Permalink to this definition"></a></dt>
<dd><p>Returns a block filled with the scalar value 0 and the given shape.</p> <dd><p>Returns a block filled with the scalar value 0 for the given <code class="code docutils literal notranslate"><span class="pre">shape</span></code> and <code class="code docutils literal notranslate"><span class="pre">dtype</span></code>.</p>
<dl class="field-list simple"> <dl class="field-list simple">
<dt class="field-odd">Parameters</dt> <dt class="field-odd">Parameters</dt>
<dd class="field-odd"><ul class="simple"> <dd class="field-odd"><ul class="simple">
<li><p><strong>shape</strong> (<em>tuple of ints</em>) Shape of the new array, e.g., (8, 16) or (8, )</p></li> <li><p><strong>shape</strong> (<em>tuple of ints</em>) Shape of the new array, e.g., (8, 16) or (8, )</p></li>
<li><p><strong>dtype</strong> (<em>triton.ir.dtype</em>) Data-type of the new array, e.g., triton.float16</p></li> <li><p><strong>dtype</strong> (<em>DType</em>) Data-type of the new array, e.g., <code class="code docutils literal notranslate"><span class="pre">triton.float16</span></code></p></li>
<li><p><strong>builder</strong> (<em>triton.ir.builder</em>) IR builder to generate code into, optional from within &#64;triton.jit functions</p></li> <li><p><strong>builder</strong> (<em>triton.ir.builder</em><em>, </em><em>optional from within JIT'ed functions</em>) IR builder to generate code into</p></li>
</ul> </ul>
</dd> </dd>
</dl> </dl>

View File

@@ -235,10 +235,10 @@
</colgroup> </colgroup>
<tbody> <tbody>
<tr class="row-odd"><td><p><a class="reference internal" href="generated/triton.program_id.html#triton.program_id" title="triton.program_id"><code class="xref py py-obj docutils literal notranslate"><span class="pre">program_id</span></code></a></p></td> <tr class="row-odd"><td><p><a class="reference internal" href="generated/triton.program_id.html#triton.program_id" title="triton.program_id"><code class="xref py py-obj docutils literal notranslate"><span class="pre">program_id</span></code></a></p></td>
<td><p>Returns the id of the current program instance along the given <cite>axis</cite>. Triton uses an SPMD model in which different &#64;triton.jit functions run in parallel with different <a href="#id1"><span class="problematic" id="id2">`</span></a>program_id`s.</p></td> <td><p>Returns the id of the current program instance along the given <code class="code docutils literal notranslate"><span class="pre">axis</span></code>.</p></td>
</tr> </tr>
<tr class="row-even"><td><p><a class="reference internal" href="generated/triton.num_programs.html#triton.num_programs" title="triton.num_programs"><code class="xref py py-obj docutils literal notranslate"><span class="pre">num_programs</span></code></a></p></td> <tr class="row-even"><td><p><a class="reference internal" href="generated/triton.num_programs.html#triton.num_programs" title="triton.num_programs"><code class="xref py py-obj docutils literal notranslate"><span class="pre">num_programs</span></code></a></p></td>
<td><p>Returns the number of program instances launched along the given <cite>axis</cite>.</p></td> <td><p>Returns the number of program instances launched along the given <code class="code docutils literal notranslate"><span class="pre">axis</span></code>.</p></td>
</tr> </tr>
</tbody> </tbody>
</table> </table>
@@ -252,10 +252,10 @@
</colgroup> </colgroup>
<tbody> <tbody>
<tr class="row-odd"><td><p><a class="reference internal" href="generated/triton.arange.html#triton.arange" title="triton.arange"><code class="xref py py-obj docutils literal notranslate"><span class="pre">arange</span></code></a></p></td> <tr class="row-odd"><td><p><a class="reference internal" href="generated/triton.arange.html#triton.arange" title="triton.arange"><code class="xref py py-obj docutils literal notranslate"><span class="pre">arange</span></code></a></p></td>
<td><p>Returns contiguous values within the open interval [start, end).</p></td> <td><p>Returns contiguous values within the open interval [<code class="code docutils literal notranslate"><span class="pre">start</span></code>, <code class="code docutils literal notranslate"><span class="pre">end</span></code>).</p></td>
</tr> </tr>
<tr class="row-even"><td><p><a class="reference internal" href="generated/triton.zeros.html#triton.zeros" title="triton.zeros"><code class="xref py py-obj docutils literal notranslate"><span class="pre">zeros</span></code></a></p></td> <tr class="row-even"><td><p><a class="reference internal" href="generated/triton.zeros.html#triton.zeros" title="triton.zeros"><code class="xref py py-obj docutils literal notranslate"><span class="pre">zeros</span></code></a></p></td>
<td><p>Returns a block filled with the scalar value 0 and the given shape.</p></td> <td><p>Returns a block filled with the scalar value 0 for the given <code class="code docutils literal notranslate"><span class="pre">shape</span></code> and <code class="code docutils literal notranslate"><span class="pre">dtype</span></code>.</p></td>
</tr> </tr>
</tbody> </tbody>
</table> </table>
@@ -269,13 +269,13 @@
</colgroup> </colgroup>
<tbody> <tbody>
<tr class="row-odd"><td><p><a class="reference internal" href="generated/triton.broadcast_to.html#triton.broadcast_to" title="triton.broadcast_to"><code class="xref py py-obj docutils literal notranslate"><span class="pre">broadcast_to</span></code></a></p></td> <tr class="row-odd"><td><p><a class="reference internal" href="generated/triton.broadcast_to.html#triton.broadcast_to" title="triton.broadcast_to"><code class="xref py py-obj docutils literal notranslate"><span class="pre">broadcast_to</span></code></a></p></td>
<td><p>Tries to broadcast a block to a new shape.</p></td> <td><p>Tries to broadcast the given block to a new <code class="code docutils literal notranslate"><span class="pre">shape</span></code>.</p></td>
</tr> </tr>
<tr class="row-even"><td><p><a class="reference internal" href="generated/triton.reshape.html#triton.reshape" title="triton.reshape"><code class="xref py py-obj docutils literal notranslate"><span class="pre">reshape</span></code></a></p></td> <tr class="row-even"><td><p><a class="reference internal" href="generated/triton.reshape.html#triton.reshape" title="triton.reshape"><code class="xref py py-obj docutils literal notranslate"><span class="pre">reshape</span></code></a></p></td>
<td><p>Reshapes a block to a new shape.</p></td> <td><p>Tries to reshape the given block to a new shape.</p></td>
</tr> </tr>
<tr class="row-odd"><td><p><a class="reference internal" href="generated/triton.ravel.html#triton.ravel" title="triton.ravel"><code class="xref py py-obj docutils literal notranslate"><span class="pre">ravel</span></code></a></p></td> <tr class="row-odd"><td><p><a class="reference internal" href="generated/triton.ravel.html#triton.ravel" title="triton.ravel"><code class="xref py py-obj docutils literal notranslate"><span class="pre">ravel</span></code></a></p></td>
<td><p></p></td> <td><p>Returns a contiguous flattened view of <code class="code docutils literal notranslate"><span class="pre">x</span></code></p></td>
</tr> </tr>
</tbody> </tbody>
</table> </table>
@@ -303,16 +303,16 @@
</colgroup> </colgroup>
<tbody> <tbody>
<tr class="row-odd"><td><p><a class="reference internal" href="generated/triton.load.html#triton.load" title="triton.load"><code class="xref py py-obj docutils literal notranslate"><span class="pre">load</span></code></a></p></td> <tr class="row-odd"><td><p><a class="reference internal" href="generated/triton.load.html#triton.load" title="triton.load"><code class="xref py py-obj docutils literal notranslate"><span class="pre">load</span></code></a></p></td>
<td><p>Return a block of data whose values are, elementwise, loaded from memory at location defined by <cite>pointer</cite>.</p></td> <td><p>Return a block of data whose values are, elementwise, loaded from memory at location defined by <code class="code docutils literal notranslate"><span class="pre">pointer</span></code>.</p></td>
</tr> </tr>
<tr class="row-even"><td><p><a class="reference internal" href="generated/triton.store.html#triton.store" title="triton.store"><code class="xref py py-obj docutils literal notranslate"><span class="pre">store</span></code></a></p></td> <tr class="row-even"><td><p><a class="reference internal" href="generated/triton.store.html#triton.store" title="triton.store"><code class="xref py py-obj docutils literal notranslate"><span class="pre">store</span></code></a></p></td>
<td><p>Stores <cite>value</cite> block of elements in memory, element-wise, at the memory locations specified by <cite>pointer</cite>.</p></td> <td><p>Stores <code class="code docutils literal notranslate"><span class="pre">value</span></code> block of elements in memory, element-wise, at the memory locations specified by <code class="code docutils literal notranslate"><span class="pre">pointer</span></code>.</p></td>
</tr> </tr>
<tr class="row-odd"><td><p><a class="reference internal" href="generated/triton.atomic_cas.html#triton.atomic_cas" title="triton.atomic_cas"><code class="xref py py-obj docutils literal notranslate"><span class="pre">atomic_cas</span></code></a></p></td> <tr class="row-odd"><td><p><a class="reference internal" href="generated/triton.atomic_cas.html#triton.atomic_cas" title="triton.atomic_cas"><code class="xref py py-obj docutils literal notranslate"><span class="pre">atomic_cas</span></code></a></p></td>
<td><p></p></td> <td><p>Performs an atomic “compare-and-swap” and the memory locations specified by <code class="code docutils literal notranslate"><span class="pre">pointer</span></code>.</p></td>
</tr> </tr>
<tr class="row-even"><td><p><a class="reference internal" href="generated/triton.atomic_xchg.html#triton.atomic_xchg" title="triton.atomic_xchg"><code class="xref py py-obj docutils literal notranslate"><span class="pre">atomic_xchg</span></code></a></p></td> <tr class="row-even"><td><p><a class="reference internal" href="generated/triton.atomic_xchg.html#triton.atomic_xchg" title="triton.atomic_xchg"><code class="xref py py-obj docutils literal notranslate"><span class="pre">atomic_xchg</span></code></a></p></td>
<td><p></p></td> <td><p>Swaps the <em>old</em> values stored at location <code class="code docutils literal notranslate"><span class="pre">pointer</span></code> with the new values given by <code class="code docutils literal notranslate"><span class="pre">val</span></code>.</p></td>
</tr> </tr>
</tbody> </tbody>
</table> </table>
@@ -326,7 +326,7 @@
</colgroup> </colgroup>
<tbody> <tbody>
<tr class="row-odd"><td><p><a class="reference internal" href="generated/triton.where.html#triton.where" title="triton.where"><code class="xref py py-obj docutils literal notranslate"><span class="pre">where</span></code></a></p></td> <tr class="row-odd"><td><p><a class="reference internal" href="generated/triton.where.html#triton.where" title="triton.where"><code class="xref py py-obj docutils literal notranslate"><span class="pre">where</span></code></a></p></td>
<td><p>Returns a block of elements from either <cite>x</cite> or <cite>y</cite>, depending on <cite>condition</cite>.</p></td> <td><p>Returns a block of elements from either <code class="code docutils literal notranslate"><span class="pre">x</span></code> or <code class="code docutils literal notranslate"><span class="pre">y</span></code>, depending on <code class="code docutils literal notranslate"><span class="pre">condition</span></code>.</p></td>
</tr> </tr>
</tbody> </tbody>
</table> </table>
@@ -340,16 +340,16 @@
</colgroup> </colgroup>
<tbody> <tbody>
<tr class="row-odd"><td><p><a class="reference internal" href="generated/triton.exp.html#triton.exp" title="triton.exp"><code class="xref py py-obj docutils literal notranslate"><span class="pre">exp</span></code></a></p></td> <tr class="row-odd"><td><p><a class="reference internal" href="generated/triton.exp.html#triton.exp" title="triton.exp"><code class="xref py py-obj docutils literal notranslate"><span class="pre">exp</span></code></a></p></td>
<td><p></p></td> <td><p>Computes the element-wise exponential of <code class="code docutils literal notranslate"><span class="pre">x</span></code></p></td>
</tr> </tr>
<tr class="row-even"><td><p><a class="reference internal" href="generated/triton.log.html#triton.log" title="triton.log"><code class="xref py py-obj docutils literal notranslate"><span class="pre">log</span></code></a></p></td> <tr class="row-even"><td><p><a class="reference internal" href="generated/triton.log.html#triton.log" title="triton.log"><code class="xref py py-obj docutils literal notranslate"><span class="pre">log</span></code></a></p></td>
<td><p></p></td> <td><p>Computes the element-wise natural logarithm of <code class="code docutils literal notranslate"><span class="pre">x</span></code></p></td>
</tr> </tr>
<tr class="row-odd"><td><p><a class="reference internal" href="generated/triton.sigmoid.html#triton.sigmoid" title="triton.sigmoid"><code class="xref py py-obj docutils literal notranslate"><span class="pre">sigmoid</span></code></a></p></td> <tr class="row-odd"><td><p><a class="reference internal" href="generated/triton.sigmoid.html#triton.sigmoid" title="triton.sigmoid"><code class="xref py py-obj docutils literal notranslate"><span class="pre">sigmoid</span></code></a></p></td>
<td><p></p></td> <td><p>Computes the element-wise sigmoid of <code class="code docutils literal notranslate"><span class="pre">x</span></code>.</p></td>
</tr> </tr>
<tr class="row-even"><td><p><a class="reference internal" href="generated/triton.softmax.html#triton.softmax" title="triton.softmax"><code class="xref py py-obj docutils literal notranslate"><span class="pre">softmax</span></code></a></p></td> <tr class="row-even"><td><p><a class="reference internal" href="generated/triton.softmax.html#triton.softmax" title="triton.softmax"><code class="xref py py-obj docutils literal notranslate"><span class="pre">softmax</span></code></a></p></td>
<td><p></p></td> <td><p>Computes the element-wise softmax of <code class="code docutils literal notranslate"><span class="pre">x</span></code>.</p></td>
</tr> </tr>
</tbody> </tbody>
</table> </table>
@@ -363,13 +363,13 @@
</colgroup> </colgroup>
<tbody> <tbody>
<tr class="row-odd"><td><p><a class="reference internal" href="generated/triton.max.html#triton.max" title="triton.max"><code class="xref py py-obj docutils literal notranslate"><span class="pre">max</span></code></a></p></td> <tr class="row-odd"><td><p><a class="reference internal" href="generated/triton.max.html#triton.max" title="triton.max"><code class="xref py py-obj docutils literal notranslate"><span class="pre">max</span></code></a></p></td>
<td><p></p></td> <td><p>Returns the maximum value of all elements in the <code class="code docutils literal notranslate"><span class="pre">input</span></code> block along the provided <code class="code docutils literal notranslate"><span class="pre">axis</span></code></p></td>
</tr> </tr>
<tr class="row-even"><td><p><a class="reference internal" href="generated/triton.min.html#triton.min" title="triton.min"><code class="xref py py-obj docutils literal notranslate"><span class="pre">min</span></code></a></p></td> <tr class="row-even"><td><p><a class="reference internal" href="generated/triton.min.html#triton.min" title="triton.min"><code class="xref py py-obj docutils literal notranslate"><span class="pre">min</span></code></a></p></td>
<td><p></p></td> <td><p>Returns the minimum value of all elements in the <code class="code docutils literal notranslate"><span class="pre">input</span></code> block along the provided <code class="code docutils literal notranslate"><span class="pre">axis</span></code></p></td>
</tr> </tr>
<tr class="row-odd"><td><p><a class="reference internal" href="generated/triton.sum.html#triton.sum" title="triton.sum"><code class="xref py py-obj docutils literal notranslate"><span class="pre">sum</span></code></a></p></td> <tr class="row-odd"><td><p><a class="reference internal" href="generated/triton.sum.html#triton.sum" title="triton.sum"><code class="xref py py-obj docutils literal notranslate"><span class="pre">sum</span></code></a></p></td>
<td><p></p></td> <td><p>Returns the sum of all elements in the <code class="code docutils literal notranslate"><span class="pre">input</span></code> block along the provided <code class="code docutils literal notranslate"><span class="pre">axis</span></code></p></td>
</tr> </tr>
</tbody> </tbody>
</table> </table>
@@ -383,10 +383,10 @@
</colgroup> </colgroup>
<tbody> <tbody>
<tr class="row-odd"><td><p><a class="reference internal" href="generated/triton.minimum.html#triton.minimum" title="triton.minimum"><code class="xref py py-obj docutils literal notranslate"><span class="pre">minimum</span></code></a></p></td> <tr class="row-odd"><td><p><a class="reference internal" href="generated/triton.minimum.html#triton.minimum" title="triton.minimum"><code class="xref py py-obj docutils literal notranslate"><span class="pre">minimum</span></code></a></p></td>
<td><p></p></td> <td><p>Computes the element-wise minimum of <code class="code docutils literal notranslate"><span class="pre">x</span></code> and <code class="code docutils literal notranslate"><span class="pre">y</span></code>.</p></td>
</tr> </tr>
<tr class="row-even"><td><p><a class="reference internal" href="generated/triton.maximum.html#triton.maximum" title="triton.maximum"><code class="xref py py-obj docutils literal notranslate"><span class="pre">maximum</span></code></a></p></td> <tr class="row-even"><td><p><a class="reference internal" href="generated/triton.maximum.html#triton.maximum" title="triton.maximum"><code class="xref py py-obj docutils literal notranslate"><span class="pre">maximum</span></code></a></p></td>
<td><p></p></td> <td><p>Computes the element-wise maximum of <code class="code docutils literal notranslate"><span class="pre">x</span></code> and <code class="code docutils literal notranslate"><span class="pre">y</span></code>.</p></td>
</tr> </tr>
</tbody> </tbody>
</table> </table>
@@ -400,7 +400,7 @@
</colgroup> </colgroup>
<tbody> <tbody>
<tr class="row-odd"><td><p><a class="reference internal" href="generated/triton.multiple_of.html#triton.multiple_of" title="triton.multiple_of"><code class="xref py py-obj docutils literal notranslate"><span class="pre">multiple_of</span></code></a></p></td> <tr class="row-odd"><td><p><a class="reference internal" href="generated/triton.multiple_of.html#triton.multiple_of" title="triton.multiple_of"><code class="xref py py-obj docutils literal notranslate"><span class="pre">multiple_of</span></code></a></p></td>
<td><p></p></td> <td><p>Let the compiler knows that the values in <code class="code docutils literal notranslate"><span class="pre">input</span></code> are all multiples of <code class="code docutils literal notranslate"><span class="pre">value</span></code>.</p></td>
</tr> </tr>
</tbody> </tbody>
</table> </table>

Binary file not shown.

File diff suppressed because one or more lines are too long