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
    atomic_cas
    atomic_xchg


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_cas
    atomic_add
    atomic_max
    atomic_min


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
