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

[DOCS] Fixed typo: triton.function -> torch.autograd.Function

Browse Source
This commit is contained in:
Philippe Tillet
2020-03-13 11:42:43 -04:00
committed by Philippe Tillet
parent a5e3397e6e
commit 55c800e632
2 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
docs/tutorials/putting-it-all-together.rst
View File

@@ -15,7 +15,7 @@ The PyTriton API provides a :code:`triton.function` class which automatically ha
import triton
# Entry point
class _dot(triton.function):
class _dot(torch.autograd.Function):
@staticmethod
# Forward Pass
@@ -170,7 +170,7 @@ Creating custom operations for Triton and PyTorch is very similar; programmers h
return da, db, None, None, None, None, None, None, None
A callable operation can be created using the :code:`apply` method of our :code:`triton.function` class.
A callable operation can be created using the :code:`apply` method of the :code:`torch.autograd.Function` class.
.. code:: python