Skip to content
Permalink
Browse files

[doc][pytriton] now showing full requirements of triton.function

  • Loading branch information
ptillet committed Oct 14, 2019
1 parent 0dc7313 commit 6f5f511a337c303d6e307a77b99c427f9cbe1a65
Showing with 10 additions and 0 deletions.
  1. +10 −0 docs/pytriton.md
@@ -73,8 +73,18 @@ The PyTriton API provides a `triton.function` class which automatically handles
```python
import triton
# Entry point
class _dot(triton.function):
@staticmethod
# Forward Pass
def forward(ctx, *args):
#...
@staticmethod
# Backward Pass
def backward(ctx, dy):
#...
```

### <span style="color:darkblue">Creation of Triton Kernels </span> <a name="creation-triton-kernel"></a>

0 comments on commit 6f5f511

Please sign in to comment.
You can’t perform that action at this time.