Dev inplace (#1879)
* OpGraph::MakePredicatorIsAllLbiConsumersReachableToOpName * refactor TaskGraph::EnableInplaceMemSharing * TaskGraph::GetSafeInplaceOpBlobArgList * InplaceLbiGraph::ForEachSafeInplaceEdgesInSourceOpSubTree * fix a typo * TaskGraph::SetTaskRegstInplaceInfo * InplaceRegstGraph * refine * refine IsLbiOnTaskEdge * fix a bug in TaskGraph::ForEachDeviceNodes * ForEachGpuDeviceNodes * remove wrong CHECK * fix wrong use of std::unordered_set::erase * fix a bug in TaskGraph::GetInplaceOpBlobArgList * fix inplace bugs * fix error CHECK between inplace in dptr and inplace out dptr
Showing
- oneflow/core/actor/actor.cpp 126 additions, 54 deletionsoneflow/core/actor/actor.cpp
- oneflow/core/actor/actor.h 24 additions, 7 deletionsoneflow/core/actor/actor.h
- oneflow/core/actor/normal_forward_compute_actor.cpp 8 additions, 1 deletiononeflow/core/actor/normal_forward_compute_actor.cpp
- oneflow/core/actor/normal_forward_compute_actor.h 1 addition, 0 deletionsoneflow/core/actor/normal_forward_compute_actor.h
- oneflow/core/graph/exec_graph.h 1 addition, 1 deletiononeflow/core/graph/exec_graph.h
- oneflow/core/graph/graph.h 15 additions, 0 deletionsoneflow/core/graph/graph.h
- oneflow/core/graph/inplace_lbi_graph.cpp 88 additions, 44 deletionsoneflow/core/graph/inplace_lbi_graph.cpp
- oneflow/core/graph/inplace_lbi_graph.h 19 additions, 13 deletionsoneflow/core/graph/inplace_lbi_graph.h
- oneflow/core/graph/inplace_regst_graph.cpp 52 additions, 0 deletionsoneflow/core/graph/inplace_regst_graph.cpp
- oneflow/core/graph/inplace_regst_graph.h 41 additions, 0 deletionsoneflow/core/graph/inplace_regst_graph.h
- oneflow/core/graph/op_graph.cpp 67 additions, 1 deletiononeflow/core/graph/op_graph.cpp
- oneflow/core/graph/op_graph.h 9 additions, 2 deletionsoneflow/core/graph/op_graph.h
- oneflow/core/graph/sharable_mem_block_graph.cpp 42 additions, 38 deletionsoneflow/core/graph/sharable_mem_block_graph.cpp
- oneflow/core/graph/sharable_mem_block_graph.h 4 additions, 8 deletionsoneflow/core/graph/sharable_mem_block_graph.h
- oneflow/core/graph/task_graph.cpp 144 additions, 23 deletionsoneflow/core/graph/task_graph.cpp
- oneflow/core/graph/task_graph.h 15 additions, 1 deletiononeflow/core/graph/task_graph.h
- oneflow/core/job/compiler.cpp 4 additions, 1 deletiononeflow/core/job/compiler.cpp
- oneflow/core/job/improver.cpp 48 additions, 22 deletionsoneflow/core/job/improver.cpp
- oneflow/core/job/job_conf.proto 1 addition, 0 deletionsoneflow/core/job/job_conf.proto
- oneflow/core/job/job_desc.h 1 addition, 0 deletionsoneflow/core/job/job_desc.h
Please register or sign in to comment