From e4eca9ec5975982f89903082532b49ec4d56da9d Mon Sep 17 00:00:00 2001 From: Jim Lin Date: Fri, 25 Oct 2024 14:13:28 -0700 Subject: [PATCH] #jax Adds a missing comma to Pallas Quickstart PiperOrigin-RevId: 689907976 --- docs/pallas/quickstart.ipynb | 2 +- docs/pallas/quickstart.md | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/docs/pallas/quickstart.ipynb b/docs/pallas/quickstart.ipynb index 50464ce8f..af34d1674 100644 --- a/docs/pallas/quickstart.ipynb +++ b/docs/pallas/quickstart.ipynb @@ -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", diff --git a/docs/pallas/quickstart.md b/docs/pallas/quickstart.md index b9acd6497..e11868f5f 100644 --- a/docs/pallas/quickstart.md +++ b/docs/pallas/quickstart.md @@ -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))`.