mirror of
https://github.com/saymrwulf/pytorch.git
synced 2026-05-15 21:00:47 +00:00
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 |
||
|---|---|---|
| .. | ||
| 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 | ||
| test_gradual_type.py | ||
| test_lazy_graph_module.py | ||
| test_matcher_utils.py | ||
| test_partitioner_order.py | ||
| test_pass_infra.py | ||
| test_shape_inference.py | ||
| test_source_matcher_utils.py | ||
| test_subgraph_rewriter.py | ||
| test_z3_gradual_types.py | ||