2021-04-23 16:42:55 -04:00
|
|
|
triton.language
|
|
|
|
|
================
|
2021-04-21 01:40:29 -04:00
|
|
|
|
2021-04-23 16:42:55 -04:00
|
|
|
.. currentmodule:: triton.language
|
2021-04-21 01:40:29 -04:00
|
|
|
|
|
|
|
|
|
|
|
|
|
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
|
2021-08-14 18:05:36 +00:00
|
|
|
cos
|
|
|
|
|
sin
|
|
|
|
|
sqrt
|
2021-04-21 01:40:29 -04:00
|
|
|
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
|
