gaspersic/triton
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

[GH-PAGES] Updated website

Browse Source
This commit is contained in:
Philippe Tillet
2021-09-11 00:13:31 +00:00
parent 224d361833
commit 8c732c2603
19 changed files with 79 additions and 79 deletions
Download Patch File Download Diff File Expand all files Collapse all files

BIN
_downloads/662999063954282841dc90b8945f85ce/tutorials_jupyter.zip
View File

Binary file not shown.

BIN
_downloads/763344228ae6bc253ed1a6cf586aa30d/tutorials_python.zip
View File

Binary file not shown.

BIN
_images/sphx_glr_01-vector-add_001.png
View File

Binary file not shown.
Side by Side Swipe Overlay

Before

Width:  |  Height:  |  Size: 25 KiB

After

Width:  |  Height:  |  Size: 25 KiB

BIN
_images/sphx_glr_01-vector-add_thumb.png
View File

Binary file not shown.
Side by Side Swipe Overlay

Before

Width:  |  Height:  |  Size: 16 KiB

After

Width:  |  Height:  |  Size: 16 KiB

BIN
_images/sphx_glr_02-fused-softmax_001.png
View File

Binary file not shown.
Side by Side Swipe Overlay

Before

Width:  |  Height:  |  Size: 38 KiB

After

Width:  |  Height:  |  Size: 38 KiB

BIN
_images/sphx_glr_02-fused-softmax_thumb.png
View File

Binary file not shown.
Side by Side Swipe Overlay

Before

Width:  |  Height:  |  Size: 24 KiB

After

Width:  |  Height:  |  Size: 24 KiB

BIN
_images/sphx_glr_03-matrix-multiplication_001.png
View File

Binary file not shown.
Side by Side Swipe Overlay

Before

Width:  |  Height:  |  Size: 56 KiB

After

Width:  |  Height:  |  Size: 56 KiB

BIN
_images/sphx_glr_03-matrix-multiplication_thumb.png
View File

Binary file not shown.
Side by Side Swipe Overlay

Before

Width:  |  Height:  |  Size: 33 KiB

After

Width:  |  Height:  |  Size: 32 KiB

8
_sources/getting-started/tutorials/01-vector-add.rst.txt
View File

@@ -231,13 +231,13 @@ We can now run the decorated function above. Pass `print_data=True` to see the p
vector-add-performance:
size Triton Torch
0 4096.0 9.540372 9.600000
0 4096.0 9.600000 9.600000
1 8192.0 19.200000 19.200000
2 16384.0 38.400001 38.400001
3 32768.0 76.800002 76.800002
4 65536.0 127.999995 127.999995
5 131072.0 219.428568 219.428568
6 262144.0 341.333321 384.000001
5 131072.0 255.999991 219.428568
6 262144.0 341.333321 341.333321
7 524288.0 472.615390 472.615390
8 1048576.0 614.400016 614.400016
9 2097152.0 722.823517 722.823517
@@ -254,7 +254,7 @@ We can now run the decorated function above. Pass `print_data=True` to see the p
.. rst-class:: sphx-glr-timing
**Total running time of the script:** ( 0 minutes 11.013 seconds)
**Total running time of the script:** ( 0 minutes 11.022 seconds)
.. _sphx_glr_download_getting-started_tutorials_01-vector-add.py:

8
_sources/getting-started/tutorials/02-fused-softmax.rst.txt
View File

@@ -290,13 +290,13 @@ We will then compare its performance against (1) :code:`torch.softmax` and (2) t
1 384.0 585.142862 585.142862 153.600004
2 512.0 630.153853 606.814814 154.566038
3 640.0 682.666684 640.000002 160.000000
4 768.0 702.171410 664.216187 162.754967
4 768.0 702.171410 664.216187 163.839992
.. ... ... ... ...
93 12160.0 810.666687 405.755985 199.038365
94 12288.0 810.754644 415.661740 199.298541
95 12416.0 809.189387 412.149375 198.854847
95 12416.0 809.189387 411.722274 198.954424
96 12544.0 807.661970 412.971190 199.209928
97 12672.0 807.776923 412.097543 199.167004
97 12672.0 807.776923 412.097543 199.264875
[98 rows x 4 columns]
@@ -314,7 +314,7 @@ In the above plot, we can see that:
.. rst-class:: sphx-glr-timing
**Total running time of the script:** ( 1 minutes 12.656 seconds)
**Total running time of the script:** ( 1 minutes 12.573 seconds)
.. _sphx_glr_download_getting-started_tutorials_02-fused-softmax.py:

