提交 1f9b2ee2 编写于 作者: 吴承辉

Merge branch 'fix_cpu_openmp' into 'master'

Fix Android OpenMP does not work issue

See merge request !479
...@@ -4,7 +4,7 @@ load("//mace:mace.bzl", "if_openmp_enabled", "if_android") ...@@ -4,7 +4,7 @@ load("//mace:mace.bzl", "if_openmp_enabled", "if_android")
cc_binary( cc_binary(
name = "mace_run", name = "mace_run",
srcs = ["mace_run.cc"], srcs = ["mace_run.cc"],
copts = if_openmp_enabled(["-fopenmp"]) + if_android([ copts = if_android([
"-DMACE_ENABLE_OPENCL", "-DMACE_ENABLE_OPENCL",
]), ]),
linkopts = if_openmp_enabled(["-fopenmp"]), linkopts = if_openmp_enabled(["-fopenmp"]),
......
...@@ -386,7 +386,6 @@ int Main(int argc, char **argv) { ...@@ -386,7 +386,6 @@ int Main(int argc, char **argv) {
} }
bool ret = false; bool ret = false;
#pragma omp parallel for
for (int i = 0; i < FLAGS_restart_round; ++i) { for (int i = 0; i < FLAGS_restart_round; ++i) {
VLOG(0) << "restart round " << i; VLOG(0) << "restart round " << i;
ret = ret =
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册