Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

[tp] improve documentation #115880

Closed
wants to merge 3 commits into from
Closed

Conversation

wanchaol
Copy link
Contributor

@wanchaol wanchaol commented Dec 14, 2023

Improve the TP documentation in terms of format and descriptions

[ghstack-poisoned]
Copy link

pytorch-bot bot commented Dec 14, 2023

🔗 Helpful Links

🧪 See artifacts and rendered test results at hud.pytorch.org/pr/115880

Note: Links to docs will display an error until the docs builds have been completed.

✅ No Failures

As of commit 1fb2ba9 with merge base 0692240 (image):
💚 Looks good so far! There are no failures yet. 💚

This comment was automatically generated by Dr. CI and updates every 15 minutes.

wanchaol added a commit that referenced this pull request Dec 14, 2023
Improve the TP documentation in terms of format and descriptions

ghstack-source-id: 83ecf45626f9f1605625de6d503cdb996d206fe8
Pull Request resolved: #115880
@github-actions github-actions bot added the oncall: distributed Add this issue/PR to distributed oncall triage queue label Dec 14, 2023
Copy link
Contributor

@XilunWu XilunWu left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

lgtm.

@@ -32,7 +32,7 @@ Tensor Parallelism supports the following parallel styles:
To simply configure the nn.Module's inputs and outputs with DTensor layouts
and perform necessary layout redistributions, without distribute the module
parameters to DTensors, the following classes can be used in
the ``parallelize_plan``of ``parallelize_module``:
the `parallelize_plan` of `parallelize_module`:
Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

I thought that we need double backtick for code type and that single backtick is italics, but I may be wrong.

Copy link
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

It's kind of weird to me too, but the lastest main docs looks wrong format so I switched back to single backtick, see lines before this API https://pytorch.org/docs/main/distributed.tensor.parallel.html#torch.distributed.tensor.parallel.PrepareModuleInput

I'll play around with the code types a bit more and see the refreshed docs before landing

Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Maybe it is missing the space between the last backtick of ``parallelize_plan`` and of, causing it to render improperly?

Copy link
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

good point, lm try that instead

Copy link
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

adjusted multiple places to double backtick , but the built doc still rendering the old version, gonna land this first and see the doc

Improve the TP documentation in terms of format and descriptions

cc mrshenli pritamdamania87 zhaojuanmao satgera rohan-varma gqchen aazzolini osalpekar jiayisuse H-Huang kwen2501 awgu penguinwu fegin XilunWu fduwjj wz337 tianyu-l wconstab yf225

[ghstack-poisoned]
wanchaol added a commit that referenced this pull request Dec 15, 2023
Improve the TP documentation in terms of format and descriptions

ghstack-source-id: 60a3ecb492552a3206efd2d21c847803e964865d
Pull Request resolved: #115880
@wanchaol wanchaol added ciflow/trunk Trigger trunk jobs on your pull request release notes: distributed (dtensor) release notes category labels Dec 15, 2023
Improve the TP documentation in terms of format and descriptions

cc mrshenli pritamdamania87 zhaojuanmao satgera rohan-varma gqchen aazzolini osalpekar jiayisuse H-Huang kwen2501 awgu penguinwu fegin XilunWu fduwjj wz337 tianyu-l wconstab yf225

[ghstack-poisoned]
wanchaol added a commit that referenced this pull request Dec 15, 2023
Improve the TP documentation in terms of format and descriptions

ghstack-source-id: d2cc1093cd0d6492af2c5815d0a19e985620fb3a
Pull Request resolved: #115880
@wanchaol
Copy link
Contributor Author

@pytorchbot merge

@pytorchmergebot
Copy link
Collaborator

Merge started

Your change will be merged once all checks pass (ETA 0-4 Hours).

Learn more about merging in the wiki.

Questions? Feedback? Please reach out to the PyTorch DevX Team

Advanced Debugging
Check the merge workflow status
here

wanchaol added a commit that referenced this pull request Dec 15, 2023
Improve the TP documentation in terms of format and descriptions

Pull Request resolved: #115880
Approved by: https://github.com/XilunWu
guilhermeleobas pushed a commit to guilhermeleobas/pytorch that referenced this pull request Dec 18, 2023
Improve the TP documentation in terms of format and descriptions

Pull Request resolved: pytorch#115880
Approved by: https://github.com/XilunWu
atalman pushed a commit that referenced this pull request Dec 18, 2023
Improve the TP documentation in terms of format and descriptions

Pull Request resolved: #115880
Approved by: https://github.com/XilunWu
@facebook-github-bot facebook-github-bot deleted the gh/wanchaol/414/head branch December 19, 2023 15:27
hyperfraise pushed a commit to hyperfraise/pytorch that referenced this pull request Dec 21, 2023
Improve the TP documentation in terms of format and descriptions

Pull Request resolved: pytorch#115880
Approved by: https://github.com/XilunWu
hyperfraise pushed a commit to hyperfraise/pytorch that referenced this pull request Dec 21, 2023
Improve the TP documentation in terms of format and descriptions

Pull Request resolved: pytorch#115880
Approved by: https://github.com/XilunWu
ZhiweiYan-96 pushed a commit to ZhiweiYan-96/pytorch that referenced this pull request Dec 22, 2023
Improve the TP documentation in terms of format and descriptions

Pull Request resolved: pytorch#115880
Approved by: https://github.com/XilunWu
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
ciflow/trunk Trigger trunk jobs on your pull request Merged oncall: distributed Add this issue/PR to distributed oncall triage queue release notes: distributed (dtensor) release notes category
Projects
None yet
Development

Successfully merging this pull request may close these issues.

None yet

4 participants