Fix ops_benchmark breakage

c41317ac · Liangliang He · 8358e491 · c41317ac
隐藏空白更改
内联并排

Showing with 1 addition and 0 deletion

mace/core/testing/test_benchmark_main.cc mace/core/testing/test_benchmark_main.cc +1 -0

未找到文件。
--- 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");