From 2181d0ac35f30202985a877950c88325ff665f6b Mon Sep 17 00:00:00 2001
From: Isabella Gottardi <isabella.gottardi@arm.com>
Date: Wed, 7 Apr 2021 09:27:38 +0100
Subject: MLECO-1766: Default build flow helper scripts added MLECO-1882:
 Anomaly Detection use-case default model download added and tests updated to
 run with it.

Test data generation cmake logic moved from use-case cmakes to top-level cmake script.

Signed-off-by: Isabella Gottardi <isabella.gottardi@arm.com>
Change-Id: Ifde469e3585c37b9a53810236a92ce52d4fbb407
---
 build_default.py | 89 ++++++++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 89 insertions(+)
 create mode 100644 build_default.py

(limited to 'build_default.py')

diff --git a/build_default.py b/build_default.py
new file mode 100644
index 0000000..318ccb1
--- /dev/null
+++ b/build_default.py
@@ -0,0 +1,89 @@
+#!env/bin/python3
+
+#  Copyright (c) 2021 Arm Limited. All rights reserved.
+#  SPDX-License-Identifier: Apache-2.0
+#
+#  Licensed under the Apache License, Version 2.0 (the "License");
+#  you may not use this file except in compliance with the License.
+#  You may obtain a copy of the License at
+#
+#      http://www.apache.org/licenses/LICENSE-2.0
+#
+#  Unless required by applicable law or agreed to in writing, software
+#  distributed under the License is distributed on an "AS IS" BASIS,
+#  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+#  See the License for the specific language governing permissions and
+#  limitations under the License.
+
+import os
+import subprocess
+import shutil
+import multiprocessing
+import logging
+import sys
+from argparse import ArgumentParser
+
+from set_up_default_resources import set_up_resources
+
+
+def run(download_resources, run_vela_on_models):
+    """
+    Run the helpers scripts.
+
+    Parameters:
+    ----------
+    download_resources (bool):    Specifies if 'Download resources' step is performed.
+    run_vela_on_models (bool):    Only if `download_resources` is True, specifies if run vela on downloaded models.
+    """
+
+    current_file_dir = os.path.dirname(os.path.abspath(__file__))
+    logging.basicConfig(filename='log_build_default.log', level=logging.DEBUG)
+    logging.getLogger().addHandler(logging.StreamHandler(sys.stdout))
+
+    # 1. Download models if specified
+    if download_resources is True:
+        logging.info("Downloading resources.")
+        set_up_resources(run_vela_on_models)
+
+    # 2. Build default configuration
+    logging.info("Building default configuration.")
+    target_platform = "mps3"
+    target_subsystem = "sse-300"
+    build_dir = os.path.join(current_file_dir, f"cmake-build-{target_platform}-{target_subsystem}-release")
+    try:
+        os.mkdir(build_dir)
+    except FileExistsError:
+        # Directory already exists, clean it
+        for filename in os.listdir(build_dir):
+            filepath = os.path.join(build_dir, filename)
+            try:
+                if os.path.isfile(filepath) or os.path.islink(filepath):
+                    os.unlink(filepath)
+                elif os.path.isdir(filepath):
+                    shutil.rmtree(filepath)
+            except Exception as e:
+                logging.error('Failed to delete %s. Reason: %s' % (filepath, e))
+    os.chdir(build_dir)
+    cmake_toolchain_file = os.path.join(current_file_dir, "scripts", "cmake", "bare-metal-toolchain.cmake")
+    cmake_command = (f"cmake .. -DTARGET_PLATFORM={target_platform} -DTARGET_SUBSYSTEM={target_subsystem} " +
+                     f"-DCMAKE_TOOLCHAIN_FILE={cmake_toolchain_file} ")
+    logging.info(cmake_command)
+    state = subprocess.run(cmake_command, shell=True, stdout=subprocess.PIPE, stderr=subprocess.STDOUT)
+    logging.info(state.stdout.decode('utf-8'))
+
+    make_command = f"make -j{multiprocessing.cpu_count()}"
+    logging.info(make_command)
+    state = subprocess.run(make_command, shell=True, stdout=subprocess.PIPE, stderr=subprocess.STDOUT)
+    logging.info(state.stdout.decode('utf-8'))
+
+
+if __name__ == '__main__':
+    parser = ArgumentParser()
+    parser.add_argument("--skip-download",
+                        help="Do not download resources: models and test vectors",
+                        action="store_true")
+    parser.add_argument("--skip-vela",
+                        help="Do not run Vela optimizer on downloaded models.",
+                        action="store_true")
+    args = parser.parse_args()
+    run(not args.skip_download, not args.skip_vela)
-- 
cgit v1.2.1