From c41317acd5abf4231f77ff64cbf8fb03be274a80 Mon Sep 17 00:00:00 2001 From: Liangliang He Date: Thu, 29 Mar 2018 09:35:29 +0800 Subject: [PATCH] Fix ops_benchmark breakage --- mace/core/testing/test_benchmark_main.cc | 1 + 1 file changed, 1 insertion(+) diff --git a/mace/core/testing/test_benchmark_main.cc b/mace/core/testing/test_benchmark_main.cc index 9c2d1daa..079a79e7 100644 --- a/mace/core/testing/test_benchmark_main.cc +++ b/mace/core/testing/test_benchmark_main.cc @@ -7,6 +7,7 @@ #include "gflags/gflags.h" #include "mace/core/testing/test_benchmark.h" #include "mace/public/mace.h" +#include "mace/public/mace_runtime.h" DEFINE_string(pattern, "all", "op benchmark pattern, eg:.*CONV.*"); DEFINE_int32(gpu_perf_hint, 3, "0:DEFAULT/1:LOW/2:NORMAL/3:HIGH"); -- GitLab