DeepSpeed/tests/torch_compile
Ma, Guokai cf41e8c4e8
[compile] Show breakdown of graph break (#6601)
This PR extends https://github.com/microsoft/DeepSpeed/pull/6570 by
showing a breakdown of graph breaks. So we can see how graph breaks are
distributed among different reasons. An example of graph break output
can be seen from the following workflow run
https://github.com/microsoft/DeepSpeed/actions/runs/11199157962
2024-10-14 17:31:34 +00:00
..
ds_config_z2.json [compile] Show breakdown of graph break (#6601) 2024-10-14 17:31:34 +00:00
ds_config_z3.json [compile] Show breakdown of graph break (#6601) 2024-10-14 17:31:34 +00:00
test_compile.py [compile] Show breakdown of graph break (#6601) 2024-10-14 17:31:34 +00:00