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

fix dygraph grad to support high differential (#36059)

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