39f4730305
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
118 lines
1.3 KiB
ReStructuredText
118 lines
1.3 KiB
ReStructuredText
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
|