From 37959522a805a5e23c930ed79aac84920c3cb208 Mon Sep 17 00:00:00 2001
From: Dmitrii Agibov <dmitrii.agibov@arm.com>
Date: Fri, 18 Nov 2022 16:34:03 +0000
Subject: Move backends functionality into separate modules

- Move backend management/executor code into module backend_core
- Create separate module for each backend in "backend" module
- Move each backend into corresponding module
- Split Vela wrapper into several submodules

Change-Id: If01b6774aab6501951212541cc5d7f5aa7c97e95
---
 tests/test_backend_vela_compat.py | 74 +++++++++++++++++++++++++++++++++++++++
 1 file changed, 74 insertions(+)
 create mode 100644 tests/test_backend_vela_compat.py

(limited to 'tests/test_backend_vela_compat.py')

diff --git a/tests/test_backend_vela_compat.py b/tests/test_backend_vela_compat.py
new file mode 100644
index 0000000..6f7a41c
--- /dev/null
+++ b/tests/test_backend_vela_compat.py
@@ -0,0 +1,74 @@
+# SPDX-FileCopyrightText: Copyright 2022, Arm Limited and/or its affiliates.
+# SPDX-License-Identifier: Apache-2.0
+"""Tests for module vela/compat."""
+from pathlib import Path
+
+import pytest
+
+from mlia.backend.vela.compat import generate_supported_operators_report
+from mlia.backend.vela.compat import NpuSupported
+from mlia.backend.vela.compat import Operator
+from mlia.backend.vela.compat import Operators
+from mlia.backend.vela.compat import supported_operators
+from mlia.devices.ethosu.config import EthosUConfiguration
+from mlia.utils.filesystem import working_directory
+
+
+@pytest.mark.parametrize(
+    "model, expected_ops",
+    [
+        (
+            "test_model.tflite",
+            Operators(
+                ops=[
+                    Operator(
+                        name="sequential/conv1/Relu;sequential/conv1/BiasAdd;"
+                        "sequential/conv2/Conv2D;sequential/conv1/Conv2D",
+                        op_type="CONV_2D",
+                        run_on_npu=NpuSupported(supported=True, reasons=[]),
+                    ),
+                    Operator(
+                        name="sequential/conv2/Relu;sequential/conv2/BiasAdd;"
+                        "sequential/conv2/Conv2D",
+                        op_type="CONV_2D",
+                        run_on_npu=NpuSupported(supported=True, reasons=[]),
+                    ),
+                    Operator(
+                        name="sequential/max_pooling2d/MaxPool",
+                        op_type="MAX_POOL_2D",
+                        run_on_npu=NpuSupported(supported=True, reasons=[]),
+                    ),
+                    Operator(
+                        name="sequential/flatten/Reshape",
+                        op_type="RESHAPE",
+                        run_on_npu=NpuSupported(supported=True, reasons=[]),
+                    ),
+                    Operator(
+                        name="Identity",
+                        op_type="FULLY_CONNECTED",
+                        run_on_npu=NpuSupported(supported=True, reasons=[]),
+                    ),
+                ]
+            ),
+        )
+    ],
+)
+def test_operators(test_models_path: Path, model: str, expected_ops: Operators) -> None:
+    """Test operators function."""
+    device = EthosUConfiguration("ethos-u55-256")
+
+    operators = supported_operators(test_models_path / model, device.compiler_options)
+    for expected, actual in zip(expected_ops.ops, operators.ops):
+        # do not compare names as they could be different on each model generation
+        assert expected.op_type == actual.op_type
+        assert expected.run_on_npu == actual.run_on_npu
+
+
+def test_generate_supported_operators_report(tmp_path: Path) -> None:
+    """Test generating supported operators report."""
+    with working_directory(tmp_path):
+        generate_supported_operators_report()
+
+        md_file = tmp_path / "SUPPORTED_OPS.md"
+        assert md_file.is_file()
+        assert md_file.stat().st_size > 0
-- 
cgit v1.2.1