Skip to content

Commit 91d32cf

Browse files
committed
Update on "[et][dim order] Makes DimOrderOpsMap as an operator to operator mapping"
This diff updates DimOrderOpsMap from name-to-operator mapping to operator-to-operator mapping, which has multiple benefits: 1. Reduce dialect ambiguity. Different dialects op may map to same name (e.g. `aten.to_copy` and `exir_ops.edge.aten._to_copy.default`). Directly using op can diminish the ambiguity. 2. Auto-maintain MemoryFormatOpsMap by reverting DimOrderOpsMap Differential Revision: [D66773612](https://our.internmc.facebook.com/intern/diff/D66773612/) [ghstack-poisoned]
1 parent 586cb29 commit 91d32cf

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

exir/passes/dim_order_ops_registry.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -62,7 +62,7 @@ def _empty_dim_order_out_impl(*args, **kwargs):
6262
"""
6363
DimOrderOpsMap = {
6464
exir_ops.edge.aten._to_copy.default: exir_ops.edge.dim_order_ops._to_dim_order_copy.default,
65-
exir_ops.edge.aten.empty.memory_forma: exir_ops.edge.dim_order_ops._empty_dim_order.default,
65+
exir_ops.edge.aten.empty.memory_format: exir_ops.edge.dim_order_ops._empty_dim_order.default,
6666
}
6767

6868
"""

0 commit comments

Comments
 (0)