Skip to content

Restructure spmd_types documentation into a Sphinx site - #4

Closed
ezyang wants to merge 1 commit into
mainfrom
export-D107149885
Closed

Restructure spmd_types documentation into a Sphinx site#4
ezyang wants to merge 1 commit into
mainfrom
export-D107149885

Conversation

@ezyang

@ezyang ezyang commented Jun 9, 2026

Copy link
Copy Markdown
Contributor

Summary:

Human summary

This starts writing some true docs for spmd_types. I slaved away on key_concepts.md and local_spmd_types.md so those would be the ones to focus in on, but there's also lots of other bits an bobs here and there.

AI summary

Reorganizes the spmd_types docs from two top-level Markdown files into a proper Sphinx documentation site under docs/.

DESIGN.md moves to docs/design.md (the full type-system specification), and the old MEGATRON_QUICKSTART.md is folded into the new docs/local_spmd_types.md, which carries the hands-on porting guide, the Megatron-to-spmd_types function mapping table, and the Invariant-vs-Replicate advice. New pages add docs/index.md, docs/key_concepts.md, docs/global_spmd_types.md, and an autodoc-backed API reference under docs/api/. docs/conf.py configures Sphinx (MyST + napoleon + autodoc, RTD theme).

README.md and CLAUDE.md references are updated to the new paths, CONTRIBUTING.md headings are normalized, and the PartitionSpec / TensorSharding docstrings in types.py are reformatted as literal blocks so they render correctly under autodoc. docs/_build/ is gitignored and a # noautodeps docs/BUCK is added so the Sphinx conf.py is not buckified.

Reviewed By: pianpwk

Differential Revision: D107149885

Summary:
# Human summary

This starts writing some true docs for `spmd_types`. I slaved away on `key_concepts.md` and `local_spmd_types.md` so those would be the ones to focus in on, but there's also lots of other bits an bobs here and there.

# AI summary

Reorganizes the spmd_types docs from two top-level Markdown files into a proper Sphinx documentation site under `docs/`.

`DESIGN.md` moves to `docs/design.md` (the full type-system specification), and the old `MEGATRON_QUICKSTART.md` is folded into the new `docs/local_spmd_types.md`, which carries the hands-on porting guide, the Megatron-to-spmd_types function mapping table, and the Invariant-vs-Replicate advice. New pages add `docs/index.md`, `docs/key_concepts.md`, `docs/global_spmd_types.md`, and an autodoc-backed API reference under `docs/api/`. `docs/conf.py` configures Sphinx (MyST + napoleon + autodoc, RTD theme).

`README.md` and `CLAUDE.md` references are updated to the new paths, `CONTRIBUTING.md` headings are normalized, and the `PartitionSpec` / `TensorSharding` docstrings in `types.py` are reformatted as literal blocks so they render correctly under autodoc. `docs/_build/` is gitignored and a `# noautodeps` `docs/BUCK` is added so the Sphinx `conf.py` is not buckified.

Reviewed By: pianpwk

Differential Revision: D107149885
@meta-cla meta-cla Bot added the CLA Signed This label is managed by the Meta Open Source bot. label Jun 9, 2026
@meta-codesync

meta-codesync Bot commented Jun 9, 2026

Copy link
Copy Markdown
Contributor

@ezyang has exported this pull request. If you are a Meta employee, you can view the originating Diff in D107149885.

@meta-codesync

meta-codesync Bot commented Jun 9, 2026

Copy link
Copy Markdown
Contributor

This pull request has been merged in 8d9235e.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

CLA Signed This label is managed by the Meta Open Source bot. Merged meta-exported

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant