Method BuildTaskPath

oneflow/core/graph/task_graph.cpp:1037–1049

}

void TaskGraph::BuildTaskPath(TaskNode* src_node, TaskNode* dst_node, const LogicalBlobId& lbi,
                              bool is_host_mem_input) {
  const MemZoneId dst_mem_zone_id = [&]() {
    if (is_host_mem_input) {
      MemZoneId mem_zone_id = dst_node->MemZoneId121();
      return MemZoneId(mem_zone_id.rank(), DeviceType::kCPU, 0);
    } else {
      return dst_node->MemZoneId121();
    }
  }();
  TaskNode* proxy_node = GetProxyNode(src_node, lbi, dst_mem_zone_id);
  ConnectWithLbi(proxy_node, dst_node, lbi);
}

Maybe<void> GlobalTaskGraph::Init() {
  OpGraph* op_graph = Singleton<OpGraph>::Get();