KEMBAR78
Don't search getitem for batch fusions by mlazos · Pull Request #112088 · pytorch/pytorch · GitHub
Skip to content

Conversation

@mlazos
Copy link
Contributor

@mlazos mlazos commented Oct 25, 2023

Batch mm fusion regressed optimizer compile time by about ~1m, excluding getitem solves this problem.

cc @voznesenskym @penguinwu @EikanWang @jgong5 @Guobing-Chen @XiaobingSuper @zhuhaozhe @blzheng @wenzhe-nrv @jiayisunx @peterbell10 @ipiszy @yf225 @chenyang78 @kadeng @muchulee8 @aakhundov @ColinPeppler

@mlazos mlazos requested a review from yanboliang October 25, 2023 22:17
@pytorch-bot
Copy link

pytorch-bot bot commented Oct 25, 2023

🔗 Helpful Links

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

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

✅ No Failures

As of commit 75e3ee6 with merge base b332200 (image):
💚 Looks good so far! There are no failures yet. 💚

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

@yanboliang
Copy link
Contributor

@jackiexu1992 Can you double check if this impacts Ads model pattern searching?

@facebook-github-bot
Copy link
Contributor

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

Copy link
Contributor

@yanboliang yanboliang left a comment

Choose a reason for hiding this comment

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

Pls land this after internal Ads model tests passed

@mlazos
Copy link
Contributor Author

mlazos commented Oct 27, 2023

@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

xuhancn pushed a commit to xuhancn/pytorch that referenced this pull request Nov 7, 2023
Batch mm fusion regressed optimizer compile time by about ~1m, excluding getitem solves this problem.

Pull Request resolved: pytorch#112088
Approved by: https://github.com/yanboliang
Skylion007 pushed a commit to Skylion007/pytorch that referenced this pull request Nov 14, 2023
Batch mm fusion regressed optimizer compile time by about ~1m, excluding getitem solves this problem.

Pull Request resolved: pytorch#112088
Approved by: https://github.com/yanboliang
@github-actions github-actions bot deleted the mlazos/batch-fuse-opt branch April 27, 2025 02:18
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.

4 participants