pytorch/test/fx
Sheng Fu ae2359638b Save DOT file of graph instead of SVG for GraphTranformObserver (#128634)
Summary:
GraphTransformObserver saves the SVG file of the input/output graph in each inductor pass. In my test with CMF model, if the graph is large, GraphViz took forever to convert DOT to SVG. That is NOT acceptable.

This DIFF is to save DOT file instead of SVG file to speed it up. Also DOT file size is order of mangitude smaller than SVG.

To view these graphs, user can run dot -Txxx inpout.dot to convert DOT to any other format you want. User can control how many iterations to layout the graph properly. Refer to https://web.archive.org/web/20170507095019/http://graphviz.org/content/attrs#dnslimit for details.

Test Plan: buck2 test mode/dev-sand caffe2/test:fx --  fx.test_fx_xform_observer.TestGraphTransformObserver

Differential Revision: D58539182

Pull Request resolved: https://github.com/pytorch/pytorch/pull/128634
Approved by: https://github.com/mengluy0125
2024-06-14 00:54:22 +00:00
..
named_tup.py
quantization.py
test_common_passes.py
test_cse_pass.py
test_dce_pass.py
test_future.py
test_fx_const_fold.py
test_fx_node_hook.py
test_fx_param_shape_control_flow.py
test_fx_split.py
test_fx_xform_observer.py Save DOT file of graph instead of SVG for GraphTranformObserver (#128634) 2024-06-14 00:54:22 +00:00
test_gradual_type.py [5/N][Easy] fix typo for usort config in pyproject.toml (kown -> known): sort torch (#127126) 2024-05-27 14:49:57 +00:00
test_lazy_graph_module.py
test_matcher_utils.py
test_partitioner_order.py [fx] Preserve Fx graph node order in partitioner across runs (#115621) 2024-06-11 14:04:52 +00:00
test_pass_infra.py
test_shape_inference.py
test_source_matcher_utils.py
test_subgraph_rewriter.py
test_z3_gradual_types.py