[GH-PAGES] Updated website
This commit is contained in:
Philippe Tillet
@@ -76,7 +76,7 @@
|
|||||||
},
|
},
|
||||||
"outputs": [],
|
"outputs": [],
|
||||||
"source": [
|
"source": [
|
||||||
"def next_power_of_2(n):\n n -= 1\n n |= n >> 1\n n |= n >> 2\n n |= n >> 4\n n |= n >> 8\n n |= n >> 16\n n += 1\n return n\n\n\ndef softmax(x):\n M, N = x.shape\n # The block size is the smallest power of two greater than the number of columns in `x`\n BLOCK = next_power_of_2(N)\n # Another trick we can use is to ask the compiler to parallelize each\n # row-normalization more aggressively -- i.e., with more warps -- vectors\n # that are longer\n # You will see in the next tutorial how to auto-tune this value in a more natural\n # way so you don't have to come up with manual heuristics yourself\n num_warps = 4\n if BLOCK >= 2048: num_warps = 8\n if BLOCK >= 4096: num_warps = 16\n # Allocate output\n y = torch.empty_like(x)\n # Enqueue kernel. The launch grid is simple: we have one kernel instance per row of the input matrix\n _softmax[(M, )](y, x, x.stride(0), y.stride(0), M, N, BLOCK=BLOCK)\n return y"
|
"def next_power_of_2(n):\n n -= 1\n n |= n >> 1\n n |= n >> 2\n n |= n >> 4\n n |= n >> 8\n n |= n >> 16\n n += 1\n return n\n\n\ndef softmax(x):\n M, N = x.shape\n # The block size is the smallest power of two greater than the number of columns in `x`\n BLOCK = next_power_of_2(N)\n # Another trick we can use is to ask the compiler to parallelize each\n # row-normalization more aggressively -- i.e., with more warps -- vectors\n # that are longer\n # You will see in the next tutorial how to auto-tune this value in a more natural\n # way so you don't have to come up with manual heuristics yourself\n num_warps = 4\n if BLOCK >= 2048: num_warps = 8\n if BLOCK >= 4096: num_warps = 16\n # Allocate output\n y = torch.empty_like(x)\n # Enqueue kernel. The launch grid is simple: we have one kernel instance per row of the input matrix\n _softmax[(M, )](y, x, x.stride(0), y.stride(0), M, N, num_warps=num_warps, BLOCK=BLOCK)\n return y"
|
||||||
]
|
]
|
||||||
},
|
},
|
||||||
{
|
{
|
||||||
|
|||||||
Binary file not shown.
Binary file not shown.
@@ -100,7 +100,7 @@ def softmax(x):
|
|||||||
# Allocate output
|
# Allocate output
|
||||||
y = torch.empty_like(x)
|
y = torch.empty_like(x)
|
||||||
# Enqueue kernel. The launch grid is simple: we have one kernel instance per row of the input matrix
|
# Enqueue kernel. The launch grid is simple: we have one kernel instance per row of the input matrix
|
||||||
_softmax[(M, )](y, x, x.stride(0), y.stride(0), M, N, BLOCK=BLOCK)
|
_softmax[(M, )](y, x, x.stride(0), y.stride(0), M, N, num_warps=num_warps, BLOCK=BLOCK)
|
||||||
return y
|
return y
|
||||||
|
|
||||||
|
|
||||||
|
|||||||
Binary file not shown.
|
Before Width: | Height: | Size: 27 KiB After Width: | Height: | Size: 26 KiB |
Binary file not shown.
|
Before Width: | Height: | Size: 17 KiB After Width: | Height: | Size: 17 KiB |
Binary file not shown.
|
Before Width: | Height: | Size: 36 KiB After Width: | Height: | Size: 34 KiB |
Binary file not shown.
|
Before Width: | Height: | Size: 23 KiB After Width: | Height: | Size: 22 KiB |
Binary file not shown.
|
Before Width: | Height: | Size: 36 KiB After Width: | Height: | Size: 36 KiB |
Binary file not shown.
|
Before Width: | Height: | Size: 24 KiB After Width: | Height: | Size: 24 KiB |
@@ -212,7 +212,7 @@ We can now run the decorated function above. Pass `show_plots=True` to see the p
|
|||||||
|
|
||||||
.. rst-class:: sphx-glr-timing
|
.. rst-class:: sphx-glr-timing
|
||||||
|
|
||||||
**Total running time of the script:** ( 0 minutes 5.812 seconds)
|
**Total running time of the script:** ( 0 minutes 7.044 seconds)
|
||||||
|
|
||||||
|
|
||||||
.. _sphx_glr_download_getting-started_tutorials_01-vector-add.py:
|
.. _sphx_glr_download_getting-started_tutorials_01-vector-add.py:
|
||||||
|
|||||||
@@ -151,7 +151,7 @@ We can create a helper function that enqueues the kernel and its (meta-)argument
|
|||||||
# Allocate output
|
# Allocate output
|
||||||
y = torch.empty_like(x)
|
y = torch.empty_like(x)
|
||||||
# Enqueue kernel. The launch grid is simple: we have one kernel instance per row of the input matrix
|
# Enqueue kernel. The launch grid is simple: we have one kernel instance per row of the input matrix
|
||||||
_softmax[(M, )](y, x, x.stride(0), y.stride(0), M, N, BLOCK=BLOCK)
|
_softmax[(M, )](y, x, x.stride(0), y.stride(0), M, N, num_warps=num_warps, BLOCK=BLOCK)
|
||||||
return y
|
return y
|
||||||
|
|
||||||
|
|
||||||
@@ -264,7 +264,7 @@ In the above plot, we can see that:
|
|||||||
|
|
||||||
.. rst-class:: sphx-glr-timing
|
.. rst-class:: sphx-glr-timing
|
||||||
|
|
||||||
**Total running time of the script:** ( 0 minutes 20.767 seconds)
|
**Total running time of the script:** ( 0 minutes 20.176 seconds)
|
||||||
|
|
||||||
|
|
||||||
.. _sphx_glr_download_getting-started_tutorials_02-fused-softmax.py:
|
.. _sphx_glr_download_getting-started_tutorials_02-fused-softmax.py:
|
||||||
|
|||||||
@@ -287,32 +287,32 @@ We can test our custom matrix multiplication operation against a native torch im
|
|||||||
|
|
||||||
.. code-block:: none
|
.. code-block:: none
|
||||||
|
|
||||||
tensor([[-0.0000e+00, 2.9438e+01, -1.3113e-06, ..., 9.7266e+00,
|
tensor([[-5.9605e-08, 5.1094e+01, -1.8477e-05, ..., 2.6547e+01,
|
||||||
-3.4237e-04, -0.0000e+00],
|
-7.2598e-05, -4.2510e-04],
|
||||||
[-1.7615e-01, -0.0000e+00, 6.1914e+00, ..., 3.7562e+01,
|
[-2.7100e-01, -3.0220e-05, 5.9414e+00, ..., 2.8340e+00,
|
||||||
-0.0000e+00, -0.0000e+00],
|
-1.8644e-04, 1.3094e+01],
|
||||||
[ 9.9531e+00, 1.9078e+01, -0.0000e+00, ..., 3.6934e+00,
|
[-1.5332e-01, 4.8125e+00, 8.4277e-01, ..., 3.6387e+00,
|
||||||
1.6578e+01, 2.1031e+01],
|
4.3375e+01, 1.6865e+00],
|
||||||
...,
|
...,
|
||||||
[ 2.6547e+01, -1.1802e-05, 7.7852e+00, ..., 5.2156e+01,
|
[-0.0000e+00, 2.9453e+01, -4.7684e-07, ..., 6.2617e+00,
|
||||||
3.5469e+01, 1.5602e+01],
|
4.1133e+00, -0.0000e+00],
|
||||||
[-0.0000e+00, -0.0000e+00, 1.6531e+01, ..., 2.1211e+00,
|
[ 1.6562e+01, -8.1539e-04, 1.3836e+01, ..., 1.9844e+00,
|
||||||
1.7412e+00, 1.1422e+01],
|
-1.1238e-02, 8.4375e+00],
|
||||||
[-2.6550e-02, -1.1325e-05, 3.0344e+01, ..., -9.1248e-03,
|
[-1.0876e-01, -2.7295e-01, 3.2156e+01, ..., -1.6907e-02,
|
||||||
-1.5199e-05, 3.8164e+00]], device='cuda:0', dtype=torch.float16)
|
-0.0000e+00, -0.0000e+00]], device='cuda:0', dtype=torch.float16)
|
||||||
tensor([[-0.0000e+00, 2.9438e+01, -1.3113e-06, ..., 9.7266e+00,
|
tensor([[-5.9605e-08, 5.1094e+01, -1.8537e-05, ..., 2.6547e+01,
|
||||||
-3.4261e-04, -0.0000e+00],
|
-7.2658e-05, -4.2605e-04],
|
||||||
[-1.7615e-01, -0.0000e+00, 6.1914e+00, ..., 3.7562e+01,
|
[-2.7100e-01, -3.0220e-05, 5.9414e+00, ..., 2.8340e+00,
|
||||||
-0.0000e+00, -0.0000e+00],
|
-1.8632e-04, 1.3094e+01],
|
||||||
[ 9.9531e+00, 1.9078e+01, -0.0000e+00, ..., 3.6934e+00,
|
[-1.5332e-01, 4.8125e+00, 8.4277e-01, ..., 3.6387e+00,
|
||||||
1.6578e+01, 2.1031e+01],
|
4.3375e+01, 1.6875e+00],
|
||||||
...,
|
...,
|
||||||
[ 2.6547e+01, -1.1802e-05, 7.7852e+00, ..., 5.2156e+01,
|
[-0.0000e+00, 2.9453e+01, -4.7684e-07, ..., 6.2617e+00,
|
||||||
3.5469e+01, 1.5602e+01],
|
4.1133e+00, -0.0000e+00],
|
||||||
[-0.0000e+00, -0.0000e+00, 1.6531e+01, ..., 2.1211e+00,
|
[ 1.6562e+01, -8.1778e-04, 1.3836e+01, ..., 1.9844e+00,
|
||||||
1.7412e+00, 1.1422e+01],
|
-1.1238e-02, 8.4375e+00],
|
||||||
[-2.6550e-02, -1.1325e-05, 3.0344e+01, ..., -9.1324e-03,
|
[-1.0876e-01, -2.7295e-01, 3.2156e+01, ..., -1.6891e-02,
|
||||||
-1.5199e-05, 3.8164e+00]], device='cuda:0', dtype=torch.float16)
|
-0.0000e+00, -0.0000e+00]], device='cuda:0', dtype=torch.float16)
|
||||||
tensor(True, device='cuda:0')
|
tensor(True, device='cuda:0')
|
||||||
|
|
||||||
|
|
||||||
@@ -373,36 +373,36 @@ We can now compare the performance of our kernel against CUTLASS. Here we focus
|
|||||||
|
|
||||||
M cuBLAS Triton
|
M cuBLAS Triton
|
||||||
0 512.0 20.164923 15.420235
|
0 512.0 20.164923 15.420235
|
||||||
1 768.0 58.982401 42.130286
|
1 768.0 58.982401 40.215272
|
||||||
2 1024.0 91.180520 72.315584
|
2 1024.0 91.180520 72.315584
|
||||||
3 1280.0 157.538463 117.028568
|
3 1280.0 157.538463 117.028568
|
||||||
4 1536.0 150.593357 147.455995
|
4 1536.0 153.867127 144.446699
|
||||||
5 1792.0 212.064605 193.783168
|
5 1792.0 208.137481 190.498706
|
||||||
6 2048.0 197.379013 151.146088
|
6 2048.0 199.728763 152.520144
|
||||||
7 2304.0 243.753804 179.608068
|
7 2304.0 246.266731 178.267699
|
||||||
8 2560.0 237.449270 217.006622
|
8 2560.0 235.741014 215.578957
|
||||||
9 2816.0 233.231062 200.987140
|
9 2816.0 231.990461 198.246398
|
||||||
10 3072.0 236.916752 221.184001
|
10 3072.0 236.916752 221.184001
|
||||||
11 3328.0 234.499328 210.500857
|
11 3328.0 239.173747 210.500857
|
||||||
12 3584.0 248.385067 230.552287
|
12 3584.0 248.385067 230.552287
|
||||||
13 3840.0 252.493157 223.418188
|
13 3840.0 251.917998 222.519114
|
||||||
14 4096.0 263.689066 244.922869
|
14 4096.0 263.172024 244.032234
|
||||||
15 4352.0 247.295210 231.639115
|
15 4352.0 249.595626 232.307632
|
||||||
16 4608.0 274.573240 254.803966
|
16 4608.0 276.560014 254.803966
|
||||||
17 4864.0 266.298229 245.366501
|
17 4864.0 266.614125 245.366501
|
||||||
18 5120.0 259.548513 238.312729
|
18 5120.0 257.003930 238.096276
|
||||||
19 5376.0 252.676487 237.081606
|
19 5376.0 252.676487 236.527241
|
||||||
20 5632.0 270.685535 249.046163
|
20 5632.0 270.057027 248.514009
|
||||||
21 5888.0 264.382140 242.069377
|
21 5888.0 264.206935 242.511113
|
||||||
22 6144.0 262.447761 240.565495
|
22 6144.0 259.441481 241.205983
|
||||||
23 6400.0 257.028108 235.078047
|
23 6400.0 257.157204 235.078047
|
||||||
24 6656.0 254.386204 232.699140
|
24 6656.0 254.161678 232.699140
|
||||||
25 6912.0 252.040861 232.926171
|
25 6912.0 251.844029 233.178785
|
||||||
26 7168.0 253.193644 231.815375
|
26 7168.0 253.282797 231.740709
|
||||||
27 7424.0 251.789150 232.860938
|
27 7424.0 251.868505 230.377264
|
||||||
28 7680.0 250.988932 231.727608
|
28 7680.0 250.988932 231.606284
|
||||||
29 7936.0 253.622108 232.094986
|
29 7936.0 253.293068 229.692102
|
||||||
30 8192.0 253.121589 231.859598
|
30 8192.0 253.002304 231.360005
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
@@ -410,7 +410,7 @@ We can now compare the performance of our kernel against CUTLASS. Here we focus
|
|||||||
|
|
||||||
.. rst-class:: sphx-glr-timing
|
.. rst-class:: sphx-glr-timing
|
||||||
|
|
||||||
**Total running time of the script:** ( 0 minutes 36.230 seconds)
|
**Total running time of the script:** ( 0 minutes 32.933 seconds)
|
||||||
|
|
||||||
|
|
||||||
.. _sphx_glr_download_getting-started_tutorials_03-matrix-multiplication.py:
|
.. _sphx_glr_download_getting-started_tutorials_03-matrix-multiplication.py:
|
||||||
|
|||||||
@@ -5,12 +5,12 @@
|
|||||||
|
|
||||||
Computation times
|
Computation times
|
||||||
=================
|
=================
|
||||||
**00:36.230** total execution time for **getting-started_tutorials** files:
|
**01:00.154** total execution time for **getting-started_tutorials** files:
|
||||||
|
|
||||||
+---------------------------------------------------------------------------------------------------------+-----------+--------+
|
+---------------------------------------------------------------------------------------------------------+-----------+--------+
|
||||||
| :ref:`sphx_glr_getting-started_tutorials_03-matrix-multiplication.py` (``03-matrix-multiplication.py``) | 00:36.230 | 0.0 MB |
|
| :ref:`sphx_glr_getting-started_tutorials_03-matrix-multiplication.py` (``03-matrix-multiplication.py``) | 00:32.933 | 0.0 MB |
|
||||||
+---------------------------------------------------------------------------------------------------------+-----------+--------+
|
+---------------------------------------------------------------------------------------------------------+-----------+--------+
|
||||||
| :ref:`sphx_glr_getting-started_tutorials_01-vector-add.py` (``01-vector-add.py``) | 00:00.000 | 0.0 MB |
|
| :ref:`sphx_glr_getting-started_tutorials_02-fused-softmax.py` (``02-fused-softmax.py``) | 00:20.176 | 0.0 MB |
|
||||||
+---------------------------------------------------------------------------------------------------------+-----------+--------+
|
+---------------------------------------------------------------------------------------------------------+-----------+--------+
|
||||||
| :ref:`sphx_glr_getting-started_tutorials_02-fused-softmax.py` (``02-fused-softmax.py``) | 00:00.000 | 0.0 MB |
|
| :ref:`sphx_glr_getting-started_tutorials_01-vector-add.py` (``01-vector-add.py``) | 00:07.044 | 0.0 MB |
|
||||||
+---------------------------------------------------------------------------------------------------------+-----------+--------+
|
+---------------------------------------------------------------------------------------------------------+-----------+--------+
|
||||||
|
|||||||
@@ -295,7 +295,7 @@ for different problem sizes.</p>
|
|||||||
</pre></div>
|
</pre></div>
|
||||||
</div>
|
</div>
|
||||||
<img alt="01 vector add" class="sphx-glr-single-img" src="../../_images/sphx_glr_01-vector-add_001.png" />
|
<img alt="01 vector add" class="sphx-glr-single-img" src="../../_images/sphx_glr_01-vector-add_001.png" />
|
||||||
<p class="sphx-glr-timing"><strong>Total running time of the script:</strong> ( 0 minutes 5.812 seconds)</p>
|
<p class="sphx-glr-timing"><strong>Total running time of the script:</strong> ( 0 minutes 7.044 seconds)</p>
|
||||||
<div class="sphx-glr-footer class sphx-glr-footer-example docutils container" id="sphx-glr-download-getting-started-tutorials-01-vector-add-py">
|
<div class="sphx-glr-footer class sphx-glr-footer-example docutils container" id="sphx-glr-download-getting-started-tutorials-01-vector-add-py">
|
||||||
<div class="sphx-glr-download sphx-glr-download-python docutils container">
|
<div class="sphx-glr-download sphx-glr-download-python docutils container">
|
||||||
<p><a class="reference download internal" download="" href="../../_downloads/62d97d49a32414049819dd8bb8378080/01-vector-add.py"><code class="xref download docutils literal notranslate"><span class="pre">Download</span> <span class="pre">Python</span> <span class="pre">source</span> <span class="pre">code:</span> <span class="pre">01-vector-add.py</span></code></a></p>
|
<p><a class="reference download internal" download="" href="../../_downloads/62d97d49a32414049819dd8bb8378080/01-vector-add.py"><code class="xref download docutils literal notranslate"><span class="pre">Download</span> <span class="pre">Python</span> <span class="pre">source</span> <span class="pre">code:</span> <span class="pre">01-vector-add.py</span></code></a></p>
|
||||||
|
|||||||
@@ -280,7 +280,7 @@ so we need to internally “pad” tiles and guard the memory operations properl
|
|||||||
<span class="c1"># Allocate output</span>
|
<span class="c1"># Allocate output</span>
|
||||||
<span class="n">y</span> <span class="o">=</span> <span class="n">torch</span><span class="o">.</span><span class="n">empty_like</span><span class="p">(</span><span class="n">x</span><span class="p">)</span>
|
<span class="n">y</span> <span class="o">=</span> <span class="n">torch</span><span class="o">.</span><span class="n">empty_like</span><span class="p">(</span><span class="n">x</span><span class="p">)</span>
|
||||||
<span class="c1"># Enqueue kernel. The launch grid is simple: we have one kernel instance per row of the input matrix</span>
|
<span class="c1"># Enqueue kernel. The launch grid is simple: we have one kernel instance per row of the input matrix</span>
|
||||||
<span class="n">_softmax</span><span class="p">[(</span><span class="n">M</span><span class="p">,</span> <span class="p">)](</span><span class="n">y</span><span class="p">,</span> <span class="n">x</span><span class="p">,</span> <span class="n">x</span><span class="o">.</span><span class="n">stride</span><span class="p">(</span><span class="mi">0</span><span class="p">),</span> <span class="n">y</span><span class="o">.</span><span class="n">stride</span><span class="p">(</span><span class="mi">0</span><span class="p">),</span> <span class="n">M</span><span class="p">,</span> <span class="n">N</span><span class="p">,</span> <span class="n">BLOCK</span><span class="o">=</span><span class="n">BLOCK</span><span class="p">)</span>
|
<span class="n">_softmax</span><span class="p">[(</span><span class="n">M</span><span class="p">,</span> <span class="p">)](</span><span class="n">y</span><span class="p">,</span> <span class="n">x</span><span class="p">,</span> <span class="n">x</span><span class="o">.</span><span class="n">stride</span><span class="p">(</span><span class="mi">0</span><span class="p">),</span> <span class="n">y</span><span class="o">.</span><span class="n">stride</span><span class="p">(</span><span class="mi">0</span><span class="p">),</span> <span class="n">M</span><span class="p">,</span> <span class="n">N</span><span class="p">,</span> <span class="n">num_warps</span><span class="o">=</span><span class="n">num_warps</span><span class="p">,</span> <span class="n">BLOCK</span><span class="o">=</span><span class="n">BLOCK</span><span class="p">)</span>
|
||||||
<span class="k">return</span> <span class="n">y</span>
|
<span class="k">return</span> <span class="n">y</span>
|
||||||
</pre></div>
|
</pre></div>
|
||||||
</div>
|
</div>
|
||||||
@@ -343,7 +343,7 @@ This means that – when temporary data is too large to fit entirely in the GPU
|
|||||||
Note that our Triton kernel is not only faster than PyTorch’s CUDA kernel, it is also <strong>easier to read, understand and maintain</strong>.</p></li>
|
Note that our Triton kernel is not only faster than PyTorch’s CUDA kernel, it is also <strong>easier to read, understand and maintain</strong>.</p></li>
|
||||||
</ul>
|
</ul>
|
||||||
</div></blockquote>
|
</div></blockquote>
|
||||||
<p class="sphx-glr-timing"><strong>Total running time of the script:</strong> ( 0 minutes 20.767 seconds)</p>
|
<p class="sphx-glr-timing"><strong>Total running time of the script:</strong> ( 0 minutes 20.176 seconds)</p>
|
||||||
<div class="sphx-glr-footer class sphx-glr-footer-example docutils container" id="sphx-glr-download-getting-started-tutorials-02-fused-softmax-py">
|
<div class="sphx-glr-footer class sphx-glr-footer-example docutils container" id="sphx-glr-download-getting-started-tutorials-02-fused-softmax-py">
|
||||||
<div class="sphx-glr-download sphx-glr-download-python docutils container">
|
<div class="sphx-glr-download sphx-glr-download-python docutils container">
|
||||||
<p><a class="reference download internal" download="" href="../../_downloads/d91442ac2982c4e0cc3ab0f43534afbc/02-fused-softmax.py"><code class="xref download docutils literal notranslate"><span class="pre">Download</span> <span class="pre">Python</span> <span class="pre">source</span> <span class="pre">code:</span> <span class="pre">02-fused-softmax.py</span></code></a></p>
|
<p><a class="reference download internal" download="" href="../../_downloads/d91442ac2982c4e0cc3ab0f43534afbc/02-fused-softmax.py"><code class="xref download docutils literal notranslate"><span class="pre">Download</span> <span class="pre">Python</span> <span class="pre">source</span> <span class="pre">code:</span> <span class="pre">02-fused-softmax.py</span></code></a></p>
|
||||||
|
|||||||
@@ -406,32 +406,32 @@ and (1) checks any shape constraint; (2) allocates the output; (3) launches the
|
|||||||
</pre></div>
|
</pre></div>
|
||||||
</div>
|
</div>
|
||||||
<p class="sphx-glr-script-out">Out:</p>
|
<p class="sphx-glr-script-out">Out:</p>
|
||||||
<div class="sphx-glr-script-out highlight-none notranslate"><div class="highlight"><pre><span></span>tensor([[-0.0000e+00, 2.9438e+01, -1.3113e-06, ..., 9.7266e+00,
|
<div class="sphx-glr-script-out highlight-none notranslate"><div class="highlight"><pre><span></span>tensor([[-5.9605e-08, 5.1094e+01, -1.8477e-05, ..., 2.6547e+01,
|
||||||
-3.4237e-04, -0.0000e+00],
|
-7.2598e-05, -4.2510e-04],
|
||||||
[-1.7615e-01, -0.0000e+00, 6.1914e+00, ..., 3.7562e+01,
|
[-2.7100e-01, -3.0220e-05, 5.9414e+00, ..., 2.8340e+00,
|
||||||
-0.0000e+00, -0.0000e+00],
|
-1.8644e-04, 1.3094e+01],
|
||||||
[ 9.9531e+00, 1.9078e+01, -0.0000e+00, ..., 3.6934e+00,
|
[-1.5332e-01, 4.8125e+00, 8.4277e-01, ..., 3.6387e+00,
|
||||||
1.6578e+01, 2.1031e+01],
|
4.3375e+01, 1.6865e+00],
|
||||||
...,
|
...,
|
||||||
[ 2.6547e+01, -1.1802e-05, 7.7852e+00, ..., 5.2156e+01,
|
[-0.0000e+00, 2.9453e+01, -4.7684e-07, ..., 6.2617e+00,
|
||||||
3.5469e+01, 1.5602e+01],
|
4.1133e+00, -0.0000e+00],
|
||||||
[-0.0000e+00, -0.0000e+00, 1.6531e+01, ..., 2.1211e+00,
|
[ 1.6562e+01, -8.1539e-04, 1.3836e+01, ..., 1.9844e+00,
|
||||||
1.7412e+00, 1.1422e+01],
|
-1.1238e-02, 8.4375e+00],
|
||||||
[-2.6550e-02, -1.1325e-05, 3.0344e+01, ..., -9.1248e-03,
|
[-1.0876e-01, -2.7295e-01, 3.2156e+01, ..., -1.6907e-02,
|
||||||
-1.5199e-05, 3.8164e+00]], device='cuda:0', dtype=torch.float16)
|
-0.0000e+00, -0.0000e+00]], device='cuda:0', dtype=torch.float16)
|
||||||
tensor([[-0.0000e+00, 2.9438e+01, -1.3113e-06, ..., 9.7266e+00,
|
tensor([[-5.9605e-08, 5.1094e+01, -1.8537e-05, ..., 2.6547e+01,
|
||||||
-3.4261e-04, -0.0000e+00],
|
-7.2658e-05, -4.2605e-04],
|
||||||
[-1.7615e-01, -0.0000e+00, 6.1914e+00, ..., 3.7562e+01,
|
[-2.7100e-01, -3.0220e-05, 5.9414e+00, ..., 2.8340e+00,
|
||||||
-0.0000e+00, -0.0000e+00],
|
-1.8632e-04, 1.3094e+01],
|
||||||
[ 9.9531e+00, 1.9078e+01, -0.0000e+00, ..., 3.6934e+00,
|
[-1.5332e-01, 4.8125e+00, 8.4277e-01, ..., 3.6387e+00,
|
||||||
1.6578e+01, 2.1031e+01],
|
4.3375e+01, 1.6875e+00],
|
||||||
...,
|
...,
|
||||||
[ 2.6547e+01, -1.1802e-05, 7.7852e+00, ..., 5.2156e+01,
|
[-0.0000e+00, 2.9453e+01, -4.7684e-07, ..., 6.2617e+00,
|
||||||
3.5469e+01, 1.5602e+01],
|
4.1133e+00, -0.0000e+00],
|
||||||
[-0.0000e+00, -0.0000e+00, 1.6531e+01, ..., 2.1211e+00,
|
[ 1.6562e+01, -8.1778e-04, 1.3836e+01, ..., 1.9844e+00,
|
||||||
1.7412e+00, 1.1422e+01],
|
-1.1238e-02, 8.4375e+00],
|
||||||
[-2.6550e-02, -1.1325e-05, 3.0344e+01, ..., -9.1324e-03,
|
[-1.0876e-01, -2.7295e-01, 3.2156e+01, ..., -1.6891e-02,
|
||||||
-1.5199e-05, 3.8164e+00]], device='cuda:0', dtype=torch.float16)
|
-0.0000e+00, -0.0000e+00]], device='cuda:0', dtype=torch.float16)
|
||||||
tensor(True, device='cuda:0')
|
tensor(True, device='cuda:0')
|
||||||
</pre></div>
|
</pre></div>
|
||||||
</div>
|
</div>
|
||||||
@@ -472,39 +472,39 @@ tensor(True, device='cuda:0')
|
|||||||
<p class="sphx-glr-script-out">Out:</p>
|
<p class="sphx-glr-script-out">Out:</p>
|
||||||
<div class="sphx-glr-script-out highlight-none notranslate"><div class="highlight"><pre><span></span> M cuBLAS Triton
|
<div class="sphx-glr-script-out highlight-none notranslate"><div class="highlight"><pre><span></span> M cuBLAS Triton
|
||||||
0 512.0 20.164923 15.420235
|
0 512.0 20.164923 15.420235
|
||||||
1 768.0 58.982401 42.130286
|
1 768.0 58.982401 40.215272
|
||||||
2 1024.0 91.180520 72.315584
|
2 1024.0 91.180520 72.315584
|
||||||
3 1280.0 157.538463 117.028568
|
3 1280.0 157.538463 117.028568
|
||||||
4 1536.0 150.593357 147.455995
|
4 1536.0 153.867127 144.446699
|
||||||
5 1792.0 212.064605 193.783168
|
5 1792.0 208.137481 190.498706
|
||||||
6 2048.0 197.379013 151.146088
|
6 2048.0 199.728763 152.520144
|
||||||
7 2304.0 243.753804 179.608068
|
7 2304.0 246.266731 178.267699
|
||||||
8 2560.0 237.449270 217.006622
|
8 2560.0 235.741014 215.578957
|
||||||
9 2816.0 233.231062 200.987140
|
9 2816.0 231.990461 198.246398
|
||||||
10 3072.0 236.916752 221.184001
|
10 3072.0 236.916752 221.184001
|
||||||
11 3328.0 234.499328 210.500857
|
11 3328.0 239.173747 210.500857
|
||||||
12 3584.0 248.385067 230.552287
|
12 3584.0 248.385067 230.552287
|
||||||
13 3840.0 252.493157 223.418188
|
13 3840.0 251.917998 222.519114
|
||||||
14 4096.0 263.689066 244.922869
|
14 4096.0 263.172024 244.032234
|
||||||
15 4352.0 247.295210 231.639115
|
15 4352.0 249.595626 232.307632
|
||||||
16 4608.0 274.573240 254.803966
|
16 4608.0 276.560014 254.803966
|
||||||
17 4864.0 266.298229 245.366501
|
17 4864.0 266.614125 245.366501
|
||||||
18 5120.0 259.548513 238.312729
|
18 5120.0 257.003930 238.096276
|
||||||
19 5376.0 252.676487 237.081606
|
19 5376.0 252.676487 236.527241
|
||||||
20 5632.0 270.685535 249.046163
|
20 5632.0 270.057027 248.514009
|
||||||
21 5888.0 264.382140 242.069377
|
21 5888.0 264.206935 242.511113
|
||||||
22 6144.0 262.447761 240.565495
|
22 6144.0 259.441481 241.205983
|
||||||
23 6400.0 257.028108 235.078047
|
23 6400.0 257.157204 235.078047
|
||||||
24 6656.0 254.386204 232.699140
|
24 6656.0 254.161678 232.699140
|
||||||
25 6912.0 252.040861 232.926171
|
25 6912.0 251.844029 233.178785
|
||||||
26 7168.0 253.193644 231.815375
|
26 7168.0 253.282797 231.740709
|
||||||
27 7424.0 251.789150 232.860938
|
27 7424.0 251.868505 230.377264
|
||||||
28 7680.0 250.988932 231.727608
|
28 7680.0 250.988932 231.606284
|
||||||
29 7936.0 253.622108 232.094986
|
29 7936.0 253.293068 229.692102
|
||||||
30 8192.0 253.121589 231.859598
|
30 8192.0 253.002304 231.360005
|
||||||
</pre></div>
|
</pre></div>
|
||||||
</div>
|
</div>
|
||||||
<p class="sphx-glr-timing"><strong>Total running time of the script:</strong> ( 0 minutes 36.230 seconds)</p>
|
<p class="sphx-glr-timing"><strong>Total running time of the script:</strong> ( 0 minutes 32.933 seconds)</p>
|
||||||
<div class="sphx-glr-footer class sphx-glr-footer-example docutils container" id="sphx-glr-download-getting-started-tutorials-03-matrix-multiplication-py">
|
<div class="sphx-glr-footer class sphx-glr-footer-example docutils container" id="sphx-glr-download-getting-started-tutorials-03-matrix-multiplication-py">
|
||||||
<div class="sphx-glr-download sphx-glr-download-python docutils container">
|
<div class="sphx-glr-download sphx-glr-download-python docutils container">
|
||||||
<p><a class="reference download internal" download="" href="../../_downloads/d5fee5b55a64e47f1b5724ec39adf171/03-matrix-multiplication.py"><code class="xref download docutils literal notranslate"><span class="pre">Download</span> <span class="pre">Python</span> <span class="pre">source</span> <span class="pre">code:</span> <span class="pre">03-matrix-multiplication.py</span></code></a></p>
|
<p><a class="reference download internal" download="" href="../../_downloads/d5fee5b55a64e47f1b5724ec39adf171/03-matrix-multiplication.py"><code class="xref download docutils literal notranslate"><span class="pre">Download</span> <span class="pre">Python</span> <span class="pre">source</span> <span class="pre">code:</span> <span class="pre">03-matrix-multiplication.py</span></code></a></p>
|
||||||
|
|||||||
@@ -169,7 +169,7 @@
|
|||||||
|
|
||||||
<div class="section" id="computation-times">
|
<div class="section" id="computation-times">
|
||||||
<span id="sphx-glr-getting-started-tutorials-sg-execution-times"></span><h1>Computation times<a class="headerlink" href="#computation-times" title="Permalink to this headline">¶</a></h1>
|
<span id="sphx-glr-getting-started-tutorials-sg-execution-times"></span><h1>Computation times<a class="headerlink" href="#computation-times" title="Permalink to this headline">¶</a></h1>
|
||||||
<p><strong>00:36.230</strong> total execution time for <strong>getting-started_tutorials</strong> files:</p>
|
<p><strong>01:00.154</strong> total execution time for <strong>getting-started_tutorials</strong> files:</p>
|
||||||
<table class="docutils align-default">
|
<table class="docutils align-default">
|
||||||
<colgroup>
|
<colgroup>
|
||||||
<col style="width: 85%" />
|
<col style="width: 85%" />
|
||||||
@@ -178,15 +178,15 @@
|
|||||||
</colgroup>
|
</colgroup>
|
||||||
<tbody>
|
<tbody>
|
||||||
<tr class="row-odd"><td><p><a class="reference internal" href="03-matrix-multiplication.html#sphx-glr-getting-started-tutorials-03-matrix-multiplication-py"><span class="std std-ref">Matrix Multiplication</span></a> (<code class="docutils literal notranslate"><span class="pre">03-matrix-multiplication.py</span></code>)</p></td>
|
<tr class="row-odd"><td><p><a class="reference internal" href="03-matrix-multiplication.html#sphx-glr-getting-started-tutorials-03-matrix-multiplication-py"><span class="std std-ref">Matrix Multiplication</span></a> (<code class="docutils literal notranslate"><span class="pre">03-matrix-multiplication.py</span></code>)</p></td>
|
||||||
<td><p>00:36.230</p></td>
|
<td><p>00:32.933</p></td>
|
||||||
<td><p>0.0 MB</p></td>
|
<td><p>0.0 MB</p></td>
|
||||||
</tr>
|
</tr>
|
||||||
<tr class="row-even"><td><p><a class="reference internal" href="01-vector-add.html#sphx-glr-getting-started-tutorials-01-vector-add-py"><span class="std std-ref">Vector Addition</span></a> (<code class="docutils literal notranslate"><span class="pre">01-vector-add.py</span></code>)</p></td>
|
<tr class="row-even"><td><p><a class="reference internal" href="02-fused-softmax.html#sphx-glr-getting-started-tutorials-02-fused-softmax-py"><span class="std std-ref">Fused Softmax</span></a> (<code class="docutils literal notranslate"><span class="pre">02-fused-softmax.py</span></code>)</p></td>
|
||||||
<td><p>00:00.000</p></td>
|
<td><p>00:20.176</p></td>
|
||||||
<td><p>0.0 MB</p></td>
|
<td><p>0.0 MB</p></td>
|
||||||
</tr>
|
</tr>
|
||||||
<tr class="row-odd"><td><p><a class="reference internal" href="02-fused-softmax.html#sphx-glr-getting-started-tutorials-02-fused-softmax-py"><span class="std std-ref">Fused Softmax</span></a> (<code class="docutils literal notranslate"><span class="pre">02-fused-softmax.py</span></code>)</p></td>
|
<tr class="row-odd"><td><p><a class="reference internal" href="01-vector-add.html#sphx-glr-getting-started-tutorials-01-vector-add-py"><span class="std std-ref">Vector Addition</span></a> (<code class="docutils literal notranslate"><span class="pre">01-vector-add.py</span></code>)</p></td>
|
||||||
<td><p>00:00.000</p></td>
|
<td><p>00:07.044</p></td>
|
||||||
<td><p>0.0 MB</p></td>
|
<td><p>0.0 MB</p></td>
|
||||||
</tr>
|
</tr>
|
||||||
</tbody>
|
</tbody>
|
||||||
|
|||||||
@@ -42,9 +42,7 @@
|
|||||||
|
|
||||||
|
|
||||||
<link rel="index" title="Index" href="../../../genindex.html" />
|
<link rel="index" title="Index" href="../../../genindex.html" />
|
||||||
<link rel="search" title="Search" href="../../../search.html" />
|
<link rel="search" title="Search" href="../../../search.html" />
|
||||||
<link rel="next" title="triton.zeros" href="triton.zeros.html" />
|
|
||||||
<link rel="prev" title="triton.num_programs" href="triton.num_programs.html" />
|
|
||||||
</head>
|
</head>
|
||||||
|
|
||||||
<body class="wy-body-for-nav">
|
<body class="wy-body-for-nav">
|
||||||
@@ -95,24 +93,8 @@
|
|||||||
<li class="toctree-l1"><a class="reference internal" href="../../../getting-started/tutorials/index.html">Tutorials</a></li>
|
<li class="toctree-l1"><a class="reference internal" href="../../../getting-started/tutorials/index.html">Tutorials</a></li>
|
||||||
</ul>
|
</ul>
|
||||||
<p class="caption"><span class="caption-text">Language Reference</span></p>
|
<p class="caption"><span class="caption-text">Language Reference</span></p>
|
||||||
<ul class="current">
|
<ul>
|
||||||
<li class="toctree-l1 current"><a class="reference internal" href="../index.html">Python API</a><ul class="current">
|
<li class="toctree-l1"><a class="reference internal" href="../index.html">Python API</a></li>
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#programming-model">Programming Model</a></li>
|
|
||||||
<li class="toctree-l2 current"><a class="reference internal" href="../index.html#creation-ops">Creation Ops</a><ul class="current">
|
|
||||||
<li class="toctree-l3 current"><a class="current reference internal" href="#">triton.arange</a></li>
|
|
||||||
<li class="toctree-l3"><a class="reference internal" href="triton.zeros.html">triton.zeros</a></li>
|
|
||||||
</ul>
|
|
||||||
</li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#shape-manipulation-ops">Shape Manipulation Ops</a></li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#linear-algebra-ops">Linear Algebra Ops</a></li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#memory-ops">Memory Ops</a></li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#indexing-ops">Indexing Ops</a></li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#math-ops">Math Ops</a></li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#reduction-ops">Reduction Ops</a></li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#comparison-ops">Comparison ops</a></li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#compiler-hint-ops">Compiler Hint Ops</a></li>
|
|
||||||
</ul>
|
|
||||||
</li>
|
|
||||||
</ul>
|
</ul>
|
||||||
<p class="caption"><span class="caption-text">Programming Guide</span></p>
|
<p class="caption"><span class="caption-text">Programming Guide</span></p>
|
||||||
<ul>
|
<ul>
|
||||||
@@ -166,8 +148,6 @@
|
|||||||
|
|
||||||
<li><a href="../../../index.html" class="icon icon-home"></a> »</li>
|
<li><a href="../../../index.html" class="icon icon-home"></a> »</li>
|
||||||
|
|
||||||
<li><a href="../index.html">Python API</a> »</li>
|
|
||||||
|
|
||||||
<li>triton.arange</li>
|
<li>triton.arange</li>
|
||||||
|
|
||||||
|
|
||||||
@@ -211,10 +191,6 @@
|
|||||||
|
|
||||||
</div>
|
</div>
|
||||||
<footer>
|
<footer>
|
||||||
<div class="rst-footer-buttons" role="navigation" aria-label="footer navigation">
|
|
||||||
<a href="triton.zeros.html" class="btn btn-neutral float-right" title="triton.zeros" accesskey="n" rel="next">Next <span class="fa fa-arrow-circle-right" aria-hidden="true"></span></a>
|
|
||||||
<a href="triton.num_programs.html" class="btn btn-neutral float-left" title="triton.num_programs" accesskey="p" rel="prev"><span class="fa fa-arrow-circle-left" aria-hidden="true"></span> Previous</a>
|
|
||||||
</div>
|
|
||||||
|
|
||||||
<hr/>
|
<hr/>
|
||||||
|
|
||||||
|
|||||||
@@ -42,9 +42,7 @@
|
|||||||
|
|
||||||
|
|
||||||
<link rel="index" title="Index" href="../../../genindex.html" />
|
<link rel="index" title="Index" href="../../../genindex.html" />
|
||||||
<link rel="search" title="Search" href="../../../search.html" />
|
<link rel="search" title="Search" href="../../../search.html" />
|
||||||
<link rel="next" title="triton.atomic_xchg" href="triton.atomic_xchg.html" />
|
|
||||||
<link rel="prev" title="triton.store" href="triton.store.html" />
|
|
||||||
</head>
|
</head>
|
||||||
|
|
||||||
<body class="wy-body-for-nav">
|
<body class="wy-body-for-nav">
|
||||||
@@ -95,26 +93,8 @@
|
|||||||
<li class="toctree-l1"><a class="reference internal" href="../../../getting-started/tutorials/index.html">Tutorials</a></li>
|
<li class="toctree-l1"><a class="reference internal" href="../../../getting-started/tutorials/index.html">Tutorials</a></li>
|
||||||
</ul>
|
</ul>
|
||||||
<p class="caption"><span class="caption-text">Language Reference</span></p>
|
<p class="caption"><span class="caption-text">Language Reference</span></p>
|
||||||
<ul class="current">
|
<ul>
|
||||||
<li class="toctree-l1 current"><a class="reference internal" href="../index.html">Python API</a><ul class="current">
|
<li class="toctree-l1"><a class="reference internal" href="../index.html">Python API</a></li>
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#programming-model">Programming Model</a></li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#creation-ops">Creation Ops</a></li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#shape-manipulation-ops">Shape Manipulation Ops</a></li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#linear-algebra-ops">Linear Algebra Ops</a></li>
|
|
||||||
<li class="toctree-l2 current"><a class="reference internal" href="../index.html#memory-ops">Memory Ops</a><ul class="current">
|
|
||||||
<li class="toctree-l3"><a class="reference internal" href="triton.load.html">triton.load</a></li>
|
|
||||||
<li class="toctree-l3"><a class="reference internal" href="triton.store.html">triton.store</a></li>
|
|
||||||
<li class="toctree-l3 current"><a class="current reference internal" href="#">triton.atomic_cas</a></li>
|
|
||||||
<li class="toctree-l3"><a class="reference internal" href="triton.atomic_xchg.html">triton.atomic_xchg</a></li>
|
|
||||||
</ul>
|
|
||||||
</li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#indexing-ops">Indexing Ops</a></li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#math-ops">Math Ops</a></li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#reduction-ops">Reduction Ops</a></li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#comparison-ops">Comparison ops</a></li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#compiler-hint-ops">Compiler Hint Ops</a></li>
|
|
||||||
</ul>
|
|
||||||
</li>
|
|
||||||
</ul>
|
</ul>
|
||||||
<p class="caption"><span class="caption-text">Programming Guide</span></p>
|
<p class="caption"><span class="caption-text">Programming Guide</span></p>
|
||||||
<ul>
|
<ul>
|
||||||
@@ -168,8 +148,6 @@
|
|||||||
|
|
||||||
<li><a href="../../../index.html" class="icon icon-home"></a> »</li>
|
<li><a href="../../../index.html" class="icon icon-home"></a> »</li>
|
||||||
|
|
||||||
<li><a href="../index.html">Python API</a> »</li>
|
|
||||||
|
|
||||||
<li>triton.atomic_cas</li>
|
<li>triton.atomic_cas</li>
|
||||||
|
|
||||||
|
|
||||||
@@ -203,10 +181,6 @@
|
|||||||
|
|
||||||
</div>
|
</div>
|
||||||
<footer>
|
<footer>
|
||||||
<div class="rst-footer-buttons" role="navigation" aria-label="footer navigation">
|
|
||||||
<a href="triton.atomic_xchg.html" class="btn btn-neutral float-right" title="triton.atomic_xchg" accesskey="n" rel="next">Next <span class="fa fa-arrow-circle-right" aria-hidden="true"></span></a>
|
|
||||||
<a href="triton.store.html" class="btn btn-neutral float-left" title="triton.store" accesskey="p" rel="prev"><span class="fa fa-arrow-circle-left" aria-hidden="true"></span> Previous</a>
|
|
||||||
</div>
|
|
||||||
|
|
||||||
<hr/>
|
<hr/>
|
||||||
|
|
||||||
|
|||||||
@@ -42,9 +42,7 @@
|
|||||||
|
|
||||||
|
|
||||||
<link rel="index" title="Index" href="../../../genindex.html" />
|
<link rel="index" title="Index" href="../../../genindex.html" />
|
||||||
<link rel="search" title="Search" href="../../../search.html" />
|
<link rel="search" title="Search" href="../../../search.html" />
|
||||||
<link rel="next" title="triton.where" href="triton.where.html" />
|
|
||||||
<link rel="prev" title="triton.atomic_cas" href="triton.atomic_cas.html" />
|
|
||||||
</head>
|
</head>
|
||||||
|
|
||||||
<body class="wy-body-for-nav">
|
<body class="wy-body-for-nav">
|
||||||
@@ -95,26 +93,8 @@
|
|||||||
<li class="toctree-l1"><a class="reference internal" href="../../../getting-started/tutorials/index.html">Tutorials</a></li>
|
<li class="toctree-l1"><a class="reference internal" href="../../../getting-started/tutorials/index.html">Tutorials</a></li>
|
||||||
</ul>
|
</ul>
|
||||||
<p class="caption"><span class="caption-text">Language Reference</span></p>
|
<p class="caption"><span class="caption-text">Language Reference</span></p>
|
||||||
<ul class="current">
|
<ul>
|
||||||
<li class="toctree-l1 current"><a class="reference internal" href="../index.html">Python API</a><ul class="current">
|
<li class="toctree-l1"><a class="reference internal" href="../index.html">Python API</a></li>
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#programming-model">Programming Model</a></li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#creation-ops">Creation Ops</a></li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#shape-manipulation-ops">Shape Manipulation Ops</a></li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#linear-algebra-ops">Linear Algebra Ops</a></li>
|
|
||||||
<li class="toctree-l2 current"><a class="reference internal" href="../index.html#memory-ops">Memory Ops</a><ul class="current">
|
|
||||||
<li class="toctree-l3"><a class="reference internal" href="triton.load.html">triton.load</a></li>
|
|
||||||
<li class="toctree-l3"><a class="reference internal" href="triton.store.html">triton.store</a></li>
|
|
||||||
<li class="toctree-l3"><a class="reference internal" href="triton.atomic_cas.html">triton.atomic_cas</a></li>
|
|
||||||
<li class="toctree-l3 current"><a class="current reference internal" href="#">triton.atomic_xchg</a></li>
|
|
||||||
</ul>
|
|
||||||
</li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#indexing-ops">Indexing Ops</a></li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#math-ops">Math Ops</a></li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#reduction-ops">Reduction Ops</a></li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#comparison-ops">Comparison ops</a></li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#compiler-hint-ops">Compiler Hint Ops</a></li>
|
|
||||||
</ul>
|
|
||||||
</li>
|
|
||||||
</ul>
|
</ul>
|
||||||
<p class="caption"><span class="caption-text">Programming Guide</span></p>
|
<p class="caption"><span class="caption-text">Programming Guide</span></p>
|
||||||
<ul>
|
<ul>
|
||||||
@@ -168,8 +148,6 @@
|
|||||||
|
|
||||||
<li><a href="../../../index.html" class="icon icon-home"></a> »</li>
|
<li><a href="../../../index.html" class="icon icon-home"></a> »</li>
|
||||||
|
|
||||||
<li><a href="../index.html">Python API</a> »</li>
|
|
||||||
|
|
||||||
<li>triton.atomic_xchg</li>
|
<li>triton.atomic_xchg</li>
|
||||||
|
|
||||||
|
|
||||||
@@ -203,10 +181,6 @@
|
|||||||
|
|
||||||
</div>
|
</div>
|
||||||
<footer>
|
<footer>
|
||||||
<div class="rst-footer-buttons" role="navigation" aria-label="footer navigation">
|
|
||||||
<a href="triton.where.html" class="btn btn-neutral float-right" title="triton.where" accesskey="n" rel="next">Next <span class="fa fa-arrow-circle-right" aria-hidden="true"></span></a>
|
|
||||||
<a href="triton.atomic_cas.html" class="btn btn-neutral float-left" title="triton.atomic_cas" accesskey="p" rel="prev"><span class="fa fa-arrow-circle-left" aria-hidden="true"></span> Previous</a>
|
|
||||||
</div>
|
|
||||||
|
|
||||||
<hr/>
|
<hr/>
|
||||||
|
|
||||||
|
|||||||
@@ -42,9 +42,7 @@
|
|||||||
|
|
||||||
|
|
||||||
<link rel="index" title="Index" href="../../../genindex.html" />
|
<link rel="index" title="Index" href="../../../genindex.html" />
|
||||||
<link rel="search" title="Search" href="../../../search.html" />
|
<link rel="search" title="Search" href="../../../search.html" />
|
||||||
<link rel="next" title="triton.reshape" href="triton.reshape.html" />
|
|
||||||
<link rel="prev" title="triton.zeros" href="triton.zeros.html" />
|
|
||||||
</head>
|
</head>
|
||||||
|
|
||||||
<body class="wy-body-for-nav">
|
<body class="wy-body-for-nav">
|
||||||
@@ -95,25 +93,8 @@
|
|||||||
<li class="toctree-l1"><a class="reference internal" href="../../../getting-started/tutorials/index.html">Tutorials</a></li>
|
<li class="toctree-l1"><a class="reference internal" href="../../../getting-started/tutorials/index.html">Tutorials</a></li>
|
||||||
</ul>
|
</ul>
|
||||||
<p class="caption"><span class="caption-text">Language Reference</span></p>
|
<p class="caption"><span class="caption-text">Language Reference</span></p>
|
||||||
<ul class="current">
|
<ul>
|
||||||
<li class="toctree-l1 current"><a class="reference internal" href="../index.html">Python API</a><ul class="current">
|
<li class="toctree-l1"><a class="reference internal" href="../index.html">Python API</a></li>
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#programming-model">Programming Model</a></li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#creation-ops">Creation Ops</a></li>
|
|
||||||
<li class="toctree-l2 current"><a class="reference internal" href="../index.html#shape-manipulation-ops">Shape Manipulation Ops</a><ul class="current">
|
|
||||||
<li class="toctree-l3 current"><a class="current reference internal" href="#">triton.broadcast_to</a></li>
|
|
||||||
<li class="toctree-l3"><a class="reference internal" href="triton.reshape.html">triton.reshape</a></li>
|
|
||||||
<li class="toctree-l3"><a class="reference internal" href="triton.ravel.html">triton.ravel</a></li>
|
|
||||||
</ul>
|
|
||||||
</li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#linear-algebra-ops">Linear Algebra Ops</a></li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#memory-ops">Memory Ops</a></li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#indexing-ops">Indexing Ops</a></li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#math-ops">Math Ops</a></li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#reduction-ops">Reduction Ops</a></li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#comparison-ops">Comparison ops</a></li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#compiler-hint-ops">Compiler Hint Ops</a></li>
|
|
||||||
</ul>
|
|
||||||
</li>
|
|
||||||
</ul>
|
</ul>
|
||||||
<p class="caption"><span class="caption-text">Programming Guide</span></p>
|
<p class="caption"><span class="caption-text">Programming Guide</span></p>
|
||||||
<ul>
|
<ul>
|
||||||
@@ -167,8 +148,6 @@
|
|||||||
|
|
||||||
<li><a href="../../../index.html" class="icon icon-home"></a> »</li>
|
<li><a href="../../../index.html" class="icon icon-home"></a> »</li>
|
||||||
|
|
||||||
<li><a href="../index.html">Python API</a> »</li>
|
|
||||||
|
|
||||||
<li>triton.broadcast_to</li>
|
<li>triton.broadcast_to</li>
|
||||||
|
|
||||||
|
|
||||||
@@ -212,10 +191,6 @@
|
|||||||
|
|
||||||
</div>
|
</div>
|
||||||
<footer>
|
<footer>
|
||||||
<div class="rst-footer-buttons" role="navigation" aria-label="footer navigation">
|
|
||||||
<a href="triton.reshape.html" class="btn btn-neutral float-right" title="triton.reshape" accesskey="n" rel="next">Next <span class="fa fa-arrow-circle-right" aria-hidden="true"></span></a>
|
|
||||||
<a href="triton.zeros.html" class="btn btn-neutral float-left" title="triton.zeros" accesskey="p" rel="prev"><span class="fa fa-arrow-circle-left" aria-hidden="true"></span> Previous</a>
|
|
||||||
</div>
|
|
||||||
|
|
||||||
<hr/>
|
<hr/>
|
||||||
|
|
||||||
|
|||||||
@@ -42,9 +42,7 @@
|
|||||||
|
|
||||||
|
|
||||||
<link rel="index" title="Index" href="../../../genindex.html" />
|
<link rel="index" title="Index" href="../../../genindex.html" />
|
||||||
<link rel="search" title="Search" href="../../../search.html" />
|
<link rel="search" title="Search" href="../../../search.html" />
|
||||||
<link rel="next" title="triton.load" href="triton.load.html" />
|
|
||||||
<link rel="prev" title="triton.ravel" href="triton.ravel.html" />
|
|
||||||
</head>
|
</head>
|
||||||
|
|
||||||
<body class="wy-body-for-nav">
|
<body class="wy-body-for-nav">
|
||||||
@@ -95,23 +93,8 @@
|
|||||||
<li class="toctree-l1"><a class="reference internal" href="../../../getting-started/tutorials/index.html">Tutorials</a></li>
|
<li class="toctree-l1"><a class="reference internal" href="../../../getting-started/tutorials/index.html">Tutorials</a></li>
|
||||||
</ul>
|
</ul>
|
||||||
<p class="caption"><span class="caption-text">Language Reference</span></p>
|
<p class="caption"><span class="caption-text">Language Reference</span></p>
|
||||||
<ul class="current">
|
<ul>
|
||||||
<li class="toctree-l1 current"><a class="reference internal" href="../index.html">Python API</a><ul class="current">
|
<li class="toctree-l1"><a class="reference internal" href="../index.html">Python API</a></li>
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#programming-model">Programming Model</a></li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#creation-ops">Creation Ops</a></li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#shape-manipulation-ops">Shape Manipulation Ops</a></li>
|
|
||||||
<li class="toctree-l2 current"><a class="reference internal" href="../index.html#linear-algebra-ops">Linear Algebra Ops</a><ul class="current">
|
|
||||||
<li class="toctree-l3 current"><a class="current reference internal" href="#">triton.dot</a></li>
|
|
||||||
</ul>
|
|
||||||
</li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#memory-ops">Memory Ops</a></li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#indexing-ops">Indexing Ops</a></li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#math-ops">Math Ops</a></li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#reduction-ops">Reduction Ops</a></li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#comparison-ops">Comparison ops</a></li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#compiler-hint-ops">Compiler Hint Ops</a></li>
|
|
||||||
</ul>
|
|
||||||
</li>
|
|
||||||
</ul>
|
</ul>
|
||||||
<p class="caption"><span class="caption-text">Programming Guide</span></p>
|
<p class="caption"><span class="caption-text">Programming Guide</span></p>
|
||||||
<ul>
|
<ul>
|
||||||
@@ -165,8 +148,6 @@
|
|||||||
|
|
||||||
<li><a href="../../../index.html" class="icon icon-home"></a> »</li>
|
<li><a href="../../../index.html" class="icon icon-home"></a> »</li>
|
||||||
|
|
||||||
<li><a href="../index.html">Python API</a> »</li>
|
|
||||||
|
|
||||||
<li>triton.dot</li>
|
<li>triton.dot</li>
|
||||||
|
|
||||||
|
|
||||||
@@ -211,10 +192,6 @@ The two blocks must be two dimensionals and have compatible inner dimensions.</p
|
|||||||
|
|
||||||
</div>
|
</div>
|
||||||
<footer>
|
<footer>
|
||||||
<div class="rst-footer-buttons" role="navigation" aria-label="footer navigation">
|
|
||||||
<a href="triton.load.html" class="btn btn-neutral float-right" title="triton.load" accesskey="n" rel="next">Next <span class="fa fa-arrow-circle-right" aria-hidden="true"></span></a>
|
|
||||||
<a href="triton.ravel.html" class="btn btn-neutral float-left" title="triton.ravel" accesskey="p" rel="prev"><span class="fa fa-arrow-circle-left" aria-hidden="true"></span> Previous</a>
|
|
||||||
</div>
|
|
||||||
|
|
||||||
<hr/>
|
<hr/>
|
||||||
|
|
||||||
|
|||||||
@@ -42,9 +42,7 @@
|
|||||||
|
|
||||||
|
|
||||||
<link rel="index" title="Index" href="../../../genindex.html" />
|
<link rel="index" title="Index" href="../../../genindex.html" />
|
||||||
<link rel="search" title="Search" href="../../../search.html" />
|
<link rel="search" title="Search" href="../../../search.html" />
|
||||||
<link rel="next" title="triton.log" href="triton.log.html" />
|
|
||||||
<link rel="prev" title="triton.where" href="triton.where.html" />
|
|
||||||
</head>
|
</head>
|
||||||
|
|
||||||
<body class="wy-body-for-nav">
|
<body class="wy-body-for-nav">
|
||||||
@@ -95,26 +93,8 @@
|
|||||||
<li class="toctree-l1"><a class="reference internal" href="../../../getting-started/tutorials/index.html">Tutorials</a></li>
|
<li class="toctree-l1"><a class="reference internal" href="../../../getting-started/tutorials/index.html">Tutorials</a></li>
|
||||||
</ul>
|
</ul>
|
||||||
<p class="caption"><span class="caption-text">Language Reference</span></p>
|
<p class="caption"><span class="caption-text">Language Reference</span></p>
|
||||||
<ul class="current">
|
<ul>
|
||||||
<li class="toctree-l1 current"><a class="reference internal" href="../index.html">Python API</a><ul class="current">
|
<li class="toctree-l1"><a class="reference internal" href="../index.html">Python API</a></li>
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#programming-model">Programming Model</a></li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#creation-ops">Creation Ops</a></li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#shape-manipulation-ops">Shape Manipulation Ops</a></li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#linear-algebra-ops">Linear Algebra Ops</a></li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#memory-ops">Memory Ops</a></li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#indexing-ops">Indexing Ops</a></li>
|
|
||||||
<li class="toctree-l2 current"><a class="reference internal" href="../index.html#math-ops">Math Ops</a><ul class="current">
|
|
||||||
<li class="toctree-l3 current"><a class="current reference internal" href="#">triton.exp</a></li>
|
|
||||||
<li class="toctree-l3"><a class="reference internal" href="triton.log.html">triton.log</a></li>
|
|
||||||
<li class="toctree-l3"><a class="reference internal" href="triton.sigmoid.html">triton.sigmoid</a></li>
|
|
||||||
<li class="toctree-l3"><a class="reference internal" href="triton.softmax.html">triton.softmax</a></li>
|
|
||||||
</ul>
|
|
||||||
</li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#reduction-ops">Reduction Ops</a></li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#comparison-ops">Comparison ops</a></li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#compiler-hint-ops">Compiler Hint Ops</a></li>
|
|
||||||
</ul>
|
|
||||||
</li>
|
|
||||||
</ul>
|
</ul>
|
||||||
<p class="caption"><span class="caption-text">Programming Guide</span></p>
|
<p class="caption"><span class="caption-text">Programming Guide</span></p>
|
||||||
<ul>
|
<ul>
|
||||||
@@ -168,8 +148,6 @@
|
|||||||
|
|
||||||
<li><a href="../../../index.html" class="icon icon-home"></a> »</li>
|
<li><a href="../../../index.html" class="icon icon-home"></a> »</li>
|
||||||
|
|
||||||
<li><a href="../index.html">Python API</a> »</li>
|
|
||||||
|
|
||||||
<li>triton.exp</li>
|
<li>triton.exp</li>
|
||||||
|
|
||||||
|
|
||||||
@@ -203,10 +181,6 @@
|
|||||||
|
|
||||||
</div>
|
</div>
|
||||||
<footer>
|
<footer>
|
||||||
<div class="rst-footer-buttons" role="navigation" aria-label="footer navigation">
|
|
||||||
<a href="triton.log.html" class="btn btn-neutral float-right" title="triton.log" accesskey="n" rel="next">Next <span class="fa fa-arrow-circle-right" aria-hidden="true"></span></a>
|
|
||||||
<a href="triton.where.html" class="btn btn-neutral float-left" title="triton.where" accesskey="p" rel="prev"><span class="fa fa-arrow-circle-left" aria-hidden="true"></span> Previous</a>
|
|
||||||
</div>
|
|
||||||
|
|
||||||
<hr/>
|
<hr/>
|
||||||
|
|
||||||
|
|||||||
@@ -42,9 +42,7 @@
|
|||||||
|
|
||||||
|
|
||||||
<link rel="index" title="Index" href="../../../genindex.html" />
|
<link rel="index" title="Index" href="../../../genindex.html" />
|
||||||
<link rel="search" title="Search" href="../../../search.html" />
|
<link rel="search" title="Search" href="../../../search.html" />
|
||||||
<link rel="next" title="triton.store" href="triton.store.html" />
|
|
||||||
<link rel="prev" title="triton.dot" href="triton.dot.html" />
|
|
||||||
</head>
|
</head>
|
||||||
|
|
||||||
<body class="wy-body-for-nav">
|
<body class="wy-body-for-nav">
|
||||||
@@ -95,26 +93,8 @@
|
|||||||
<li class="toctree-l1"><a class="reference internal" href="../../../getting-started/tutorials/index.html">Tutorials</a></li>
|
<li class="toctree-l1"><a class="reference internal" href="../../../getting-started/tutorials/index.html">Tutorials</a></li>
|
||||||
</ul>
|
</ul>
|
||||||
<p class="caption"><span class="caption-text">Language Reference</span></p>
|
<p class="caption"><span class="caption-text">Language Reference</span></p>
|
||||||
<ul class="current">
|
<ul>
|
||||||
<li class="toctree-l1 current"><a class="reference internal" href="../index.html">Python API</a><ul class="current">
|
<li class="toctree-l1"><a class="reference internal" href="../index.html">Python API</a></li>
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#programming-model">Programming Model</a></li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#creation-ops">Creation Ops</a></li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#shape-manipulation-ops">Shape Manipulation Ops</a></li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#linear-algebra-ops">Linear Algebra Ops</a></li>
|
|
||||||
<li class="toctree-l2 current"><a class="reference internal" href="../index.html#memory-ops">Memory Ops</a><ul class="current">
|
|
||||||
<li class="toctree-l3 current"><a class="current reference internal" href="#">triton.load</a></li>
|
|
||||||
<li class="toctree-l3"><a class="reference internal" href="triton.store.html">triton.store</a></li>
|
|
||||||
<li class="toctree-l3"><a class="reference internal" href="triton.atomic_cas.html">triton.atomic_cas</a></li>
|
|
||||||
<li class="toctree-l3"><a class="reference internal" href="triton.atomic_xchg.html">triton.atomic_xchg</a></li>
|
|
||||||
</ul>
|
|
||||||
</li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#indexing-ops">Indexing Ops</a></li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#math-ops">Math Ops</a></li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#reduction-ops">Reduction Ops</a></li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#comparison-ops">Comparison ops</a></li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#compiler-hint-ops">Compiler Hint Ops</a></li>
|
|
||||||
</ul>
|
|
||||||
</li>
|
|
||||||
</ul>
|
</ul>
|
||||||
<p class="caption"><span class="caption-text">Programming Guide</span></p>
|
<p class="caption"><span class="caption-text">Programming Guide</span></p>
|
||||||
<ul>
|
<ul>
|
||||||
@@ -168,8 +148,6 @@
|
|||||||
|
|
||||||
<li><a href="../../../index.html" class="icon icon-home"></a> »</li>
|
<li><a href="../../../index.html" class="icon icon-home"></a> »</li>
|
||||||
|
|
||||||
<li><a href="../index.html">Python API</a> »</li>
|
|
||||||
|
|
||||||
<li>triton.load</li>
|
<li>triton.load</li>
|
||||||
|
|
||||||
|
|
||||||
@@ -214,10 +192,6 @@
|
|||||||
|
|
||||||
</div>
|
</div>
|
||||||
<footer>
|
<footer>
|
||||||
<div class="rst-footer-buttons" role="navigation" aria-label="footer navigation">
|
|
||||||
<a href="triton.store.html" class="btn btn-neutral float-right" title="triton.store" accesskey="n" rel="next">Next <span class="fa fa-arrow-circle-right" aria-hidden="true"></span></a>
|
|
||||||
<a href="triton.dot.html" class="btn btn-neutral float-left" title="triton.dot" accesskey="p" rel="prev"><span class="fa fa-arrow-circle-left" aria-hidden="true"></span> Previous</a>
|
|
||||||
</div>
|
|
||||||
|
|
||||||
<hr/>
|
<hr/>
|
||||||
|
|
||||||
|
|||||||
@@ -42,9 +42,7 @@
|
|||||||
|
|
||||||
|
|
||||||
<link rel="index" title="Index" href="../../../genindex.html" />
|
<link rel="index" title="Index" href="../../../genindex.html" />
|
||||||
<link rel="search" title="Search" href="../../../search.html" />
|
<link rel="search" title="Search" href="../../../search.html" />
|
||||||
<link rel="next" title="triton.sigmoid" href="triton.sigmoid.html" />
|
|
||||||
<link rel="prev" title="triton.exp" href="triton.exp.html" />
|
|
||||||
</head>
|
</head>
|
||||||
|
|
||||||
<body class="wy-body-for-nav">
|
<body class="wy-body-for-nav">
|
||||||
@@ -95,26 +93,8 @@
|
|||||||
<li class="toctree-l1"><a class="reference internal" href="../../../getting-started/tutorials/index.html">Tutorials</a></li>
|
<li class="toctree-l1"><a class="reference internal" href="../../../getting-started/tutorials/index.html">Tutorials</a></li>
|
||||||
</ul>
|
</ul>
|
||||||
<p class="caption"><span class="caption-text">Language Reference</span></p>
|
<p class="caption"><span class="caption-text">Language Reference</span></p>
|
||||||
<ul class="current">
|
<ul>
|
||||||
<li class="toctree-l1 current"><a class="reference internal" href="../index.html">Python API</a><ul class="current">
|
<li class="toctree-l1"><a class="reference internal" href="../index.html">Python API</a></li>
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#programming-model">Programming Model</a></li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#creation-ops">Creation Ops</a></li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#shape-manipulation-ops">Shape Manipulation Ops</a></li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#linear-algebra-ops">Linear Algebra Ops</a></li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#memory-ops">Memory Ops</a></li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#indexing-ops">Indexing Ops</a></li>
|
|
||||||
<li class="toctree-l2 current"><a class="reference internal" href="../index.html#math-ops">Math Ops</a><ul class="current">
|
|
||||||
<li class="toctree-l3"><a class="reference internal" href="triton.exp.html">triton.exp</a></li>
|
|
||||||
<li class="toctree-l3 current"><a class="current reference internal" href="#">triton.log</a></li>
|
|
||||||
<li class="toctree-l3"><a class="reference internal" href="triton.sigmoid.html">triton.sigmoid</a></li>
|
|
||||||
<li class="toctree-l3"><a class="reference internal" href="triton.softmax.html">triton.softmax</a></li>
|
|
||||||
</ul>
|
|
||||||
</li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#reduction-ops">Reduction Ops</a></li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#comparison-ops">Comparison ops</a></li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#compiler-hint-ops">Compiler Hint Ops</a></li>
|
|
||||||
</ul>
|
|
||||||
</li>
|
|
||||||
</ul>
|
</ul>
|
||||||
<p class="caption"><span class="caption-text">Programming Guide</span></p>
|
<p class="caption"><span class="caption-text">Programming Guide</span></p>
|
||||||
<ul>
|
<ul>
|
||||||
@@ -168,8 +148,6 @@
|
|||||||
|
|
||||||
<li><a href="../../../index.html" class="icon icon-home"></a> »</li>
|
<li><a href="../../../index.html" class="icon icon-home"></a> »</li>
|
||||||
|
|
||||||
<li><a href="../index.html">Python API</a> »</li>
|
|
||||||
|
|
||||||
<li>triton.log</li>
|
<li>triton.log</li>
|
||||||
|
|
||||||
|
|
||||||
@@ -203,10 +181,6 @@
|
|||||||
|
|
||||||
</div>
|
</div>
|
||||||
<footer>
|
<footer>
|
||||||
<div class="rst-footer-buttons" role="navigation" aria-label="footer navigation">
|
|
||||||
<a href="triton.sigmoid.html" class="btn btn-neutral float-right" title="triton.sigmoid" accesskey="n" rel="next">Next <span class="fa fa-arrow-circle-right" aria-hidden="true"></span></a>
|
|
||||||
<a href="triton.exp.html" class="btn btn-neutral float-left" title="triton.exp" accesskey="p" rel="prev"><span class="fa fa-arrow-circle-left" aria-hidden="true"></span> Previous</a>
|
|
||||||
</div>
|
|
||||||
|
|
||||||
<hr/>
|
<hr/>
|
||||||
|
|
||||||
|
|||||||
@@ -42,9 +42,7 @@
|
|||||||
|
|
||||||
|
|
||||||
<link rel="index" title="Index" href="../../../genindex.html" />
|
<link rel="index" title="Index" href="../../../genindex.html" />
|
||||||
<link rel="search" title="Search" href="../../../search.html" />
|
<link rel="search" title="Search" href="../../../search.html" />
|
||||||
<link rel="next" title="triton.min" href="triton.min.html" />
|
|
||||||
<link rel="prev" title="triton.softmax" href="triton.softmax.html" />
|
|
||||||
</head>
|
</head>
|
||||||
|
|
||||||
<body class="wy-body-for-nav">
|
<body class="wy-body-for-nav">
|
||||||
@@ -95,25 +93,8 @@
|
|||||||
<li class="toctree-l1"><a class="reference internal" href="../../../getting-started/tutorials/index.html">Tutorials</a></li>
|
<li class="toctree-l1"><a class="reference internal" href="../../../getting-started/tutorials/index.html">Tutorials</a></li>
|
||||||
</ul>
|
</ul>
|
||||||
<p class="caption"><span class="caption-text">Language Reference</span></p>
|
<p class="caption"><span class="caption-text">Language Reference</span></p>
|
||||||
<ul class="current">
|
<ul>
|
||||||
<li class="toctree-l1 current"><a class="reference internal" href="../index.html">Python API</a><ul class="current">
|
<li class="toctree-l1"><a class="reference internal" href="../index.html">Python API</a></li>
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#programming-model">Programming Model</a></li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#creation-ops">Creation Ops</a></li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#shape-manipulation-ops">Shape Manipulation Ops</a></li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#linear-algebra-ops">Linear Algebra Ops</a></li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#memory-ops">Memory Ops</a></li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#indexing-ops">Indexing Ops</a></li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#math-ops">Math Ops</a></li>
|
|
||||||
<li class="toctree-l2 current"><a class="reference internal" href="../index.html#reduction-ops">Reduction Ops</a><ul class="current">
|
|
||||||
<li class="toctree-l3 current"><a class="current reference internal" href="#">triton.max</a></li>
|
|
||||||
<li class="toctree-l3"><a class="reference internal" href="triton.min.html">triton.min</a></li>
|
|
||||||
<li class="toctree-l3"><a class="reference internal" href="triton.sum.html">triton.sum</a></li>
|
|
||||||
</ul>
|
|
||||||
</li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#comparison-ops">Comparison ops</a></li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#compiler-hint-ops">Compiler Hint Ops</a></li>
|
|
||||||
</ul>
|
|
||||||
</li>
|
|
||||||
</ul>
|
</ul>
|
||||||
<p class="caption"><span class="caption-text">Programming Guide</span></p>
|
<p class="caption"><span class="caption-text">Programming Guide</span></p>
|
||||||
<ul>
|
<ul>
|
||||||
@@ -167,8 +148,6 @@
|
|||||||
|
|
||||||
<li><a href="../../../index.html" class="icon icon-home"></a> »</li>
|
<li><a href="../../../index.html" class="icon icon-home"></a> »</li>
|
||||||
|
|
||||||
<li><a href="../index.html">Python API</a> »</li>
|
|
||||||
|
|
||||||
<li>triton.max</li>
|
<li>triton.max</li>
|
||||||
|
|
||||||
|
|
||||||
@@ -202,10 +181,6 @@
|
|||||||
|
|
||||||
</div>
|
</div>
|
||||||
<footer>
|
<footer>
|
||||||
<div class="rst-footer-buttons" role="navigation" aria-label="footer navigation">
|
|
||||||
<a href="triton.min.html" class="btn btn-neutral float-right" title="triton.min" accesskey="n" rel="next">Next <span class="fa fa-arrow-circle-right" aria-hidden="true"></span></a>
|
|
||||||
<a href="triton.softmax.html" class="btn btn-neutral float-left" title="triton.softmax" accesskey="p" rel="prev"><span class="fa fa-arrow-circle-left" aria-hidden="true"></span> Previous</a>
|
|
||||||
</div>
|
|
||||||
|
|
||||||
<hr/>
|
<hr/>
|
||||||
|
|
||||||
|
|||||||
@@ -42,9 +42,7 @@
|
|||||||
|
|
||||||
|
|
||||||
<link rel="index" title="Index" href="../../../genindex.html" />
|
<link rel="index" title="Index" href="../../../genindex.html" />
|
||||||
<link rel="search" title="Search" href="../../../search.html" />
|
<link rel="search" title="Search" href="../../../search.html" />
|
||||||
<link rel="next" title="triton.multiple_of" href="triton.multiple_of.html" />
|
|
||||||
<link rel="prev" title="triton.minimum" href="triton.minimum.html" />
|
|
||||||
</head>
|
</head>
|
||||||
|
|
||||||
<body class="wy-body-for-nav">
|
<body class="wy-body-for-nav">
|
||||||
@@ -95,24 +93,8 @@
|
|||||||
<li class="toctree-l1"><a class="reference internal" href="../../../getting-started/tutorials/index.html">Tutorials</a></li>
|
<li class="toctree-l1"><a class="reference internal" href="../../../getting-started/tutorials/index.html">Tutorials</a></li>
|
||||||
</ul>
|
</ul>
|
||||||
<p class="caption"><span class="caption-text">Language Reference</span></p>
|
<p class="caption"><span class="caption-text">Language Reference</span></p>
|
||||||
<ul class="current">
|
<ul>
|
||||||
<li class="toctree-l1 current"><a class="reference internal" href="../index.html">Python API</a><ul class="current">
|
<li class="toctree-l1"><a class="reference internal" href="../index.html">Python API</a></li>
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#programming-model">Programming Model</a></li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#creation-ops">Creation Ops</a></li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#shape-manipulation-ops">Shape Manipulation Ops</a></li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#linear-algebra-ops">Linear Algebra Ops</a></li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#memory-ops">Memory Ops</a></li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#indexing-ops">Indexing Ops</a></li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#math-ops">Math Ops</a></li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#reduction-ops">Reduction Ops</a></li>
|
|
||||||
<li class="toctree-l2 current"><a class="reference internal" href="../index.html#comparison-ops">Comparison ops</a><ul class="current">
|
|
||||||
<li class="toctree-l3"><a class="reference internal" href="triton.minimum.html">triton.minimum</a></li>
|
|
||||||
<li class="toctree-l3 current"><a class="current reference internal" href="#">triton.maximum</a></li>
|
|
||||||
</ul>
|
|
||||||
</li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#compiler-hint-ops">Compiler Hint Ops</a></li>
|
|
||||||
</ul>
|
|
||||||
</li>
|
|
||||||
</ul>
|
</ul>
|
||||||
<p class="caption"><span class="caption-text">Programming Guide</span></p>
|
<p class="caption"><span class="caption-text">Programming Guide</span></p>
|
||||||
<ul>
|
<ul>
|
||||||
@@ -166,8 +148,6 @@
|
|||||||
|
|
||||||
<li><a href="../../../index.html" class="icon icon-home"></a> »</li>
|
<li><a href="../../../index.html" class="icon icon-home"></a> »</li>
|
||||||
|
|
||||||
<li><a href="../index.html">Python API</a> »</li>
|
|
||||||
|
|
||||||
<li>triton.maximum</li>
|
<li>triton.maximum</li>
|
||||||
|
|
||||||
|
|
||||||
@@ -201,10 +181,6 @@
|
|||||||
|
|
||||||
</div>
|
</div>
|
||||||
<footer>
|
<footer>
|
||||||
<div class="rst-footer-buttons" role="navigation" aria-label="footer navigation">
|
|
||||||
<a href="triton.multiple_of.html" class="btn btn-neutral float-right" title="triton.multiple_of" accesskey="n" rel="next">Next <span class="fa fa-arrow-circle-right" aria-hidden="true"></span></a>
|
|
||||||
<a href="triton.minimum.html" class="btn btn-neutral float-left" title="triton.minimum" accesskey="p" rel="prev"><span class="fa fa-arrow-circle-left" aria-hidden="true"></span> Previous</a>
|
|
||||||
</div>
|
|
||||||
|
|
||||||
<hr/>
|
<hr/>
|
||||||
|
|
||||||
|
|||||||
@@ -42,9 +42,7 @@
|
|||||||
|
|
||||||
|
|
||||||
<link rel="index" title="Index" href="../../../genindex.html" />
|
<link rel="index" title="Index" href="../../../genindex.html" />
|
||||||
<link rel="search" title="Search" href="../../../search.html" />
|
<link rel="search" title="Search" href="../../../search.html" />
|
||||||
<link rel="next" title="triton.sum" href="triton.sum.html" />
|
|
||||||
<link rel="prev" title="triton.max" href="triton.max.html" />
|
|
||||||
</head>
|
</head>
|
||||||
|
|
||||||
<body class="wy-body-for-nav">
|
<body class="wy-body-for-nav">
|
||||||
@@ -95,25 +93,8 @@
|
|||||||
<li class="toctree-l1"><a class="reference internal" href="../../../getting-started/tutorials/index.html">Tutorials</a></li>
|
<li class="toctree-l1"><a class="reference internal" href="../../../getting-started/tutorials/index.html">Tutorials</a></li>
|
||||||
</ul>
|
</ul>
|
||||||
<p class="caption"><span class="caption-text">Language Reference</span></p>
|
<p class="caption"><span class="caption-text">Language Reference</span></p>
|
||||||
<ul class="current">
|
<ul>
|
||||||
<li class="toctree-l1 current"><a class="reference internal" href="../index.html">Python API</a><ul class="current">
|
<li class="toctree-l1"><a class="reference internal" href="../index.html">Python API</a></li>
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#programming-model">Programming Model</a></li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#creation-ops">Creation Ops</a></li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#shape-manipulation-ops">Shape Manipulation Ops</a></li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#linear-algebra-ops">Linear Algebra Ops</a></li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#memory-ops">Memory Ops</a></li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#indexing-ops">Indexing Ops</a></li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#math-ops">Math Ops</a></li>
|
|
||||||
<li class="toctree-l2 current"><a class="reference internal" href="../index.html#reduction-ops">Reduction Ops</a><ul class="current">
|
|
||||||
<li class="toctree-l3"><a class="reference internal" href="triton.max.html">triton.max</a></li>
|
|
||||||
<li class="toctree-l3 current"><a class="current reference internal" href="#">triton.min</a></li>
|
|
||||||
<li class="toctree-l3"><a class="reference internal" href="triton.sum.html">triton.sum</a></li>
|
|
||||||
</ul>
|
|
||||||
</li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#comparison-ops">Comparison ops</a></li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#compiler-hint-ops">Compiler Hint Ops</a></li>
|
|
||||||
</ul>
|
|
||||||
</li>
|
|
||||||
</ul>
|
</ul>
|
||||||
<p class="caption"><span class="caption-text">Programming Guide</span></p>
|
<p class="caption"><span class="caption-text">Programming Guide</span></p>
|
||||||
<ul>
|
<ul>
|
||||||
@@ -167,8 +148,6 @@
|
|||||||
|
|
||||||
<li><a href="../../../index.html" class="icon icon-home"></a> »</li>
|
<li><a href="../../../index.html" class="icon icon-home"></a> »</li>
|
||||||
|
|
||||||
<li><a href="../index.html">Python API</a> »</li>
|
|
||||||
|
|
||||||
<li>triton.min</li>
|
<li>triton.min</li>
|
||||||
|
|
||||||
|
|
||||||
@@ -202,10 +181,6 @@
|
|||||||
|
|
||||||
</div>
|
</div>
|
||||||
<footer>
|
<footer>
|
||||||
<div class="rst-footer-buttons" role="navigation" aria-label="footer navigation">
|
|
||||||
<a href="triton.sum.html" class="btn btn-neutral float-right" title="triton.sum" accesskey="n" rel="next">Next <span class="fa fa-arrow-circle-right" aria-hidden="true"></span></a>
|
|
||||||
<a href="triton.max.html" class="btn btn-neutral float-left" title="triton.max" accesskey="p" rel="prev"><span class="fa fa-arrow-circle-left" aria-hidden="true"></span> Previous</a>
|
|
||||||
</div>
|
|
||||||
|
|
||||||
<hr/>
|
<hr/>
|
||||||
|
|
||||||
|
|||||||
@@ -42,9 +42,7 @@
|
|||||||
|
|
||||||
|
|
||||||
<link rel="index" title="Index" href="../../../genindex.html" />
|
<link rel="index" title="Index" href="../../../genindex.html" />
|
||||||
<link rel="search" title="Search" href="../../../search.html" />
|
<link rel="search" title="Search" href="../../../search.html" />
|
||||||
<link rel="next" title="triton.maximum" href="triton.maximum.html" />
|
|
||||||
<link rel="prev" title="triton.sum" href="triton.sum.html" />
|
|
||||||
</head>
|
</head>
|
||||||
|
|
||||||
<body class="wy-body-for-nav">
|
<body class="wy-body-for-nav">
|
||||||
@@ -95,24 +93,8 @@
|
|||||||
<li class="toctree-l1"><a class="reference internal" href="../../../getting-started/tutorials/index.html">Tutorials</a></li>
|
<li class="toctree-l1"><a class="reference internal" href="../../../getting-started/tutorials/index.html">Tutorials</a></li>
|
||||||
</ul>
|
</ul>
|
||||||
<p class="caption"><span class="caption-text">Language Reference</span></p>
|
<p class="caption"><span class="caption-text">Language Reference</span></p>
|
||||||
<ul class="current">
|
<ul>
|
||||||
<li class="toctree-l1 current"><a class="reference internal" href="../index.html">Python API</a><ul class="current">
|
<li class="toctree-l1"><a class="reference internal" href="../index.html">Python API</a></li>
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#programming-model">Programming Model</a></li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#creation-ops">Creation Ops</a></li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#shape-manipulation-ops">Shape Manipulation Ops</a></li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#linear-algebra-ops">Linear Algebra Ops</a></li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#memory-ops">Memory Ops</a></li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#indexing-ops">Indexing Ops</a></li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#math-ops">Math Ops</a></li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#reduction-ops">Reduction Ops</a></li>
|
|
||||||
<li class="toctree-l2 current"><a class="reference internal" href="../index.html#comparison-ops">Comparison ops</a><ul class="current">
|
|
||||||
<li class="toctree-l3 current"><a class="current reference internal" href="#">triton.minimum</a></li>
|
|
||||||
<li class="toctree-l3"><a class="reference internal" href="triton.maximum.html">triton.maximum</a></li>
|
|
||||||
</ul>
|
|
||||||
</li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#compiler-hint-ops">Compiler Hint Ops</a></li>
|
|
||||||
</ul>
|
|
||||||
</li>
|
|
||||||
</ul>
|
</ul>
|
||||||
<p class="caption"><span class="caption-text">Programming Guide</span></p>
|
<p class="caption"><span class="caption-text">Programming Guide</span></p>
|
||||||
<ul>
|
<ul>
|
||||||
@@ -166,8 +148,6 @@
|
|||||||
|
|
||||||
<li><a href="../../../index.html" class="icon icon-home"></a> »</li>
|
<li><a href="../../../index.html" class="icon icon-home"></a> »</li>
|
||||||
|
|
||||||
<li><a href="../index.html">Python API</a> »</li>
|
|
||||||
|
|
||||||
<li>triton.minimum</li>
|
<li>triton.minimum</li>
|
||||||
|
|
||||||
|
|
||||||
@@ -201,10 +181,6 @@
|
|||||||
|
|
||||||
</div>
|
</div>
|
||||||
<footer>
|
<footer>
|
||||||
<div class="rst-footer-buttons" role="navigation" aria-label="footer navigation">
|
|
||||||
<a href="triton.maximum.html" class="btn btn-neutral float-right" title="triton.maximum" accesskey="n" rel="next">Next <span class="fa fa-arrow-circle-right" aria-hidden="true"></span></a>
|
|
||||||
<a href="triton.sum.html" class="btn btn-neutral float-left" title="triton.sum" accesskey="p" rel="prev"><span class="fa fa-arrow-circle-left" aria-hidden="true"></span> Previous</a>
|
|
||||||
</div>
|
|
||||||
|
|
||||||
<hr/>
|
<hr/>
|
||||||
|
|
||||||
|
|||||||
@@ -42,9 +42,7 @@
|
|||||||
|
|
||||||
|
|
||||||
<link rel="index" title="Index" href="../../../genindex.html" />
|
<link rel="index" title="Index" href="../../../genindex.html" />
|
||||||
<link rel="search" title="Search" href="../../../search.html" />
|
<link rel="search" title="Search" href="../../../search.html" />
|
||||||
<link rel="next" title="Introduction" href="../../../programming-guide/chapter-1/introduction.html" />
|
|
||||||
<link rel="prev" title="triton.maximum" href="triton.maximum.html" />
|
|
||||||
</head>
|
</head>
|
||||||
|
|
||||||
<body class="wy-body-for-nav">
|
<body class="wy-body-for-nav">
|
||||||
@@ -95,23 +93,8 @@
|
|||||||
<li class="toctree-l1"><a class="reference internal" href="../../../getting-started/tutorials/index.html">Tutorials</a></li>
|
<li class="toctree-l1"><a class="reference internal" href="../../../getting-started/tutorials/index.html">Tutorials</a></li>
|
||||||
</ul>
|
</ul>
|
||||||
<p class="caption"><span class="caption-text">Language Reference</span></p>
|
<p class="caption"><span class="caption-text">Language Reference</span></p>
|
||||||
<ul class="current">
|
<ul>
|
||||||
<li class="toctree-l1 current"><a class="reference internal" href="../index.html">Python API</a><ul class="current">
|
<li class="toctree-l1"><a class="reference internal" href="../index.html">Python API</a></li>
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#programming-model">Programming Model</a></li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#creation-ops">Creation Ops</a></li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#shape-manipulation-ops">Shape Manipulation Ops</a></li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#linear-algebra-ops">Linear Algebra Ops</a></li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#memory-ops">Memory Ops</a></li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#indexing-ops">Indexing Ops</a></li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#math-ops">Math Ops</a></li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#reduction-ops">Reduction Ops</a></li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#comparison-ops">Comparison ops</a></li>
|
|
||||||
<li class="toctree-l2 current"><a class="reference internal" href="../index.html#compiler-hint-ops">Compiler Hint Ops</a><ul class="current">
|
|
||||||
<li class="toctree-l3 current"><a class="current reference internal" href="#">triton.multiple_of</a></li>
|
|
||||||
</ul>
|
|
||||||
</li>
|
|
||||||
</ul>
|
|
||||||
</li>
|
|
||||||
</ul>
|
</ul>
|
||||||
<p class="caption"><span class="caption-text">Programming Guide</span></p>
|
<p class="caption"><span class="caption-text">Programming Guide</span></p>
|
||||||
<ul>
|
<ul>
|
||||||
@@ -165,8 +148,6 @@
|
|||||||
|
|
||||||
<li><a href="../../../index.html" class="icon icon-home"></a> »</li>
|
<li><a href="../../../index.html" class="icon icon-home"></a> »</li>
|
||||||
|
|
||||||
<li><a href="../index.html">Python API</a> »</li>
|
|
||||||
|
|
||||||
<li>triton.multiple_of</li>
|
<li>triton.multiple_of</li>
|
||||||
|
|
||||||
|
|
||||||
@@ -200,10 +181,6 @@
|
|||||||
|
|
||||||
</div>
|
</div>
|
||||||
<footer>
|
<footer>
|
||||||
<div class="rst-footer-buttons" role="navigation" aria-label="footer navigation">
|
|
||||||
<a href="../../../programming-guide/chapter-1/introduction.html" class="btn btn-neutral float-right" title="Introduction" accesskey="n" rel="next">Next <span class="fa fa-arrow-circle-right" aria-hidden="true"></span></a>
|
|
||||||
<a href="triton.maximum.html" class="btn btn-neutral float-left" title="triton.maximum" accesskey="p" rel="prev"><span class="fa fa-arrow-circle-left" aria-hidden="true"></span> Previous</a>
|
|
||||||
</div>
|
|
||||||
|
|
||||||
<hr/>
|
<hr/>
|
||||||
|
|
||||||
|
|||||||
@@ -42,9 +42,7 @@
|
|||||||
|
|
||||||
|
|
||||||
<link rel="index" title="Index" href="../../../genindex.html" />
|
<link rel="index" title="Index" href="../../../genindex.html" />
|
||||||
<link rel="search" title="Search" href="../../../search.html" />
|
<link rel="search" title="Search" href="../../../search.html" />
|
||||||
<link rel="next" title="triton.arange" href="triton.arange.html" />
|
|
||||||
<link rel="prev" title="triton.program_id" href="triton.program_id.html" />
|
|
||||||
</head>
|
</head>
|
||||||
|
|
||||||
<body class="wy-body-for-nav">
|
<body class="wy-body-for-nav">
|
||||||
@@ -95,24 +93,8 @@
|
|||||||
<li class="toctree-l1"><a class="reference internal" href="../../../getting-started/tutorials/index.html">Tutorials</a></li>
|
<li class="toctree-l1"><a class="reference internal" href="../../../getting-started/tutorials/index.html">Tutorials</a></li>
|
||||||
</ul>
|
</ul>
|
||||||
<p class="caption"><span class="caption-text">Language Reference</span></p>
|
<p class="caption"><span class="caption-text">Language Reference</span></p>
|
||||||
<ul class="current">
|
<ul>
|
||||||
<li class="toctree-l1 current"><a class="reference internal" href="../index.html">Python API</a><ul class="current">
|
<li class="toctree-l1"><a class="reference internal" href="../index.html">Python API</a></li>
|
||||||
<li class="toctree-l2 current"><a class="reference internal" href="../index.html#programming-model">Programming Model</a><ul class="current">
|
|
||||||
<li class="toctree-l3"><a class="reference internal" href="triton.program_id.html">triton.program_id</a></li>
|
|
||||||
<li class="toctree-l3 current"><a class="current reference internal" href="#">triton.num_programs</a></li>
|
|
||||||
</ul>
|
|
||||||
</li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#creation-ops">Creation Ops</a></li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#shape-manipulation-ops">Shape Manipulation Ops</a></li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#linear-algebra-ops">Linear Algebra Ops</a></li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#memory-ops">Memory Ops</a></li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#indexing-ops">Indexing Ops</a></li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#math-ops">Math Ops</a></li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#reduction-ops">Reduction Ops</a></li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#comparison-ops">Comparison ops</a></li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#compiler-hint-ops">Compiler Hint Ops</a></li>
|
|
||||||
</ul>
|
|
||||||
</li>
|
|
||||||
</ul>
|
</ul>
|
||||||
<p class="caption"><span class="caption-text">Programming Guide</span></p>
|
<p class="caption"><span class="caption-text">Programming Guide</span></p>
|
||||||
<ul>
|
<ul>
|
||||||
@@ -166,8 +148,6 @@
|
|||||||
|
|
||||||
<li><a href="../../../index.html" class="icon icon-home"></a> »</li>
|
<li><a href="../../../index.html" class="icon icon-home"></a> »</li>
|
||||||
|
|
||||||
<li><a href="../index.html">Python API</a> »</li>
|
|
||||||
|
|
||||||
<li>triton.num_programs</li>
|
<li>triton.num_programs</li>
|
||||||
|
|
||||||
|
|
||||||
@@ -210,10 +190,6 @@
|
|||||||
|
|
||||||
</div>
|
</div>
|
||||||
<footer>
|
<footer>
|
||||||
<div class="rst-footer-buttons" role="navigation" aria-label="footer navigation">
|
|
||||||
<a href="triton.arange.html" class="btn btn-neutral float-right" title="triton.arange" accesskey="n" rel="next">Next <span class="fa fa-arrow-circle-right" aria-hidden="true"></span></a>
|
|
||||||
<a href="triton.program_id.html" class="btn btn-neutral float-left" title="triton.program_id" accesskey="p" rel="prev"><span class="fa fa-arrow-circle-left" aria-hidden="true"></span> Previous</a>
|
|
||||||
</div>
|
|
||||||
|
|
||||||
<hr/>
|
<hr/>
|
||||||
|
|
||||||
|
|||||||
@@ -42,9 +42,7 @@
|
|||||||
|
|
||||||
|
|
||||||
<link rel="index" title="Index" href="../../../genindex.html" />
|
<link rel="index" title="Index" href="../../../genindex.html" />
|
||||||
<link rel="search" title="Search" href="../../../search.html" />
|
<link rel="search" title="Search" href="../../../search.html" />
|
||||||
<link rel="next" title="triton.num_programs" href="triton.num_programs.html" />
|
|
||||||
<link rel="prev" title="Python API" href="../index.html" />
|
|
||||||
</head>
|
</head>
|
||||||
|
|
||||||
<body class="wy-body-for-nav">
|
<body class="wy-body-for-nav">
|
||||||
@@ -95,24 +93,8 @@
|
|||||||
<li class="toctree-l1"><a class="reference internal" href="../../../getting-started/tutorials/index.html">Tutorials</a></li>
|
<li class="toctree-l1"><a class="reference internal" href="../../../getting-started/tutorials/index.html">Tutorials</a></li>
|
||||||
</ul>
|
</ul>
|
||||||
<p class="caption"><span class="caption-text">Language Reference</span></p>
|
<p class="caption"><span class="caption-text">Language Reference</span></p>
|
||||||
<ul class="current">
|
<ul>
|
||||||
<li class="toctree-l1 current"><a class="reference internal" href="../index.html">Python API</a><ul class="current">
|
<li class="toctree-l1"><a class="reference internal" href="../index.html">Python API</a></li>
|
||||||
<li class="toctree-l2 current"><a class="reference internal" href="../index.html#programming-model">Programming Model</a><ul class="current">
|
|
||||||
<li class="toctree-l3 current"><a class="current reference internal" href="#">triton.program_id</a></li>
|
|
||||||
<li class="toctree-l3"><a class="reference internal" href="triton.num_programs.html">triton.num_programs</a></li>
|
|
||||||
</ul>
|
|
||||||
</li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#creation-ops">Creation Ops</a></li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#shape-manipulation-ops">Shape Manipulation Ops</a></li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#linear-algebra-ops">Linear Algebra Ops</a></li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#memory-ops">Memory Ops</a></li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#indexing-ops">Indexing Ops</a></li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#math-ops">Math Ops</a></li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#reduction-ops">Reduction Ops</a></li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#comparison-ops">Comparison ops</a></li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#compiler-hint-ops">Compiler Hint Ops</a></li>
|
|
||||||
</ul>
|
|
||||||
</li>
|
|
||||||
</ul>
|
</ul>
|
||||||
<p class="caption"><span class="caption-text">Programming Guide</span></p>
|
<p class="caption"><span class="caption-text">Programming Guide</span></p>
|
||||||
<ul>
|
<ul>
|
||||||
@@ -166,8 +148,6 @@
|
|||||||
|
|
||||||
<li><a href="../../../index.html" class="icon icon-home"></a> »</li>
|
<li><a href="../../../index.html" class="icon icon-home"></a> »</li>
|
||||||
|
|
||||||
<li><a href="../index.html">Python API</a> »</li>
|
|
||||||
|
|
||||||
<li>triton.program_id</li>
|
<li>triton.program_id</li>
|
||||||
|
|
||||||
|
|
||||||
@@ -211,10 +191,6 @@ Triton uses an SPMD model in which different @triton.jit functions run in pa
|
|||||||
|
|
||||||
</div>
|
</div>
|
||||||
<footer>
|
<footer>
|
||||||
<div class="rst-footer-buttons" role="navigation" aria-label="footer navigation">
|
|
||||||
<a href="triton.num_programs.html" class="btn btn-neutral float-right" title="triton.num_programs" accesskey="n" rel="next">Next <span class="fa fa-arrow-circle-right" aria-hidden="true"></span></a>
|
|
||||||
<a href="../index.html" class="btn btn-neutral float-left" title="Python API" accesskey="p" rel="prev"><span class="fa fa-arrow-circle-left" aria-hidden="true"></span> Previous</a>
|
|
||||||
</div>
|
|
||||||
|
|
||||||
<hr/>
|
<hr/>
|
||||||
|
|
||||||
|
|||||||
@@ -42,9 +42,7 @@
|
|||||||
|
|
||||||
|
|
||||||
<link rel="index" title="Index" href="../../../genindex.html" />
|
<link rel="index" title="Index" href="../../../genindex.html" />
|
||||||
<link rel="search" title="Search" href="../../../search.html" />
|
<link rel="search" title="Search" href="../../../search.html" />
|
||||||
<link rel="next" title="triton.dot" href="triton.dot.html" />
|
|
||||||
<link rel="prev" title="triton.reshape" href="triton.reshape.html" />
|
|
||||||
</head>
|
</head>
|
||||||
|
|
||||||
<body class="wy-body-for-nav">
|
<body class="wy-body-for-nav">
|
||||||
@@ -95,25 +93,8 @@
|
|||||||
<li class="toctree-l1"><a class="reference internal" href="../../../getting-started/tutorials/index.html">Tutorials</a></li>
|
<li class="toctree-l1"><a class="reference internal" href="../../../getting-started/tutorials/index.html">Tutorials</a></li>
|
||||||
</ul>
|
</ul>
|
||||||
<p class="caption"><span class="caption-text">Language Reference</span></p>
|
<p class="caption"><span class="caption-text">Language Reference</span></p>
|
||||||
<ul class="current">
|
<ul>
|
||||||
<li class="toctree-l1 current"><a class="reference internal" href="../index.html">Python API</a><ul class="current">
|
<li class="toctree-l1"><a class="reference internal" href="../index.html">Python API</a></li>
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#programming-model">Programming Model</a></li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#creation-ops">Creation Ops</a></li>
|
|
||||||
<li class="toctree-l2 current"><a class="reference internal" href="../index.html#shape-manipulation-ops">Shape Manipulation Ops</a><ul class="current">
|
|
||||||
<li class="toctree-l3"><a class="reference internal" href="triton.broadcast_to.html">triton.broadcast_to</a></li>
|
|
||||||
<li class="toctree-l3"><a class="reference internal" href="triton.reshape.html">triton.reshape</a></li>
|
|
||||||
<li class="toctree-l3 current"><a class="current reference internal" href="#">triton.ravel</a></li>
|
|
||||||
</ul>
|
|
||||||
</li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#linear-algebra-ops">Linear Algebra Ops</a></li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#memory-ops">Memory Ops</a></li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#indexing-ops">Indexing Ops</a></li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#math-ops">Math Ops</a></li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#reduction-ops">Reduction Ops</a></li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#comparison-ops">Comparison ops</a></li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#compiler-hint-ops">Compiler Hint Ops</a></li>
|
|
||||||
</ul>
|
|
||||||
</li>
|
|
||||||
</ul>
|
</ul>
|
||||||
<p class="caption"><span class="caption-text">Programming Guide</span></p>
|
<p class="caption"><span class="caption-text">Programming Guide</span></p>
|
||||||
<ul>
|
<ul>
|
||||||
@@ -167,8 +148,6 @@
|
|||||||
|
|
||||||
<li><a href="../../../index.html" class="icon icon-home"></a> »</li>
|
<li><a href="../../../index.html" class="icon icon-home"></a> »</li>
|
||||||
|
|
||||||
<li><a href="../index.html">Python API</a> »</li>
|
|
||||||
|
|
||||||
<li>triton.ravel</li>
|
<li>triton.ravel</li>
|
||||||
|
|
||||||
|
|
||||||
@@ -202,10 +181,6 @@
|
|||||||
|
|
||||||
</div>
|
</div>
|
||||||
<footer>
|
<footer>
|
||||||
<div class="rst-footer-buttons" role="navigation" aria-label="footer navigation">
|
|
||||||
<a href="triton.dot.html" class="btn btn-neutral float-right" title="triton.dot" accesskey="n" rel="next">Next <span class="fa fa-arrow-circle-right" aria-hidden="true"></span></a>
|
|
||||||
<a href="triton.reshape.html" class="btn btn-neutral float-left" title="triton.reshape" accesskey="p" rel="prev"><span class="fa fa-arrow-circle-left" aria-hidden="true"></span> Previous</a>
|
|
||||||
</div>
|
|
||||||
|
|
||||||
<hr/>
|
<hr/>
|
||||||
|
|
||||||
|
|||||||
@@ -42,9 +42,7 @@
|
|||||||
|
|
||||||
|
|
||||||
<link rel="index" title="Index" href="../../../genindex.html" />
|
<link rel="index" title="Index" href="../../../genindex.html" />
|
||||||
<link rel="search" title="Search" href="../../../search.html" />
|
<link rel="search" title="Search" href="../../../search.html" />
|
||||||
<link rel="next" title="triton.ravel" href="triton.ravel.html" />
|
|
||||||
<link rel="prev" title="triton.broadcast_to" href="triton.broadcast_to.html" />
|
|
||||||
</head>
|
</head>
|
||||||
|
|
||||||
<body class="wy-body-for-nav">
|
<body class="wy-body-for-nav">
|
||||||
@@ -95,25 +93,8 @@
|
|||||||
<li class="toctree-l1"><a class="reference internal" href="../../../getting-started/tutorials/index.html">Tutorials</a></li>
|
<li class="toctree-l1"><a class="reference internal" href="../../../getting-started/tutorials/index.html">Tutorials</a></li>
|
||||||
</ul>
|
</ul>
|
||||||
<p class="caption"><span class="caption-text">Language Reference</span></p>
|
<p class="caption"><span class="caption-text">Language Reference</span></p>
|
||||||
<ul class="current">
|
<ul>
|
||||||
<li class="toctree-l1 current"><a class="reference internal" href="../index.html">Python API</a><ul class="current">
|
<li class="toctree-l1"><a class="reference internal" href="../index.html">Python API</a></li>
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#programming-model">Programming Model</a></li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#creation-ops">Creation Ops</a></li>
|
|
||||||
<li class="toctree-l2 current"><a class="reference internal" href="../index.html#shape-manipulation-ops">Shape Manipulation Ops</a><ul class="current">
|
|
||||||
<li class="toctree-l3"><a class="reference internal" href="triton.broadcast_to.html">triton.broadcast_to</a></li>
|
|
||||||
<li class="toctree-l3 current"><a class="current reference internal" href="#">triton.reshape</a></li>
|
|
||||||
<li class="toctree-l3"><a class="reference internal" href="triton.ravel.html">triton.ravel</a></li>
|
|
||||||
</ul>
|
|
||||||
</li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#linear-algebra-ops">Linear Algebra Ops</a></li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#memory-ops">Memory Ops</a></li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#indexing-ops">Indexing Ops</a></li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#math-ops">Math Ops</a></li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#reduction-ops">Reduction Ops</a></li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#comparison-ops">Comparison ops</a></li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#compiler-hint-ops">Compiler Hint Ops</a></li>
|
|
||||||
</ul>
|
|
||||||
</li>
|
|
||||||
</ul>
|
</ul>
|
||||||
<p class="caption"><span class="caption-text">Programming Guide</span></p>
|
<p class="caption"><span class="caption-text">Programming Guide</span></p>
|
||||||
<ul>
|
<ul>
|
||||||
@@ -167,8 +148,6 @@
|
|||||||
|
|
||||||
<li><a href="../../../index.html" class="icon icon-home"></a> »</li>
|
<li><a href="../../../index.html" class="icon icon-home"></a> »</li>
|
||||||
|
|
||||||
<li><a href="../index.html">Python API</a> »</li>
|
|
||||||
|
|
||||||
<li>triton.reshape</li>
|
<li>triton.reshape</li>
|
||||||
|
|
||||||
|
|
||||||
@@ -205,10 +184,6 @@
|
|||||||
|
|
||||||
</div>
|
</div>
|
||||||
<footer>
|
<footer>
|
||||||
<div class="rst-footer-buttons" role="navigation" aria-label="footer navigation">
|
|
||||||
<a href="triton.ravel.html" class="btn btn-neutral float-right" title="triton.ravel" accesskey="n" rel="next">Next <span class="fa fa-arrow-circle-right" aria-hidden="true"></span></a>
|
|
||||||
<a href="triton.broadcast_to.html" class="btn btn-neutral float-left" title="triton.broadcast_to" accesskey="p" rel="prev"><span class="fa fa-arrow-circle-left" aria-hidden="true"></span> Previous</a>
|
|
||||||
</div>
|
|
||||||
|
|
||||||
<hr/>
|
<hr/>
|
||||||
|
|
||||||
|
|||||||
@@ -42,9 +42,7 @@
|
|||||||
|
|
||||||
|
|
||||||
<link rel="index" title="Index" href="../../../genindex.html" />
|
<link rel="index" title="Index" href="../../../genindex.html" />
|
||||||
<link rel="search" title="Search" href="../../../search.html" />
|
<link rel="search" title="Search" href="../../../search.html" />
|
||||||
<link rel="next" title="triton.softmax" href="triton.softmax.html" />
|
|
||||||
<link rel="prev" title="triton.log" href="triton.log.html" />
|
|
||||||
</head>
|
</head>
|
||||||
|
|
||||||
<body class="wy-body-for-nav">
|
<body class="wy-body-for-nav">
|
||||||
@@ -95,26 +93,8 @@
|
|||||||
<li class="toctree-l1"><a class="reference internal" href="../../../getting-started/tutorials/index.html">Tutorials</a></li>
|
<li class="toctree-l1"><a class="reference internal" href="../../../getting-started/tutorials/index.html">Tutorials</a></li>
|
||||||
</ul>
|
</ul>
|
||||||
<p class="caption"><span class="caption-text">Language Reference</span></p>
|
<p class="caption"><span class="caption-text">Language Reference</span></p>
|
||||||
<ul class="current">
|
<ul>
|
||||||
<li class="toctree-l1 current"><a class="reference internal" href="../index.html">Python API</a><ul class="current">
|
<li class="toctree-l1"><a class="reference internal" href="../index.html">Python API</a></li>
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#programming-model">Programming Model</a></li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#creation-ops">Creation Ops</a></li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#shape-manipulation-ops">Shape Manipulation Ops</a></li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#linear-algebra-ops">Linear Algebra Ops</a></li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#memory-ops">Memory Ops</a></li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#indexing-ops">Indexing Ops</a></li>
|
|
||||||
<li class="toctree-l2 current"><a class="reference internal" href="../index.html#math-ops">Math Ops</a><ul class="current">
|
|
||||||
<li class="toctree-l3"><a class="reference internal" href="triton.exp.html">triton.exp</a></li>
|
|
||||||
<li class="toctree-l3"><a class="reference internal" href="triton.log.html">triton.log</a></li>
|
|
||||||
<li class="toctree-l3 current"><a class="current reference internal" href="#">triton.sigmoid</a></li>
|
|
||||||
<li class="toctree-l3"><a class="reference internal" href="triton.softmax.html">triton.softmax</a></li>
|
|
||||||
</ul>
|
|
||||||
</li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#reduction-ops">Reduction Ops</a></li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#comparison-ops">Comparison ops</a></li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#compiler-hint-ops">Compiler Hint Ops</a></li>
|
|
||||||
</ul>
|
|
||||||
</li>
|
|
||||||
</ul>
|
</ul>
|
||||||
<p class="caption"><span class="caption-text">Programming Guide</span></p>
|
<p class="caption"><span class="caption-text">Programming Guide</span></p>
|
||||||
<ul>
|
<ul>
|
||||||
@@ -168,8 +148,6 @@
|
|||||||
|
|
||||||
<li><a href="../../../index.html" class="icon icon-home"></a> »</li>
|
<li><a href="../../../index.html" class="icon icon-home"></a> »</li>
|
||||||
|
|
||||||
<li><a href="../index.html">Python API</a> »</li>
|
|
||||||
|
|
||||||
<li>triton.sigmoid</li>
|
<li>triton.sigmoid</li>
|
||||||
|
|
||||||
|
|
||||||
@@ -203,10 +181,6 @@
|
|||||||
|
|
||||||
</div>
|
</div>
|
||||||
<footer>
|
<footer>
|
||||||
<div class="rst-footer-buttons" role="navigation" aria-label="footer navigation">
|
|
||||||
<a href="triton.softmax.html" class="btn btn-neutral float-right" title="triton.softmax" accesskey="n" rel="next">Next <span class="fa fa-arrow-circle-right" aria-hidden="true"></span></a>
|
|
||||||
<a href="triton.log.html" class="btn btn-neutral float-left" title="triton.log" accesskey="p" rel="prev"><span class="fa fa-arrow-circle-left" aria-hidden="true"></span> Previous</a>
|
|
||||||
</div>
|
|
||||||
|
|
||||||
<hr/>
|
<hr/>
|
||||||
|
|
||||||
|
|||||||
@@ -42,9 +42,7 @@
|
|||||||
|
|
||||||
|
|
||||||
<link rel="index" title="Index" href="../../../genindex.html" />
|
<link rel="index" title="Index" href="../../../genindex.html" />
|
||||||
<link rel="search" title="Search" href="../../../search.html" />
|
<link rel="search" title="Search" href="../../../search.html" />
|
||||||
<link rel="next" title="triton.max" href="triton.max.html" />
|
|
||||||
<link rel="prev" title="triton.sigmoid" href="triton.sigmoid.html" />
|
|
||||||
</head>
|
</head>
|
||||||
|
|
||||||
<body class="wy-body-for-nav">
|
<body class="wy-body-for-nav">
|
||||||
@@ -95,26 +93,8 @@
|
|||||||
<li class="toctree-l1"><a class="reference internal" href="../../../getting-started/tutorials/index.html">Tutorials</a></li>
|
<li class="toctree-l1"><a class="reference internal" href="../../../getting-started/tutorials/index.html">Tutorials</a></li>
|
||||||
</ul>
|
</ul>
|
||||||
<p class="caption"><span class="caption-text">Language Reference</span></p>
|
<p class="caption"><span class="caption-text">Language Reference</span></p>
|
||||||
<ul class="current">
|
<ul>
|
||||||
<li class="toctree-l1 current"><a class="reference internal" href="../index.html">Python API</a><ul class="current">
|
<li class="toctree-l1"><a class="reference internal" href="../index.html">Python API</a></li>
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#programming-model">Programming Model</a></li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#creation-ops">Creation Ops</a></li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#shape-manipulation-ops">Shape Manipulation Ops</a></li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#linear-algebra-ops">Linear Algebra Ops</a></li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#memory-ops">Memory Ops</a></li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#indexing-ops">Indexing Ops</a></li>
|
|
||||||
<li class="toctree-l2 current"><a class="reference internal" href="../index.html#math-ops">Math Ops</a><ul class="current">
|
|
||||||
<li class="toctree-l3"><a class="reference internal" href="triton.exp.html">triton.exp</a></li>
|
|
||||||
<li class="toctree-l3"><a class="reference internal" href="triton.log.html">triton.log</a></li>
|
|
||||||
<li class="toctree-l3"><a class="reference internal" href="triton.sigmoid.html">triton.sigmoid</a></li>
|
|
||||||
<li class="toctree-l3 current"><a class="current reference internal" href="#">triton.softmax</a></li>
|
|
||||||
</ul>
|
|
||||||
</li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#reduction-ops">Reduction Ops</a></li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#comparison-ops">Comparison ops</a></li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#compiler-hint-ops">Compiler Hint Ops</a></li>
|
|
||||||
</ul>
|
|
||||||
</li>
|
|
||||||
</ul>
|
</ul>
|
||||||
<p class="caption"><span class="caption-text">Programming Guide</span></p>
|
<p class="caption"><span class="caption-text">Programming Guide</span></p>
|
||||||
<ul>
|
<ul>
|
||||||
@@ -168,8 +148,6 @@
|
|||||||
|
|
||||||
<li><a href="../../../index.html" class="icon icon-home"></a> »</li>
|
<li><a href="../../../index.html" class="icon icon-home"></a> »</li>
|
||||||
|
|
||||||
<li><a href="../index.html">Python API</a> »</li>
|
|
||||||
|
|
||||||
<li>triton.softmax</li>
|
<li>triton.softmax</li>
|
||||||
|
|
||||||
|
|
||||||
@@ -203,10 +181,6 @@
|
|||||||
|
|
||||||
</div>
|
</div>
|
||||||
<footer>
|
<footer>
|
||||||
<div class="rst-footer-buttons" role="navigation" aria-label="footer navigation">
|
|
||||||
<a href="triton.max.html" class="btn btn-neutral float-right" title="triton.max" accesskey="n" rel="next">Next <span class="fa fa-arrow-circle-right" aria-hidden="true"></span></a>
|
|
||||||
<a href="triton.sigmoid.html" class="btn btn-neutral float-left" title="triton.sigmoid" accesskey="p" rel="prev"><span class="fa fa-arrow-circle-left" aria-hidden="true"></span> Previous</a>
|
|
||||||
</div>
|
|
||||||
|
|
||||||
<hr/>
|
<hr/>
|
||||||
|
|
||||||
|
|||||||
@@ -42,9 +42,7 @@
|
|||||||
|
|
||||||
|
|
||||||
<link rel="index" title="Index" href="../../../genindex.html" />
|
<link rel="index" title="Index" href="../../../genindex.html" />
|
||||||
<link rel="search" title="Search" href="../../../search.html" />
|
<link rel="search" title="Search" href="../../../search.html" />
|
||||||
<link rel="next" title="triton.atomic_cas" href="triton.atomic_cas.html" />
|
|
||||||
<link rel="prev" title="triton.load" href="triton.load.html" />
|
|
||||||
</head>
|
</head>
|
||||||
|
|
||||||
<body class="wy-body-for-nav">
|
<body class="wy-body-for-nav">
|
||||||
@@ -95,26 +93,8 @@
|
|||||||
<li class="toctree-l1"><a class="reference internal" href="../../../getting-started/tutorials/index.html">Tutorials</a></li>
|
<li class="toctree-l1"><a class="reference internal" href="../../../getting-started/tutorials/index.html">Tutorials</a></li>
|
||||||
</ul>
|
</ul>
|
||||||
<p class="caption"><span class="caption-text">Language Reference</span></p>
|
<p class="caption"><span class="caption-text">Language Reference</span></p>
|
||||||
<ul class="current">
|
<ul>
|
||||||
<li class="toctree-l1 current"><a class="reference internal" href="../index.html">Python API</a><ul class="current">
|
<li class="toctree-l1"><a class="reference internal" href="../index.html">Python API</a></li>
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#programming-model">Programming Model</a></li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#creation-ops">Creation Ops</a></li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#shape-manipulation-ops">Shape Manipulation Ops</a></li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#linear-algebra-ops">Linear Algebra Ops</a></li>
|
|
||||||
<li class="toctree-l2 current"><a class="reference internal" href="../index.html#memory-ops">Memory Ops</a><ul class="current">
|
|
||||||
<li class="toctree-l3"><a class="reference internal" href="triton.load.html">triton.load</a></li>
|
|
||||||
<li class="toctree-l3 current"><a class="current reference internal" href="#">triton.store</a></li>
|
|
||||||
<li class="toctree-l3"><a class="reference internal" href="triton.atomic_cas.html">triton.atomic_cas</a></li>
|
|
||||||
<li class="toctree-l3"><a class="reference internal" href="triton.atomic_xchg.html">triton.atomic_xchg</a></li>
|
|
||||||
</ul>
|
|
||||||
</li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#indexing-ops">Indexing Ops</a></li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#math-ops">Math Ops</a></li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#reduction-ops">Reduction Ops</a></li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#comparison-ops">Comparison ops</a></li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#compiler-hint-ops">Compiler Hint Ops</a></li>
|
|
||||||
</ul>
|
|
||||||
</li>
|
|
||||||
</ul>
|
</ul>
|
||||||
<p class="caption"><span class="caption-text">Programming Guide</span></p>
|
<p class="caption"><span class="caption-text">Programming Guide</span></p>
|
||||||
<ul>
|
<ul>
|
||||||
@@ -168,8 +148,6 @@
|
|||||||
|
|
||||||
<li><a href="../../../index.html" class="icon icon-home"></a> »</li>
|
<li><a href="../../../index.html" class="icon icon-home"></a> »</li>
|
||||||
|
|
||||||
<li><a href="../index.html">Python API</a> »</li>
|
|
||||||
|
|
||||||
<li>triton.store</li>
|
<li>triton.store</li>
|
||||||
|
|
||||||
|
|
||||||
@@ -214,10 +192,6 @@
|
|||||||
|
|
||||||
</div>
|
</div>
|
||||||
<footer>
|
<footer>
|
||||||
<div class="rst-footer-buttons" role="navigation" aria-label="footer navigation">
|
|
||||||
<a href="triton.atomic_cas.html" class="btn btn-neutral float-right" title="triton.atomic_cas" accesskey="n" rel="next">Next <span class="fa fa-arrow-circle-right" aria-hidden="true"></span></a>
|
|
||||||
<a href="triton.load.html" class="btn btn-neutral float-left" title="triton.load" accesskey="p" rel="prev"><span class="fa fa-arrow-circle-left" aria-hidden="true"></span> Previous</a>
|
|
||||||
</div>
|
|
||||||
|
|
||||||
<hr/>
|
<hr/>
|
||||||
|
|
||||||
|
|||||||
@@ -42,9 +42,7 @@
|
|||||||
|
|
||||||
|
|
||||||
<link rel="index" title="Index" href="../../../genindex.html" />
|
<link rel="index" title="Index" href="../../../genindex.html" />
|
||||||
<link rel="search" title="Search" href="../../../search.html" />
|
<link rel="search" title="Search" href="../../../search.html" />
|
||||||
<link rel="next" title="triton.minimum" href="triton.minimum.html" />
|
|
||||||
<link rel="prev" title="triton.min" href="triton.min.html" />
|
|
||||||
</head>
|
</head>
|
||||||
|
|
||||||
<body class="wy-body-for-nav">
|
<body class="wy-body-for-nav">
|
||||||
@@ -95,25 +93,8 @@
|
|||||||
<li class="toctree-l1"><a class="reference internal" href="../../../getting-started/tutorials/index.html">Tutorials</a></li>
|
<li class="toctree-l1"><a class="reference internal" href="../../../getting-started/tutorials/index.html">Tutorials</a></li>
|
||||||
</ul>
|
</ul>
|
||||||
<p class="caption"><span class="caption-text">Language Reference</span></p>
|
<p class="caption"><span class="caption-text">Language Reference</span></p>
|
||||||
<ul class="current">
|
<ul>
|
||||||
<li class="toctree-l1 current"><a class="reference internal" href="../index.html">Python API</a><ul class="current">
|
<li class="toctree-l1"><a class="reference internal" href="../index.html">Python API</a></li>
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#programming-model">Programming Model</a></li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#creation-ops">Creation Ops</a></li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#shape-manipulation-ops">Shape Manipulation Ops</a></li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#linear-algebra-ops">Linear Algebra Ops</a></li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#memory-ops">Memory Ops</a></li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#indexing-ops">Indexing Ops</a></li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#math-ops">Math Ops</a></li>
|
|
||||||
<li class="toctree-l2 current"><a class="reference internal" href="../index.html#reduction-ops">Reduction Ops</a><ul class="current">
|
|
||||||
<li class="toctree-l3"><a class="reference internal" href="triton.max.html">triton.max</a></li>
|
|
||||||
<li class="toctree-l3"><a class="reference internal" href="triton.min.html">triton.min</a></li>
|
|
||||||
<li class="toctree-l3 current"><a class="current reference internal" href="#">triton.sum</a></li>
|
|
||||||
</ul>
|
|
||||||
</li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#comparison-ops">Comparison ops</a></li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#compiler-hint-ops">Compiler Hint Ops</a></li>
|
|
||||||
</ul>
|
|
||||||
</li>
|
|
||||||
</ul>
|
</ul>
|
||||||
<p class="caption"><span class="caption-text">Programming Guide</span></p>
|
<p class="caption"><span class="caption-text">Programming Guide</span></p>
|
||||||
<ul>
|
<ul>
|
||||||
@@ -167,8 +148,6 @@
|
|||||||
|
|
||||||
<li><a href="../../../index.html" class="icon icon-home"></a> »</li>
|
<li><a href="../../../index.html" class="icon icon-home"></a> »</li>
|
||||||
|
|
||||||
<li><a href="../index.html">Python API</a> »</li>
|
|
||||||
|
|
||||||
<li>triton.sum</li>
|
<li>triton.sum</li>
|
||||||
|
|
||||||
|
|
||||||
@@ -202,10 +181,6 @@
|
|||||||
|
|
||||||
</div>
|
</div>
|
||||||
<footer>
|
<footer>
|
||||||
<div class="rst-footer-buttons" role="navigation" aria-label="footer navigation">
|
|
||||||
<a href="triton.minimum.html" class="btn btn-neutral float-right" title="triton.minimum" accesskey="n" rel="next">Next <span class="fa fa-arrow-circle-right" aria-hidden="true"></span></a>
|
|
||||||
<a href="triton.min.html" class="btn btn-neutral float-left" title="triton.min" accesskey="p" rel="prev"><span class="fa fa-arrow-circle-left" aria-hidden="true"></span> Previous</a>
|
|
||||||
</div>
|
|
||||||
|
|
||||||
<hr/>
|
<hr/>
|
||||||
|
|
||||||
|
|||||||
@@ -42,9 +42,7 @@
|
|||||||
|
|
||||||
|
|
||||||
<link rel="index" title="Index" href="../../../genindex.html" />
|
<link rel="index" title="Index" href="../../../genindex.html" />
|
||||||
<link rel="search" title="Search" href="../../../search.html" />
|
<link rel="search" title="Search" href="../../../search.html" />
|
||||||
<link rel="next" title="triton.exp" href="triton.exp.html" />
|
|
||||||
<link rel="prev" title="triton.atomic_xchg" href="triton.atomic_xchg.html" />
|
|
||||||
</head>
|
</head>
|
||||||
|
|
||||||
<body class="wy-body-for-nav">
|
<body class="wy-body-for-nav">
|
||||||
@@ -95,23 +93,8 @@
|
|||||||
<li class="toctree-l1"><a class="reference internal" href="../../../getting-started/tutorials/index.html">Tutorials</a></li>
|
<li class="toctree-l1"><a class="reference internal" href="../../../getting-started/tutorials/index.html">Tutorials</a></li>
|
||||||
</ul>
|
</ul>
|
||||||
<p class="caption"><span class="caption-text">Language Reference</span></p>
|
<p class="caption"><span class="caption-text">Language Reference</span></p>
|
||||||
<ul class="current">
|
<ul>
|
||||||
<li class="toctree-l1 current"><a class="reference internal" href="../index.html">Python API</a><ul class="current">
|
<li class="toctree-l1"><a class="reference internal" href="../index.html">Python API</a></li>
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#programming-model">Programming Model</a></li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#creation-ops">Creation Ops</a></li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#shape-manipulation-ops">Shape Manipulation Ops</a></li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#linear-algebra-ops">Linear Algebra Ops</a></li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#memory-ops">Memory Ops</a></li>
|
|
||||||
<li class="toctree-l2 current"><a class="reference internal" href="../index.html#indexing-ops">Indexing Ops</a><ul class="current">
|
|
||||||
<li class="toctree-l3 current"><a class="current reference internal" href="#">triton.where</a></li>
|
|
||||||
</ul>
|
|
||||||
</li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#math-ops">Math Ops</a></li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#reduction-ops">Reduction Ops</a></li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#comparison-ops">Comparison ops</a></li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#compiler-hint-ops">Compiler Hint Ops</a></li>
|
|
||||||
</ul>
|
|
||||||
</li>
|
|
||||||
</ul>
|
</ul>
|
||||||
<p class="caption"><span class="caption-text">Programming Guide</span></p>
|
<p class="caption"><span class="caption-text">Programming Guide</span></p>
|
||||||
<ul>
|
<ul>
|
||||||
@@ -165,8 +148,6 @@
|
|||||||
|
|
||||||
<li><a href="../../../index.html" class="icon icon-home"></a> »</li>
|
<li><a href="../../../index.html" class="icon icon-home"></a> »</li>
|
||||||
|
|
||||||
<li><a href="../index.html">Python API</a> »</li>
|
|
||||||
|
|
||||||
<li>triton.where</li>
|
<li>triton.where</li>
|
||||||
|
|
||||||
|
|
||||||
@@ -215,10 +196,6 @@ The shape of <cite>x</cite> and <cite>y</cite> are both broadcast to the shape o
|
|||||||
|
|
||||||
</div>
|
</div>
|
||||||
<footer>
|
<footer>
|
||||||
<div class="rst-footer-buttons" role="navigation" aria-label="footer navigation">
|
|
||||||
<a href="triton.exp.html" class="btn btn-neutral float-right" title="triton.exp" accesskey="n" rel="next">Next <span class="fa fa-arrow-circle-right" aria-hidden="true"></span></a>
|
|
||||||
<a href="triton.atomic_xchg.html" class="btn btn-neutral float-left" title="triton.atomic_xchg" accesskey="p" rel="prev"><span class="fa fa-arrow-circle-left" aria-hidden="true"></span> Previous</a>
|
|
||||||
</div>
|
|
||||||
|
|
||||||
<hr/>
|
<hr/>
|
||||||
|
|
||||||
|
|||||||
@@ -42,9 +42,7 @@
|
|||||||
|
|
||||||
|
|
||||||
<link rel="index" title="Index" href="../../../genindex.html" />
|
<link rel="index" title="Index" href="../../../genindex.html" />
|
||||||
<link rel="search" title="Search" href="../../../search.html" />
|
<link rel="search" title="Search" href="../../../search.html" />
|
||||||
<link rel="next" title="triton.broadcast_to" href="triton.broadcast_to.html" />
|
|
||||||
<link rel="prev" title="triton.arange" href="triton.arange.html" />
|
|
||||||
</head>
|
</head>
|
||||||
|
|
||||||
<body class="wy-body-for-nav">
|
<body class="wy-body-for-nav">
|
||||||
@@ -95,24 +93,8 @@
|
|||||||
<li class="toctree-l1"><a class="reference internal" href="../../../getting-started/tutorials/index.html">Tutorials</a></li>
|
<li class="toctree-l1"><a class="reference internal" href="../../../getting-started/tutorials/index.html">Tutorials</a></li>
|
||||||
</ul>
|
</ul>
|
||||||
<p class="caption"><span class="caption-text">Language Reference</span></p>
|
<p class="caption"><span class="caption-text">Language Reference</span></p>
|
||||||
<ul class="current">
|
<ul>
|
||||||
<li class="toctree-l1 current"><a class="reference internal" href="../index.html">Python API</a><ul class="current">
|
<li class="toctree-l1"><a class="reference internal" href="../index.html">Python API</a></li>
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#programming-model">Programming Model</a></li>
|
|
||||||
<li class="toctree-l2 current"><a class="reference internal" href="../index.html#creation-ops">Creation Ops</a><ul class="current">
|
|
||||||
<li class="toctree-l3"><a class="reference internal" href="triton.arange.html">triton.arange</a></li>
|
|
||||||
<li class="toctree-l3 current"><a class="current reference internal" href="#">triton.zeros</a></li>
|
|
||||||
</ul>
|
|
||||||
</li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#shape-manipulation-ops">Shape Manipulation Ops</a></li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#linear-algebra-ops">Linear Algebra Ops</a></li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#memory-ops">Memory Ops</a></li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#indexing-ops">Indexing Ops</a></li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#math-ops">Math Ops</a></li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#reduction-ops">Reduction Ops</a></li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#comparison-ops">Comparison ops</a></li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="../index.html#compiler-hint-ops">Compiler Hint Ops</a></li>
|
|
||||||
</ul>
|
|
||||||
</li>
|
|
||||||
</ul>
|
</ul>
|
||||||
<p class="caption"><span class="caption-text">Programming Guide</span></p>
|
<p class="caption"><span class="caption-text">Programming Guide</span></p>
|
||||||
<ul>
|
<ul>
|
||||||
@@ -166,8 +148,6 @@
|
|||||||
|
|
||||||
<li><a href="../../../index.html" class="icon icon-home"></a> »</li>
|
<li><a href="../../../index.html" class="icon icon-home"></a> »</li>
|
||||||
|
|
||||||
<li><a href="../index.html">Python API</a> »</li>
|
|
||||||
|
|
||||||
<li>triton.zeros</li>
|
<li>triton.zeros</li>
|
||||||
|
|
||||||
|
|
||||||
@@ -211,10 +191,6 @@
|
|||||||
|
|
||||||
</div>
|
</div>
|
||||||
<footer>
|
<footer>
|
||||||
<div class="rst-footer-buttons" role="navigation" aria-label="footer navigation">
|
|
||||||
<a href="triton.broadcast_to.html" class="btn btn-neutral float-right" title="triton.broadcast_to" accesskey="n" rel="next">Next <span class="fa fa-arrow-circle-right" aria-hidden="true"></span></a>
|
|
||||||
<a href="triton.arange.html" class="btn btn-neutral float-left" title="triton.arange" accesskey="p" rel="prev"><span class="fa fa-arrow-circle-left" aria-hidden="true"></span> Previous</a>
|
|
||||||
</div>
|
|
||||||
|
|
||||||
<hr/>
|
<hr/>
|
||||||
|
|
||||||
|
|||||||
@@ -43,7 +43,7 @@
|
|||||||
|
|
||||||
<link rel="index" title="Index" href="../../genindex.html" />
|
<link rel="index" title="Index" href="../../genindex.html" />
|
||||||
<link rel="search" title="Search" href="../../search.html" />
|
<link rel="search" title="Search" href="../../search.html" />
|
||||||
<link rel="next" title="triton.program_id" href="generated/triton.program_id.html" />
|
<link rel="next" title="Introduction" href="../../programming-guide/chapter-1/introduction.html" />
|
||||||
<link rel="prev" title="Matrix Multiplication" href="../../getting-started/tutorials/03-matrix-multiplication.html" />
|
<link rel="prev" title="Matrix Multiplication" href="../../getting-started/tutorials/03-matrix-multiplication.html" />
|
||||||
</head>
|
</head>
|
||||||
|
|
||||||
@@ -97,59 +97,16 @@
|
|||||||
<p class="caption"><span class="caption-text">Language Reference</span></p>
|
<p class="caption"><span class="caption-text">Language Reference</span></p>
|
||||||
<ul class="current">
|
<ul class="current">
|
||||||
<li class="toctree-l1 current"><a class="current reference internal" href="#">Python API</a><ul>
|
<li class="toctree-l1 current"><a class="current reference internal" href="#">Python API</a><ul>
|
||||||
<li class="toctree-l2"><a class="reference internal" href="#programming-model">Programming Model</a><ul>
|
<li class="toctree-l2"><a class="reference internal" href="#programming-model">Programming Model</a></li>
|
||||||
<li class="toctree-l3"><a class="reference internal" href="generated/triton.program_id.html">triton.program_id</a></li>
|
<li class="toctree-l2"><a class="reference internal" href="#creation-ops">Creation Ops</a></li>
|
||||||
<li class="toctree-l3"><a class="reference internal" href="generated/triton.num_programs.html">triton.num_programs</a></li>
|
<li class="toctree-l2"><a class="reference internal" href="#shape-manipulation-ops">Shape Manipulation Ops</a></li>
|
||||||
</ul>
|
<li class="toctree-l2"><a class="reference internal" href="#linear-algebra-ops">Linear Algebra Ops</a></li>
|
||||||
</li>
|
<li class="toctree-l2"><a class="reference internal" href="#memory-ops">Memory Ops</a></li>
|
||||||
<li class="toctree-l2"><a class="reference internal" href="#creation-ops">Creation Ops</a><ul>
|
<li class="toctree-l2"><a class="reference internal" href="#indexing-ops">Indexing Ops</a></li>
|
||||||
<li class="toctree-l3"><a class="reference internal" href="generated/triton.arange.html">triton.arange</a></li>
|
<li class="toctree-l2"><a class="reference internal" href="#math-ops">Math Ops</a></li>
|
||||||
<li class="toctree-l3"><a class="reference internal" href="generated/triton.zeros.html">triton.zeros</a></li>
|
<li class="toctree-l2"><a class="reference internal" href="#reduction-ops">Reduction Ops</a></li>
|
||||||
</ul>
|
<li class="toctree-l2"><a class="reference internal" href="#comparison-ops">Comparison ops</a></li>
|
||||||
</li>
|
<li class="toctree-l2"><a class="reference internal" href="#compiler-hint-ops">Compiler Hint Ops</a></li>
|
||||||
<li class="toctree-l2"><a class="reference internal" href="#shape-manipulation-ops">Shape Manipulation Ops</a><ul>
|
|
||||||
<li class="toctree-l3"><a class="reference internal" href="generated/triton.broadcast_to.html">triton.broadcast_to</a></li>
|
|
||||||
<li class="toctree-l3"><a class="reference internal" href="generated/triton.reshape.html">triton.reshape</a></li>
|
|
||||||
<li class="toctree-l3"><a class="reference internal" href="generated/triton.ravel.html">triton.ravel</a></li>
|
|
||||||
</ul>
|
|
||||||
</li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="#linear-algebra-ops">Linear Algebra Ops</a><ul>
|
|
||||||
<li class="toctree-l3"><a class="reference internal" href="generated/triton.dot.html">triton.dot</a></li>
|
|
||||||
</ul>
|
|
||||||
</li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="#memory-ops">Memory Ops</a><ul>
|
|
||||||
<li class="toctree-l3"><a class="reference internal" href="generated/triton.load.html">triton.load</a></li>
|
|
||||||
<li class="toctree-l3"><a class="reference internal" href="generated/triton.store.html">triton.store</a></li>
|
|
||||||
<li class="toctree-l3"><a class="reference internal" href="generated/triton.atomic_cas.html">triton.atomic_cas</a></li>
|
|
||||||
<li class="toctree-l3"><a class="reference internal" href="generated/triton.atomic_xchg.html">triton.atomic_xchg</a></li>
|
|
||||||
</ul>
|
|
||||||
</li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="#indexing-ops">Indexing Ops</a><ul>
|
|
||||||
<li class="toctree-l3"><a class="reference internal" href="generated/triton.where.html">triton.where</a></li>
|
|
||||||
</ul>
|
|
||||||
</li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="#math-ops">Math Ops</a><ul>
|
|
||||||
<li class="toctree-l3"><a class="reference internal" href="generated/triton.exp.html">triton.exp</a></li>
|
|
||||||
<li class="toctree-l3"><a class="reference internal" href="generated/triton.log.html">triton.log</a></li>
|
|
||||||
<li class="toctree-l3"><a class="reference internal" href="generated/triton.sigmoid.html">triton.sigmoid</a></li>
|
|
||||||
<li class="toctree-l3"><a class="reference internal" href="generated/triton.softmax.html">triton.softmax</a></li>
|
|
||||||
</ul>
|
|
||||||
</li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="#reduction-ops">Reduction Ops</a><ul>
|
|
||||||
<li class="toctree-l3"><a class="reference internal" href="generated/triton.max.html">triton.max</a></li>
|
|
||||||
<li class="toctree-l3"><a class="reference internal" href="generated/triton.min.html">triton.min</a></li>
|
|
||||||
<li class="toctree-l3"><a class="reference internal" href="generated/triton.sum.html">triton.sum</a></li>
|
|
||||||
</ul>
|
|
||||||
</li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="#comparison-ops">Comparison ops</a><ul>
|
|
||||||
<li class="toctree-l3"><a class="reference internal" href="generated/triton.minimum.html">triton.minimum</a></li>
|
|
||||||
<li class="toctree-l3"><a class="reference internal" href="generated/triton.maximum.html">triton.maximum</a></li>
|
|
||||||
</ul>
|
|
||||||
</li>
|
|
||||||
<li class="toctree-l2"><a class="reference internal" href="#compiler-hint-ops">Compiler Hint Ops</a><ul>
|
|
||||||
<li class="toctree-l3"><a class="reference internal" href="generated/triton.multiple_of.html">triton.multiple_of</a></li>
|
|
||||||
</ul>
|
|
||||||
</li>
|
|
||||||
</ul>
|
</ul>
|
||||||
</li>
|
</li>
|
||||||
</ul>
|
</ul>
|
||||||
@@ -234,12 +191,6 @@
|
|||||||
<col style="width: 90%" />
|
<col style="width: 90%" />
|
||||||
</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>
|
|
||||||
<td><p>Returns the id of the current program instance along the given <cite>axis</cite>. Triton uses an SPMD model in which different @triton.jit functions run in parallel with different <a href="#id1"><span class="problematic" id="id2">`</span></a>program_id`s.</p></td>
|
|
||||||
</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>
|
|
||||||
<td><p>Returns the number of program instances launched along the given <cite>axis</cite>.</p></td>
|
|
||||||
</tr>
|
|
||||||
</tbody>
|
</tbody>
|
||||||
</table>
|
</table>
|
||||||
</div>
|
</div>
|
||||||
@@ -251,12 +202,6 @@
|
|||||||
<col style="width: 90%" />
|
<col style="width: 90%" />
|
||||||
</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>
|
|
||||||
<td><p>Returns contiguous values within the open interval [start, end).</p></td>
|
|
||||||
</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>
|
|
||||||
<td><p>Returns a block filled with the scalar value 0 and the given shape.</p></td>
|
|
||||||
</tr>
|
|
||||||
</tbody>
|
</tbody>
|
||||||
</table>
|
</table>
|
||||||
</div>
|
</div>
|
||||||
@@ -268,15 +213,6 @@
|
|||||||
<col style="width: 90%" />
|
<col style="width: 90%" />
|
||||||
</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>
|
|
||||||
<td><p>Tries to broadcast a block to a new shape.</p></td>
|
|
||||||
</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>
|
|
||||||
<td><p>Reshapes a block to a new shape.</p></td>
|
|
||||||
</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>
|
|
||||||
<td><p></p></td>
|
|
||||||
</tr>
|
|
||||||
</tbody>
|
</tbody>
|
||||||
</table>
|
</table>
|
||||||
</div>
|
</div>
|
||||||
@@ -288,9 +224,6 @@
|
|||||||
<col style="width: 90%" />
|
<col style="width: 90%" />
|
||||||
</colgroup>
|
</colgroup>
|
||||||
<tbody>
|
<tbody>
|
||||||
<tr class="row-odd"><td><p><a class="reference internal" href="generated/triton.dot.html#triton.dot" title="triton.dot"><code class="xref py py-obj docutils literal notranslate"><span class="pre">dot</span></code></a></p></td>
|
|
||||||
<td><p>Returns the matrix product of two blocks.</p></td>
|
|
||||||
</tr>
|
|
||||||
</tbody>
|
</tbody>
|
||||||
</table>
|
</table>
|
||||||
</div>
|
</div>
|
||||||
@@ -302,18 +235,6 @@
|
|||||||
<col style="width: 90%" />
|
<col style="width: 90%" />
|
||||||
</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>
|
|
||||||
<td><p>Return a block of data whose values are, elementwise, loaded from memory at location defined by <cite>pointer</cite>.</p></td>
|
|
||||||
</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>
|
|
||||||
<td><p>Stores <cite>value</cite> block of elements in memory, element-wise, at the memory locations specified by <cite>pointer</cite>.</p></td>
|
|
||||||
</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>
|
|
||||||
<td><p></p></td>
|
|
||||||
</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>
|
|
||||||
<td><p></p></td>
|
|
||||||
</tr>
|
|
||||||
</tbody>
|
</tbody>
|
||||||
</table>
|
</table>
|
||||||
</div>
|
</div>
|
||||||
@@ -325,9 +246,6 @@
|
|||||||
<col style="width: 90%" />
|
<col style="width: 90%" />
|
||||||
</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>
|
|
||||||
<td><p>Returns a block of elements from either <cite>x</cite> or <cite>y</cite>, depending on <cite>condition</cite>.</p></td>
|
|
||||||
</tr>
|
|
||||||
</tbody>
|
</tbody>
|
||||||
</table>
|
</table>
|
||||||
</div>
|
</div>
|
||||||
@@ -339,18 +257,6 @@
|
|||||||
<col style="width: 90%" />
|
<col style="width: 90%" />
|
||||||
</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>
|
|
||||||
<td><p></p></td>
|
|
||||||
</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>
|
|
||||||
<td><p></p></td>
|
|
||||||
</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>
|
|
||||||
<td><p></p></td>
|
|
||||||
</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>
|
|
||||||
<td><p></p></td>
|
|
||||||
</tr>
|
|
||||||
</tbody>
|
</tbody>
|
||||||
</table>
|
</table>
|
||||||
</div>
|
</div>
|
||||||
@@ -362,15 +268,6 @@
|
|||||||
<col style="width: 90%" />
|
<col style="width: 90%" />
|
||||||
</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>
|
|
||||||
<td><p></p></td>
|
|
||||||
</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>
|
|
||||||
<td><p></p></td>
|
|
||||||
</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>
|
|
||||||
<td><p></p></td>
|
|
||||||
</tr>
|
|
||||||
</tbody>
|
</tbody>
|
||||||
</table>
|
</table>
|
||||||
</div>
|
</div>
|
||||||
@@ -382,12 +279,6 @@
|
|||||||
<col style="width: 90%" />
|
<col style="width: 90%" />
|
||||||
</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>
|
|
||||||
<td><p></p></td>
|
|
||||||
</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>
|
|
||||||
<td><p></p></td>
|
|
||||||
</tr>
|
|
||||||
</tbody>
|
</tbody>
|
||||||
</table>
|
</table>
|
||||||
</div>
|
</div>
|
||||||
@@ -399,9 +290,6 @@
|
|||||||
<col style="width: 90%" />
|
<col style="width: 90%" />
|
||||||
</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>
|
|
||||||
<td><p></p></td>
|
|
||||||
</tr>
|
|
||||||
</tbody>
|
</tbody>
|
||||||
</table>
|
</table>
|
||||||
</div>
|
</div>
|
||||||
@@ -413,7 +301,7 @@
|
|||||||
</div>
|
</div>
|
||||||
<footer>
|
<footer>
|
||||||
<div class="rst-footer-buttons" role="navigation" aria-label="footer navigation">
|
<div class="rst-footer-buttons" role="navigation" aria-label="footer navigation">
|
||||||
<a href="generated/triton.program_id.html" class="btn btn-neutral float-right" title="triton.program_id" accesskey="n" rel="next">Next <span class="fa fa-arrow-circle-right" aria-hidden="true"></span></a>
|
<a href="../../programming-guide/chapter-1/introduction.html" class="btn btn-neutral float-right" title="Introduction" accesskey="n" rel="next">Next <span class="fa fa-arrow-circle-right" aria-hidden="true"></span></a>
|
||||||
<a href="../../getting-started/tutorials/03-matrix-multiplication.html" class="btn btn-neutral float-left" title="Matrix Multiplication" accesskey="p" rel="prev"><span class="fa fa-arrow-circle-left" aria-hidden="true"></span> Previous</a>
|
<a href="../../getting-started/tutorials/03-matrix-multiplication.html" class="btn btn-neutral float-left" title="Matrix Multiplication" accesskey="p" rel="prev"><span class="fa fa-arrow-circle-left" aria-hidden="true"></span> Previous</a>
|
||||||
</div>
|
</div>
|
||||||
|
|
||||||
|
|||||||
@@ -44,7 +44,7 @@
|
|||||||
<link rel="index" title="Index" href="../../genindex.html" />
|
<link rel="index" title="Index" href="../../genindex.html" />
|
||||||
<link rel="search" title="Search" href="../../search.html" />
|
<link rel="search" title="Search" href="../../search.html" />
|
||||||
<link rel="next" title="Related Work" href="../chapter-2/related-work.html" />
|
<link rel="next" title="Related Work" href="../chapter-2/related-work.html" />
|
||||||
<link rel="prev" title="triton.multiple_of" href="../../language-reference/python-api/generated/triton.multiple_of.html" />
|
<link rel="prev" title="Python API" href="../../language-reference/python-api/index.html" />
|
||||||
</head>
|
</head>
|
||||||
|
|
||||||
<body class="wy-body-for-nav">
|
<body class="wy-body-for-nav">
|
||||||
@@ -294,7 +294,7 @@
|
|||||||
<footer>
|
<footer>
|
||||||
<div class="rst-footer-buttons" role="navigation" aria-label="footer navigation">
|
<div class="rst-footer-buttons" role="navigation" aria-label="footer navigation">
|
||||||
<a href="../chapter-2/related-work.html" class="btn btn-neutral float-right" title="Related Work" accesskey="n" rel="next">Next <span class="fa fa-arrow-circle-right" aria-hidden="true"></span></a>
|
<a href="../chapter-2/related-work.html" class="btn btn-neutral float-right" title="Related Work" accesskey="n" rel="next">Next <span class="fa fa-arrow-circle-right" aria-hidden="true"></span></a>
|
||||||
<a href="../../language-reference/python-api/generated/triton.multiple_of.html" class="btn btn-neutral float-left" title="triton.multiple_of" accesskey="p" rel="prev"><span class="fa fa-arrow-circle-left" aria-hidden="true"></span> Previous</a>
|
<a href="../../language-reference/python-api/index.html" class="btn btn-neutral float-left" title="Python API" accesskey="p" rel="prev"><span class="fa fa-arrow-circle-left" aria-hidden="true"></span> Previous</a>
|
||||||
</div>
|
</div>
|
||||||
|
|
||||||
<hr/>
|
<hr/>
|
||||||
|
|||||||
File diff suppressed because one or more lines are too long
Reference in New Issue
Block a user