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

[DTensor][Shampoo] add _tenso.zero function #95863

Closed
wants to merge 1 commit into from

Conversation

minddrummer
Copy link
Contributor

Summary:
implement zeros function inside DTensor API

  • user specify the zeros tensor shape, and the function will create local zero tensor given the placement information

Test Plan:
{F889157756} - unit test for util function for compute_local_tensor_size

  • unit test for _tensor.zeros

Reviewed By: wanchaol

Differential Revision: D43630718

@pytorch-bot
Copy link

pytorch-bot bot commented Mar 2, 2023

🔗 Helpful Links

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

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

✅ No Failures

As of commit 9431180:
💚 Looks good so far! There are no failures yet. 💚

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

@linux-foundation-easycla
Copy link

linux-foundation-easycla bot commented Mar 2, 2023

CLA Signed

The committers listed above are authorized under a signed CLA.

  • ✅ login: minddrummer / name: Ke Sang (51e1778)

@facebook-github-bot
Copy link
Contributor

This pull request was exported from Phabricator. Differential Revision: D43630718

minddrummer added a commit to minddrummer/pytorch that referenced this pull request Mar 2, 2023
Summary:
Pull Request resolved: pytorch#95863

implement zeros function inside DTensor API
- user specify the zeros tensor shape, and the function will create local zero tensor given the placement information

Test Plan:
{F889157756} - unit test for util function for compute_local_tensor_size
- unit test for _tensor.zeros

Reviewed By: wanchaol

Differential Revision: D43630718

fbshipit-source-id: 893578b7eb0e3993a24a2f4c5ae0ffc7713c70e2
@facebook-github-bot
Copy link
Contributor

This pull request was exported from Phabricator. Differential Revision: D43630718

1 similar comment
@facebook-github-bot
Copy link
Contributor

This pull request was exported from Phabricator. Differential Revision: D43630718

minddrummer added a commit to minddrummer/pytorch that referenced this pull request Mar 2, 2023
Summary:
Pull Request resolved: pytorch#95863

implement zeros function inside DTensor API
- user specify the zeros tensor shape, and the function will create local zero tensor given the placement information

Test Plan:
{F889157756} - unit test for util function for compute_local_tensor_size
- unit test for _tensor.zeros

Reviewed By: wanchaol

Differential Revision: D43630718

fbshipit-source-id: 14b40863b40df696bc87a92166953f5ba2c5425d
minddrummer added a commit to minddrummer/pytorch that referenced this pull request Mar 2, 2023
Summary:
Pull Request resolved: pytorch#95863

implement zeros function inside DTensor API
- user specify the zeros tensor shape, and the function will create local zero tensor given the placement information

Test Plan:
{F889157756} - unit test for util function for compute_local_tensor_size
- unit test for _tensor.zeros

Reviewed By: wanchaol

Differential Revision: D43630718

fbshipit-source-id: 85adfdf0f2c6f5e2fcbc9df84e90e8d6abc7d2c5
@facebook-github-bot
Copy link
Contributor

This pull request was exported from Phabricator. Differential Revision: D43630718

@wanchaol wanchaol added the release notes: distributed (dtensor) release notes category label Mar 2, 2023
@minddrummer
Copy link
Contributor Author

/easycla

@facebook-github-bot
Copy link
Contributor

This pull request was exported from Phabricator. Differential Revision: D43630718

minddrummer added a commit to minddrummer/pytorch that referenced this pull request Mar 3, 2023
Summary:
Pull Request resolved: pytorch#95863

implement zeros function inside DTensor API
- user specify the zeros tensor shape, and the function will create local zero tensor given the placement information

Test Plan:
{F889157756} - unit test for util function for compute_local_tensor_size
- unit test for _tensor.zeros

Reviewed By: wanchaol

Differential Revision: D43630718

fbshipit-source-id: d9406f30c2bd5919cbc63f398d66e10de43248d6
@minddrummer
Copy link
Contributor Author

@pytorchbot merge -f

@pytorch-bot
Copy link

pytorch-bot bot commented Mar 3, 2023

