Lazy job stream type (#5389)
* replace ForeignJobInstance using JobInstance * LazyJobStreamType * NNGraphIf * NNGraph -> NNGraphIf * fix compile bugs * add unit tests for instruction RunLazyJob * GetInputBufferName/GetOutputBufferName * address pr comments * add cuda complie switch * refine Co-authored-by:liufengwei <2472937968@qq.com> Co-authored-by:
oneflow-ci-bot <69100618+oneflow-ci-bot@users.noreply.github.com> Co-authored-by:
clackhan <han_binbin@163.com>
Showing
- oneflow/core/common/buffer_manager.h 18 additions, 1 deletiononeflow/core/common/buffer_manager.h
- oneflow/core/eager/lazy_job_device_context.h 98 additions, 0 deletionsoneflow/core/eager/lazy_job_device_context.h
- oneflow/core/eager/lazy_job_instruction_type.cpp 166 additions, 0 deletionsoneflow/core/eager/lazy_job_instruction_type.cpp
- oneflow/core/eager/lazy_job_instruction_type_test.cpp 255 additions, 0 deletionsoneflow/core/eager/lazy_job_instruction_type_test.cpp
- oneflow/core/eager/lazy_job_stream_type.cpp 68 additions, 0 deletionsoneflow/core/eager/lazy_job_stream_type.cpp
- oneflow/core/eager/lazy_job_stream_type.h 53 additions, 0 deletionsoneflow/core/eager/lazy_job_stream_type.h
- oneflow/core/eager/local_call_opkernel_phy_instr_operand.h 0 additions, 1 deletiononeflow/core/eager/local_call_opkernel_phy_instr_operand.h
- oneflow/core/eager/opkernel_instruction_type.cpp 1 addition, 1 deletiononeflow/core/eager/opkernel_instruction_type.cpp
- oneflow/core/eager/run_lazy_job_phy_instr_operand.cpp 54 additions, 0 deletionsoneflow/core/eager/run_lazy_job_phy_instr_operand.cpp
- oneflow/core/eager/run_lazy_job_phy_instr_operand.h 72 additions, 0 deletionsoneflow/core/eager/run_lazy_job_phy_instr_operand.h
- oneflow/core/framework/instructions_builder.cpp 15 additions, 3 deletionsoneflow/core/framework/instructions_builder.cpp
- oneflow/core/framework/instructions_builder.h 8 additions, 0 deletionsoneflow/core/framework/instructions_builder.h
- oneflow/core/job/job_instance.h 6 additions, 0 deletionsoneflow/core/job/job_instance.h
- oneflow/core/register/ofblob.h 4 additions, 0 deletionsoneflow/core/register/ofblob.h
- oneflow/core/vm/cpu_stream_type.cpp 1 addition, 2 deletionsoneflow/core/vm/cpu_stream_type.cpp
Please register or sign in to comment