pytorch/torch/export
chilli c486e2ab64 Add coloring to fx graph print out (#128476)
Note: Won't land immediately, at least I'll need to add a color option to the field. But curious if any tests fail.

Old:
<img width="1294" alt="image" src="https://github.com/pytorch/pytorch/assets/6355099/c3a750ed-5e54-4621-b2e4-be5481be15b6">

New:
<img width="1303" alt="image" src="https://github.com/pytorch/pytorch/assets/6355099/3a1f1adc-6f3a-413e-8b87-ee53da9bf4ed">

Pull Request resolved: https://github.com/pytorch/pytorch/pull/128476
Approved by: https://github.com/ezyang
2024-06-13 23:39:04 +00:00
..
__init__.py [RELAND] Switch default behavoir of export IR to be predispatch (#125860) 2024-05-10 17:36:53 +00:00
_remove_auto_functionalized_pass.py Flip default value for mypy disallow_untyped_defs [6/11] (#127843) 2024-06-08 18:49:29 +00:00
_remove_effect_tokens_pass.py Flip default value for mypy disallow_untyped_defs [6/11] (#127843) 2024-06-08 18:49:29 +00:00
_safeguard.py Flip default value for mypy disallow_untyped_defs [6/11] (#127843) 2024-06-08 18:49:29 +00:00
_trace.py Flip default value for mypy disallow_untyped_defs [6/11] (#127843) 2024-06-08 18:49:29 +00:00
_tree_utils.py
_unlift.py Support aten operations with out tensor (#124926) 2024-06-12 22:31:59 +00:00
custom_obj.py
dynamic_shapes.py Introduce int_oo (#127693) 2024-06-13 04:08:20 +00:00
exported_program.py Add coloring to fx graph print out (#128476) 2024-06-13 23:39:04 +00:00
graph_signature.py Flip default value for mypy disallow_untyped_defs [7/11] (#127844) 2024-06-08 18:49:45 +00:00
unflatten.py [export] FIx unflattener for preserving modules containing unused inputs (#128260) 2024-06-11 22:32:08 +00:00