diff --git a/mace/core/testing/test_benchmark_main.cc b/mace/core/testing/test_benchmark_main.cc
index 9c2d1daa916513054796ec8ad22caca59869c772..079a79e79afb35ed3319380409ae6155ec4112d9 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");