From 50af052b3129164e28efa8b9321d733311b7b459 Mon Sep 17 00:00:00 2001 From: awaelchli Date: Tue, 9 Jul 2024 18:18:35 +0200 Subject: [PATCH] Add reference to the `torch.compile` manual (#20066) --- docs/source-fabric/advanced/compile.rst | 3 +++ docs/source-pytorch/advanced/compile.rst | 3 +++ 2 files changed, 6 insertions(+) diff --git a/docs/source-fabric/advanced/compile.rst b/docs/source-fabric/advanced/compile.rst index ed46a1f822..17ba6e4ca9 100644 --- a/docs/source-fabric/advanced/compile.rst +++ b/docs/source-fabric/advanced/compile.rst @@ -272,6 +272,8 @@ As a note, the compilation phase itself will take some time, taking up to severa For these reasons, we recommend that you don't invest too much time trying to apply ``torch.compile`` during development, and rather evaluate its effectiveness toward the end when you are about to launch long-running, expensive experiments. Always compare the speed and memory usage of the compiled model against the original model! +For a thorough troubleshooting guide, see `Torch.compile: the missing manual `_. + ---- @@ -313,5 +315,6 @@ Here are a few resources for further reading after you complete this tutorial: - `GenAI with PyTorch 2.0 blog post series `_ - `Training Production AI Models with PyTorch 2.0 `_ - `Empowering Models with Performance: The Art of Generalized Model Transformation Approach `_ +- `Torch.compile: the missing manual `_ | diff --git a/docs/source-pytorch/advanced/compile.rst b/docs/source-pytorch/advanced/compile.rst index 484559e111..d5bd333c04 100644 --- a/docs/source-pytorch/advanced/compile.rst +++ b/docs/source-pytorch/advanced/compile.rst @@ -241,6 +241,8 @@ As a note, the compilation phase itself will take some time, taking up to severa For these reasons, we recommend that you don't invest too much time trying to apply ``torch.compile`` during development, and rather evaluate its effectiveness toward the end when you are about to launch long-running, expensive experiments. Always compare the speed and memory usage of the compiled model against the original model! +For a thorough troubleshooting guide, see `Torch.compile: the missing manual `_. + ---- @@ -282,5 +284,6 @@ Here are a few resources for further reading after you complete this tutorial: - `GenAI with PyTorch 2.0 blog post series `_ - `Training Production AI Models with PyTorch 2.0 `_ - `Empowering Models with Performance: The Art of Generalized Model Transformation Approach `_ +- `Torch.compile: the missing manual `_ |