Skip to content

Commit

Permalink
Update tiles.rst
Browse files Browse the repository at this point in the history
  • Loading branch information
mmacklin committed Nov 29, 2024
1 parent cc4f284 commit fa0add3
Showing 1 changed file with 5 additions and 7 deletions.
12 changes: 5 additions & 7 deletions docs/modules/tiles.rst
Original file line number Diff line number Diff line change
Expand Up @@ -3,9 +3,7 @@ Tiles

.. warning:: Tile-based operations in Warp are under preview, APIs are subject to change.

Block-based programming models such as those in OpenAI Triton have proved to be effective ways of expressing high-performance kernels that can leverage cooperative operations on modern GPUs.

Warp 1.5.0 introduces tile extensions that expose a block-based programming to Warp kernels.
Block-based programming models such as those in OpenAI Triton have proved to be effective ways of expressing high-performance kernels that can leverage cooperative operations on modern GPUs. With Warp 1.5.0 developers now have access to new tile-based programming primitives in Warp kernels. Leveraging cuBLASDx and cuFFTDx, these new tools provide developers with efficient matrix multiplication and Fourier transforms for accelerated simulation and scientific computing.

Requirements
------------
Expand Down Expand Up @@ -35,7 +33,7 @@ In the following example, we launch a grid of threads where each block is respon
# load a row from global memory
t = wp.tile_load(array[i], i, TILE_SIZE)
s = wp.sum(t)
s = wp.tile_sum(t)
...
wp.launch_tiled(compute, dim=[a.shape[0]], inputs=[a], block_dim=TILE_THREADS)
Expand All @@ -62,7 +60,7 @@ In Warp, tile objects are 2D arrays of data where the tile elements may be scala
# load a 2d tile from global memory
t = wp.tile_load(array, i, j, m=TILE_M, n=TILE_N)
s = wp.sum(t)
s = wp.tile_sum(t)
...
wp.launch_tiled(compute, dim=[a.shape[0]/TILE_M, a.shape[1]/TILE_N], inputs=[a], block_dim=TILE_THREADS)
Expand Down Expand Up @@ -193,8 +191,8 @@ Maps/Reductions
* :func:`warp.tile_min`
* :func:`warp.tile_max`

MathDx
^^^^^^
Linear Algebra
^^^^^^^^^^^^^^

* :func:`warp.tile_matmul`
* :func:`warp.tile_transpose`
Expand Down

0 comments on commit fa0add3

Please sign in to comment.