Tick per machine (#4204)
* source subset tick
* remove useless header files
* insert DstSubsetTickOp
* remove incorrect CHECK
* add tick op for each machine
* TryBindBnWithOneofRegst
* fix typo in task_graph
* rename and refactor DstSubsetTick::InferBlobDescs and SrcSubsetTick::InferBlobDescs
* 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 10 additions, 6 deletionsoneflow/core/graph_impl/logical_node.cpp
- oneflow/core/graph_impl/src_subset_tick_compute_task_node.cpp 1 addition, 1 deletion...low/core/graph_impl/src_subset_tick_compute_task_node.cpp
- oneflow/core/job_rewriter/autotick.cpp 34 additions, 24 deletionsoneflow/core/job_rewriter/autotick.cpp
- oneflow/core/operator/op_conf.proto 1 addition, 1 deletiononeflow/core/operator/op_conf.proto
- oneflow/core/operator/src_subset_tick_op.cpp 1 addition, 1 deletiononeflow/core/operator/src_subset_tick_op.cpp
Please register or sign in to comment