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