What problem does it solve?
This Skill helps diagnose PyTorch Dynamo compilation problems such as graph breaks, unsupported operations, unexpected FX graph transformations, excessive recompilation, and missing pre-grad optimizations.
Core Features & Use Cases
- Dynamo tracing: Inspect bytecode capture, FX graph construction, graph breaks, and unsupported operations.
- Pre-grad analysis: Compare FX graphs before and after transformations such as Conv-BN fusion and split-cat elimination.
- Dynamic shape debugging: Investigate guards, recompilation causes, and shape specialization behavior.
- Compilation verification: Reproduce captured graphs, compare eager and compiled behavior, and validate optimization results.
- Use Case: When a model produces multiple small compiled graphs or fails to fuse Conv-BN operations, use the Skill to enable targeted logs, inspect generated FX graph files, identify the cause, and verify the fix.
Quick Start
Ask the AI to trace a PyTorch compilation failure through Dynamo, inspect graph breaks and FX graphs, and verify whether the expected pre-grad optimizations occurred.