From 918aae3b609db323f5909e860c3d87a99149ec07 Mon Sep 17 00:00:00 2001 From: liuqi Date: Mon, 30 Oct 2017 20:10:01 +0800 Subject: [PATCH] Revert the cpu and noen benchmark. --- mace/ops/batch_norm_benchmark.cc | 2 ++ 1 file changed, 2 insertions(+) diff --git a/mace/ops/batch_norm_benchmark.cc b/mace/ops/batch_norm_benchmark.cc index 4994dad3..e1b53b92 100644 --- a/mace/ops/batch_norm_benchmark.cc +++ b/mace/ops/batch_norm_benchmark.cc @@ -60,6 +60,8 @@ static void BatchNorm( BENCHMARK(BM_BATCH_NORM_##N##_##C##_##H##_##W##_##TYPE##_##DEVICE) #define BM_BATCH_NORM(N, C, H, W, TYPE) \ + BM_BATCH_NORM_MACRO(N, C, H, W, TYPE, CPU); \ + BM_BATCH_NORM_MACRO(N, C, H, W, TYPE, NEON);\ BM_BATCH_NORM_MACRO(N, C, H, W, TYPE, OPENCL); BM_BATCH_NORM(1, 1, 512, 512, float); -- GitLab