提交 599f7a87 编写于 作者: Y Yu Yang

Refine code

上级 43e54079
...@@ -389,10 +389,6 @@ struct ComputationOpHandle : public OpHandle { ...@@ -389,10 +389,6 @@ struct ComputationOpHandle : public OpHandle {
for (auto *in : inputs_) { for (auto *in : inputs_) {
bool need_wait = bool need_wait =
in->generated_op_ && in->generated_op_->dev_ctx_[place_] != cur_ctx; in->generated_op_ && in->generated_op_->dev_ctx_[place_] != cur_ctx;
if (dynamic_cast<NCCLAllReduceOpHandle *>(in->generated_op_)) {
VLOG(3) << "Input is nccl all reduce, need to wait" << need_wait;
}
if (need_wait) { if (need_wait) {
in->generated_op_->Wait(cur_ctx); in->generated_op_->Wait(cur_ctx);
} }
...@@ -545,13 +541,6 @@ void ParallelExecutor::ConstructDependencyGraph( ...@@ -545,13 +541,6 @@ void ParallelExecutor::ConstructDependencyGraph(
harzaeds need to be handled. harzaeds need to be handled.
*/ */
PolishGraphToSupportDataHazards(); PolishGraphToSupportDataHazards();
for (auto &g : grads) {
LOG(INFO) << member_->vars_.begin()
->second[g]
.rbegin()
->second.pending_ops_.size();
}
} }
/** /**
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册