提交 59cf96ec 编写于 作者: Q Qiao Longfei

add log

上级 3bd54ed7
...@@ -357,6 +357,9 @@ class AdamOpKernel : public framework::OpKernel<T> { ...@@ -357,6 +357,9 @@ class AdamOpKernel : public framework::OpKernel<T> {
if (inner_op_parallelism > 1 && if (inner_op_parallelism > 1 &&
FLAGS_min_param_size_to_use_multithread > 0 && FLAGS_min_param_size_to_use_multithread > 0 &&
param.numel() > FLAGS_min_param_size_to_use_multithread) { param.numel() > FLAGS_min_param_size_to_use_multithread) {
VLOG(3) << "use multi thread, inner_op_parallelism="
<< inner_op_parallelism << " min_param_size_to_use_multithread"
<< FLAGS_min_param_size_to_use_multithread;
std::vector<std::future<void>> fs; std::vector<std::future<void>> fs;
int64_t block_size = param.numel() / inner_op_parallelism; int64_t block_size = param.numel() / inner_op_parallelism;
for (int i = 0; i < inner_op_parallelism; ++i) { for (int i = 0; i < inner_op_parallelism; ++i) {
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册