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-21 00:30:26 +00:00
parent 0f2f9d5e84
commit ecca67079b
19 changed files with 89 additions and 89 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: 24 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: 15 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: 37 KiB

After

Width:  |  Height:  |  Size: 37 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: 57 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: 33 KiB

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

@@ -233,8 +233,8 @@ We can now run the decorated function above. Pass `print_data=True` to see the p
size Triton Torch
0 4096.0 9.600000 9.600000
1 8192.0 19.200000 19.200000
2 16384.0 38.400001 31.999999
3 32768.0 63.999998 76.800002
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
@@ -244,7 +244,7 @@ We can now run the decorated function above. Pass `print_data=True` to see the p
10 4194304.0 780.190482 780.190482
11 8388608.0 812.429770 812.429770
12 16777216.0 833.084721 833.084721
13 33554432.0 842.004273 843.811163
13 33554432.0 842.004273 842.004273
14 67108864.0 847.448255 848.362445
15 134217728.0 849.737435 850.656574
@@ -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:** ( 1 minutes 52.287 seconds)
**Total running time of the script:** ( 1 minutes 52.819 seconds)
.. _sphx_glr_download_getting-started_tutorials_01-vector-add.py:

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

@@ -286,17 +286,17 @@ We will then compare its performance against (1) :code:`torch.softmax` and (2) t
softmax-performance:
N Triton Torch (native) Torch (jit)
0 256.0 512.000001 546.133347 186.181817
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 163.839992
0 256.0 512.000001 546.133347 188.321838
1 384.0 585.142862 585.142862 151.703707
2 512.0 655.360017 606.814814 154.566038
3 640.0 660.645170 640.000002 160.000000
4 768.0 702.171410 664.216187 162.754967
.. ... ... ... ...
93 12160.0 810.666687 406.179533 199.140227
94 12288.0 810.754644 415.661740 199.399583
93 12160.0 810.666687 405.755985 199.038365
94 12288.0 810.754644 415.661740 199.197579
95 12416.0 809.189387 412.149375 198.954424
96 12544.0 807.661970 412.971190 199.209928
97 12672.0 807.776923 412.097543 199.264875
96 12544.0 807.661970 412.971190 199.111113
97 12672.0 807.776923 412.516771 199.167004
[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:** ( 3 minutes 29.511 seconds)
**Total running time of the script:** ( 3 minutes 29.348 seconds)
.. _sphx_glr_download_getting-started_tutorials_02-fused-softmax.py:

48
_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 7.899428
1 384.0 7.372800 ... 8.507077 8.507077
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 ... 39.025776 37.971025
6 1024.0 49.932191 ... 52.428801 52.428801
5 896.0 37.971025 ... 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 66.485074
10 1536.0 79.526831 ... 79.526831 78.643199
11 1664.0 62.929456 ... 62.492442 62.061463
12 1792.0 72.983276 ... 72.512412 72.047592
13 1920.0 69.120002 ... 70.530615 70.172588
10 1536.0 80.430545 ... 78.643199 78.643199
11 1664.0 63.372618 ... 62.492442 62.061463
12 1792.0 72.983276 ... 72.047592 71.588687
13 1920.0 69.120002 ... 70.530615 70.530615
14 2048.0 73.908442 ... 76.959706 76.608294
15 2176.0 83.500614 ... 85.632545 85.269692
16 2304.0 68.446623 ... 76.809875 76.809875
17 2432.0 71.305746 ... 84.621881 84.877538
18 2560.0 77.649287 ... 80.709358 81.310171
19 2688.0 83.552988 ... 89.464755 89.464755
20 2816.0 83.392363 ... 83.392363 82.916747
21 2944.0 82.509987 ... 82.373605 82.237674
22 3072.0 82.301023 ... 88.750943 88.335577
23 3200.0 84.993363 ... 95.380032 94.955488
24 3328.0 83.130825 ... 84.200347 84.895397
25 3456.0 81.108217 ... 85.858966 85.494768
26 3584.0 86.707226 ... 97.947050 98.429551
27 3712.0 81.548851 ... 88.015279 88.326564
28 3840.0 80.139129 ... 89.187096 91.097196
29 3968.0 85.992909 ... 87.347124 88.103928
30 4096.0 91.805174 ... 92.948562 85.216761
15 2176.0 83.500614 ... 85.998493 85.632545
16 2304.0 68.446623 ... 76.809875 76.563695
17 2432.0 71.305746 ... 74.719317 84.367759
18 2560.0 77.833728 ... 80.709358 80.313727
19 2688.0 83.552988 ... 88.836198 89.676257
20 2816.0 84.035084 ... 83.552120 83.392363
21 2944.0 81.967162 ... 83.198715 81.298583
22 3072.0 81.943708 ... 89.170242 88.335577
23 3200.0 83.116885 ... 94.814812 94.395283
24 3328.0 82.275764 ... 84.895397 84.397770
25 3456.0 82.519518 ... 91.407671 88.304015
26 3584.0 86.540320 ... 87.548840 94.448944
27 3712.0 83.947349 ... 86.117243 87.590836
28 3840.0 83.465663 ... 84.679936 91.701494
29 3968.0 86.053553 ... 90.859224 85.932350
30 4096.0 93.271527 ... 88.359266 86.313653
[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:** ( 6 minutes 28.511 seconds)
**Total running time of the script:** ( 6 minutes 28.869 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.354 seconds)
**Total running time of the script:** ( 0 minutes 0.016 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
=================
**11:50.662** total execution time for **getting-started_tutorials** files:
**11:51.052** total execution time for **getting-started_tutorials** files:
+---------------------------------------------------------------------------------------------------------+-----------+--------+
| :ref:`sphx_glr_getting-started_tutorials_03-matrix-multiplication.py` (``03-matrix-multiplication.py``) | 06:28.511 | 0.0 MB |
| :ref:`sphx_glr_getting-started_tutorials_03-matrix-multiplication.py` (``03-matrix-multiplication.py``) | 06:28.869 | 0.0 MB |
+---------------------------------------------------------------------------------------------------------+-----------+--------+
| :ref:`sphx_glr_getting-started_tutorials_02-fused-softmax.py` (``02-fused-softmax.py``) | 03:29.511 | 0.0 MB |
| :ref:`sphx_glr_getting-started_tutorials_02-fused-softmax.py` (``02-fused-softmax.py``) | 03:29.348 | 0.0 MB |
+---------------------------------------------------------------------------------------------------------+-----------+--------+
| :ref:`sphx_glr_getting-started_tutorials_01-vector-add.py` (``01-vector-add.py``) | 01:52.287 | 0.0 MB |
| :ref:`sphx_glr_getting-started_tutorials_01-vector-add.py` (``01-vector-add.py``) | 01:52.819 | 0.0 MB |
+---------------------------------------------------------------------------------------------------------+-----------+--------+
| :ref:`sphx_glr_getting-started_tutorials_04-low-memory-dropout.py` (``04-low-memory-dropout.py``) | 00:00.354 | 0.0 MB |
| :ref:`sphx_glr_getting-started_tutorials_04-low-memory-dropout.py` (``04-low-memory-dropout.py``) | 00:00.016 | 0.0 MB |
+---------------------------------------------------------------------------------------------------------+-----------+--------+

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

@@ -322,8 +322,8 @@ for different problem sizes.</p>
size Triton Torch
0 4096.0 9.600000 9.600000
1 8192.0 19.200000 19.200000
2 16384.0 38.400001 31.999999
3 32768.0 63.999998 76.800002
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
@@ -333,12 +333,12 @@ for different problem sizes.</p>
10 4194304.0 780.190482 780.190482
11 8388608.0 812.429770 812.429770
12 16777216.0 833.084721 833.084721
13 33554432.0 842.004273 843.811163
13 33554432.0 842.004273 842.004273
14 67108864.0 847.448255 848.362445
15 134217728.0 849.737435 850.656574
</pre></div>
</div>
<p class="sphx-glr-timing"><strong>Total running time of the script:</strong> ( 1 minutes 52.287 seconds)</p>
<p class="sphx-glr-timing"><strong>Total running time of the script:</strong> ( 1 minutes 52.819 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>

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

@@ -373,17 +373,17 @@ We will then compare its performance against (1) <code class="code docutils lite
<p class="sphx-glr-script-out">Out:</p>
<div class="sphx-glr-script-out highlight-none notranslate"><div class="highlight"><pre><span></span>softmax-performance:
N Triton Torch (native) Torch (jit)
0 256.0 512.000001 546.133347 186.181817
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 163.839992
0 256.0 512.000001 546.133347 188.321838
1 384.0 585.142862 585.142862 151.703707
2 512.0 655.360017 606.814814 154.566038
3 640.0 660.645170 640.000002 160.000000
4 768.0 702.171410 664.216187 162.754967
.. ... ... ... ...
93 12160.0 810.666687 406.179533 199.140227
94 12288.0 810.754644 415.661740 199.399583
93 12160.0 810.666687 405.755985 199.038365
94 12288.0 810.754644 415.661740 199.197579
95 12416.0 809.189387 412.149375 198.954424
96 12544.0 807.661970 412.971190 199.209928
97 12672.0 807.776923 412.097543 199.264875
96 12544.0 807.661970 412.971190 199.111113
97 12672.0 807.776923 412.516771 199.167004
[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> ( 3 minutes 29.511 seconds)</p>
<p class="sphx-glr-timing"><strong>Total running time of the script:</strong> ( 3 minutes 29.348 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>

48
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 7.899428
1 384.0 7.372800 ... 8.507077 8.507077
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 ... 39.025776 37.971025
6 1024.0 49.932191 ... 52.428801 52.428801
5 896.0 37.971025 ... 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 66.485074
10 1536.0 79.526831 ... 79.526831 78.643199
11 1664.0 62.929456 ... 62.492442 62.061463
12 1792.0 72.983276 ... 72.512412 72.047592
13 1920.0 69.120002 ... 70.530615 70.172588
10 1536.0 80.430545 ... 78.643199 78.643199
11 1664.0 63.372618 ... 62.492442 62.061463
12 1792.0 72.983276 ... 72.047592 71.588687
13 1920.0 69.120002 ... 70.530615 70.530615
14 2048.0 73.908442 ... 76.959706 76.608294
15 2176.0 83.500614 ... 85.632545 85.269692
16 2304.0 68.446623 ... 76.809875 76.809875
17 2432.0 71.305746 ... 84.621881 84.877538
18 2560.0 77.649287 ... 80.709358 81.310171
19 2688.0 83.552988 ... 89.464755 89.464755
20 2816.0 83.392363 ... 83.392363 82.916747
21 2944.0 82.509987 ... 82.373605 82.237674
22 3072.0 82.301023 ... 88.750943 88.335577
23 3200.0 84.993363 ... 95.380032 94.955488
24 3328.0 83.130825 ... 84.200347 84.895397
25 3456.0 81.108217 ... 85.858966 85.494768
26 3584.0 86.707226 ... 97.947050 98.429551
27 3712.0 81.548851 ... 88.015279 88.326564
28 3840.0 80.139129 ... 89.187096 91.097196
29 3968.0 85.992909 ... 87.347124 88.103928
30 4096.0 91.805174 ... 92.948562 85.216761
15 2176.0 83.500614 ... 85.998493 85.632545
16 2304.0 68.446623 ... 76.809875 76.563695
17 2432.0 71.305746 ... 74.719317 84.367759
18 2560.0 77.833728 ... 80.709358 80.313727
19 2688.0 83.552988 ... 88.836198 89.676257
20 2816.0 84.035084 ... 83.552120 83.392363
21 2944.0 81.967162 ... 83.198715 81.298583
22 3072.0 81.943708 ... 89.170242 88.335577
23 3200.0 83.116885 ... 94.814812 94.395283
24 3328.0 82.275764 ... 84.895397 84.397770
25 3456.0 82.519518 ... 91.407671 88.304015
26 3584.0 86.540320 ... 87.548840 94.448944
27 3712.0 83.947349 ... 86.117243 87.590836
28 3840.0 83.465663 ... 84.679936 91.701494
29 3968.0 86.053553 ... 90.859224 85.932350
30 4096.0 93.271527 ... 88.359266 86.313653
[31 rows x 5 columns]
</pre></div>
</div>
<p class="sphx-glr-timing"><strong>Total running time of the script:</strong> ( 6 minutes 28.511 seconds)</p>
<p class="sphx-glr-timing"><strong>Total running time of the script:</strong> ( 6 minutes 28.869 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.354 seconds)</p>
<p class="sphx-glr-timing"><strong>Total running time of the script:</strong> ( 0 minutes 0.016 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>11:50.662</strong> total execution time for <strong>getting-started_tutorials</strong> files:</p>
<p><strong>11:51.052</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>06:28.511</p></td>
<td><p>06:28.869</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>03:29.511</p></td>
<td><p>03:29.348</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>01:52.287</p></td>
<td><p>01:52.819</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.354</p></td>
<td><p>00:00.016</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