Skip to content

[tracking] Refactor aten_native_batch_norm #1125

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.

Sign up for GitHub

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

Jump to bottom
Open
titaiwangms opened this issue Nov 2, 2023 · 0 comments
Open

[tracking] Refactor aten_native_batch_norm #1125

titaiwangms opened this issue Nov 2, 2023 · 0 comments
Labels
module: torchlib Related to the torch/aten function lib in development

Comments

@titaiwangms
Copy link
Contributor

titaiwangms commented Nov 2, 2023 •
edited
Loading

In https://github.com/pytorch/pytorch/blob/a44f8894fa6d973693aab44a3dda079a168b05c1/torch/_decomp/decompositions.py#L1501-L1510, _native_batch_norm_legit_no_training and _native_batch_norm_legit are meant to replace native_batch_norm within unknown time period. When that comes, we should retire aten::native_batch_norm.

### Tasks
- [ ] Retire `aten::native_batch_norm`
- [ ] Refactor `aten::_native_batch_norm_legit ` and `aten::_native_batch_norm_legit_functional` by merging them to a higher level TracedFunction.
@titaiwangms titaiwangms added the module: torchlib Related to the torch/aten function lib in development label Nov 2, 2023
@titaiwangms titaiwangms mentioned this issue Nov 2, 2023
Merged
@titaiwangms titaiwangms changed the title Retire aten::native_batch_norm Refactor aten_native_batch_norm Nov 10, 2023
@titaiwangms titaiwangms mentioned this issue Nov 10, 2023
Merged
@justinchuby justinchuby added the contribution welcome We welcome code contributions for this label Feb 8, 2024
@justinchuby justinchuby changed the title Refactor aten_native_batch_norm [tracking] Refactor aten_native_batch_norm Feb 8, 2024
@justinchuby justinchuby removed the contribution welcome We welcome code contributions for this label Feb 8, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees
No one assigned
Labels
module: torchlib Related to the torch/aten function lib in development
Projects
None yet
Milestone
No milestone
Development

No branches or pull requests
2 participants
@justinchuby @titaiwangms