Adding Examples to the Helion Docs/site #330
Conversation
meta-cla
bot
added
the
CLA Signed
| @@ -0,0 +1,736 @@ | |||
| Helion Puzzles | |||
Co-authored-by: Jason Ansel <[email protected]>
Co-authored-by: Jason Ansel <[email protected]>
Changing example generation to use sphinx-gallery
There was a problem hiding this comment.
Can we make this a markdown file?
There was a problem hiding this comment.
sphinx gallery has issues with markdown files but I am investigating myst sphinx gallery which should work with markdown, will update
docs/requirements.txt
Outdated
There was a problem hiding this comment.
Our docs requirements are here:
Lines 31 to 38 in 6c5c4ca
Installed with pip install ".[docs]" (see docs/README.md)
Let's update that rather than creating a new file.
There was a problem hiding this comment.
Oh sure, will update that, did not see this. Thanks
| @@ -1,3 +1,14 @@ | |||
| """ | |||
| All-Gather Matrix Multiplication Example | |||
| ===============================>>>>>>> REPLACE | |||
There was a problem hiding this comment.
Thanks for catching that, will update it, not sure how I missed it but might explain a random issue I was having with this file rendering
| - ``jagged_dense_add.py``: Addition between a jagged tensor and a dense tensor | ||
| - ``jagged_mean.py``: Computing the mean of each row in a jagged tensor | ||
| - ``segment_reduction.py``: Segmented reduction operation | ||
| - ``moe_matmul_ogs.py``: Mixture-of-Experts matrix multiplication using Outer-Gather-Scatter |
There was a problem hiding this comment.
@yf225 is this the right section?
| @@ -1,3 +1,13 @@ | |||
| """ | |||
| Attention Mechanism Example | |||
There was a problem hiding this comment.
(thoughout) I think "Mechanism" is the wrong word here.
| @@ -1,4 +1,14 @@ | |||
| # Code based on https://github.com/pytorch-labs/helion/issues/237 | |||
There was a problem hiding this comment.
We should keep this citation
|
I think you also need to rebase to fix merge conflict. |
Co-authored-by: Jason Ansel <[email protected]>
Co-authored-by: Jason Ansel <[email protected]>
|
Lints/tests are failing |
PR adds examples to the helion docs site:
Helion Puzzles is taken from a bento notebook created internally
The rest are examples taken from the .pys taken from examples/ folder.
Updates include:
Some mock ups below built locally