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
2022-02-11 00:40:00 +00:00
parent 30f8e01add
commit 7c47f3e043
158 changed files with 262 additions and 262 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -328,19 +328,19 @@ for different problem sizes.</p>
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
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
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 43.473 seconds)</p>
<p class="sphx-glr-timing"><strong>Total running time of the script:</strong> ( 1 minutes 40.155 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>

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

@@ -369,17 +369,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 188.321838
0 256.0 512.000001 546.133347 186.181817
1 384.0 614.400016 585.142862 153.600004
2 512.0 655.360017 606.814814 154.566038
3 640.0 706.206879 640.000002 160.000000
4 768.0 722.823517 664.216187 162.754967
.. ... ... ... ...
93 12160.0 815.765209 406.179533 198.936606
94 12288.0 814.111783 415.661740 199.197579
93 12160.0 814.058574 406.179533 198.936606
94 12288.0 814.111783 415.222812 199.197579
95 12416.0 814.163950 412.149375 198.755369
96 12544.0 814.214963 412.971190 199.012395
97 12672.0 814.265046 412.097543 198.971549
96 12544.0 814.214963 412.546756 199.012395
97 12672.0 814.265046 411.679167 199.069228
[98 rows x 4 columns]
</pre></div>
@@ -392,7 +392,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 21.420 seconds)</p>
<p class="sphx-glr-timing"><strong>Total running time of the script:</strong> ( 3 minutes 20.786 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
master/getting-started/tutorials/03-matrix-multiplication.html
View File

@@ -564,42 +564,42 @@ torch_output=tensor([[ 1.1045, -36.9688, 31.4688, ..., -11.3906, 24.4531, -3
<p class="sphx-glr-script-out">Out:</p>
<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
2 512.0 14.563555 ... 16.384000 15.420235
0 256.0 2.730667 ... 3.276800 2.978909
1 384.0 7.372800 ... 8.507077 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
5 896.0 39.025776 ... 39.025776 39.025776
6 1024.0 49.932191 ... 52.428801 52.428801
7 1152.0 45.242181 ... 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 79.526831
11 1664.0 62.929456 ... 62.061463 62.061463
12 1792.0 72.512412 ... 72.047592 72.047592
13 1920.0 69.467336 ... 70.172588 70.172588
10 1536.0 79.526831 ... 79.526831 79.526831
11 1664.0 62.929456 ... 62.492442 62.061463
12 1792.0 72.983276 ... 72.047592 72.047592
13 1920.0 68.776119 ... 70.172588 70.172588
14 2048.0 73.262953 ... 76.608294 76.260072
15 2176.0 83.155572 ... 85.998493 85.269692
16 2304.0 68.251065 ... 76.319081 76.809875
17 2432.0 71.487187 ... 74.521127 84.877538
18 2560.0 78.019048 ... 81.310171 81.108913
19 2688.0 83.369354 ... 89.254248 89.676257
20 2816.0 83.392363 ... 82.135981 83.392363
21 2944.0 81.967162 ... 81.967162 81.698415
22 3072.0 80.659693 ... 88.750943 88.750943
23 3200.0 84.432717 ... 94.955488 95.096582
24 3328.0 82.939284 ... 84.695641 81.254285
25 3456.0 80.300370 ... 82.435141 88.497878
26 3584.0 83.177979 ... 88.586589 94.947616
27 3712.0 85.675250 ... 89.035062 87.170458
28 3840.0 78.433173 ... 84.355978 91.511791
29 3968.0 86.849777 ... 90.791620 84.040329
30 4096.0 93.077479 ... 86.480498 83.365047
15 2176.0 83.155572 ... 85.998493 85.632545
16 2304.0 68.643310 ... 76.319081 75.834511
17 2432.0 71.125224 ... 74.521127 84.621881
18 2560.0 77.833728 ... 81.108913 81.108913
19 2688.0 83.186525 ... 89.888756 89.464755
20 2816.0 83.074685 ... 82.916747 81.293956
21 2944.0 81.967162 ... 82.237674 81.298583
22 3072.0 81.238312 ... 88.612060 88.890270
23 3200.0 84.656085 ... 95.665176 94.674553
24 3328.0 83.468170 ... 82.843841 84.695641
25 3456.0 81.683457 ... 90.281712 90.994998
26 3584.0 84.986191 ... 87.296493 95.047985
27 3712.0 82.355598 ... 88.718781 83.040189
28 3840.0 84.228485 ... 91.701494 85.465227
29 3968.0 91.062642 ... 85.271796 89.394823
30 4096.0 87.352901 ... 88.011627 87.154371
[31 rows x 5 columns]
</pre></div>
</div>
<p class="sphx-glr-timing"><strong>Total running time of the script:</strong> ( 6 minutes 1.040 seconds)</p>
<p class="sphx-glr-timing"><strong>Total running time of the script:</strong> ( 5 minutes 55.972 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
master/getting-started/tutorials/04-low-memory-dropout.html
View File

@@ -372,7 +372,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.482 seconds)</p>
<p class="sphx-glr-timing"><strong>Total running time of the script:</strong> ( 0 minutes 0.480 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>

56
master/getting-started/tutorials/05-layer-norm.html
View File

@@ -194,36 +194,36 @@ to download the full example code</p>
<p class="sphx-glr-script-out">Out:</p>
<div class="sphx-glr-script-out highlight-none notranslate"><div class="highlight"><pre><span></span>layer-norm-backward:
N Triton Torch Apex
0 1024.0 307.200008 98.303995 307.200008
0 1024.0 307.200008 98.303995 303.407414
1 1536.0 347.773587 134.540150 341.333333
2 2048.0 420.102553 161.684218 325.509933
3 2560.0 458.507457 181.775141 325.079368
4 3072.0 511.999982 192.501302 320.556515
5 3584.0 551.384634 208.271186 311.652167
2 2048.0 420.102553 161.684218 323.368435
3 2560.0 455.111129 181.775141 325.079368
4 3072.0 511.999982 192.501302 319.168834
5 3584.0 547.872604 208.271186 311.652167
6 4096.0 568.231237 220.412561 299.707322
7 4608.0 507.302750 232.825259 287.999990
8 5120.0 529.655159 242.845844 287.775181
9 5632.0 545.032265 243.545956 289.438969
10 6144.0 548.163546 248.661056 286.879370
7 4608.0 504.986315 232.825259 286.507772
8 5120.0 527.381977 242.845844 284.444444
9 5632.0 542.843364 243.545956 289.438969
10 6144.0 546.133354 248.661056 286.322318
11 6656.0 534.260858 256.000009 285.767438
12 7168.0 507.469040 260.654538 286.242939
13 7680.0 479.999983 262.564106 278.850215
14 8192.0 462.607053 267.493874 284.939124
15 8704.0 417.791980 267.815384 284.599455
16 9216.0 431.157889 272.729961 289.129410
17 9728.0 439.683593 280.278512 290.027323
18 10240.0 450.109870 286.433562 290.153487
19 10752.0 426.525614 247.172406 290.922209
20 11264.0 427.071098 245.760001 286.676558
21 11776.0 423.089806 249.888595 288.981596
22 12288.0 419.504980 254.673582 294.323369
23 12800.0 414.016170 253.674644 288.180121
24 13312.0 411.181478 252.759501 289.916513
25 13824.0 404.112047 257.190689 292.056329
26 14336.0 393.215988 254.485198 286.719986
27 14848.0 385.245405 257.665934 289.246765
28 15360.0 373.874218 257.970599 286.211174
29 15872.0 370.913333 261.806182 289.899545
12 7168.0 507.469040 260.260201 285.767449
13 7680.0 479.999983 262.564106 279.272719
14 8192.0 462.607053 267.493874 284.526763
15 8704.0 417.791980 267.815384 284.987724
16 9216.0 431.157889 272.059034 288.751954
17 9728.0 438.857162 280.278512 290.027323
18 10240.0 449.287041 286.767793 290.496460
19 10752.0 426.525614 247.172406 290.594591
20 11264.0 426.397479 245.536784 286.676558
21 11776.0 423.089806 249.667843 288.981596
22 12288.0 419.504980 254.673582 294.029924
23 12800.0 414.016170 253.674644 289.538159
24 13312.0 411.181478 252.559690 289.916513
25 13824.0 404.604870 257.390218 292.056329
26 14336.0 393.665895 254.297107 286.959121
27 14848.0 385.245405 257.665934 289.481735
28 15360.0 373.495460 257.970599 287.102804
29 15872.0 371.637071 261.626369 289.899545
</pre></div>
</div>
<div class="line-block">
@@ -487,7 +487,7 @@ to download the full example code</p>
<span class="n">bench_layer_norm</span><span class="o">.</span><span class="n">run</span><span class="p">(</span><span class="n">save_path</span><span class="o">=</span><span class="s1">&#39;.&#39;</span><span class="p">,</span> <span class="n">print_data</span><span class="o">=</span><span class="kc">True</span><span class="p">)</span>
</pre></div>
</div>
<p class="sphx-glr-timing"><strong>Total running time of the script:</strong> ( 2 minutes 12.002 seconds)</p>
<p class="sphx-glr-timing"><strong>Total running time of the script:</strong> ( 2 minutes 11.400 seconds)</p>
<div class="sphx-glr-footer class sphx-glr-footer-example docutils container" id="sphx-glr-download-getting-started-tutorials-05-layer-norm-py">
<div class="sphx-glr-download sphx-glr-download-python docutils container">
<p><a class="reference download internal" download="" href="../../_downloads/935c0dd0fbeb4b2e69588471cbb2d4b2/05-layer-norm.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">05-layer-norm.py</span></code></a></p>

12
master/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>13:18.417</strong> total execution time for <strong>getting-started_tutorials</strong> files:</p>
<p><strong>13:08.793</strong> total execution time for <strong>getting-started_tutorials</strong> files:</p>
<table class="docutils align-default">
<colgroup>
<col style="width: 85%" />
@@ -183,23 +183,23 @@
</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:01.040</p></td>
<td><p>05:55.972</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:21.420</p></td>
<td><p>03:20.786</p></td>
<td><p>0.0 MB</p></td>
</tr>
<tr class="row-odd"><td><p><a class="reference internal" href="05-layer-norm.html#sphx-glr-getting-started-tutorials-05-layer-norm-py"><span class="std std-ref">Layer Normalization</span></a> (<code class="docutils literal notranslate"><span class="pre">05-layer-norm.py</span></code>)</p></td>
<td><p>02:12.002</p></td>
<td><p>02:11.400</p></td>
<td><p>0.0 MB</p></td>
</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>
<td><p>01:43.473</p></td>
<td><p>01:40.155</p></td>
<td><p>0.0 MB</p></td>
</tr>
<tr class="row-odd"><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.482</p></td>
<td><p>00:00.480</p></td>
<td><p>0.0 MB</p></td>
</tr>
</tbody>