diff options
Diffstat (limited to 'scripts')
-rw-r--r-- | scripts/cmake/common_user_options.cmake | 6 | ||||
-rw-r--r-- | scripts/cmake/tensorflow.cmake | 3 |
2 files changed, 7 insertions, 2 deletions
diff --git a/scripts/cmake/common_user_options.cmake b/scripts/cmake/common_user_options.cmake index 3e1206d..97e9b40 100644 --- a/scripts/cmake/common_user_options.cmake +++ b/scripts/cmake/common_user_options.cmake @@ -88,9 +88,13 @@ if (NOT TARGET_PLATFORM STREQUAL native) "${DEPENDENCY_ROOT_DIR}/cmsis" PATH) USER_OPTION(CMSIS_DSP_SRC_PATH - "Path to CMSIS-5 DSP sources" + "Path to CMSIS-5 DSP sources" "${DEPENDENCY_ROOT_DIR}/cmsis-dsp" PATH) + USER_OPTION(CMSIS_NN_SRC_PATH + "Path to CMSIS-5 NN sources" + "${DEPENDENCY_ROOT_DIR}/cmsis-nn" + PATH) # If we need NPU libraries: if (ETHOS_U_NPU_ENABLED) diff --git a/scripts/cmake/tensorflow.cmake b/scripts/cmake/tensorflow.cmake index 4b50ed0..d807416 100644 --- a/scripts/cmake/tensorflow.cmake +++ b/scripts/cmake/tensorflow.cmake @@ -1,5 +1,5 @@ #---------------------------------------------------------------------------- -# SPDX-FileCopyrightText: Copyright 2021 - 2022 Arm Limited and/or its affiliates <open-source-office@arm.com> +# SPDX-FileCopyrightText: Copyright 2021-2022 Arm Limited and/or its affiliates <open-source-office@arm.com> # SPDX-License-Identifier: Apache-2.0 # # Licensed under the Apache License, Version 2.0 (the "License"); @@ -101,6 +101,7 @@ add_custom_target(tensorflow_build ALL TARGET_ARCH=${TENSORFLOW_LITE_MICRO_TARGET_ARCH} BUILD_TYPE=${TENSORFLOW_LITE_MICRO_BUILD_TYPE} CMSIS_PATH=${CMSIS_SRC_PATH} + CMSIS_NN_PATH=${CMSIS_NN_SRC_PATH} # Conditional arguments $<$<BOOL:${ETHOS_U_NPU_ENABLED}>:ETHOSU_ARCH=${ETHOS_U_NPU_ID}> $<$<BOOL:${ETHOS_U_NPU_ENABLED}>:ETHOSU_DRIVER_PATH=${ETHOS_U_NPU_DRIVER_SRC_PATH}> |