Skip to content

[Inductor] Support fallback for all gemm like ops#165755

Closed
nandesuka wants to merge 1 commit intopytorch:mainfrom
nandesuka:export-D84840948
Closed

[Inductor] Support fallback for all gemm like ops#165755
nandesuka wants to merge 1 commit intopytorch:mainfrom
nandesuka:export-D84840948

Conversation

@nandesuka
Copy link
Contributor

@nandesuka nandesuka commented Oct 17, 2025

Summary: Fill op_override field for bmm aten ops so they can be converted properly in the wrapper_fxir backend

Reviewed By: StellarrZ

Differential Revision: D84840948

cc @voznesenskym @penguinwu @EikanWang @jgong5 @Guobing-Chen @XiaobingSuper @zhuhaozhe @blzheng @wenzhe-nrv @jiayisunx @ipiszy @chenyang78 @kadeng @muchulee8 @amjames @chauhang @aakhundov @coconutruben

Summary: Allow all gemm like ops to fallback appropriatly for full inductor backend

Test Plan:
buck2 run //mtia/compiler/graph_compiler/test_library:afg_inductor_test -- -k __addmm__basic_
buck2 run //mtia/compiler/graph_compiler/test_library:afg_inductor_test -- -k __bmm__basic_
buck2 run //mtia/compiler/graph_compiler/test_library:afg_inductor_test -- -k __matmul__basic_
buck2 run //mtia/compiler/graph_compiler/test_library:afg_inductor_test -- -k __mm__basic_
buck2 run

Reviewed By: StellarrZ

Differential Revision: D84840948
@pytorch-bot
Copy link

pytorch-bot bot commented Oct 17, 2025

🔗 Helpful Links

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

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

✅ No Failures

As of commit 68fcdc3 with merge base 3af2f0c (image):
💚 Looks good so far! There are no failures yet. 💚

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

@meta-codesync
Copy link

meta-codesync bot commented Oct 17, 2025

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

Copy link
Contributor

@blaine-rister blaine-rister left a comment

Choose a reason for hiding this comment

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

LGTM.

@pytorch-bot pytorch-bot bot added the ciflow/trunk Trigger trunk jobs on your pull request label Oct 17, 2025
@nandesuka
Copy link
Contributor Author

@pytorchbot label "topic: not user facing"

@pytorch-bot pytorch-bot bot added the topic: not user facing topic category label Oct 17, 2025
@facebook-github-bot
Copy link
Contributor

@pytorchbot merge

(Initiating merge automatically since Phabricator Diff has merged)

@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

Chao1Han pushed a commit to Chao1Han/pytorch that referenced this pull request Oct 21, 2025
Summary: Fill op_override field for bmm aten ops so they can be converted properly in the wrapper_fxir backend

Reviewed By: StellarrZ

Differential Revision: D84840948

Pull Request resolved: pytorch#165755
Approved by: https://github.com/blaine-rister
zhudada0120 pushed a commit to zhudada0120/pytorch that referenced this pull request Oct 22, 2025
Summary: Fill op_override field for bmm aten ops so they can be converted properly in the wrapper_fxir backend

Reviewed By: StellarrZ

Differential Revision: D84840948

Pull Request resolved: pytorch#165755
Approved by: https://github.com/blaine-rister
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