gaspersic/triton
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

[TUTORIALS] fixed formatting

Browse Source
This commit is contained in:
Philippe Tillet
2022-05-12 12:28:15 -07:00
parent d87435e536
commit cd30a99aa2
1 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
python/tutorials/05-layer-norm.py
View File

@@ -3,9 +3,11 @@ Layer Normalization
==================== ====================
""" """
import torch
import triton import triton
import triton.language as tl import triton.language as tl
import torch
try: try:
# This is https://github.com/NVIDIA/apex, NOT the apex on PyPi, so it # This is https://github.com/NVIDIA/apex, NOT the apex on PyPi, so it
# should not be added to extras_require in setup.py. # should not be added to extras_require in setup.py.