change the limit of thead num

recover_files
Qiao Longfei 7 years ago
parent 39a400345e
commit 0e747e8d02

@ -490,9 +490,10 @@ class AdamOpKernel : public framework::OpKernel<T> {
<< FLAGS_inner_op_parallelism << FLAGS_inner_op_parallelism
<< " min_param_size_to_use_multithread=" << " min_param_size_to_use_multithread="
<< FLAGS_min_param_size_to_use_multithread; << FLAGS_min_param_size_to_use_multithread;
PADDLE_ENFORCE_LE( if (FLAGS_inner_op_parallelism > 10) {
FLAGS_inner_op_parallelism, 8, LOG(WARNING) << "FLAGS_inner_op_parallelism "
"FLAGS_inner_op_parallelism should not be larger then 8"); << FLAGS_inner_op_parallelism << " is two large!";
}
auto& grad_rows = grad_merge.rows(); auto& grad_rows = grad_merge.rows();
std::unordered_map<size_t, int> row_id_to_grad_row_offset; std::unordered_map<size_t, int> row_id_to_grad_row_offset;
size_t param_row_count = param.numel() / row_numel; size_t param_row_count = param.numel() / row_numel;

Loading…
Cancel
Save