❌ 🤖 pytorchbot command failed:

@pytorchbot merge: error: argument -f/--force: expected one argument

usage: @pytorchbot merge [-g | -f MESSAGE | -l] [-r [{viable/strict,master}]]

Try @pytorchbot --help for more info.

Copy link
Contributor

@wanchaol wanchaol left a comment

Choose a reason for hiding this comment

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

lgtm

@facebook-github-bot
Copy link
Contributor

This pull request was exported from Phabricator. Differential Revision: D43630718

minddrummer added a commit to minddrummer/pytorch that referenced this pull request Mar 3, 2023
Summary:
Pull Request resolved: pytorch#95863

implement zeros function inside DTensor API
- user specify the zeros tensor shape, and the function will create local zero tensor given the placement information

Test Plan:
{F889157756} - unit test for util function for compute_local_tensor_size
- unit test for _tensor.zeros

Reviewed By: wanchaol

Differential Revision: D43630718

fbshipit-source-id: 341ecb2940ca1f5d4cebfc0fd69abe09c5c9959d
Summary:
Pull Request resolved: pytorch#95863

implement zeros function inside DTensor API
- user specify the zeros tensor shape, and the function will create local zero tensor given the placement information

Test Plan:
{F889157756} - unit test for util function for compute_local_tensor_size
- unit test for _tensor.zeros

Reviewed By: wanchaol

Differential Revision: D43630718

fbshipit-source-id: a2564cce961f898089f946245eac550db5f8d20b
@facebook-github-bot
Copy link
Contributor

This pull request was exported from Phabricator. Differential Revision: D43630718

@facebook-github-bot
Copy link
Contributor

@pytorchbot merge -f 'Landed internally'

(Initiating merge automatically since Phabricator Diff has merged, using force because this PR might not pass merge_rules.json but landed internally)

@pytorchmergebot
Copy link
Collaborator

Merge started

Your change will be merged immediately since you used the force (-f) flag, bypassing any CI checks (ETA: 1-5 minutes).

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

cyyever pushed a commit to cyyever/pytorch_private that referenced this pull request Mar 5, 2023
Summary:
implement zeros function inside DTensor API
- user specify the zeros tensor shape, and the function will create local zero tensor given the placement information

Test Plan:
{F889157756} - unit test for util function for compute_local_tensor_size
- unit test for _tensor.zeros

Reviewed By: wanchaol

Differential Revision: D43630718

Pull Request resolved: pytorch/pytorch#95863
Approved by: https://github.com/wanchaol
cyyever pushed a commit to cyyever/pytorch_private that referenced this pull request Mar 5, 2023
Summary:
implement zeros function inside DTensor API
- user specify the zeros tensor shape, and the function will create local zero tensor given the placement information

Test Plan:
{F889157756} - unit test for util function for compute_local_tensor_size
- unit test for _tensor.zeros

Reviewed By: wanchaol

Differential Revision: D43630718

Pull Request resolved: pytorch/pytorch#95863
Approved by: https://github.com/wanchaol
ydwu4 pushed a commit to ydwu4/pytorch that referenced this pull request Mar 10, 2023
Summary:
implement zeros function inside DTensor API
- user specify the zeros tensor shape, and the function will create local zero tensor given the placement information

Test Plan:
{F889157756} - unit test for util function for compute_local_tensor_size
- unit test for _tensor.zeros

Reviewed By: wanchaol

Differential Revision: D43630718

Pull Request resolved: pytorch#95863
Approved by: https://github.com/wanchaol
ydwu4 added a commit to ydwu4/pytorch that referenced this pull request Mar 13, 2023
Summary:
implement zeros function inside DTensor API
- user specify the zeros tensor shape, and the function will create local zero tensor given the placement information

Test Plan:
{F889157756} - unit test for util function for compute_local_tensor_size
- unit test for _tensor.zeros

Reviewed By: wanchaol

Differential Revision: D43630718

Pull Request resolved: pytorch#95863
Approved by: https://github.com/wanchaol
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
Development

Successfully merging this pull request may close these issues.

None yet

4 participants