未验证 提交 994e73f6 编写于 作者: W wopeizl 提交者: GitHub

Merge pull request #15351 from wopeizl/fixbuildissue

disable the parallel mode for adam op on windows test=develop
...@@ -486,7 +486,9 @@ class AdamOpKernel : public framework::OpKernel<T> { ...@@ -486,7 +486,9 @@ class AdamOpKernel : public framework::OpKernel<T> {
functor.adam_update(i, grad_data[row_index * row_numel + offset]); functor.adam_update(i, grad_data[row_index * row_numel + offset]);
} }
} }
} else if (FLAGS_inner_op_parallelism > 1 && }
#ifndef _WIN32
else if (FLAGS_inner_op_parallelism > 1 &&
min_row_size_to_use_multithread > 0 && min_row_size_to_use_multithread > 0 &&
param.dims()[0] > min_row_size_to_use_multithread) { param.dims()[0] > min_row_size_to_use_multithread) {
VLOG(3) << "use multi thread, inner_op_parallelism=" VLOG(3) << "use multi thread, inner_op_parallelism="
...@@ -542,7 +544,9 @@ class AdamOpKernel : public framework::OpKernel<T> { ...@@ -542,7 +544,9 @@ class AdamOpKernel : public framework::OpKernel<T> {
})); }));
} }
for (size_t i = 0; i < fs.size(); ++i) fs[i].wait(); for (size_t i = 0; i < fs.size(); ++i) fs[i].wait();
} else { }
#endif // !_WIN32
else {
functor(param.numel()); functor(param.numel());
} }
} else if (platform::is_gpu_place(ctx.GetPlace())) { } else if (platform::is_gpu_place(ctx.GetPlace())) {
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册