* the Free Software Foundation, either version 3 of the License, or * (at your option) any later version. * This program is distributed in the hope that it will be useful, * but WITHOUT ANY WARRANTY; ...
Example demonstrating the Triton execution model with 2D grids. Shows how to use 2D program IDs for grid mapping. offs_x = pid_x * BLOCK_N + tl.arange(0, BLOCK_N) offs_y = pid_y * BLOCK_M + ...