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