Merge branch 'dev_bert' of https://github.com/Oneflow-Inc/oneflow into dev_multi-socket

Former-commit-id: 8689609b90abed7538a728395ea9880cd5af9178

oneflow/core/graph/chain_graph.cpp

@@ -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;