Python API¶
Programming Model¶
Returns the id of the current program instance along the given axis. Triton uses an SPMD model in which different @triton.jit functions run in parallel with different `program_id`s. |
|
Returns the number of program instances launched along the given axis. |
Creation Ops¶
Returns contiguous values within the open interval [start, end). |
|
Returns a block filled with the scalar value 0 and the given shape. |
Shape Manipulation Ops¶
Tries to broadcast a block to a new shape. |
|
Reshapes a block to a new shape. |
|
Memory Ops¶
Return a block of data whose values are, elementwise, loaded from memory at location defined by pointer. |
|
Stores value block of elements in memory, element-wise, at the memory locations specified by pointer. |
|