Sink tick in main job (#4207)
* source subset tick
* remove useless header files
* insert DstSubsetTickOp
* remove incorrect CHECK
* add tick op for each machine
* TryBindBnWithOneofRegst
* add sink tick op in main_job
* refactor LinkMainJob
* fix typo in task_graph
* refactor AddGlobalCriticalSection
* rename and refactor DstSubsetTick::InferBlobDescs and SrcSubsetTick::InferBlobDescs
* add src_subset_tick for input-output critical section
* refactor AutoSourceTick and AutoSinkTick
* SrcSubsetTickCompTaskNode: bind bns and in_regst if bns is valid in current device
* refactor optional input to repeated inputs for SrcSubsetTickOpConf
Co-authored-by:
oneflow-ci-bot <69100618+oneflow-ci-bot@users.noreply.github.com>
Showing
- oneflow/core/graph_impl/logical_node.cpp 1 addition, 1 deletiononeflow/core/graph_impl/logical_node.cpp
- oneflow/core/job/critical_section.proto 2 additions, 2 deletionsoneflow/core/job/critical_section.proto
- oneflow/core/job/critical_section_desc.cpp 5 additions, 1 deletiononeflow/core/job/critical_section_desc.cpp
- oneflow/core/job/critical_section_desc.h 1 addition, 1 deletiononeflow/core/job/critical_section_desc.h
- oneflow/core/job/oneflow.cpp 37 additions, 18 deletionsoneflow/core/job/oneflow.cpp
- oneflow/core/job_rewriter/autotick.cpp 47 additions, 63 deletionsoneflow/core/job_rewriter/autotick.cpp
- oneflow/core/job_rewriter/autotick.h 2 additions, 3 deletionsoneflow/core/job_rewriter/autotick.h
- oneflow/core/job_rewriter/job_completer.cpp 2 additions, 3 deletionsoneflow/core/job_rewriter/job_completer.cpp
Please register or sign in to comment