pytorch
e3451385 - [retake2][mobile] Fix lightweight dispatch OOM error by introducing selective build (#80791)

Commit
3 years ago
[retake2][mobile] Fix lightweight dispatch OOM error by introducing selective build (#80791) To fix #78540 I committed #78983 which is reverted due to internal CI failure. Then I comitted #79215 which was only fixing the failure but didn't have the full feature of #78983. This PR is another try. This PR adds script to dump all operators from test models and automatically write into `lightweight_dispatch_ops.yaml`. This way we don't have to manually update the yaml file. Pull Request resolved: https://github.com/pytorch/pytorch/pull/80791 Approved by: https://github.com/raziel
Author
Committer
Parents
Loading