Use GraphBuilder in reorder unit tests.

Summary: Use GraphBuilder in reorder unit tests.

Differential Revision: D75257323

Author: eigen-k

backends/cadence/aot/TARGETS

@@ -388,6 +388,7 @@ python_unittest(
         "//caffe2:torch",
         "//executorch/backends/cadence/aot:compiler",
         "//executorch/backends/cadence/aot:fuse_ops",
+        "//executorch/backends/cadence/aot:graph_builder",
         "//executorch/backends/cadence/aot:ops_registrations",
         "//executorch/backends/cadence/aot:pass_utils",
         "//executorch/backends/cadence/aot:reorder_ops",

backends/cadence/aot/compiler.py

@@ -343,6 +343,9 @@ def quantize_and_export_to_cadence(
     """
     quantized_model = quantize_pt2(model, inputs)
 
+    import fbvscode
+    fbvscode.set_trace()
+
     return _lower_ep_to_cadence(
         quantized_model,
         opt_level=opt_level,

backends/cadence/aot/pass_utils.py

@@ -143,6 +143,18 @@ def nodes_not_connected_in_gm(
                 return False
     return True
 
+# Returns the position of the first entry of a node of a given kind in the graph.
+def get_node_pos(
+    graph_module: torch.fx.GraphModule,
+    target: torch.fx.Node,
+) -> int:
+    pos = 0
+    for node in graph_module.graph.nodes:
+        if node.target == target:
+            return pos
+        pos += 1
+    return -1
+
 
 # Returns true if there is no instance of a node with target succ_target
 # positioned immediately after a node with target pred_target in the graph