Update on "[et][dim order] aot support for dim order variant empty op"

Gasoonjia · Gasoonjia · commit a7b1559e145f · 2024-12-03T18:30:02.000-08:00
This diff added aot support for dim order variant empty op, including operator impl and registration, memory_format_pass update, and end2end tests on both aten and lean mode. Differential Revision: [D66738618](https://our.internmc.facebook.com/intern/diff/D66738618/) [ghstack-poisoned]
diff --git a/exir/passes/memory_format_ops_pass.py b/exir/passes/memory_format_ops_pass.py
@@ -56,7 +56,7 @@ def call_operator(self, op, args, kwargs, meta):
         else:
             assert (
                 0
-            ), f"Expecting a Tensor a ProxyValue or a Sequence, but got {type(args[0])}"
+            ), f"Expecting a Tensor, a ProxyValue, or a Sequence, but got {type(args[0])}"
 
         nkwargs["dim_order"] = get_dim_order(mem_format, ndim)
         logger.debug(

Original file line number	Diff line number	Diff line change
`@@ -56,7 +56,7 @@ def call_operator(self, op, args, kwargs, meta):`
`56`	`56`	`else:`
`57`	`57`	`assert (`
`58`	`58`	`0`
`59`		`- ), f"Expecting a Tensor a ProxyValue or a Sequence, but got {type(args[0])}"`
	`59`	`+ ), f"Expecting a Tensor, a ProxyValue, or a Sequence, but got {type(args[0])}"`
`60`	`60`
`61`	`61`	`nkwargs["dim_order"] = get_dim_order(mem_format, ndim)`
`62`	`62`	`logger.debug(`