diff options
| author | Per Åstrand <per.astrand@arm.com> | 2020-05-12 13:16:37 +0200 |
|---|---|---|
| committer | Per Åstrand <per.astrand@arm.com> | 2020-05-12 13:16:46 +0200 |
| commit | 3f6e07ac33c7bd320527e556d8a9fdbd343c1a91 (patch) | |
| tree | e69d8ed4a4749afe7b23e43609e0850a78603c34 | |
| parent | 43fd4deb94ec32426bfdc64b6c4804d75c5cbc6b (diff) | |
| download | ethos-u-core-software-20.07.tar.gz | |
Pass on compiler information depending on used toolchain.
Change-Id: Icf97f6c4d2b6f889b2cc3b87b7db356c097fc1a4
| -rw-r--r-- | tensorflow.cmake | 14 |
1 files changed, 11 insertions, 3 deletions
diff --git a/tensorflow.cmake b/tensorflow.cmake index d304a84..312b783 100644 --- a/tensorflow.cmake +++ b/tensorflow.cmake @@ -19,9 +19,17 @@ include(ProcessorCount) ProcessorCount(J) -set(TFLU_CC "${CMAKE_C_COMPILER} --target=${CMAKE_C_COMPILER_TARGET} -mcpu=${CMAKE_SYSTEM_PROCESSOR}${CPU_FEATURES}") -set(TFLU_CXX "${CMAKE_CXX_COMPILER} --target=${CMAKE_C_COMPILER_TARGET} -mcpu=${CMAKE_SYSTEM_PROCESSOR}${CPU_FEATURES}") -set(TFLU_AR ${CMAKE_AR}) +if (CMAKE_CXX_COMPILER_ID STREQUAL "ARMClang") + set(TFLU_CC "${CMAKE_C_COMPILER} --target=${CMAKE_C_COMPILER_TARGET} -mcpu=${CMAKE_SYSTEM_PROCESSOR}${CPU_FEATURES}") + set(TFLU_CXX "${CMAKE_CXX_COMPILER} --target=${CMAKE_C_COMPILER_TARGET} -mcpu=${CMAKE_SYSTEM_PROCESSOR}${CPU_FEATURES}") + set(TFLU_AR ${CMAKE_AR}) +elseif (CMAKE_CXX_COMPILER_ID STREQUAL "GNU") + set(TFLU_CC "${CMAKE_C_COMPILER} -mcpu=${CMAKE_SYSTEM_PROCESSOR}${CPU_FEATURES}") + set(TFLU_CXX "${CMAKE_CXX_COMPILER} -mcpu=${CMAKE_SYSTEM_PROCESSOR}${CPU_FEATURES}") + set(TFLU_AR ${CMAKE_AR}) +else () + message(FATAL_ERROR "No compiler ID is set") +endif() set(TFLU_PATH "${TENSORFLOW_PATH}/tensorflow/lite/micro") set(TFLU_GENDIR ${CMAKE_CURRENT_BINARY_DIR}/tensorflow/) |