mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-06 12:20:52 +01:00
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/67817 Implementation of build features as a useable feature. Includes tracing support and selectivity support. Follow up of Dhruv's prototype in D30076214. The general idea is to allow selectivity of arbitrary sections of the codebase through the 2 apis, BUILD_FEATURE_REQUIRED(NAME), and BUILD_FEATURE_AVAILABLE(NAME) References PyTorch Edge Team Workplace group post link: https://fb.workplace.com/groups/pytorch.edge.team/posts/905584476662959/ Quip talking about some early ideas related to build features: https://fb.quip.com/iur3ApU9q29v Google Doc about most recent discussion and details: https://docs.google.com/document/d/1533zuN_9pwpQBa4RhtstUjT5B7guowblqJz35QYWPE0/edit Will remove the copy kernel example after. Its just here as an example. ghstack-source-id: 142850218 Test Plan: CI, dummy traced a model, and played around with its unit test if i removed the traced value from the yaml Reviewed By: dhruvbird Differential Revision: D32151856 fbshipit-source-id: 33764c1f6902a025e53807b784792a83c8385984 |
||
|---|---|---|
| .. | ||
| __init__.py | ||
| gen_selected_mobile_ops_header.py | ||