From cd30a99aa2efe6af67967c9d5cbd4342768d21b8 Mon Sep 17 00:00:00 2001 From: Philippe Tillet Date: Thu, 12 May 2022 12:28:15 -0700 Subject: [PATCH] [TUTORIALS] fixed formatting --- python/tutorials/05-layer-norm.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/python/tutorials/05-layer-norm.py b/python/tutorials/05-layer-norm.py index 6581c809e..a03fa2cf6 100644 --- a/python/tutorials/05-layer-norm.py +++ b/python/tutorials/05-layer-norm.py @@ -3,9 +3,11 @@ Layer Normalization ==================== """ +import torch + import triton import triton.language as tl -import torch + try: # This is https://github.com/NVIDIA/apex, NOT the apex on PyPi, so it # should not be added to extras_require in setup.py.