mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-15 19:36:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

#jax Adds a missing comma to Pallas Quickstart

Browse Source 
PiperOrigin-RevId: 689907976
This commit is contained in:
Jim Lin 2024-10-25 14:13:28 -07:00 committed by jax authors
parent 94440c74c8
commit e4eca9ec59
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
docs/pallas/quickstart.ipynb
View File

@ -412,7 +412,7 @@
"\n",
"For `x`, we use `BlockSpec((512, 1024), lambda i, j: (i, 0))` -- this\n",
"carves `x` up into \"row\" blocks.\n",
"To see this see how both program instances\n",
"To see this, see how both program instances\n",
"`(1, 0)` and `(1, 1)` pick the `(1, 0)` block in `x`.\n",
"For `y`, we use a transposed version `BlockSpec((1024, 512), lambda i, j: (0, j))`.\n",
"Finally, for `z` we use `BlockSpec((512, 512), lambda i, j: (i, j))`.\n",

2
docs/pallas/quickstart.md
View File

@ -294,7 +294,7 @@ To express this, we'd first use a `(2, 2)` grid (one block for each program).
For `x`, we use `BlockSpec((512, 1024), lambda i, j: (i, 0))` -- this
carves `x` up into "row" blocks.
To see this see how both program instances
To see this, see how both program instances
`(1, 0)` and `(1, 1)` pick the `(1, 0)` block in `x`.
For `y`, we use a transposed version `BlockSpec((1024, 512), lambda i, j: (0, j))`.
Finally, for `z` we use `BlockSpec((512, 512), lambda i, j: (i, j))`.