From 3f6e07ac33c7bd320527e556d8a9fdbd343c1a91 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Per=20=C3=85strand?= <per.astrand@arm.com>
Date: Tue, 12 May 2020 13:16:37 +0200
Subject: MLBEDSW-1732 Add support for arm-none-eabi-gcc

Pass on compiler information depending on used toolchain.

Change-Id: Icf97f6c4d2b6f889b2cc3b87b7db356c097fc1a4
---
 tensorflow.cmake | 14 +++++++++++---
 1 file 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/)
-- 
cgit v1.2.1