| MICROLITE_BENCHMARK_ROOT_DIR := $(TENSORFLOW_ROOT)tensorflow/lite/micro/tools/benchmarking |
| |
| ifneq ($(BENCHMARK_MODEL_PATH),) |
| GENERIC_BENCHMARK_MODEL_DIR := $(dir $(BENCHMARK_MODEL_PATH)) |
| GENERIC_BENCHMARK_MODEL_NAME := $(notdir $(basename $(BENCHMARK_MODEL_PATH))) |
| CXXFLAGS += -DMODEL_HEADER_PATH=\"$(GENERIC_BENCHMARK_MODEL_DIR)$(GENERIC_BENCHMARK_MODEL_NAME)_model_data.h\" |
| CXXFLAGS += -DMODEL_NAME=$(GENERIC_BENCHMARK_MODEL_NAME) |
| ifneq ($(BENCHMARK_ARENA_SIZE),) |
| CXXFLAGS += -DTENSOR_ARENA_SIZE=$(BENCHMARK_ARENA_SIZE) |
| endif |
| |
| GENERIC_BENCHMARK_GENERATOR_INPUTS := $(TENSORFLOW_ROOT)$(BENCHMARK_MODEL_PATH) |
| |
| GENERIC_BENCHMARK_GENERATED_SRCS := \ |
| $(GENERATED_SRCS_DIR)$(TENSORFLOW_ROOT)$(GENERIC_BENCHMARK_MODEL_DIR)$(GENERIC_BENCHMARK_MODEL_NAME)_model_data.cc |
| |
| GENERIC_BENCHMARK_GENERATED_HDRS := \ |
| $(GENERATED_SRCS_DIR)$(TENSORFLOW_ROOT)$(GENERIC_BENCHMARK_MODEL_DIR)$(GENERIC_BENCHMARK_MODEL_NAME)_model_data.h |
| endif |
| |
| GENERIC_BENCHMARK_SRCS := \ |
| $(MICROLITE_BENCHMARK_ROOT_DIR)/generic_model_benchmark.cc \ |
| $(MICROLITE_BENCHMARK_ROOT_DIR)/metrics.cc |
| |
| GENERIC_BENCHMARK_HDRS := \ |
| $(MICROLITE_BENCHMARK_ROOT_DIR)/op_resolver.h \ |
| $(MICROLITE_BENCHMARK_ROOT_DIR)/metrics.h |
| |
| ifneq ($(TARGET),bluepill) |
| ifneq ($(TARGET_ARCH), $(filter $(TARGET_ARCH), hifimini)) |
| $(eval $(call microlite_test,tflm_benchmark,\ |
| $(GENERIC_BENCHMARK_SRCS),$(GENERIC_BENCHMARK_HDRS),\ |
| $(GENERIC_BENCHMARK_GENERATOR_INPUTS))) |
| endif |
| endif |
