Fea/nn graph/graph build ctx (#5420)
* graph api * add graph dummy test * add test * add recursive module mode * graph.build test pass * add detail check on graph inner node * support config and train * add repr for debug * test buffer * test buffer add * refine test * add comment * refine test * refactor Node to Block * add named_state * refine Graph.named_state() * add state_tensortuple * graph._compile() * add mc session 0 * nn.graph: state tuple to private var; add BlockType; add simple multi client session * NNGraphIf * rm old graph.cpp * nn.graph: add cpp NNGraph; export and call NNGraph * add comment * nn.Graph: rm prototype MultiClientSession * nn.Graph: rm prototype MultiClientSession test * nn.Graph: add TODO * nn.Graph: format for review * nn.Graph: format * nn.Graph: format * nn.Graph: pass flake8 check * job_build_ctx * support lazy context * format * lazy mode * format * format * lazy mode add test * debug session * init session and job_build_context * rm temp code * build default scope * add default scope * add scope proto for debug * chech scope * format * refine MultiClientSession.resource * address review * lazy init scope stack in single-client, instantly init scope stack after MultiClientSession created in multi-client * fix typo * address review * fix clear default session * merger and test Co-authored-by:Xinqi Li <lixinqi0703106@163.com> Co-authored-by:
oneflow-ci-bot <69100618+oneflow-ci-bot@users.noreply.github.com> Co-authored-by:
leaves-zwx <kunta0932@gmail.com> Co-authored-by:
cheng cheng <472491134@qq.com>
Showing
- oneflow/api/python/framework/framework.h 1 addition, 6 deletionsoneflow/api/python/framework/framework.h
- oneflow/api/python/functional/common.h 1 addition, 1 deletiononeflow/api/python/functional/common.h
- oneflow/api/python/job_build/lazy_mode.cpp 36 additions, 0 deletionsoneflow/api/python/job_build/lazy_mode.cpp
- oneflow/api/python/symbol/scope_symbol.cpp 3 additions, 1 deletiononeflow/api/python/symbol/scope_symbol.cpp
- oneflow/core/framework/attr_value_accessor.cpp 4 additions, 4 deletionsoneflow/core/framework/attr_value_accessor.cpp
- oneflow/core/framework/consistent_tensor_infer_cache.cpp 1 addition, 1 deletiononeflow/core/framework/consistent_tensor_infer_cache.cpp
- oneflow/core/framework/device.cpp 4 additions, 4 deletionsoneflow/core/framework/device.cpp
- oneflow/core/framework/op_interpreter/op_interpreter_util.cpp 12 additions, 12 deletions...low/core/framework/op_interpreter/op_interpreter_util.cpp
- oneflow/core/framework/random_generator_impl.cpp 1 addition, 1 deletiononeflow/core/framework/random_generator_impl.cpp
- oneflow/core/framework/tensor.cpp 2 additions, 2 deletionsoneflow/core/framework/tensor.cpp
- oneflow/core/framework/tensor_impl.cpp 4 additions, 4 deletionsoneflow/core/framework/tensor_impl.cpp
- oneflow/core/framework/tensor_name_scope.cpp 1 addition, 1 deletiononeflow/core/framework/tensor_name_scope.cpp
- oneflow/core/job/job.proto 2 additions, 2 deletionsoneflow/core/job/job.proto
- oneflow/core/job/job_build_and_infer_ctx_mgr.cpp 4 additions, 2 deletionsoneflow/core/job/job_build_and_infer_ctx_mgr.cpp
- oneflow/core/job/lazy_mode.cpp 29 additions, 0 deletionsoneflow/core/job/lazy_mode.cpp
- oneflow/core/job/lazy_mode.h 47 additions, 0 deletionsoneflow/core/job/lazy_mode.h
- oneflow/python/framework/graph_build_util.py 62 additions, 0 deletionsoneflow/python/framework/graph_build_util.py
- oneflow/python/framework/scope_util.py 9 additions, 1 deletiononeflow/python/framework/scope_util.py
- oneflow/python/nn/graph.py 49 additions, 13 deletionsoneflow/python/nn/graph.py
- oneflow/python/test/graph/test_graph.py 60 additions, 1 deletiononeflow/python/test/graph/test_graph.py
Please register or sign in to comment