pytorch/torch/csrc/jit/python
Elias Ellison aab4beb87f [JIT] Pass To Safely Remove Aten Inplace Ops (#33186)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/33186

This helps create larger functional graphs. It has the potential to increase memory use, so in order to land this on by default we would probably also do a reuse of buffers pass.

This is currently O(n * | Removed Nodes | ) because we have to rebuild the alias Db each time we make a change. This pass is critical to creating functional graphs, so this might be a compelling use case to build incremental updates to alias Db.

Test Plan: Imported from OSS

Differential Revision: D20603189

Pulled By: eellison

fbshipit-source-id: 105db52bf38e02188ca6df6d36294466d3309a0a
2020-03-24 23:45:58 -07:00
..
init.cpp [JIT] Pass To Safely Remove Aten Inplace Ops (#33186) 2020-03-24 23:45:58 -07:00
init.h
module_python.h [jit] kill script namespace (#34515) 2020-03-11 23:32:48 -07:00
pybind_utils.h [JIT] add id function (#34975) 2020-03-23 17:10:13 -07:00
pybind.h
python_arg_flatten.cpp
python_arg_flatten.h
python_custom_class.cpp [reapply][JIT] Namespaces for TorchBind (#35254) 2020-03-24 00:39:48 -07:00
python_custom_class.h
python_interpreter.cpp Delete OperatorOptions, absorb AliasAnalysisKind into FunctionSchema. (#34588) 2020-03-11 20:59:46 -07:00
python_ir.cpp [jit] make Future type annotation available in Python (#27637) 2020-03-24 14:36:05 -07:00
python_ir.h
python_ivalue.h
python_sugared_value.cpp [jit] kill script namespace (#34515) 2020-03-11 23:32:48 -07:00
python_sugared_value.h [jit] kill script namespace (#34515) 2020-03-11 23:32:48 -07:00
python_tracer.cpp [jit] kill script namespace (#34515) 2020-03-11 23:32:48 -07:00
python_tracer.h [jit] kill script namespace (#34515) 2020-03-11 23:32:48 -07:00
python_tree_views.cpp [jit] kill script namespace (#34515) 2020-03-11 23:32:48 -07:00
python_tree_views.h [jit] kill script namespace (#34515) 2020-03-11 23:32:48 -07:00
script_init.cpp [jit] Include call stack in OSError message (#34669) 2020-03-18 15:10:23 -07:00
script_init.h [jit] kill script namespace (#34515) 2020-03-11 23:32:48 -07:00
update_graph_executor_opt.cpp
update_graph_executor_opt.h