Skip to content

Commit 233c651

Browse files
committed
Update on "Add pass to tag external constants for delegates"
generate pte+ptd file for a delegated linear example Differential Revision: [D73281924](https://our.internmc.facebook.com/intern/diff/D73281924/) [ghstack-poisoned]
1 parent 9018bd3 commit 233c651

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

exir/passes/external_constants_pass.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -97,8 +97,12 @@ def delegate_external_constants_pass(
9797
Tag external constants before to_backend. Tagged constants will be saved
9898
to an external file.
9999
100+
Note: this pass must be run after run_decompositions(), as tags on
101+
constants are removed then.
102+
100103
Args:
101104
gm: GraphModule to tag.
105+
ep: ExportedProgram, to distinguish if a node is a constant.
102106
filter_fn: node -> str callable indicating the file (str) that a node should be saved to.
103107
Returns:
104108
PassResult: The resulting gm, and if it was mutated or not.

0 commit comments

Comments
 (0)