diff --git a/_downloads/662999063954282841dc90b8945f85ce/tutorials_jupyter.zip b/_downloads/662999063954282841dc90b8945f85ce/tutorials_jupyter.zip index 1488600ad..a3ebafcb8 100644 Binary files a/_downloads/662999063954282841dc90b8945f85ce/tutorials_jupyter.zip and b/_downloads/662999063954282841dc90b8945f85ce/tutorials_jupyter.zip differ diff --git a/_downloads/763344228ae6bc253ed1a6cf586aa30d/tutorials_python.zip b/_downloads/763344228ae6bc253ed1a6cf586aa30d/tutorials_python.zip index c0dccebcb..55c0a9f38 100644 Binary files a/_downloads/763344228ae6bc253ed1a6cf586aa30d/tutorials_python.zip and b/_downloads/763344228ae6bc253ed1a6cf586aa30d/tutorials_python.zip differ diff --git a/_images/sphx_glr_01-vector-add_001.png b/_images/sphx_glr_01-vector-add_001.png index d103ad29b..423b366d6 100644 Binary files a/_images/sphx_glr_01-vector-add_001.png and b/_images/sphx_glr_01-vector-add_001.png differ diff --git a/_images/sphx_glr_01-vector-add_thumb.png b/_images/sphx_glr_01-vector-add_thumb.png index d2558f2c4..feac82077 100644 Binary files a/_images/sphx_glr_01-vector-add_thumb.png and b/_images/sphx_glr_01-vector-add_thumb.png differ diff --git a/_images/sphx_glr_02-fused-softmax_001.png b/_images/sphx_glr_02-fused-softmax_001.png index 4adad176a..d1386546d 100644 Binary files a/_images/sphx_glr_02-fused-softmax_001.png and b/_images/sphx_glr_02-fused-softmax_001.png differ diff --git a/_images/sphx_glr_02-fused-softmax_thumb.png b/_images/sphx_glr_02-fused-softmax_thumb.png index 95427a2a6..b2c2e08b6 100644 Binary files a/_images/sphx_glr_02-fused-softmax_thumb.png and b/_images/sphx_glr_02-fused-softmax_thumb.png differ diff --git a/_images/sphx_glr_03-matrix-multiplication_001.png b/_images/sphx_glr_03-matrix-multiplication_001.png index bb8bcb7dc..5366ee1a6 100644 Binary files a/_images/sphx_glr_03-matrix-multiplication_001.png and b/_images/sphx_glr_03-matrix-multiplication_001.png differ diff --git a/_images/sphx_glr_03-matrix-multiplication_thumb.png b/_images/sphx_glr_03-matrix-multiplication_thumb.png index 78282fd2c..72225d786 100644 Binary files a/_images/sphx_glr_03-matrix-multiplication_thumb.png and b/_images/sphx_glr_03-matrix-multiplication_thumb.png differ diff --git a/_sources/getting-started/tutorials/01-vector-add.rst.txt b/_sources/getting-started/tutorials/01-vector-add.rst.txt index 46c7371a3..d38c65ca4 100644 --- a/_sources/getting-started/tutorials/01-vector-add.rst.txt +++ b/_sources/getting-started/tutorials/01-vector-add.rst.txt @@ -234,7 +234,7 @@ We can now run the decorated function above. Pass `print_data=True` to see the p 0 4096.0 9.600000 9.600000 1 8192.0 19.200000 19.200000 2 16384.0 38.400001 38.400001 - 3 32768.0 63.999998 76.800002 + 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 341.333321 @@ -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 44.504 seconds) + **Total running time of the script:** ( 1 minutes 47.617 seconds) .. _sphx_glr_download_getting-started_tutorials_01-vector-add.py: diff --git a/_sources/getting-started/tutorials/02-fused-softmax.rst.txt b/_sources/getting-started/tutorials/02-fused-softmax.rst.txt index 73cc4b463..15c82cecb 100644 --- a/_sources/getting-started/tutorials/02-fused-softmax.rst.txt +++ b/_sources/getting-started/tutorials/02-fused-softmax.rst.txt @@ -286,16 +286,16 @@ 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 184.089886 - 1 384.0 585.142862 558.545450 151.703707 + 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 .. ... ... ... ... - 93 12160.0 810.666687 406.179533 199.038365 - 94 12288.0 810.754644 415.222812 199.298541 - 95 12416.0 809.189387 412.577363 198.854847 - 96 12544.0 807.661970 412.971190 199.061730 + 93 12160.0 810.666687 406.179533 199.140227 + 94 12288.0 810.754644 416.101597 199.399583 + 95 12416.0 809.189387 412.149375 199.054102 + 96 12544.0 807.661970 412.546756 199.308841 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:** ( 3 minutes 26.029 seconds) + **Total running time of the script:** ( 3 minutes 26.788 seconds) .. _sphx_glr_download_getting-started_tutorials_02-fused-softmax.py: diff --git a/_sources/getting-started/tutorials/03-matrix-multiplication.rst.txt b/_sources/getting-started/tutorials/03-matrix-multiplication.rst.txt index 04727690b..c0c53695e 100644 --- a/_sources/getting-started/tutorials/03-matrix-multiplication.rst.txt +++ b/_sources/getting-started/tutorials/03-matrix-multiplication.rst.txt @@ -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.978909 ... 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 37.971025 ... 39.025776 39.025776 + 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 66.485074 + 7 1152.0 44.566925 ... 46.656000 46.656000 + 8 1280.0 51.200001 ... 56.888887 56.109587 + 9 1408.0 64.138541 ... 67.305878 66.485074 10 1536.0 79.526831 ... 78.643199 78.643199 - 11 1664.0 62.929456 ... 62.492442 62.492442 - 12 1792.0 72.983276 ... 72.047592 72.047592 + 11 1664.0 62.929456 ... 62.061463 62.061463 + 12 1792.0 72.983276 ... 71.588687 72.047592 13 1920.0 69.120002 ... 70.530615 70.172588 14 2048.0 73.908442 ... 76.959706 76.608294 - 15 2176.0 83.155572 ... 85.998493 85.998493 - 16 2304.0 68.446623 ... 76.809875 76.809875 - 17 2432.0 71.305746 ... 74.918570 85.393507 - 18 2560.0 77.833728 ... 81.310171 80.709358 - 19 2688.0 83.552988 ... 89.676257 89.254248 - 20 2816.0 82.759409 ... 83.074685 83.392363 - 21 2944.0 82.784108 ... 81.832567 82.237674 - 22 3072.0 81.943708 ... 87.924073 89.030036 - 23 3200.0 82.368085 ... 89.012517 95.025983 - 24 3328.0 83.613586 ... 81.346098 83.905938 - 25 3456.0 81.766291 ... 90.943675 91.097818 - 26 3584.0 86.540320 ... 91.655413 87.381330 - 27 3712.0 85.163978 ... 84.088676 88.561477 - 28 3840.0 80.960466 ... 86.400002 91.701494 - 29 3968.0 86.083907 ... 91.198760 84.154440 - 30 4096.0 93.498941 ... 93.336389 89.181212 + 15 2176.0 83.500614 ... 86.367588 85.632545 + 16 2304.0 68.446623 ... 76.809875 77.057651 + 17 2432.0 71.305746 ... 74.521127 85.393507 + 18 2560.0 78.019048 ... 81.512437 81.108913 + 19 2688.0 83.922689 ... 89.464755 89.676257 + 20 2816.0 84.035084 ... 83.233226 83.552120 + 21 2944.0 82.373605 ... 82.784108 81.967162 + 22 3072.0 81.707223 ... 89.170242 87.924073 + 23 3200.0 84.656085 ... 95.451158 95.096582 + 24 3328.0 84.003845 ... 82.181847 83.710812 + 25 3456.0 81.849303 ... 91.097818 91.097818 + 26 3584.0 86.790921 ... 93.661869 95.047985 + 27 3712.0 84.159518 ... 86.044224 89.273764 + 28 3840.0 83.528704 ... 90.649182 87.011801 + 29 3968.0 89.003603 ... 86.480463 86.911637 + 30 4096.0 91.741443 ... 91.428970 86.258181 [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 4.767 seconds) + **Total running time of the script:** ( 5 minutes 38.718 seconds) .. _sphx_glr_download_getting-started_tutorials_03-matrix-multiplication.py: diff --git a/_sources/getting-started/tutorials/04-low-memory-dropout.rst.txt b/_sources/getting-started/tutorials/04-low-memory-dropout.rst.txt index f7149d658..fe5ecad7d 100644 --- a/_sources/getting-started/tutorials/04-low-memory-dropout.rst.txt +++ b/_sources/getting-started/tutorials/04-low-memory-dropout.rst.txt @@ -238,7 +238,7 @@ References .. rst-class:: sphx-glr-timing - **Total running time of the script:** ( 0 minutes 0.271 seconds) + **Total running time of the script:** ( 0 minutes 0.010 seconds) .. _sphx_glr_download_getting-started_tutorials_04-low-memory-dropout.py: diff --git a/_sources/getting-started/tutorials/sg_execution_times.rst.txt b/_sources/getting-started/tutorials/sg_execution_times.rst.txt index a6feaad6c..6f36d0542 100644 --- a/_sources/getting-started/tutorials/sg_execution_times.rst.txt +++ b/_sources/getting-started/tutorials/sg_execution_times.rst.txt @@ -5,14 +5,14 @@ Computation times ================= -**11:15.571** total execution time for **getting-started_tutorials** files: +**10:53.132** total execution time for **getting-started_tutorials** files: +---------------------------------------------------------------------------------------------------------+-----------+--------+ -| :ref:`sphx_glr_getting-started_tutorials_03-matrix-multiplication.py` (``03-matrix-multiplication.py``) | 06:04.767 | 0.0 MB | +| :ref:`sphx_glr_getting-started_tutorials_03-matrix-multiplication.py` (``03-matrix-multiplication.py``) | 05:38.718 | 0.0 MB | +---------------------------------------------------------------------------------------------------------+-----------+--------+ -| :ref:`sphx_glr_getting-started_tutorials_02-fused-softmax.py` (``02-fused-softmax.py``) | 03:26.029 | 0.0 MB | +| :ref:`sphx_glr_getting-started_tutorials_02-fused-softmax.py` (``02-fused-softmax.py``) | 03:26.788 | 0.0 MB | +---------------------------------------------------------------------------------------------------------+-----------+--------+ -| :ref:`sphx_glr_getting-started_tutorials_01-vector-add.py` (``01-vector-add.py``) | 01:44.504 | 0.0 MB | +| :ref:`sphx_glr_getting-started_tutorials_01-vector-add.py` (``01-vector-add.py``) | 01:47.617 | 0.0 MB | +---------------------------------------------------------------------------------------------------------+-----------+--------+ -| :ref:`sphx_glr_getting-started_tutorials_04-low-memory-dropout.py` (``04-low-memory-dropout.py``) | 00:00.271 | 0.0 MB | +| :ref:`sphx_glr_getting-started_tutorials_04-low-memory-dropout.py` (``04-low-memory-dropout.py``) | 00:00.010 | 0.0 MB | +---------------------------------------------------------------------------------------------------------+-----------+--------+ diff --git a/getting-started/tutorials/01-vector-add.html b/getting-started/tutorials/01-vector-add.html index 5a8e9df35..32bac86fb 100644 --- a/getting-started/tutorials/01-vector-add.html +++ b/getting-started/tutorials/01-vector-add.html @@ -323,7 +323,7 @@ for different problem sizes.
0 4096.0 9.600000 9.600000 1 8192.0 19.200000 19.200000 2 16384.0 38.400001 38.400001 -3 32768.0 63.999998 76.800002 +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 341.333321 @@ -338,7 +338,7 @@ for different problem sizes. 15 134217728.0 849.737435 850.656574 -Total running time of the script: ( 1 minutes 44.504 seconds)
+Total running time of the script: ( 1 minutes 47.617 seconds)