pytorch-lightning icon indicating copy to clipboard operation
pytorch-lightning copied to clipboard
verified publisher icon Lightning-AI

Add `torch.compile` guide to docs

Open carmocca opened this issue 2 years ago • 1 comments

📚 Documentation

This applies to

  • [x] Fabric (https://github.com/Lightning-AI/pytorch-lightning/pull/19330)
  • [ ] PyTorch Lightning

Things to mention:

  • Compiling nn.Module vs LightningModule or fabric.setup(module) (#17250)
  • Known limitations.
  • Known sources of graph breaks.

Related: #12398

cc @borda @carmocca @justusschock @awaelchli

carmocca avatar Oct 24 '23 13:10 carmocca

This would also close #17250

awaelchli avatar Oct 24 '23 15:10 awaelchli