pytorch/tools/jit
Larry Liu e345138591 [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
2022-07-15 18:04:25 +00:00
..
templates Generate aten_interned_strings.h automatically (#69407) 2022-01-18 08:29:54 -08:00
test Apply ufmt linter to all py files under tools (#81285) 2022-07-13 07:59:22 +00:00
__init__.py remediation of S205607 2020-07-17 17:19:47 -07:00
gen_unboxing.py [retake2][mobile] Fix lightweight dispatch OOM error by introducing selective build (#80791) 2022-07-15 18:04:25 +00:00