Rename folder, add cross-reference

Lightning-AI · Nov 25, 2024 · 55a6fde · 55a6fde
1 parent ca3c4cf
commit 55a6fde
Show file tree

Hide file tree

Showing 8 changed files with 4 additions and 1 deletion.
diff --git a/docs/source-fabric/advanced/compile.rst b/docs/source-fabric/advanced/compile.rst
@@ -115,7 +115,6 @@ always exclude the first call to ``forward()`` from your measurements, since it
         Compile median time: 0.0185 seconds
         Speedup: 1.4x
 
-
 ----
 
 *********************************
@@ -220,6 +219,8 @@ needed to achieve the equivalent of `compile(distributed(quantized(model)))`:
 
         model = fabric.setup(model)
 
+For a full example, see our `FP8 Distributed Transformer example <https://github.com/Lightning-AI/lightning/blob/master/examples/fabric/fp8_distributed_transformer>`_.
+
 ----
 
 ******************

diff --git a/docs/source-pytorch/advanced/compile.rst b/docs/source-pytorch/advanced/compile.rst
@@ -249,6 +249,8 @@ needed to achieve the equivalent of `compile(distributed(quantized(model)))`:
 
             self.model = torch.compile(model)
 
+For a full example, see our `FP8 Distributed Transformer example <https://github.com/Lightning-AI/lightning/blob/master/examples/pytorch/fp8_distributed_transformer>`_.
+
 ----
 
 ******************

diff --git a/examples/fabric/fp8_fsdp2_compile/README.md → ...ric/fp8_distributed_transformer/README.md b/examples/fabric/fp8_fsdp2_compile/README.md → ...ric/fp8_distributed_transformer/README.md
diff --git a/...fabric/fp8_fsdp2_compile/requirements.txt → ..._distributed_transformer/requirements.txt b/...fabric/fp8_fsdp2_compile/requirements.txt → ..._distributed_transformer/requirements.txt
diff --git a/examples/fabric/fp8_fsdp2_compile/train.py → ...bric/fp8_distributed_transformer/train.py b/examples/fabric/fp8_fsdp2_compile/train.py → ...bric/fp8_distributed_transformer/train.py
diff --git a/examples/pytorch/fp8_fsdp2_compile/README.md → ...rch/fp8_distributed_transformer/README.md b/examples/pytorch/fp8_fsdp2_compile/README.md → ...rch/fp8_distributed_transformer/README.md
diff --git a/...ytorch/fp8_fsdp2_compile/requirements.txt → ..._distributed_transformer/requirements.txt b/...ytorch/fp8_fsdp2_compile/requirements.txt → ..._distributed_transformer/requirements.txt
diff --git a/examples/pytorch/fp8_fsdp2_compile/train.py → ...orch/fp8_distributed_transformer/train.py b/examples/pytorch/fp8_fsdp2_compile/train.py → ...orch/fp8_distributed_transformer/train.py
-Original file line number
+Diff line change
@@ Expand Up @@
                 self.model = torch.compile(model)
+    For a full example, see our `FP8 Distributed Transformer example <https://github.com/Lightning-AI/lightning/blob/master/examples/pytorch/fp8_distributed_transformer>`_.
     ----
     ******************
@@ Expand Down @@