提交 88a04c86 编写于 作者: J Juncheng 提交者: Li Xinqi

fix fw_consumer nullptr (#1622)



Former-commit-id: 92b9ff1ae7efc90391581eafe8253347a5712b9c
上级 52a6c519
......@@ -140,6 +140,7 @@ bool NoOutRegstConsumedByBwNode(TaskNode* node) {
if (fw_node == nullptr) { return false; }
for (TaskEdge* edge : fw_node->out_edges()) {
auto* fw_consumer = dynamic_cast<NormalForwardCompTaskNode*>(edge->dst_node());
if (fw_consumer == nullptr) { return false; }
if (fw_consumer->HasBackwardCompTaskNode()) { return false; }
}
return true;
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册