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

[DataPipe] Lazily generate exception message for performance #78673

Closed
wants to merge 2 commits into from

Conversation

NivekT
Copy link
Contributor

@NivekT NivekT commented Jun 1, 2022

Stack from ghstack:

With this PR, exception message for IterDataPipe are lazily generated as needed rather than at every call. This greatly improves the performance while iterating IterDataPipe.

I plan to cherry-pick this for 1.12 to prevent performance regression.

Differential Revision: D36830075

@facebook-github-bot
Copy link
Contributor

facebook-github-bot commented Jun 1, 2022

🔗 Helpful links

✅ No Failures (0 Pending)

As of commit 70fa3e8 (more details on the Dr. CI page):

Expand to see more

💚 💚 Looks good so far! There are no failures yet. 💚 💚


This comment was automatically generated by Dr. CI (expand for details).

Please report bugs/suggestions to the (internal) Dr. CI Users group.

Click here to manually regenerate this comment.

@NivekT NivekT requested a review from ejguan June 1, 2022 21:12
@NivekT NivekT added module: data torch.utils.data release notes: dataloader release notes category topic: performance topic category labels Jun 1, 2022
@NivekT
Copy link
Contributor Author

NivekT commented Jun 1, 2022

@NivekT has imported this pull request. If you are a Facebook employee, you can view this diff on Phabricator.

Copy link
Contributor

@ejguan ejguan left a comment

Choose a reason for hiding this comment

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

Thank you!

@NivekT
Copy link
Contributor Author

NivekT commented Jun 1, 2022

@pytorchbot merge -g

dzdang added a commit that referenced this pull request Jun 2, 2022
Pull Request resolved: #78673

Approved by: https://github.com/ejguan
ghstack-source-id: 24087821ebe7833ece1e8a7b7cb6802e14070dfd
NivekT added a commit that referenced this pull request Jun 2, 2022
atalman pushed a commit that referenced this pull request Jun 2, 2022
facebook-github-bot pushed a commit that referenced this pull request Jun 2, 2022
Summary:
Pull Request resolved: #78673

Approved by: https://github.com/ejguan

Test Plan: contbuild & OSS CI, see https://hud.pytorch.org/commit/pytorch/pytorch/575c42028727db4428dc6bcd03c8631f7798b9d3

Reviewed By: ejguan

Differential Revision: D36830075

Pulled By: NivekT

fbshipit-source-id: 6bd07a76aa4718364e50b7b9cca4d011067874d7
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
Projects
None yet
Development

Successfully merging this pull request may close these issues.

4 participants