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:
committed by
Philippe Tillet
parent
1fdb465b71
commit
39f4730305
117
docs/language-reference/python-api/index.rst
Normal file
117
docs/language-reference/python-api/index.rst
Normal 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
|
Reference in New Issue
Block a user