summaryrefslogtreecommitdiff
path: root/build_default.py
diff options
context:
space:
mode:
authorIsabella Gottardi <isabella.gottardi@arm.com>2021-04-07 09:27:38 +0100
committerIsabella Gottardi <isabella.gottardi@arm.com>2021-05-07 12:19:19 +0100
commit2181d0ac35f30202985a877950c88325ff665f6b (patch)
treee16c50d41d85945e0c2c864323ac1769b02af64f /build_default.py
parentd580eee180be219e118152cedabc9637da8574d6 (diff)
downloadml-embedded-evaluation-kit-2181d0ac35f30202985a877950c88325ff665f6b.tar.gz
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
Diffstat (limited to 'build_default.py')
-rw-r--r--build_default.py89
1 files changed, 89 insertions, 0 deletions
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)