Skip to content

Commit 51b2385

Browse files
mcr229facebook-github-bot
authored andcommitted
Operator Support uses Exported Program (#357)
Summary: Pull Request resolved: #357 When using supported operators for partitioning, we need to know which nodes are attributes. With the new torch.export, we need the exported program in order to find this information. We add ExportedProgram to OperatorSupport so constraints have access to ExportedProgram and can use information of attributes to determine how to partition Reviewed By: digantdesai Differential Revision: D49301134 fbshipit-source-id: fa685d7f94ec15443cc2e364189a83d210f204bf
1 parent d5ed0e8 commit 51b2385

File tree

1 file changed

+99
-67
lines changed

1 file changed

+99
-67
lines changed

0 commit comments

Comments
 (0)