nn.Graph call and launch impl (#5580)
* nn.Graph call and launch impl
* Fix bug of 1.pybind11 list->vector; 2.GlobalJobDesc in NNGraph Compile
* Fix bug: AutoTick Multi-Client callback notifier lbn
* Fix bug: callback notifier op check in multi-client
* Fix bug: CheckOpGraph in multi-client
* Fix bug of MultiClientSessionContext new ProfileConf
* Fix bug of create NONE type kernel; add note; refine PlanUtil::PopulateOpAttibute
* Fix bug: CallbackNotifier kernel dtype with uint8; change tick dtype; Feat: add Kernel create fault-tolerant
* inputs tensor list
* rollback of hack
* Using TensorTuple for RunLazyNNGraph
* Fix bug: WaitAndSendIdsKernel handle for Multi-Client
* skip graph runnable test
* refine
* skip test graph optimizer for nn.Graph train job completer error
* Skip SetDefaultVariableConf JobPass in Multi-Client
* Fix bug: register_variable_op_names_and_tensors args with TensorTuple
* refine test graphs
* rm experimental in new scripts
Co-authored-by:
oneflow-ci-bot <69100618+oneflow-ci-bot@users.noreply.github.com>
Showing
- oneflow/api/python/framework/nn_graph.cpp 5 additions, 6 deletionsoneflow/api/python/framework/nn_graph.cpp
- oneflow/core/common/auto_registration_factory.h 3 additions, 1 deletiononeflow/core/common/auto_registration_factory.h
- oneflow/core/framework/multi_client_session_context.cpp 2 additions, 0 deletionsoneflow/core/framework/multi_client_session_context.cpp
- oneflow/core/framework/nn_graph.cpp 9 additions, 4 deletionsoneflow/core/framework/nn_graph.cpp
- oneflow/core/framework/nn_graph.h 3 additions, 3 deletionsoneflow/core/framework/nn_graph.h
- oneflow/core/job/job_build_and_infer_ctx.cpp 2 additions, 0 deletionsoneflow/core/job/job_build_and_infer_ctx.cpp
- oneflow/core/job/oneflow.cpp 1 addition, 25 deletionsoneflow/core/job/oneflow.cpp
- oneflow/core/job/plan_util.cpp 24 additions, 0 deletionsoneflow/core/job/plan_util.cpp
- oneflow/core/job/plan_util.h 5 additions, 0 deletionsoneflow/core/job/plan_util.h
- oneflow/core/job_rewriter/autotick.cpp 1 addition, 1 deletiononeflow/core/job_rewriter/autotick.cpp
- oneflow/core/job_rewriter/job_completer.cpp 22 additions, 2 deletionsoneflow/core/job_rewriter/job_completer.cpp
- oneflow/core/job_rewriter/set_default_variable_conf.cpp 5 additions, 0 deletionsoneflow/core/job_rewriter/set_default_variable_conf.cpp
- oneflow/core/kernel/kernel.cpp 2 additions, 0 deletionsoneflow/core/kernel/kernel.cpp
- oneflow/core/kernel/kernel.h 28 additions, 4 deletionsoneflow/core/kernel/kernel.h
- oneflow/core/kernel/wait_and_send_ids_kernel.cpp 6 additions, 1 deletiononeflow/core/kernel/wait_and_send_ids_kernel.cpp
- oneflow/core/operator/device_tick_op.cpp 1 addition, 1 deletiononeflow/core/operator/device_tick_op.cpp
- oneflow/core/operator/dst_subset_tick_op.cpp 1 addition, 1 deletiononeflow/core/operator/dst_subset_tick_op.cpp
- oneflow/core/operator/sink_tick_op.cpp 1 addition, 1 deletiononeflow/core/operator/sink_tick_op.cpp
- oneflow/core/operator/source_tick_op.cpp 2 additions, 2 deletionsoneflow/core/operator/source_tick_op.cpp
- 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