50
_sources/getting-started/tutorials/03-matrix-multiplication.rst.txt
View File

@@ -463,36 +463,36 @@ We can now compare the performance of our kernel against that of cuBLAS. Here we
matmul-performance:
M cuBLAS ... Triton Triton (+ LeakyReLU)
0 256.0 2.730667 ... 2.978909 2.978909
1 384.0 7.372800 ... 8.507077 8.507077
1 384.0 7.372800 ... 7.899428 7.899428
2 512.0 14.563555 ... 16.384000 16.384000
3 640.0 22.260869 ... 24.380953 24.380953
4 768.0 32.768000 ... 34.028308 34.028308
5 896.0 39.025776 ... 40.140799 39.025776
6 1024.0 49.932191 ... 53.773130 52.428801
7 1152.0 44.566925 ... 46.656000 45.938215
8 1280.0 51.200001 ... 56.109587 56.109587
9 1408.0 64.138541 ... 66.485074 65.684049
10 1536.0 79.526831 ... 78.643199 78.643199
7 1152.0 44.566925 ... 46.656000 46.656000
8 1280.0 51.200001 ... 56.888887 56.888887
9 1408.0 64.138541 ... 67.305878 66.485074
10 1536.0 80.430545 ... 79.526831 78.643199
11 1664.0 63.372618 ... 62.929456 62.492442
12 1792.0 72.983276 ... 72.512412 72.512412
13 1920.0 69.467336 ... 67.764707 68.776119
14 2048.0 73.262953 ... 76.959706 76.608294
15 2176.0 81.803444 ... 83.848535 85.269692
16 2304.0 68.056616 ... 77.057651 77.057651
17 2432.0 71.305746 ... 83.864074 85.134737
18 2560.0 78.019048 ... 79.727497 79.533982
19 2688.0 83.552988 ... 84.861423 88.422041
20 2816.0 83.873477 ... 83.392363 83.712490
21 2944.0 83.060049 ... 83.198715 80.380696
22 3072.0 81.825298 ... 89.451983 87.787755
23 3200.0 79.800499 ... 87.791493 94.534716
24 3328.0 80.707733 ... 81.346098 84.596116
25 3456.0 82.015834 ... 87.727494 87.536988
26 3584.0 87.381330 ... 97.100854 89.376337
27 3712.0 83.386762 ... 84.301560 83.947349
28 3840.0 83.465663 ... 90.058629 91.322872
29 3968.0 93.006050 ... 91.678389 91.403695
30 4096.0 93.727466 ... 88.944818 93.466385
12 1792.0 72.983276 ... 72.512412 72.047592
13 1920.0 68.098521 ... 68.098521 69.818184
14 2048.0 73.908442 ... 76.959706 76.608294
15 2176.0 83.500614 ... 82.473969 85.632545
16 2304.0 68.446623 ... 77.307030 77.057651
17 2432.0 71.305746 ... 85.134737 85.134737
18 2560.0 77.833728 ... 79.727497 72.335543
19 2688.0 82.823267 ... 88.422041 85.627666
20 2816.0 83.552120 ... 80.320825 83.074685
21 2944.0 81.431424 ... 83.198715 82.102191
22 3072.0 82.540970 ... 86.712254 86.978653
23 3200.0 84.432717 ... 95.380032 89.510493
24 3328.0 83.130825 ... 84.695641 84.995628
25 3456.0 80.220468 ... 85.494768 89.480098
26 3584.0 87.893835 ... 90.549237 96.372338
27 3712.0 85.822459 ... 86.416391 87.783251
28 3840.0 84.744825 ... 85.070769 84.744825
29 3968.0 86.539882 ... 84.502282 84.270676
30 4096.0 93.727466 ... 91.428970 91.242506
[31 rows x 5 columns]
@@ -502,7 +502,7 @@ We can now compare the performance of our kernel against that of cuBLAS. Here we
.. rst-class:: sphx-glr-timing
**Total running time of the script:** ( 2 minutes 20.024 seconds)
**Total running time of the script:** ( 2 minutes 10.371 seconds)
.. _sphx_glr_download_getting-started_tutorials_03-matrix-multiplication.py:

