mirror of
https://github.com/ROCm/ROCm.git
synced 2026-02-21 03:00:39 -05:00
71a8544ce7
- Move atomic_cas and atomic_xchg to "atomic ops" section of documentation. - Don't talk about the `cmp` operand for operations which don't have it. - Document the `sem` operand. - :code:`foo` and ``foo`` don't work inside a :type: annotation, apparently. (They are rendered literally, instead of being treated as a formatting command.) Get rid of them. - Format the bulleted lists in the load/store operations as intended.
195 lines
2.2 KiB
ReStructuredText
195 lines
2.2 KiB
ReStructuredText
triton.language
|
|
===============
|
|
|
|
.. currentmodule:: triton.language
|
|
|
|
|
|
Programming Model
|
|
-----------------
|
|
|
|
.. autosummary::
|
|
:toctree: generated
|
|
:nosignatures:
|
|
|
|
program_id
|
|
num_programs
|
|
|
|
|
|
Creation Ops
|
|
------------
|
|
|
|
.. autosummary::
|
|
:toctree: generated
|
|
:nosignatures:
|
|
|
|
arange
|
|
cat
|
|
full
|
|
zeros
|
|
|
|
|
|
Shape Manipulation Ops
|
|
----------------------
|
|
|
|
.. autosummary::
|
|
:toctree: generated
|
|
:nosignatures:
|
|
|
|
broadcast
|
|
broadcast_to
|
|
expand_dims
|
|
ravel
|
|
reshape
|
|
trans
|
|
view
|
|
|
|
|
|
Linear Algebra Ops
|
|
------------------
|
|
|
|
.. autosummary::
|
|
:toctree: generated
|
|
:nosignatures:
|
|
|
|
dot
|
|
|
|
|
|
Memory Ops
|
|
----------
|
|
|
|
.. autosummary::
|
|
:toctree: generated
|
|
:nosignatures:
|
|
|
|
load
|
|
store
|
|
|
|
|
|
Indexing Ops
|
|
------------
|
|
|
|
.. autosummary::
|
|
:toctree: generated
|
|
:nosignatures:
|
|
|
|
where
|
|
|
|
|
|
Math Ops
|
|
--------
|
|
|
|
.. autosummary::
|
|
:toctree: generated
|
|
:nosignatures:
|
|
|
|
abs
|
|
exp
|
|
log
|
|
fdiv
|
|
cos
|
|
sin
|
|
sqrt
|
|
sigmoid
|
|
softmax
|
|
umulhi
|
|
|
|
|
|
Reduction Ops
|
|
-------------
|
|
|
|
.. autosummary::
|
|
:toctree: generated
|
|
:nosignatures:
|
|
|
|
argmax
|
|
argmin
|
|
max
|
|
min
|
|
reduce
|
|
sum
|
|
xor_sum
|
|
|
|
Scan Ops
|
|
-------------
|
|
|
|
.. autosummary::
|
|
:toctree: generated
|
|
:nosignatures:
|
|
|
|
associative_scan
|
|
cumsum
|
|
cumprod
|
|
|
|
Atomic Ops
|
|
----------
|
|
|
|
.. autosummary::
|
|
:toctree: generated
|
|
:nosignatures:
|
|
|
|
atomic_add
|
|
atomic_cas
|
|
atomic_max
|
|
atomic_min
|
|
atomic_xchg
|
|
|
|
|
|
Comparison ops
|
|
--------------
|
|
|
|
.. autosummary::
|
|
:toctree: generated
|
|
:nosignatures:
|
|
|
|
minimum
|
|
maximum
|
|
|
|
.. _Random Number Generation:
|
|
|
|
Random Number Generation
|
|
------------------------
|
|
|
|
.. autosummary::
|
|
:toctree: generated
|
|
:nosignatures:
|
|
|
|
randint4x
|
|
randint
|
|
rand
|
|
randn
|
|
|
|
|
|
Compiler Hint Ops
|
|
-----------------
|
|
|
|
.. autosummary::
|
|
:toctree: generated
|
|
:nosignatures:
|
|
|
|
debug_barrier
|
|
max_constancy
|
|
max_contiguous
|
|
multiple_of
|
|
|
|
Debug Ops
|
|
-----------------
|
|
|
|
.. autosummary::
|
|
:toctree: generated
|
|
:nosignatures:
|
|
|
|
static_print
|
|
static_assert
|
|
device_print
|
|
device_assert
|
|
|
|
Iterators
|
|
-----------------
|
|
|
|
.. autosummary::
|
|
:toctree: generated
|
|
:nosignatures:
|
|
|
|
static_range
|
|
multiple_of
|