We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 1f1a96f commit e2a8936Copy full SHA for e2a8936
backends/cadence/aot/compiler.py
@@ -8,7 +8,7 @@
8
9
import logging
10
from pathlib import Path
11
-from typing import Callable, cast, Optional
+from typing import Any, Callable, cast, Optional
12
13
import executorch.backends.cadence.aot.ops_registrations # noqa
14
import torch
@@ -174,6 +174,7 @@ def export_to_edge(
174
model: torch.nn.Module,
175
inputs: tuple[object, ...],
176
dump_graphs: bool = False,
177
+ constant_methods: Optional[dict[str, Any]] = None,
178
) -> EdgeProgramManager:
179
assert isinstance(model, torch.nn.Module), "model should be an nn.Module"
180
@@ -195,6 +196,7 @@ def export_to_edge(
195
196
torch.ops.aten.angle.default,
197
],
198
),
199
+ constant_methods=constant_methods,
200
)
201
202
if dump_graphs:
0 commit comments