2
_sources/getting-started/tutorials/04-low-memory-dropout.rst.txt
View File

@@ -238,7 +238,7 @@ References
.. rst-class:: sphx-glr-timing
**Total running time of the script:** ( 0 minutes 0.184 seconds)
**Total running time of the script:** ( 0 minutes 0.209 seconds)
.. _sphx_glr_download_getting-started_tutorials_04-low-memory-dropout.py:

10
_sources/getting-started/tutorials/sg_execution_times.rst.txt
View File

@@ -5,14 +5,14 @@
Computation times
=================
**03:43.876** total execution time for **getting-started_tutorials** files:
**03:34.175** total execution time for **getting-started_tutorials** files:
+---------------------------------------------------------------------------------------------------------+-----------+--------+
| :ref:`sphx_glr_getting-started_tutorials_03-matrix-multiplication.py` (``03-matrix-multiplication.py``) | 02:20.024 | 0.0 MB |
| :ref:`sphx_glr_getting-started_tutorials_03-matrix-multiplication.py` (``03-matrix-multiplication.py``) | 02:10.371 | 0.0 MB |
+---------------------------------------------------------------------------------------------------------+-----------+--------+
| :ref:`sphx_glr_getting-started_tutorials_02-fused-softmax.py` (``02-fused-softmax.py``) | 01:12.656 | 0.0 MB |
| :ref:`sphx_glr_getting-started_tutorials_02-fused-softmax.py` (``02-fused-softmax.py``) | 01:12.573 | 0.0 MB |
+---------------------------------------------------------------------------------------------------------+-----------+--------+
| :ref:`sphx_glr_getting-started_tutorials_01-vector-add.py` (``01-vector-add.py``) | 00:11.013 | 0.0 MB |
| :ref:`sphx_glr_getting-started_tutorials_01-vector-add.py` (``01-vector-add.py``) | 00:11.022 | 0.0 MB |
+---------------------------------------------------------------------------------------------------------+-----------+--------+
| :ref:`sphx_glr_getting-started_tutorials_04-low-memory-dropout.py` (``04-low-memory-dropout.py``) | 00:00.184 | 0.0 MB |
| :ref:`sphx_glr_getting-started_tutorials_04-low-memory-dropout.py` (``04-low-memory-dropout.py``) | 00:00.209 | 0.0 MB |
+---------------------------------------------------------------------------------------------------------+-----------+--------+

8
getting-started/tutorials/01-vector-add.html
View File

@@ -320,13 +320,13 @@ for different problem sizes.</p>
<p class="sphx-glr-script-out">Out:</p>
<div class="sphx-glr-script-out highlight-none notranslate"><div class="highlight"><pre><span></span>vector-add-performance:
size Triton Torch
0 4096.0 9.540372 9.600000
0 4096.0 9.600000 9.600000
1 8192.0 19.200000 19.200000
2 16384.0 38.400001 38.400001
3 32768.0 76.800002 76.800002
4 65536.0 127.999995 127.999995
5 131072.0 219.428568 219.428568
6 262144.0 341.333321 384.000001
5 131072.0 255.999991 219.428568
6 262144.0 341.333321 341.333321
7 524288.0 472.615390 472.615390
8 1048576.0 614.400016 614.400016
9 2097152.0 722.823517 722.823517
@@ -338,7 +338,7 @@ for different problem sizes.</p>
15 134217728.0 849.737435 850.656574
</pre></div>
</div>
<p class="sphx-glr-timing"><strong>Total running time of the script:</strong> ( 0 minutes 11.013 seconds)</p>
<p class="sphx-glr-timing"><strong>Total running time of the script:</strong> ( 0 minutes 11.022 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-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>

8
getting-started/tutorials/02-fused-softmax.html
View File

@@ -377,13 +377,13 @@ We will then compare its performance against (1) <code class="code docutils lite
1 384.0 585.142862 585.142862 153.600004
2 512.0 630.153853 606.814814 154.566038
3 640.0 682.666684 640.000002 160.000000
4 768.0 702.171410 664.216187 162.754967
4 768.0 702.171410 664.216187 163.839992
.. ... ... ... ...
93 12160.0 810.666687 405.755985 199.038365
94 12288.0 810.754644 415.661740 199.298541
95 12416.0 809.189387 412.149375 198.854847
95 12416.0 809.189387 411.722274 198.954424
96 12544.0 807.661970 412.971190 199.209928
97 12672.0 807.776923 412.097543 199.167004
97 12672.0 807.776923 412.097543 199.264875
[98 rows x 4 columns]
</pre></div>
@@ -396,7 +396,7 @@ We will then compare its performance against (1) <code class="code docutils lite
Note however that the PyTorch <cite>softmax</cite> operation is more general and will works on tensors of any shape.</p></li>
</ul>
</div></blockquote>
<p class="sphx-glr-timing"><strong>Total running time of the script:</strong> ( 1 minutes 12.656 seconds)</p>
<p class="sphx-glr-timing"><strong>Total running time of the script:</strong> ( 1 minutes 12.573 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-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>

50
getting-started/tutorials/03-matrix-multiplication.html
View File

@@ -568,41 +568,41 @@ torch_output=tensor([[ 1.1045, -36.9688, 31.4688, ..., -11.3906, 24.4531, -3
<div class="sphx-glr-script-out highlight-none notranslate"><div class="highlight"><pre><span></span>matmul-performance:
M cuBLAS ... Triton Triton (+ LeakyReLU)
0 256.0 2.730667 ... 2.978909 2.978909
1 384.0 7.372800 ... 8.507077 8.507077
1 384.0 7.372800 ... 7.899428 7.899428
2 512.0 14.563555 ... 16.384000 16.384000
3 640.0 22.260869 ... 24.380953 24.380953
4 768.0 32.768000 ... 34.028308 34.028308
5 896.0 39.025776 ... 40.140799 39.025776
6 1024.0 49.932191 ... 53.773130 52.428801
7 1152.0 44.566925 ... 46.656000 45.938215
8 1280.0 51.200001 ... 56.109587 56.109587
9 1408.0 64.138541 ... 66.485074 65.684049
10 1536.0 79.526831 ... 78.643199 78.643199
7 1152.0 44.566925 ... 46.656000 46.656000
8 1280.0 51.200001 ... 56.888887 56.888887
9 1408.0 64.138541 ... 67.305878 66.485074
10 1536.0 80.430545 ... 79.526831 78.643199
11 1664.0 63.372618 ... 62.929456 62.492442
12 1792.0 72.983276 ... 72.512412 72.512412
13 1920.0 69.467336 ... 67.764707 68.776119
14 2048.0 73.262953 ... 76.959706 76.608294
15 2176.0 81.803444 ... 83.848535 85.269692
16 2304.0 68.056616 ... 77.057651 77.057651
17 2432.0 71.305746 ... 83.864074 85.134737
18 2560.0 78.019048 ... 79.727497 79.533982
19 2688.0 83.552988 ... 84.861423 88.422041
20 2816.0 83.873477 ... 83.392363 83.712490
21 2944.0 83.060049 ... 83.198715 80.380696
22 3072.0 81.825298 ... 89.451983 87.787755
23 3200.0 79.800499 ... 87.791493 94.534716
24 3328.0 80.707733 ... 81.346098 84.596116
25 3456.0 82.015834 ... 87.727494 87.536988
26 3584.0 87.381330 ... 97.100854 89.376337
27 3712.0 83.386762 ... 84.301560 83.947349
28 3840.0 83.465663 ... 90.058629 91.322872
29 3968.0 93.006050 ... 91.678389 91.403695
30 4096.0 93.727466 ... 88.944818 93.466385
12 1792.0 72.983276 ... 72.512412 72.047592
13 1920.0 68.098521 ... 68.098521 69.818184
14 2048.0 73.908442 ... 76.959706 76.608294
15 2176.0 83.500614 ... 82.473969 85.632545
16 2304.0 68.446623 ... 77.307030 77.057651
17 2432.0 71.305746 ... 85.134737 85.134737
18 2560.0 77.833728 ... 79.727497 72.335543
19 2688.0 82.823267 ... 88.422041 85.627666
20 2816.0 83.552120 ... 80.320825 83.074685
21 2944.0 81.431424 ... 83.198715 82.102191
22 3072.0 82.540970 ... 86.712254 86.978653
23 3200.0 84.432717 ... 95.380032 89.510493
24 3328.0 83.130825 ... 84.695641 84.995628
25 3456.0 80.220468 ... 85.494768 89.480098
26 3584.0 87.893835 ... 90.549237 96.372338
27 3712.0 85.822459 ... 86.416391 87.783251
28 3840.0 84.744825 ... 85.070769 84.744825
29 3968.0 86.539882 ... 84.502282 84.270676
30 4096.0 93.727466 ... 91.428970 91.242506
[31 rows x 5 columns]
</pre></div>
</div>
<p class="sphx-glr-timing"><strong>Total running time of the script:</strong> ( 2 minutes 20.024 seconds)</p>
<p class="sphx-glr-timing"><strong>Total running time of the script:</strong> ( 2 minutes 10.371 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-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>

2
getting-started/tutorials/04-low-memory-dropout.html
View File

@@ -370,7 +370,7 @@ to explore the <cite>triton/language/random</cite> folder!</p>
<dd><p>Nitish Srivastava and Geoffrey Hinton and Alex Krizhevsky and Ilya Sutskever and Ruslan Salakhutdinov, “Dropout: A Simple Way to Prevent Neural Networks from Overfitting”, JMLR 2014</p>
</dd>
</dl>
<p class="sphx-glr-timing"><strong>Total running time of the script:</strong> ( 0 minutes 0.184 seconds)</p>
<p class="sphx-glr-timing"><strong>Total running time of the script:</strong> ( 0 minutes 0.209 seconds)</p>
<div class="sphx-glr-footer class sphx-glr-footer-example docutils container" id="sphx-glr-download-getting-started-tutorials-04-low-memory-dropout-py">
<div class="sphx-glr-download sphx-glr-download-python docutils container">
<p><a class="reference download internal" download="" href="../../_downloads/c9aed78977a4c05741d675a38dde3d7d/04-low-memory-dropout.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">04-low-memory-dropout.py</span></code></a></p>

10
getting-started/tutorials/sg_execution_times.html
View File

@@ -174,7 +174,7 @@
<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>
<p><strong>03:43.876</strong> total execution time for <strong>getting-started_tutorials</strong> files:</p>
<p><strong>03:34.175</strong> total execution time for <strong>getting-started_tutorials</strong> files:</p>
<table class="docutils align-default">
<colgroup>
<col style="width: 85%" />
@@ -183,19 +183,19 @@
</colgroup>
<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>
<td><p>02:20.024</p></td>
<td><p>02:10.371</p></td>
<td><p>0.0 MB</p></td>
</tr>
<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>01:12.656</p></td>
<td><p>01:12.573</p></td>
<td><p>0.0 MB</p></td>
</tr>
<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:11.013</p></td>
<td><p>00:11.022</p></td>
<td><p>0.0 MB</p></td>
</tr>
<tr class="row-even"><td><p><a class="reference internal" href="04-low-memory-dropout.html#sphx-glr-getting-started-tutorials-04-low-memory-dropout-py"><span class="std std-ref">Low-Memory Dropout</span></a> (<code class="docutils literal notranslate"><span class="pre">04-low-memory-dropout.py</span></code>)</p></td>
<td><p>00:00.184</p></td>
<td><p>00:00.209</p></td>
<td><p>0.0 MB</p></td>
</tr>
</tbody>

2
searchindex.js
View File

File diff suppressed because one or more lines are too long