diff --git a/mace/tools/validation/BUILD b/mace/tools/validation/BUILD index 151423ef1690a991fba29b1487cab54fff120acb..236b05434004d52f2e11748aed2465430121e067 100644 --- a/mace/tools/validation/BUILD +++ b/mace/tools/validation/BUILD @@ -4,7 +4,7 @@ load("//mace:mace.bzl", "if_openmp_enabled", "if_android") cc_binary( name = "mace_run", srcs = ["mace_run.cc"], - copts = if_openmp_enabled(["-fopenmp"]) + if_android([ + copts = if_android([ "-DMACE_ENABLE_OPENCL", ]), linkopts = if_openmp_enabled(["-fopenmp"]), diff --git a/mace/tools/validation/mace_run.cc b/mace/tools/validation/mace_run.cc index dc3faaf07ab115f8ab5a30151878bdf4a4fd83eb..e912071525b4e3adef108ae5ff55301c0792dd9b 100644 --- a/mace/tools/validation/mace_run.cc +++ b/mace/tools/validation/mace_run.cc @@ -386,7 +386,6 @@ int Main(int argc, char **argv) { } bool ret = false; -#pragma omp parallel for for (int i = 0; i < FLAGS_restart_round; ++i) { VLOG(0) << "restart round " << i; ret =