Enable constant methods in cadence export.

Differential Revision: D68637824 Pull Request resolved: #7948
pytorch · Jan 25, 2025 · e78ed83 · e78ed83
1 parent 25395b9
commit e78ed83
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/backends/cadence/aot/compiler.py b/backends/cadence/aot/compiler.py
@@ -174,6 +174,7 @@ def export_to_edge(
     model: torch.nn.Module,
     inputs: tuple[object, ...],
     dump_graphs: bool = False,
+    constant_methods: Optional[dict[str, object]] = None,
 ) -> EdgeProgramManager:
     assert isinstance(model, torch.nn.Module), "model should be an nn.Module"
 
@@ -195,6 +196,7 @@ def export_to_edge(
                 torch.ops.aten.angle.default,
             ],
         ),
+        constant_methods=constant_methods,
     )
 
     if dump_graphs: