experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

cmake: Fix Android build error on missing MaybeCreateATraceProfiler()

Browse Source 
atrace_profiler.cc should be included in tflite library.
This CL fixes Github issue #44916.

PiperOrigin-RevId: 343968021
Change-Id: I1ae9715ef66e015254ec65701b8b630d0cf63f24
This commit is contained in:
Terry Heo 2020-11-23 18:28:12 -08:00
parent 5d44405066
commit e3cfa2c0be
1 changed files with 7 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
tensorflow/lite/CMakeLists.txt
View File

@ -319,6 +319,12 @@ populate_tflite_source_vars("kernels/internal/reference/integer_ops"
populate_tflite_source_vars("kernels/internal/reference/sparse_ops"
TFLITE_KERNEL_INTERNAL_REF_SPARSE_OPS_SRCS
)
set(TFLITE_PROFILER_SRCS ${TFLITE_SOURCE_DIR}/profiling/platform_profiler.cc)
if(CMAKE_SYSTEM_NAME MATCHES "Android")
list(APPEND TFLITE_PROFILER_SRCS
${TFLITE_SOURCE_DIR}/profiling/atrace_profiler.cc
)
endif()
# Common include directories
set(TFLITE_INCLUDE_DIRS
@ -353,7 +359,7 @@ add_library(tensorflow-lite
${TFLITE_KERNEL_SRCS}
${TFLITE_NNAPI_SRCS}
${TFLITE_SRCS}
${TFLITE_SOURCE_DIR}/profiling/platform_profiler.cc
${TFLITE_PROFILER_SRCS}
${TFLITE_SOURCE_DIR}/schema/schema_utils.cc
${TFLITE_SOURCE_DIR}/tools/optimize/sparsity/format_converter.cc
)
@ -392,7 +398,6 @@ populate_source_vars("${TFLITE_SOURCE_DIR}/tools/benchmark"
list(APPEND TFLITE_BENCHMARK_SRCS
${TF_SOURCE_DIR}/core/util/stats_calculator.cc
${TFLITE_SOURCE_DIR}/profiling/memory_info.cc
${TFLITE_SOURCE_DIR}/profiling/platform_profiler.cc
${TFLITE_SOURCE_DIR}/profiling/profile_summarizer.cc
${TFLITE_SOURCE_DIR}/profiling/profile_summary_formatter.cc
${TFLITE_SOURCE_DIR}/profiling/time.cc
@ -424,9 +429,6 @@ else()
endif() # TFLITE_ENABLE_XNNPACK
if(CMAKE_SYSTEM_NAME MATCHES "Android")
list(APPEND TFLITE_BENCHMARK_SRCS
${TFLITE_SOURCE_DIR}/profiling/atrace_profiler.cc
)
if(_TFLITE_ENABLE_NNAPI)
list(APPEND TFLITE_BENCHMARK_SRCS
${TFLITE_SOURCE_DIR}/tools/delegates/nnapi_delegate_provider.cc