diff --git a/tensorflow/lite/tools/make/Makefile b/tensorflow/lite/tools/make/Makefile index 26d3ce99223..ad3832f9962 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