diff --git a/mace/ops/resize_bilinear_benchmark.cc b/mace/ops/resize_bilinear_benchmark.cc index 37c07cd270b64adf80ceb34fe3a2b4c6a0f9f1a0..c36fd7c9140d369f1c8df2ae8f2b4a2abf05f149 100644 --- a/mace/ops/resize_bilinear_benchmark.cc +++ b/mace/ops/resize_bilinear_benchmark.cc @@ -40,6 +40,7 @@ static void ResizeBilinearBenchmark(int iters, while (iters--) { net.RunOp(D); } + net.Sync(); } #define BM_RESIZE_BILINEAR_MACRO(N, C, H0, W0, H1, W1, TYPE, DEVICE) \ diff --git a/mace/ops/resize_bilinear_test.cc b/mace/ops/resize_bilinear_test.cc index d569ad71b08804a7ed8e948afd00d6b90eef9a43..92f7ec4671dcd4bbc73f03ab6c4459bfa3ecd85a 100644 --- a/mace/ops/resize_bilinear_test.cc +++ b/mace/ops/resize_bilinear_test.cc @@ -86,12 +86,6 @@ void TestRandomResizeBilinear() { net.AddRandomInput("Input", {batch, channels, in_height, in_width}); net.AddInputFromArray("OutSize", {2}, {height, width}); - /* - vector input(24); - std::iota(begin(input), end(input), 0); - net.AddInputFromArray("Input", {1, 3, 2, 4}, input); - net.AddInputFromArray("OutSize", {2}, {1, 2}); - */ // Run net.RunOp(D);