未验证 提交 e123b873 编写于 作者: J Jiabin Yang 提交者: GitHub

fix dygraph grad to support high differential (#36059)

上级 ac72f974
......@@ -137,10 +137,12 @@ static void GetGraphInfoBetweenTargets(
}
for (auto &pending_node : node->GradPendingNodes()) {
for (auto &pending_op : *pending_node) {
preceding_ops[&pending_op].insert(op);
}
if (visited.count(pending_node.get()) == 0) {
visited.insert(pending_node.get());
for (auto &pending_op : *pending_node) {
preceding_ops[&pending_op].insert(op);
q.emplace(&pending_op, pending_node.get());
}
}
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册