Deprecation of Triton-C and Replacement by decorated Python functions (#86)

This PR implements a major overhaul of the frontend for Triton, and replaces Triton-C by a pure Python API in which kernels are defined as @triton.jit decorated functions. The documentation and tutorials have also been updated to accommodate these changes.

See documentations for more information on the new API
This commit is contained in:
Philippe Tillet
2021-04-20 22:29:40 -04:00
committed by Philippe Tillet
parent 1fdb465b71
commit 39f4730305
91 changed files with 4500 additions and 13008 deletions

View File

@@ -0,0 +1,117 @@
Python API
===========
.. currentmodule:: triton
Programming Model
-------------------
.. autosummary::
:toctree: generated
:nosignatures:
program_id
num_programs
Creation Ops
-------------
.. autosummary::
:toctree: generated
:nosignatures:
arange
zeros
Shape Manipulation Ops
-----------------------
.. autosummary::
:toctree: generated
:nosignatures:
broadcast_to
reshape
ravel
Linear Algebra Ops
-------------------
.. autosummary::
:toctree: generated
:nosignatures:
dot
Memory Ops
--------------------
.. autosummary::
:toctree: generated
:nosignatures:
load
store
atomic_cas
atomic_xchg
Indexing Ops
--------------
.. autosummary::
:toctree: generated
:nosignatures:
where
Math Ops
----------
.. autosummary::
:toctree: generated
:nosignatures:
exp
log
sigmoid
softmax
Reduction Ops
---------------
.. autosummary::
:toctree: generated
:nosignatures:
max
min
sum
Comparison ops
---------------
.. autosummary::
:toctree: generated
:nosignatures:
minimum
maximum
Compiler Hint Ops
-------------------
.. autosummary::
:toctree: generated
:nosignatures:
multiple_of