File tree Expand file tree Collapse file tree 2 files changed +3
-3
lines changed Expand file tree Collapse file tree 2 files changed +3
-3
lines changed Original file line number Diff line number Diff line change @@ -36,6 +36,7 @@ def __init__(self):
3636class OpenvinoOperatorsSupport (OperatorSupportBase ):
3737 extended_support_dict = {
3838 "torch.ops.dim_order_ops._clone_dim_order.default" : None ,
39+ "torch.ops.dim_order_ops._to_dim_order_copy.default" : None ,
3940 }
4041
4142 def __init__ (
Original file line number Diff line number Diff line change 88
99from typing import final , List
1010
11- from executorch .backends . transforms . remove_clone_ops import RemoveCloneOpsTransform
11+ from executorch .exir . passes . memory_format_ops_pass import DimOrderOpsRevertPass
1212from executorch .exir .backend .backend_details import (
1313 BackendDetails ,
1414 ExportedProgram ,
@@ -37,8 +37,7 @@ def preprocess(
3737 Returns:
3838 PreprocessResult: The result of preprocessing, including the compiled model bytes.
3939 """
40- # Apply RemoveCloneOpsTransform to eliminate unnecessary clone operations
41- transformed_ep = RemoveCloneOpsTransform ()(edge_program .graph_module )
40+ transformed_ep = DimOrderOpsRevertPass ()(edge_program .graph_module )
4241
4342 # Update the edge_program with the transformed graph
4443 if transformed_ep and transformed_ep .graph_module :
You can’t perform that action at this time.
0 commit comments