OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-07 12:21:27 +01:00
Code Issues Actions 38 Packages Projects Releases Wiki Activity
3d7c22a2ce
pytorch/torch/csrc/jit/python
History
neginraoof 3d7c22a2ce [ONNX] Enable new scripting passes for functionalization and remove_mutation (#43791) 
Summary:
Duplicate of https://github.com/pytorch/pytorch/issues/41413
This PR initiates the process of updating the torchsciprt backend interface used by ONNX exporter.

Replace jit lower graph pass by freeze module pass

Enable ScriptModule tests for ONNX operator tests (ORT backend) and model tests by default.

Replace jit remove_inplace_ops pass with remove_mutation and consolidation all passes for handling inplace ops.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/43791

Reviewed By: houseroad

Differential Revision: D23421872

Pulled By: bzinodev

fbshipit-source-id: a98710c45ee905748ec58385e2a232de2486331b
2020-09-04 15:21:45 -07:00
..
init.cpp [ONNX] Enable new scripting passes for functionalization and remove_mutation (#43791) 2020-09-04 15:21:45 -07:00
init.h
module_python.h
pybind_utils.h Add Enum convert back to Python object support (#43121) 2020-08-21 10:36:51 -07:00
pybind.h
python_arg_flatten.cpp
python_arg_flatten.h Move torch/csrc/utils/hash.h to c10/util/hash.h. (#42503) 2020-08-29 17:47:00 -07:00
python_custom_class.cpp s/torch::jit::class_/torch::class_/ (#40795) 2020-07-06 15:53:33 -07:00
python_custom_class.h
python_interpreter.cpp Unify boxed function signature between jit and c10 (#37034) 2020-06-29 19:24:26 -07:00
python_ir.cpp Implement JIT Enum type serialization and deserialization (#43460) 2020-08-24 12:04:31 -07:00
python_ir.h
python_ivalue.h Explicitly decref py::object in ConcretePyObjectHolder and PythonFunctionGuard (#38364) 2020-05-12 20:55:53 -07:00
python_sugared_value.cpp [jit] Better match behavior of loaded ScriptModules vs. freshly created ones (#43298) 2020-09-03 15:03:39 -07:00
python_sugared_value.h Implement JIT Enum type serialization and deserialization (#43460) 2020-08-24 12:04:31 -07:00
python_tracer.cpp [jit] tracer flag to guard risky behaivors (#36277) 2020-04-13 22:35:03 -07:00
python_tracer.h [jit] tracer flag to guard risky behaivors (#36277) 2020-04-13 22:35:03 -07:00
python_tree_views.cpp Add __str__ and __repr__ bindings to SourceRange (#43601) 2020-08-27 12:30:47 -07:00
python_tree_views.h
script_init.cpp [jit] Better match behavior of loaded ScriptModules vs. freshly created ones (#43298) 2020-09-03 15:03:39 -07:00
script_init.h
update_graph_executor_opt.cpp Canonicalize includes in torch, and add tests for it (#36303) 2020-04-23 08:09:21 -07:00
update_graph_executor_opt.h