mirror of
https://github.com/saymrwulf/pytorch.git
synced 2026-05-14 20:57:59 +00:00
Summary: Add `node_mapping = create_node_mapping(pre_grad_graph_id, inductor_post_to_pre_grad_nodes, debug_info)`, to produce a `inductor_provenance_tracking_node_mappings.json` file. This file will be used by the provenance tracking highlighter tool to create provenance visualization. `inductor_triton_kernel_to_post_grad_nodes.json` and `inductor_provenance_tracking_node_mappings.json` files are not dumped if they are both empty. So it's removed from some of the `test_structured_trace` tests. Test Plan: CI ``` buck run mode/dev-nosan fbcode//caffe2/test:fx -- -r graph_provenance buck run mode/dev-nosan fbcode//caffe2/test/inductor:provenance_tracing python test/dynamo/test_structured_trace.py ``` Differential Revision: D68190173 Pull Request resolved: https://github.com/pytorch/pytorch/pull/146103 Approved by: https://github.com/chenyang78 |
||
|---|---|---|
| .. | ||
| 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_traceback.py | ||
| test_fx_xform_observer.py | ||
| test_gradual_type.py | ||
| test_graph_pickler.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 | ||