commit 3f6e07ac33c7bd320527e556d8a9fdbd343c1a91
author Per Åstrand <per.astrand@arm.com> Tue May 12 13:16:37 2020 +0200
committer Per Åstrand <per.astrand@arm.com> Tue May 12 13:16:46 2020 +0200
tree e69d8ed4a4749afe7b23e43609e0850a78603c34
parent 43fd4deb94ec32426bfdc64b6c4804d75c5cbc6b

MLBEDSW-1732 Add support for arm-none-eabi-gcc

Pass on compiler information depending on used toolchain.

Change-Id: Icf97f6c4d2b6f889b2cc3b87b7db356c097fc1a4

tensorflow.cmake

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/)