File tree Expand file tree Collapse file tree 1 file changed +4
-1
lines changed Expand file tree Collapse file tree 1 file changed +4
-1
lines changed Original file line number Diff line number Diff line change 24
24
from executorch .exir .dynamic_shape import DynamicMemoryPlanningMode
25
25
from executorch .exir .error import InternalError
26
26
from executorch .exir .operator .convert import (
27
+ _get_overload_schema ,
27
28
get_out_args_from_opoverload ,
28
29
is_out_variant ,
29
30
to_out_variant ,
63
64
from torch ._subclasses import FakeTensor
64
65
from torch .fx .passes .infra .pass_base import PassBase , PassResult
65
66
from torch .fx .passes .shape_prop import TensorMetadata
67
+ from torchgen .model import SchemaKind
66
68
67
69
__all__ = [
68
70
"ExportPass" ,
@@ -257,7 +259,6 @@ def callWithLoggerEnabled(self, graph_module: torch.fx.GraphModule) -> None:
257
259
memory .alloc ,
258
260
memory .view ,
259
261
executorch_call_delegate ,
260
- torch .ops .aten .copy_ .default ,
261
262
}
262
263
to_out_var_skiplist .update (_EXECUTORCH_SYM_OPS )
263
264
@@ -347,6 +348,8 @@ def get_submodule(node: torch.fx.Node) -> torch.fx.GraphModule:
347
348
continue
348
349
elif target in to_out_var_skiplist :
349
350
continue
351
+ elif _get_overload_schema (target ).kind () == SchemaKind .inplace :
352
+ continue
350
353
if not isinstance (
351
354
target , (torch ._ops .OpOverload , EdgeOpOverload , BackendOpOverload )
352
355
):
You can’t perform that action at this time.
0 commit comments