Zhicheng Yan 01b662bafe [gen_operators_yaml] add arguments to control include_all_overloads (#108396) ... Summary: In SelectiveBuildOperator, we can specify argument `include_all_overloads`. If True, all overloaded operators (for example, `aten::to.dtype_layout`, `aten::to.prim_Device"` are considered as overloaded operators of `aten::to`), will be built and linked to the final binary. This can significantly increases the final binary size, which could be a deal breaker for on-device deployment. In this diff, we make back-compatible changes to add new arguments `--not-include-all-overloads-static-root-ops` and `--not-include-all-overloads-closure-ops`. When they are set, we set `include_all_overloads` flag to False for static root ops and closure ops, and rely on code analyzer to decide the actual used overloaded operator. Test Plan: - unit test ``` buck test //xplat/caffe2/tools:gen_operators_yaml_test ``` - See test plan in D48771544 where we reduce the shared lib file `libmrengine.lib` from 16653072 bytes to 13686032 bytes. - See detailed document: https://fburl.com/gdoc/mc93h6kb Reviewed By: larryliu0820 Differential Revision: D48772302 Pull Request resolved: https://github.com/pytorch/pytorch/pull/108396 Approved by: https://github.com/larryliu0820		2023-09-02 17:37:36 +00:00
..
gen_op_registration_allowlist.py	[BE] Enable ruff's UP rules and autoformat tools and scripts (#105428)	2023-07-19 01:24:44 +00:00
gen_operators_yaml.py	[gen_operators_yaml] add arguments to control include_all_overloads (#108396)	2023-09-02 17:37:36 +00:00
gen_oplist.py	[BE]: Update Ruff to 0.0.280 (#105724)	2023-07-22 23:03:34 +00:00