Skip to content

Commit e78ed83

Browse files
authored
Enable constant methods in cadence export.
Differential Revision: D68637824 Pull Request resolved: #7948
1 parent 25395b9 commit e78ed83

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

backends/cadence/aot/compiler.py

+2
Original file line numberDiff line numberDiff line change
@@ -174,6 +174,7 @@ def export_to_edge(
174174
model: torch.nn.Module,
175175
inputs: tuple[object, ...],
176176
dump_graphs: bool = False,
177+
constant_methods: Optional[dict[str, object]] = None,
177178
) -> EdgeProgramManager:
178179
assert isinstance(model, torch.nn.Module), "model should be an nn.Module"
179180

@@ -195,6 +196,7 @@ def export_to_edge(
195196
torch.ops.aten.angle.default,
196197
],
197198
),
199+
constant_methods=constant_methods,
198200
)
199201

200202
if dump_graphs:

0 commit comments

Comments
 (0)