Whenever I am trying to do something fun (read: poorly considered) with torch.compile I find myself googling for the same handful of references, most of which are in the PyTorch Drive folder, so putting them here for my own reference.
If you are doing regular compile usage you shouldn’t need these, but they are helpful when debugging more complex shape issues or investigating performance issues or graph breaks.
- The dynamic shapes manual
- GuardOnDataDependent aka Ed helps you put torch.check everywhere
- torch.compile the missing manual
- Dynamic shapes manual (this one has the good guidance on meta functions)
- Various dynamo logging settings
Leave a Reply