[DOCS] Updates and improvements (#87)

2021-04-22 10:27:02 -04:00
parent 39f4730305
commit 29e33e50b7
8 changed files with 195 additions and 70 deletions
--- a/python/tutorials/01-vector-add.py
+++ b/python/tutorials/01-vector-add.py
@@ -1,5 +1,3 @@
-import torch
-import triton
 """
 Vector Addition
 =================
@@ -14,6 +12,9 @@ In this tutorial, you will write a simple vector addition using Triton and learn
 # Compute Kernel
 # --------------------------

+import torch
+import triton
+

@triton.jit
 def _add(