From 37959522a805a5e23c930ed79aac84920c3cb208 Mon Sep 17 00:00:00 2001 From: Dmitrii Agibov 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 --- src/mlia/backend/tosa_checker/__init__.py | 3 +++ src/mlia/backend/tosa_checker/install.py | 19 +++++++++++++++++++ 2 files changed, 22 insertions(+) create mode 100644 src/mlia/backend/tosa_checker/__init__.py create mode 100644 src/mlia/backend/tosa_checker/install.py (limited to 'src/mlia/backend/tosa_checker') diff --git a/src/mlia/backend/tosa_checker/__init__.py b/src/mlia/backend/tosa_checker/__init__.py new file mode 100644 index 0000000..cec210d --- /dev/null +++ b/src/mlia/backend/tosa_checker/__init__.py @@ -0,0 +1,3 @@ +# SPDX-FileCopyrightText: Copyright 2022, Arm Limited and/or its affiliates. +# SPDX-License-Identifier: Apache-2.0 +"""TOSA checker backend module.""" diff --git a/src/mlia/backend/tosa_checker/install.py b/src/mlia/backend/tosa_checker/install.py new file mode 100644 index 0000000..72454bc --- /dev/null +++ b/src/mlia/backend/tosa_checker/install.py @@ -0,0 +1,19 @@ +# SPDX-FileCopyrightText: Copyright 2022, Arm Limited and/or its affiliates. +# SPDX-License-Identifier: Apache-2.0 +"""Module for python package based installations.""" +from __future__ import annotations + +from mlia.backend.install import Installation +from mlia.backend.install import PyPackageBackendInstallation + + +def get_tosa_backend_installation() -> Installation: + """Get TOSA backend installation.""" + return PyPackageBackendInstallation( + name="tosa-checker", + description="Tool to check if a ML model is compatible " + "with the TOSA specification", + packages_to_install=["mlia[tosa]"], + packages_to_uninstall=["tosa-checker"], + expected_packages=["tosa-checker"], + ) -- cgit v1.2.1