提交 d0572bf0 编写于 作者: Q Qiao Longfei

add log for lazy mode test=develop

上级 1177b0bc
...@@ -509,6 +509,7 @@ class AdamOpKernel : public framework::OpKernel<T> { ...@@ -509,6 +509,7 @@ 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 { } else {
if (lazy_mode) { if (lazy_mode) {
VLOG(3) << "run cpu lazy mode";
size_t row_count = grad_merge.rows().size(); size_t row_count = grad_merge.rows().size();
std::vector<int64_t> cpu_rows(grad_merge.rows()); std::vector<int64_t> cpu_rows(grad_merge.rows());
for (size_t row_index = 0; row_index < row_count; ++row_index) { for (size_t row_index = 0; row_index < row_count; ++row_index) {
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册