Address CR comments

Signed-off-by: Erik Lundell <[email protected]>
Change-Id: If733617374683765bf1d49ff8f64e4c7ab9bc42d

Author: Erik-Lundell

backends/arm/arm_backend.py

@@ -284,6 +284,12 @@ def preprocess(  # noqa: C901
                 # any checking of compatibility.
                 dbg_fail(node, tosa_graph, artifact_path)
 
+        if len(input_order) > 0:
+            if input_count != len(input_order):
+                raise RuntimeError(
+                    "The rank of the input order is not equal to amount of input tensors"
+                )
+
         if artifact_path:
             tag = _get_first_delegation_tag(graph_module)
             dbg_tosa_dump(

backends/arm/test/common.py

@@ -76,7 +76,7 @@ def get_tosa_compile_spec_unbuilt(
     if not custom_path:
         custom_path = maybe_get_tosa_collate_path()
 
-    if custom_path is not None and not os.path.exists(custom_path):
+    if custom_path is not None:
         os.makedirs(custom_path, exist_ok=True)
     compile_spec_builder = (
         ArmCompileSpecBuilder()

examples/arm/setup.sh

@@ -226,7 +226,7 @@ function setup_tosa_reference_model() {
 
     # reference_model flatbuffers version clashes with Vela.
     # go with Vela's since it newer.
-    # Could cause issues down the line, beware..
+    # Vela's flatbuffer requirement is expected to loosen, then remove this. MLETORCH-565
     pip install tosa-tools@git+${tosa_reference_model_url}@${tosa_reference_model_rev} --no-dependencies flatbuffers
 
 }