1 files changed, 113 insertions, 0 deletions
diff --git a/src/mlia/cli/command_validators.py b/src/mlia/cli/command_validators.py
new file mode 100644
index 0000000..1974a1d
--- /dev/null
+++ b/src/mlia/cli/command_validators.py
@@ -0,0 +1,113 @@
+# SPDX-FileCopyrightText: Copyright 2023, Arm Limited and/or its affiliates.
+# SPDX-License-Identifier: Apache-2.0
+"""CLI command validators module."""
+from __future__ import annotations
+
+import argparse
+import logging
+import sys
+
+from mlia.cli.config import get_default_backends
+from mlia.target.registry import supported_backends
+from mlia.utils.filesystem import get_target
+
+logger = logging.getLogger(__name__)
+
+
+def validate_backend(
+    target_profile: str, backend: list[str] | None
+) -> list[str] | None:
+    """Validate backend with given target profile.
+
+    This validator checks whether the given target-profile and backend are
+    compatible with each other.
+    It assumes that prior checks where made on the validity of the target-profile.
+    """
+    target_map = {
+        "ethos-u55": "Ethos-U55",
+        "ethos-u65": "Ethos-U65",
+        "cortex-a": "Cortex-A",
+        "tosa": "TOSA",
+    }
+    target = get_target(target_profile)
+
+    if not backend:
+        return get_default_backends()[target]
+
+    compatible_backends = supported_backends(target_map[target])
+
+    nor_backend = list(map(normalize_string, backend))
+    nor_compat_backend = list(map(normalize_string, compatible_backends))
+
+    incompatible_backends = [
+        backend[i] for i, x in enumerate(nor_backend) if x not in nor_compat_backend
+    ]
+    # Throw an error if any unsupported backends are used
+    if incompatible_backends:
+        raise argparse.ArgumentError(
+            None,
+            f"{', '.join(incompatible_backends)} backend not supported "
+            f"with target-profile {target_profile}.",
+        )
+    return backend
+
+
+def validate_check_target_profile(target_profile: str, category: set[str]) -> None:
+    """Validate whether advice category is compatible with the provided target_profile.
+
+    This validator function raises warnings if any desired advice category is not
+    compatible with the selected target profile. If no operation can be
+    performed as a result of the validation, MLIA exits with error code 0.
+    """
+    incompatible_targets_performance: list[str] = ["tosa", "cortex-a"]
+    incompatible_targets_compatibility: list[str] = []
+
+    # Check which check operation should be performed
+    try_performance = "performance" in category
+    try_compatibility = "compatibility" in category
+
+    # Cross check which of the desired operations can be performed on given
+    # target-profile
+    do_performance = (
+        try_performance and target_profile not in incompatible_targets_performance
+    )
+    do_compatibility = (
+        try_compatibility and target_profile not in incompatible_targets_compatibility
+    )
+
+    # Case: desired operations can be performed with given target profile
+    if (try_performance == do_performance) and (try_compatibility == do_compatibility):
+        return
+
+    warning_message = "\nWARNING: "
+    # Case: performance operation to be skipped
+    if try_performance and not do_performance:
+        warning_message += (
+            "Performance checks skipped as they cannot be "
+            f"performed with target profile {target_profile}."
+        )
+
+    # Case: compatibility operation to be skipped
+    if try_compatibility and not do_compatibility:
+        warning_message += (
+            "Compatibility checks skipped as they cannot be "
+            f"performed with target profile {target_profile}."
+        )
+
+    # Case: at least one operation will be performed
+    if do_compatibility or do_performance:
+        logger.warning(warning_message)
+        return
+
+    # Case: no operation will be performed
+    warning_message += " No operation was performed."
+    logger.warning(warning_message)
+    sys.exit(0)
+
+
+def normalize_string(value: str) -> str:
+    """Given a string return the normalized version.
+
+    E.g. Given "ToSa-cHecker" -> "tosachecker"
+    """
+    return value.lower().replace("-", "")