triton.language¶
Programming Model¶
Returns the id of the current program instance along the given |
|
Returns the number of program instances launched along the given |
Creation Ops¶
Returns contiguous values within the open interval [ |
|
Returns a tensor filled with the scalar value 0 for the given |
Shape Manipulation Ops¶
Tries to broadcast the given tensor to a new |
|
Tries to reshape the given tensor to a new shape. |
|
Returns a contiguous flattened view of |
Memory Ops¶
Return a tensor of data whose values are, elementwise, loaded from memory at location defined by |
|
Stores |
|
Performs an atomic compare-and-swap at the memory location specified by |
|
Performs an atomic exchange at the memory location specified by |
Math Ops¶
Computes the element-wise exponential of |
|
Computes the element-wise natural logarithm of |
|
Computes the element-wise cosine of |
|
Computes the element-wise sine of |
|
Computes the element-wise square root of |
|
Computes the element-wise sigmoid of |
|
Computes the element-wise softmax of |
Reduction Ops¶
Returns the maximum of all elements in the |
|
Returns the minimum of all elements in the |
|
Returns the sum of all elements in the |
Atomic Ops¶
Performs an atomic compare-and-swap at the memory location specified by |
|
Performs an atomic add at the memory location specified by |
|
Performs an atomic max at the memory location specified by |
|
Performs an atomic min at the memory location specified by |
Comparison ops¶
Computes the element-wise minimum of |
|
Computes the element-wise maximum of |
Random Number Generation¶
Given a |
|
Given a |
|
Given a |
|
Given a |
Compiler Hint Ops¶
Let the compiler knows that the values in |