pytorch/test/fx
Gabriel Ferns 1e73842029 Refactor FxGraphDrawer to use HTML-like labels (#137726)
Fixes https://github.com/pytorch/pytorch/issues/137499
Testing: Added a new unit test to make sure that the regression case succeeds.
I'm debating about whether to make the borders visible. I'm partial to no borders, but it might make it harder for some people to read?
![68a2b0e3-orig_fx_graph_diagram](https://github.com/user-attachments/assets/fbc2fd98-9e76-488e-8ebe-c64fbf206932)
Vs.
![2bfe1c4f-orig_fx_graph_diagram](https://github.com/user-attachments/assets/b6bc88ba-dda2-4cf7-84ac-a615e1e03a74)

Pull Request resolved: https://github.com/pytorch/pytorch/pull/137726
Approved by: https://github.com/eellison, https://github.com/malfet
2024-11-01 23:19:50 +00:00
..
named_tup.py Enable UFMT on all of test/fx (#123622) 2024-04-09 15:59:17 +00:00
quantization.py [BE][Easy][9/19] enforce style for empty lines in import segments in test/[e-h]*/ (#129760) 2024-07-17 14:25:29 +00:00
test_common_passes.py [BE][Easy][9/19] enforce style for empty lines in import segments in test/[e-h]*/ (#129760) 2024-07-17 14:25:29 +00:00
test_cse_pass.py [BE][Easy][9/19] enforce style for empty lines in import segments in test/[e-h]*/ (#129760) 2024-07-17 14:25:29 +00:00
test_dce_pass.py [BE]: Add better optional typing (#138426) 2024-10-27 14:19:00 +00:00
test_future.py Enable UFMT on all of test/fx (#123622) 2024-04-09 15:59:17 +00:00
test_fx_const_fold.py Add None return type to init -- tests (#132352) 2024-08-01 15:44:51 +00:00
test_fx_node_hook.py Register creak_node_hook (#126671) 2024-05-24 23:32:15 +00:00
test_fx_param_shape_control_flow.py [BE][Easy][9/19] enforce style for empty lines in import segments in test/[e-h]*/ (#129760) 2024-07-17 14:25:29 +00:00
test_fx_split.py Add None return type to init -- tests (#132352) 2024-08-01 15:44:51 +00:00
test_fx_xform_observer.py [Easy] GraphTransformObserver Refactoring (#139292) 2024-10-31 00:33:28 +00:00
test_gradual_type.py Add None return type to init -- tests (#132352) 2024-08-01 15:44:51 +00:00
test_graph_drawer.py Refactor FxGraphDrawer to use HTML-like labels (#137726) 2024-11-01 23:19:50 +00:00
test_lazy_graph_module.py make dyanmo work with _LazyGraphModule.lazy_forward (#121259) 2024-03-08 01:37:39 +00:00
test_matcher_utils.py Replace capture_pre_autograd_graph with export_for_training in torch tests (#135623) 2024-09-11 19:23:08 +00:00
test_partitioner_order.py [BE][Easy][9/19] enforce style for empty lines in import segments in test/[e-h]*/ (#129760) 2024-07-17 14:25:29 +00:00
test_pass_infra.py [BE][Easy][9/19] enforce style for empty lines in import segments in test/[e-h]*/ (#129760) 2024-07-17 14:25:29 +00:00
test_shape_inference.py Add None return type to init -- tests (#132352) 2024-08-01 15:44:51 +00:00
test_source_matcher_utils.py [export] Temporarily bypass torch_fn in partitioner (#134292) 2024-08-24 05:50:18 +00:00
test_subgraph_rewriter.py [torch][fx] Add new replacement_callback to materialize a replacement just in time (#135553) 2024-09-12 18:52:14 +00:00
test_z3_gradual_types.py Add None return type to init -- tests (#132352) 2024-08-01 15:44:51 +00:00