Method BuildExecGphAndRegst

oneflow/core/graph/copy_task_node.cpp:30–48

void CopyTaskNode::ConsumeAllRegsts() { ConsumeRegst("copy_in", SoleInDataEdge()->GetSoleRegst()); }

void CopyTaskNode::BuildExecGphAndRegst() {
  auto out_regst = GetProducedRegst("copy_out");
  auto in_regst = GetSoleConsumedRegst("copy_in");
  out_regst->CopyBlobDescFrom(in_regst.get());
  ExecNode* node = mut_exec_gph().NewNode();
  auto constructed = CHECK_JUST(ConstructOp(NewCopyOpConf()));

  // prevent filling parallel desc for copy commnet
  if (constructed->op_conf().has_user_conf()) {
    std::shared_ptr<Shape> hierarchy = std::make_shared<Shape>(Shape({1}));
    auto parallel_desc =
        ParallelDesc::New(constructed->op_conf().device_tag(), {"0:0-0"}, hierarchy).GetOrThrow();
    CHECK_JUST(constructed->FillOpParallelDesc(parallel_desc));
  }

  node->mut_op() = constructed;
  node->BindBnWithRegst(node->op()->SoleIbn(), in_regst);
  node->BindBnWithRegst(node->op()->SoleObn(), out_regst);
}

void CopyTaskNode::InferProducedDataRegstTimeShape() { NaiveInferProducedDataRegstTimeShape(); }