Add dim order guard for mediatek backend AOT part

Author: neuropilot-captain

backends/mediatek/partitioner.py

@@ -44,6 +44,11 @@ def is_node_supported(self, _, node: torch.fx.Node) -> bool:
             return False
 
         op_type = node.target.__name__
+
+        # Skip until we can handle the dimension order representation
+        if op_type == 'aten._to_copy.default':
+            return False
+
         if op_type in self._op_types_to_skip or node.name in self._op_names_to_skip:
             print(
                 f"[Neuropilot Backend] The {op_type} operator with name '{node.name}' is skipped."

backends/mediatek/preprocess.py

@@ -22,6 +22,23 @@
 SKIP_COMPILE_SPEC_KEYS = {"ImportForever"}
 
 
+def assert_default_dim_order(edge_graph_module: torch.fx.GraphModule) -> None:
+    for node in edge_graph_module.graph.nodes:
+        if node.op != "placeholder":
+            continue
+
+        # We expect the default dim order for all tensor-like inputs i.e. inputs, buffers, and params
+        t = node.meta.get("val", None)
+        if t is not None and getattr(t, "dim_order", None) is not None:
+            default_dim_order = tuple(range(t.dim()))
+            if t.dim_order() != default_dim_order:
+                raise RuntimeError(
+                    f"Neuropilot backend only supports contiguous memory format for inputs."
+                    f"Expecting dim_order: {default_dim_order}, but got "
+                    f"{node.meta['val'].dim_order()} for a placeholder node {node}."
+                )
+
+
 @final
 class NeuropilotBackend(BackendDetails):
 
@@ -30,6 +47,10 @@ def preprocess(
         cls, edge_program: ExportedProgram, module_compile_spec: List[CompileSpec]
     ) -> PreprocessResult:
 
+        # Make sure all inputs are contiguous_format or NCHW or default dim order
+        print('here')
+        assert_default_dim_order(edge_program.graph_module)
+
         name_to_node_mappings = {node.name: node for node in edge_program.graph.nodes}
         input_names = edge_program.graph_signature.user_inputs
         output_names = edge_program.graph_signature.user_outputs