From 3e6030a241b8cb6fdc9953d7853c209fe2ee1a06 Mon Sep 17 00:00:00 2001 From: periannath Date: Mon, 20 Apr 2020 21:49:25 +0900 Subject: [PATCH] [TFLite] Introduce ruy_profiler to makefile Introduce `BUILD_WITH_RUY_PROFILER` option to support ruy_profiler --- tensorflow/lite/tools/make/Makefile | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/tensorflow/lite/tools/make/Makefile b/tensorflow/lite/tools/make/Makefile index 426ed63b482..87a76cea69f 100644 --- a/tensorflow/lite/tools/make/Makefile +++ b/tensorflow/lite/tools/make/Makefile @@ -184,6 +184,14 @@ ifeq ($(BUILD_WITH_RUY),true) CXXFLAGS += -DTFLITE_WITH_RUY endif +BUILD_WITH_RUY_PROFILER ?= false +ifeq ($(BUILD_WITH_RUY_PROFILER),true) + CORE_CC_ALL_SRCS += tensorflow/lite/tools/make/downloads/ruy/ruy/profiler/instrumentation.cc + CORE_CC_ALL_SRCS += tensorflow/lite/tools/make/downloads/ruy/ruy/profiler/profiler.cc + CORE_CC_ALL_SRCS += tensorflow/lite/tools/make/downloads/ruy/ruy/profiler/treeview.cc + CXXFLAGS += -DRUY_PROFILER +endif + # Not to include XNNPACK. CXXFLAGS += -DTFLITE_WITHOUT_XNNPACK